import { ReyaTheme } from '.';

type ReyaThemeGetterParams = {
  theme: ReyaTheme;
};

type ColorName = keyof ReyaTheme['colors'];

export const getColor =
  (color: ColorName) =>
  ({ theme }: ReyaThemeGetterParams): string =>
    theme.colors[color];

type BooleanMatcher = { false: ColorName; true: ColorName };
type Matcher<T extends string | number | symbol> = { [key in T]: ColorName };
type MatcherType<T> = Extract<T, 'string' | 'number' | 'symbol'>;
export const matchColor =
  <T extends ReyaThemeGetterParams, K extends keyof T, U = T[K]>(
    variable: K,
    matcher: U extends boolean ? BooleanMatcher : Matcher<MatcherType<U>>,
  ) =>
  (params: T) => {
    if (typeof params[variable] === 'boolean') {
      const booleanMatcher = matcher as BooleanMatcher;
      const color = params[variable] ? booleanMatcher.true : booleanMatcher.false;

      return getColor(color);
    }

    const color = (matcher as Matcher<MatcherType<U>>)[
      params[variable] as unknown as MatcherType<U>
    ];

    return getColor(color);
  };