/* eslint-disable @typescript-eslint/no-explicit-any */
import { get } from 'lodash-es';
import { ModelTypes } from '@/zeus/index.js';
import { FieldFrontType } from '@/graphql/core/selectors.js';

type RecordFieldsAll = Record<string, any>;
type FieldWithVisuals = Omit<ModelTypes['CMSField'], 'visual'> & {
  visual?: {
    [P in keyof ModelTypes['Visual']]: ModelTypes['Visual'][P] extends infer R | null ? R : ModelTypes['Visual'][P];
  };
};

const fieldsToObject = (fields: ModelTypes['CMSField'][]): RecordFieldsAll => {
  return Object.fromEntries(
    fields.map((f) => {
      return [
        f.name,
        {
          ...(f.fields ? fieldsToObject(f.fields) : undefined),
          __field: f,
        },
      ];
    }),
  );
};

export const useHusar = () => {
  const cmsFields =
    (fields: ModelTypes['CMSField'][]) =>
    <T>(query: FlattenKeys<T>) => {
      const obj = fieldsToObject(fields);
      return noNulls(get(obj, `${String(query)}.__field`)) as FieldWithVisuals;
    };
  return {
    cmsFields,
  };
};
type TrimDot<S extends string> = S extends `${infer R}.` ? R : S;
export type RemoveArrays<T> = T extends Array<infer R> ? R : T;

export type FlattenKeysStart<T> = T extends object
  ? {
      [K in keyof T]-?: T[K] extends infer U
        ? U extends undefined
          ? never
          : `${Extract<K, string>}${FlattenKeysStart<RemoveArrays<U>> extends '' ? '' : '.'}${FlattenKeysStart<RemoveArrays<U>>}`
        : never;
    }[keyof T]
  : '';

export type FlattenKeys<T> = TrimDot<FlattenKeysStart<T>>;

export type RemoveNullKeys<T> = {
  [K in keyof T]: null extends T[K] ? never : K;
}[keyof T];
type CleanedType<T> = Pick<T, RemoveNullKeys<T>>;

export const noNulls = <T extends Record<string, any>>(t?: T) => {
  if (!t) return t;
  return Object.fromEntries(Object.entries(t).filter(([, v]) => v !== null)) as CleanedType<T>;
};

// type Example = FlattenedKeyType<ModelTypes['homepage'], 'components.integrations_section.combo.logo1'>;
export type FlattenedKeyType<T, P extends string> = T extends (infer E)[]
  ? FlattenedKeyType<E, P>
  : P extends `${infer K}.${infer Rest}`
    ? K extends keyof T
      ? FlattenedKeyType<T[K], Rest> extends infer R
        ? undefined extends T[K]
          ? R | undefined
          : R
        : never
      : never
    : P extends keyof T
      ? T[P]
      : never;

export type ComponentsOverride<T> = Partial<{
  [P in FlattenKeys<T>]: (props: {
    attrs: {
      [P in keyof ModelTypes['Visual']]: ModelTypes['Visual'][P] extends infer R | null ? R : ModelTypes['Visual'][P];
    };
    data?: FlattenedKeyType<T, P> | null;
    component?: React.JSX.Element;
    f: FieldFrontType;
  }) => React.JSX.Element;
}>;

export type AttributesOverride<T> = Partial<{
  [P in FlattenKeys<T>]: <Y, X = false>(props: {
    attrs: {
      [P in keyof ModelTypes['Visual']]: ModelTypes['Visual'][P] extends infer R | null ? R : ModelTypes['Visual'][P];
    };
    data?: X extends true
      ? FlattenedKeyType<T, P> extends (infer E)[] | undefined | null
        ? E | null
        : FlattenedKeyType<T, P> | null
      : FlattenedKeyType<T, P> | null;
  }) => React.DetailedHTMLProps<React.HTMLAttributes<Y>, Y>;
}>;