import { FlowfulError } from './error';

export const ACTION_MODE_SINGLE = 'single';
export type ActionModeSingle = typeof ACTION_MODE_SINGLE;

export const ACTION_MODE_MULTI = 'multi';
export type ActionModeMulti = typeof ACTION_MODE_MULTI;

export type ActionMode = ActionModeSingle | ActionModeMulti;

export interface IAction<Payload = any, Mode extends ActionMode = ActionMode, CreatorArguments extends any[] = any[]> {
  readonly type: string;
  readonly mode: Mode;
  readonly creator: IActionCreator<Payload, Mode, CreatorArguments>;
  readonly payload: Payload;
}

export type Payload<T> = T extends IActionCreator<infer P> ? P : never;

export interface IActionCreator<
  Payload = any,
  Mode extends ActionMode = ActionMode,
  Arguments extends any[] = any[],
  A extends IAction<Payload, Mode, Arguments> = IAction<Payload, Mode, Arguments>
> {
  (...args: Arguments): A;
  readonly type: string;
  readonly mode: Mode;
}

export function actionCreatorSingle<Payload = any>(type: string): IActionCreator<Payload, ActionModeSingle, [Payload]> {
  function ActionCreator(payload: Payload) {
    return {
      type,
      mode: ACTION_MODE_SINGLE as ActionModeSingle,
      payload,
      creator: ActionCreator,
    };
  }
  ActionCreator.type = type;
  ActionCreator.mode = ACTION_MODE_SINGLE as ActionModeSingle;
  return ActionCreator;
}

export function actionCreatorMulti<Payload = any>(type: string): IActionCreator<Payload, ActionModeMulti, [Payload]> {
  function ActionCreator(payload: Payload) {
    return {
      type,
      mode: ACTION_MODE_MULTI as ActionModeMulti,
      payload,
      creator: ActionCreator,
    };
  }
  ActionCreator.type = type;
  ActionCreator.mode = ACTION_MODE_MULTI as ActionModeMulti;
  return ActionCreator;
}

export function actionCreator<Payload = any>(
  type: string,
  mode?: ActionModeSingle
): IActionCreator<Payload, ActionModeSingle, [Payload]>;
export function actionCreator<Payload = any>(
  type: string,
  mode: ActionModeMulti
): IActionCreator<Payload, ActionModeMulti, [Payload]>;
export function actionCreator<Payload = any>(
  type: string,
  mode: ActionMode = ACTION_MODE_SINGLE
): IActionCreator<Payload, ActionMode, [Payload]> {
  if (mode === ACTION_MODE_SINGLE) {
    return actionCreatorSingle<Payload>(type);
  } else if (mode === ACTION_MODE_MULTI) {
    return actionCreatorMulti<Payload>(type);
  } else {
    throw new FlowfulError(
      `[Flowful][actionCreator] Invalid mode: "${mode}" for action "${type}". Must be "${ACTION_MODE_SINGLE}" or "${ACTION_MODE_MULTI}".`,
      { type, mode }
    );
  }
}

export function actionCreatorDefault<Payload = any>(
  type: string,
  defaultPayload: Payload,
  mode?: ActionModeSingle
): IActionCreator<Payload, ActionModeSingle, [Payload] | []>;
export function actionCreatorDefault<Payload = any>(
  type: string,
  defaultPayload: Payload,
  mode: ActionModeMulti
): IActionCreator<Payload, ActionModeMulti, [Payload] | []>;
export function actionCreatorDefault<Payload = any, Mode extends ActionMode = ActionModeSingle>(
  type: string,
  defaultPayload: Payload,
  mode?: Mode
): IActionCreator<Payload, Mode, [Payload] | []> {
  if (mode === undefined) {
    mode = ACTION_MODE_SINGLE as Mode;
  }
  if (mode !== ACTION_MODE_SINGLE && mode !== ACTION_MODE_MULTI) {
    throw new FlowfulError(
      `[Flowful][actionCreatorDefault] Invalid action mode: "${mode}" for action "${type}". Must be "${ACTION_MODE_SINGLE}" or "${ACTION_MODE_MULTI}".`,
      { type, mode }
    );
  }

  function ActionCreatorDefault(payload?: Payload) {
    return {
      type,
      mode,
      payload: payload !== undefined ? payload : defaultPayload,
      creator: ActionCreatorDefault,
    };
  }
  ActionCreatorDefault.type = type;
  ActionCreatorDefault.mode = mode;
  return ActionCreatorDefault as IActionCreator<Payload, Mode, any[]>;
}