import {
  EnumTypeDefinitionNode,
  EnumTypeExtensionNode,
  GraphQLEnumType,
} from 'graphql';
import { Maybe } from 'graphql/jsutils/Maybe';

interface GraphqlEnumTypeConfig<T = object> {
  name: string;
  description?: Maybe<string>;
  values: T;
  extensions?: Maybe<Readonly<Record<string, never>>>;
  astNode?: Maybe<EnumTypeDefinitionNode>;
  extensionASTNodes?: Maybe<ReadonlyArray<EnumTypeExtensionNode>>;
}

export const enumToGraphqlEnum = <T>(values: T) =>
  Object.keys(values)
    .filter((value) => isNaN(Number(value)) === false)
    .reduce((acc, value) => {
      acc[(values as unknown)[value]] = { value: Number(value) };
      return acc;
    }, {} as T);

export class GraphqlEnumType<T> extends GraphQLEnumType {
  constructor(config: GraphqlEnumTypeConfig<T>) {
    config.values = enumToGraphqlEnum(config.values);
    super(config as never);
  }
}