import React, {Component} from 'react';
import {AppRegistry} from 'react-native';
import * as PropsStore from './PropsStore';
import * as ContainerStore from './ContainerStore';

export function registerContainer(containerKey, getContainerFunc) {
  const OriginalContainer = getContainerFunc();
  const NavigationContainer = wrapContainer(containerKey, OriginalContainer);
  ContainerStore.saveContainerClass(containerKey, NavigationContainer);
  AppRegistry.registerComponent(containerKey, () => NavigationContainer);
}

function wrapContainer(containerKey, OriginalContainer) {
  return class extends Component {
    constructor(props) {
      super(props);
      if (!props.screenId) {
        throw new Error(`Screen ${containerKey} does not have a screenId!`);
      }
      this.state = {
        screenId: props.screenId,
        allProps: {...props, ...PropsStore.getPropsForScreenId(props.screenId)}
      };
    }

    componentWillReceiveProps(nextProps) {
      this.setState({
        allProps: {...nextProps, ...PropsStore.getPropsForScreenId(this.state.screenId)}
      });
    }

    render() {
      return (
        <OriginalContainer {...this.state.allProps} screenId={this.state.screenId}/>
      );
    }
  };
}