import {
  Dynadux,
  IDynaduxConfig,
  TDynaduxReducerDispatch,
  IDynaduxReducerDic,
} from "../Dynadux/Dynadux";
import {EventEmitter} from "../tools/EventEmitter";
import {createSection} from "./createSection";

export interface ICreateStoreConfig<TState> extends IDynaduxConfig<TState> {
}

export interface ICreateStoreAPI<TState = any> {
  state: TState;
  dispatch: TDynaduxReducerDispatch;

  createSection: <TSectionState>(createSectionConfig: ICreateSectionConfig<TSectionState>) => ICreateSectionAPI<TState, TSectionState>;

  addChangeEventListener: (cb: (sectionState: TState, action: string, payload?: any) => void) => void;
  removeChangeEventListener: (cb: (sectionState: TState, action: string, payload?: any) => void) => void;

  provider: IStoreProviderAPI<TState>;
}

export interface IStoreProviderAPI<TState> {
  store: ICreateStoreAPI<TState>;
  addChangeEventListener: (cb: (storeState: TState, action: string, payload?: any) => void) => void;
  removeChangeEventListener: (cb: (storeState: TState, action: string, payload?: any) => void) => void;
}

export interface ICreateSectionConfig<TSectionState> {
  section: string;
  initialState: TSectionState;
  reducers: IDynaduxReducerDic<TSectionState>;
  onChange?: (sectionState: TSectionState, action: string, payload?: any) => void;
}

export interface ICreateSectionAPI<TState, TSectionState> {
  storeState: TState;
  state: TSectionState;
  dispatch: TDynaduxReducerDispatch;
  addChangeEventListener: (cb: (sectionState: TSectionState, action: string, payload?: any) => void) => void;
  removeChangeEventListener: (cb: (sectionState: TSectionState, action: string, payload?: any) => void) => void;
}

export const createStore = <TState = any>(config?: ICreateStoreConfig<TState>): ICreateStoreAPI<TState> => {
  const dynadux = new Dynadux<TState>(config);
  const storeChangeEventEmitter = new EventEmitter();

  const dynaduxOnChange = dynadux._onChange;
  dynadux._onChange = ((state: TState, action: string, payload?: any) => {
    storeChangeEventEmitter.trigger(state, action, payload);
    dynaduxOnChange(state, action, payload);
  });

  const store: ICreateStoreAPI<TState> = {
    get state() {
      return dynadux.state;
    },

    dispatch: dynadux.dispatch,

    addChangeEventListener: (cb) => storeChangeEventEmitter.addEventListener(cb),
    removeChangeEventListener: (cb) => storeChangeEventEmitter.removeEventListener(cb),

    provider: {
      get store() {
        return store;
      },
      addChangeEventListener: (cb) => storeChangeEventEmitter.addEventListener(cb),
      removeChangeEventListener: (cb) => storeChangeEventEmitter.removeEventListener(cb),
    },

    createSection: <TSectionState>(createSectionConfig: ICreateSectionConfig<TSectionState>): ICreateSectionAPI<TState, TSectionState> => {
      return createSection<TState, TSectionState>({
        dynadux,
        createSectionConfig,
      });
    },
  };

  return store;
};