import { Squid } from '@0xsquid/sdk';
import { PurchaseItem, TokenInfo } from '@imtbl/checkout-sdk';
import { createContext, useContext } from 'react';
import { Chain, Token, RouteData } from '../../../lib/squid/types';
import { OrderQuoteResponse, SignResponse } from '../../../lib/primary-sales';

export interface PurchaseState {
  id: string;
  squid: {
    squid: Squid | null;
    chains: Chain[] | null;
    tokens: Token[] | null;
  };
  items: PurchaseItem[];
  selectedToken: TokenInfo | undefined;
  selectedRouteData: RouteData | undefined;
  quote: OrderQuoteResponse | null;
  signResponse: SignResponse | undefined;
}

export const initialPurchaseState: PurchaseState = {
  id: '',
  squid: {
    squid: null,
    chains: null,
    tokens: null,
  },
  items: [],
  selectedToken: undefined,
  selectedRouteData: undefined,
  quote: null,
  signResponse: undefined,
};

export interface PurchaseContextState {
  purchaseState: PurchaseState;
  purchaseDispatch: React.Dispatch<PurchaseAction>;
}

export interface PurchaseAction {
  payload: ActionPayload;
}

type ActionPayload =
  | SetId
  | SetSquid
  | SetItems
  | SetSelectedToken
  | SetSelectedRouteData
  | SetSquidChains
  | SetSquidTokens
  | SetItems
  | SetQuote
  | SetSignResponse;

export enum PurchaseActions {
  SET_ID = 'SET_ID',
  SET_SQUID = 'SET_SQUID',
  SET_SQUID_CHAINS = 'SET_SQUID_CHAINS',
  SET_SQUID_TOKENS = 'SET_SQUID_TOKENS',
  SET_ITEMS = 'SET_ITEMS',
  SET_SELECTED_TOKEN = 'SET_SELECTED_TOKEN',
  SET_SELECTED_ROUTE_DATA = 'SET_SELECTED_ROUTE_DATA',
  SET_QUOTE = 'SET_QUOTE',
  SET_SIGN_RESPONSE = 'SET_SIGN_RESPONSE',
}

export interface SetId {
  type: PurchaseActions.SET_ID;
  id: string;
}

export interface SetSquid {
  type: PurchaseActions.SET_SQUID;
  squid: Squid;
}

export interface SetSquidChains {
  type: PurchaseActions.SET_SQUID_CHAINS;
  chains: Chain[];
}

export interface SetSquidTokens {
  type: PurchaseActions.SET_SQUID_TOKENS;
  tokens: Token[];
}

export interface SetItems {
  type: PurchaseActions.SET_ITEMS;
  items: PurchaseItem[];
}

export interface SetSignResponse {
  type: PurchaseActions.SET_SIGN_RESPONSE;
  signResponse: SignResponse;
}

export interface SetSelectedToken {
  type: PurchaseActions.SET_SELECTED_TOKEN;
  selectedToken: TokenInfo;
}

export interface SetSelectedRouteData {
  type: PurchaseActions.SET_SELECTED_ROUTE_DATA;
  selectedRouteData: RouteData;
}

export interface SetQuote {
  type: PurchaseActions.SET_QUOTE;
  quote: OrderQuoteResponse;
}

// eslint-disable-next-line @typescript-eslint/naming-convention
export const PurchaseContext = createContext<PurchaseContextState>({
  purchaseState: initialPurchaseState,
  purchaseDispatch: () => {},
});

PurchaseContext.displayName = 'PurchaseContext';

export type Reducer<S, A> = (prevState: S, action: A) => S;

export const purchaseReducer: Reducer<PurchaseState, PurchaseAction> = (
  state: PurchaseState,
  action: PurchaseAction,
) => {
  switch (action.payload.type) {
    case PurchaseActions.SET_ID:
      return {
        ...state,
        id: action.payload.id,
      };
    case PurchaseActions.SET_SQUID:
      return {
        ...state,
        squid: {
          ...state.squid,
          squid: action.payload.squid,
        },
      };
    case PurchaseActions.SET_SQUID_CHAINS:
      return {
        ...state,
        squid: {
          ...state.squid,
          chains: action.payload.chains,
        },
      };
    case PurchaseActions.SET_SQUID_TOKENS:
      return {
        ...state,
        squid: {
          ...state.squid,
          tokens: action.payload.tokens,
        },
      };
    case PurchaseActions.SET_ITEMS:
      return {
        ...state,
        items: action.payload.items,
      };
    case PurchaseActions.SET_SELECTED_TOKEN:
      return {
        ...state,
        selectedToken: action.payload.selectedToken,
      };
    case PurchaseActions.SET_SELECTED_ROUTE_DATA:
      return {
        ...state,
        selectedRouteData: action.payload.selectedRouteData,
      };
    case PurchaseActions.SET_QUOTE:
      return {
        ...state,
        quote: action.payload.quote,
      };
    case PurchaseActions.SET_SIGN_RESPONSE:
      return {
        ...state,
        signResponse: action.payload.signResponse,
      };
    default:
      return state;
  }
};

export function usePurchaseContext() {
  return useContext(PurchaseContext);
}