import {
  AddCustomTokenAction,
  CustomTokenAction,
  RemoveCustomTokenAction
} from 'actions/customTokens';
import { TypeKeys } from 'actions/customTokens/constants';
import { Token } from 'types/network';

export type State = Token[];

export const INITIAL_STATE: State = [];

function addCustomToken(state: State, action: AddCustomTokenAction): State {
  if (state.find(token => token.symbol === action.payload.symbol)) {
    return state;
  }
  return [...state, action.payload];
}

function removeCustomToken(state: State, action: RemoveCustomTokenAction): State {
  return state.filter(token => token.symbol !== action.payload);
}

export function customTokens(state: State = INITIAL_STATE, action: CustomTokenAction): State {
  switch (action.type) {
    case TypeKeys.CUSTOM_TOKEN_ADD:
      return addCustomToken(state, action);
    case TypeKeys.CUSTOM_TOKEN_REMOVE:
      return removeCustomToken(state, action);
    default:
      return state;
  }
}