/* eslint-disable @typescript-eslint/no-explicit-any */
import { Unpacked } from "types/utility-types";

type CustomEnum<Keys extends string, Y = unknown> = Record<Keys, string> & {
  is: (value: string) => boolean;
  _all: Array<Keys>;
} & Y;

export function makeEnum<
  Values extends string,
  OtherProps extends Record<string, unknown>,
>(
  values: Array<Values> | Record<string, Values>,
  otherProps: OtherProps,
): CustomEnum<Unpacked<Values>, OtherProps> | CustomEnum<Values, OtherProps> {
  let enumObj: any;
  let set: any;

  if (Array.isArray(values)) {
    enumObj = {};
    for (const value of values) {
      enumObj[value] = value;
    }
    set = new Set(values);
    enumObj._all = values;
  } else {
    enumObj = values;
    set = new Set(Object.values(values));
    enumObj._all = Array.from(set);
  }

  enumObj.is = (value: string): boolean => {
    return set.has(value);
  };

  if (otherProps) {
    for (const prop in otherProps) {
      enumObj[prop] = otherProps[prop];
    }
  }

  return enumObj;
}