Mailspring/packages/client-app/internal_packages/thread-list/lib/injects-toolbar-buttons.jsx

import React, {Component, PropTypes} from 'react'
import {ListensToObservable, InjectedComponentSet} from 'nylas-component-kit'
import ThreadListStore from './thread-list-store'


export const ToolbarRole = 'ThreadActionsToolbarButton'


function defaultObservable() {
  return ThreadListStore.selectionObservable()
}

function InjectsToolbarButtons(ToolbarComponent, {getObservable, extraRoles = []}) {
  const roles = [ToolbarRole].concat(extraRoles)

  class ComposedComponent extends Component {
    static displayName = ToolbarComponent.displayName;

    static propTypes = {
      items: PropTypes.array,
    };

    static containerRequired = false;

    render() {
      const {items} = this.props;
      const {selection} = ThreadListStore.dataSource()

      // Keep all of the exposed props from deprecated regions that now map to this one
      const exposedProps = {
        items,
        selection,
        thread: items[0],
      }
      const injectedButtons = (
        <InjectedComponentSet
          key="injected"
          matching={{roles}}
          exposedProps={exposedProps}
        />
      )
      return (
        <ToolbarComponent
          items={items}
          selection={selection}
          injectedButtons={injectedButtons}
        />
      )
    }
  }

  const getStateFromObservable = (items) => {
    if (!items) {
      return {items: []}
    }
    return {items}
  }
  return ListensToObservable(ComposedComponent, {
    getObservable: getObservable || defaultObservable,
    getStateFromObservable,
  })
}

export default InjectsToolbarButtons