Function Injector

src/Injector.js:13–74

({ into } = {})

let injectorIndex = 0;

const Injector = ({ into } = {}) => {
  invariant(
    into &&
    typeof into === `function` &&
    into.injectionId &&
    into.contextTypes &&
    into.contextTypes.registerInjectable &&
    into.contextTypes.removeInjectable,
    // Error message
    invalidTargetMsg);

  return function WrapComponent(InjectionComponent) {
    invariant(
      InjectionComponent &&
      typeof InjectionComponent === `function`,
      invalidInjectMsg
    );

    injectorIndex++;
    const injectorId = `injector_${injectorIndex}`;

    class InjectorComponent extends Component {
      static contextTypes = {
        registerInjector: PropTypes.func.isRequired,
        updateInjector: PropTypes.func.isRequired,
        removeInjector: PropTypes.func.isRequired
      };

      componentWillMount() {
        this.context.registerInjector({
          injectionId: into.injectionId,
          injectorId,
          injectorInstance: this,
          inject: () => <InjectionComponent {...this.props} />
        });
      }

      componentWillUpdate(nextProps) {
        this.context.updateInjector({
          injectionId: into.injectionId,
          injectorId,
          injectorInstance: this,
          inject: () => <InjectionComponent {...nextProps} />
        });
      }

      componentWillUnmount() {
        this.context.removeInjector({
          injectionId: into.injectionId,
          injectorId,
          injectorInstance: this
        });
      }

      render() {
        return null;
      }
    }