import type {
  FieldChoiceWithLabel,
  PlainInputField,
} from './schemas.generated';
import type { ZObject, Bundle } from './custom';
import type { FromSchema, JSONSchema } from 'json-schema-to-ts';

// #region UTILITIES

/**
 * Squash complex intersections into a flat type. Improves readability
 * of the generated types.
 *
 * @example
 * type result = Simplify<{ a: string } & { b: boolean }>;
 * // { a: string, b: boolean }
 */
type Simplify<T> = T extends infer O ? { [K in keyof O]: O[K] } : never;

/**
 * Determine if an array type is generic, like T[]. This lets us
 * distinguish between tuples that have positional information, and a
 * generic array that treats all elements the same.
 *
 * @example
 * type result1 = JustArray<string[]>;
 * // true
 *
 * type result2 = JustArray<[string, number]>;
 * // false
 */
type JustArray<T> = T extends unknown[]
  ? T extends readonly [unknown, ...unknown[]]
    ? false
    : true
  : false;

/**
 * @link https://github.com/type-challenges/type-challenges/issues/737
 * @example
 * type result = UnionToIntersection<{ foo: string } | { bar: string }>;
 * // { foo: string } & { bar: string }
 */
type UnionToIntersection<U> = (
  U extends unknown ? (arg: U) => 0 : never
) extends (arg: infer I) => 0
  ? I
  : never;

/**
 * @link https://github.com/type-challenges/type-challenges/issues/737
 * @example
 * type result = LastInUnion<1 | 2>;
 * // 2
 */
type LastInUnion<U> =
  UnionToIntersection<U extends unknown ? (x: U) => 0 : never> extends (
    x: infer L,
  ) => 0
    ? L
    : never;

/**
 * @link https://github.com/type-challenges/type-challenges/issues/737
 * @example
 * type result = UnionToTuple<1 | 2>;
 * // [1, 2]
 */
type UnionToTuple<U, Last = LastInUnion<U>> = [U] extends [never]
  ? []
  : [...UnionToTuple<Exclude<U, Last>>, Last];

/**
 * Flatten a tuple of tuples into a single tuple.
 *
 * @example
 * type result = Flatten<[[1, 2], [3, 4]]>;
 * // [1, 2, 3, 4]
 */
type Flatten<S extends unknown[], T extends unknown[] = []> = S extends [
  infer X,
  ...infer Y,
]
  ? X extends unknown[]
    ? Flatten<[...X, ...Y], T>
    : Flatten<[...Y], [...T, X]>
  : T;

/**
 * Merge an array (tuple) of object types into a single type. This is
 * the critical step in going from list of field definitions to a
 * bundle's inputData object.
 *
 * @example
 * type result = Merge<[{ a: string }, { b: boolean }]>
 * // { a: string, b: boolean }
 */
type Merge<T extends object[]> = T extends [infer F, ...infer R]
  ? F & Merge<Extract<R, object[]>>
  : {};

// #endregion

// MAIN BITS
// =========

/**
 * Any value that is valid in JSON, including primitives. Used for
 * nested values inside a json field's object or array.
 */
export type JsonValue =
  | string
  | number
  | boolean
  | null
  | JsonValue[]
  | { [key: string]: JsonValue };

/**
 * The root-level type of a `json` input field. Per the platform's
 * functional constraint, a json field's root schema.type must be
 * "object" or "array" — bare primitives are not allowed at the top
 * level. Nested values inside can be any JsonValue.
 */
export type JsonFieldValue = JsonValue[] | { [key: string]: JsonValue };

type InputDataConstraint = Record<string, unknown>;
type InputDataDefault = Record<string, undefined>;

/**
 * All of the field types a Zapier field can have, as defined by
 * zapier-platform-schema.
 */
type SchemaFieldTypes = NonNullable<PlainInputField['type']>;

/**
 * Lookup of zapier field types (as defined by zapier-platform-schema)
 * to their corresponding TypeScript types. Dates and files are just
 * strings.
 */
type FieldResultTypes = {
  [$FieldType in SchemaFieldTypes]: $FieldType extends
    | 'string'
    | 'text'
    | 'datetime'
    | 'file'
    | 'password'
    | 'code'
    ? string
    : $FieldType extends 'number' | 'integer'
      ? number
      : $FieldType extends 'boolean'
        ? boolean
        : $FieldType extends 'json'
          ? JsonFieldValue
          : never; // Ignore `copy` and any other non-value types
};

/**
 * Get the TypeScript type that corresponds to the zapier field type. If
 * `type` is not set, the field defaults to `string`. Does not pay
 * attention to `list` `, `dict`, or `required` statuses. Does not work
 * for fields with `children` subfields either.
 *
 * @example
 * type result = PrimitiveFieldResultType<{ }>;
 * // string (string is default when `type` is not set)
 *
 * type result = PrimitiveFieldResultType<{ type: 'string' }>;
 * // string
 *
 * type result = PrimitiveFieldResultType<{ type: 'number' }>;
 * // number
 *
 * type result = PrimitiveFieldResultType<{ type: 'boolean' }>;
 * // boolean
 */
type PrimitiveFieldResultType<$Field extends PlainInputField> = $Field extends {
  type: infer $T extends PlainInputField['type'];
}
  ? $T extends string
    ? $T extends 'json'
      ? $Field extends { schema: infer $S extends JSONSchema }
        ? FromSchema<$S>
        : JsonFieldValue
      : FieldResultTypes[$T]
    : string
  : string;

/**
 * Capture a union of string literals while also allowing the string
 * type. This allows the available choices to show in autocomplete,
 * while also allowing arbitrary strings to still be valid.
 *
 * @example
 * type result = StringHints<'a' | 'b'>;
 * const A: result = 'a'; // Ok and autocomplete shows 'a' and 'b'.
 * const C: result = 'c'; // Any string still ok.
 */
export type StringHints<S> = S | (string & {});

type PrimitiveFieldResultTypesWithChoiceHints<$Field extends PlainInputField> =
  $Field extends { choices: infer $Choices }
    ? $Choices extends Record<string, string>
      ? StringHints<keyof $Choices>
      : $Choices extends FieldChoiceWithLabel[]
        ? StringHints<$Choices[number]['value']>
        : $Choices extends string[]
          ? StringHints<$Choices[number]>
          : PrimitiveFieldResultType<$Field>
    : PrimitiveFieldResultType<$Field>;

/**
 * A function that returns a list of plain fields, sync or async.
 * Can be used as a member of an array of input fields itself.
 */
export type InputFieldFunction<
  $InputData extends InputDataConstraint = InputDataDefault,
> = (
  z: ZObject,
  bundle: Bundle<$InputData>,
) => PlainInputField[] | Promise<PlainInputField[]>;

/**
 * Input fields can be plain fields, or functions that return plain
 * fields, async or not.
 */
export type InputField<
  $InputData extends InputDataConstraint = InputDataDefault,
> = PlainInputField | InputFieldFunction<$InputData>;
export type InputFields = InputField[];

/**
 * Extract the "contribution" of a plain field to the bundle. Just the
 * field's key mapped to its corresponding TypeScript type. The result
 * will be optionally defined if the `required` flag is false or
 * omitted. Fields that have an undefined value type will be removed
 * from the type, and thus contribute nothing to the bundle.
 *
 * @example
 * type result1 = PlainFieldContribution<{ key: "a"; type: "string", required: true }>;
 * // { a: string }
 *
 * type result2 = PlainFieldContribution<{ key: "b"; type: "integer", required: false }>;
 * // { b?: number | undefined }
 *
 * type result3 = PlainFieldContribution<{ key: "c"; type: "boolean" }>;
 * // { c?: boolean | undefined }
 */
export type PlainFieldContribution<$Field extends PlainInputField> =
  $Field extends { children: PlainInputField[] }
    ? ParentFieldContribution<$Field>
    : $Field extends { dict: true }
      ? DictFieldContribution<$Field>
      : $Field extends { list: true }
        ? ListFieldContribution<$Field>
        : PrimitiveFieldContribution<$Field>;

/**
 * Extract the contribution of a parent field to the input data. A parent
 * Field has a `children` field array. The parent's own `key` will be
 * ignored, and the children's contributions will be merged into the top
 * level inputData object.
 *
 * @example
 * type result = ParentFieldContribution<{ key: "a"; children: [{ key: "b"; required: true }] }>;
 * // { b: string }
 */
type ParentFieldContribution<
  $Field extends PlainInputField & { children: PlainInputField[] },
> = Simplify<
  PlainFieldArrayContribution<$Field['children']> &
    // When line items are mapped into any child input, then the parent will also
    // be included as an array of objects containing the children's inputs'
    // contributions. This may or may not apply on a per-zap basis, so this parent
    // array is an optional member of inputData.
    Partial<
      Record<
        $Field['key'],
        Array<PlainFieldArrayContribution<$Field['children']>>
      >
    >
>;

/**
 * Extract the contribution of a dictionary field to the input data. A
 * dictionary field has a `dict:true` field. The type for this `key`
 * field in the inputData object will therefore be a record of the key.
 * Currently, the value type is always `string`, but this may change in
 * the future.
 *
 * @see https://zapier.atlassian.net/browse/PDE-6547 for when non-string
 * values will be supported.
 *
 * @example
 * type result = DictFieldContribution<{ key: "a"; dict: true, required: true }>;
 * // { a: Record<string, string> }
 */
type DictFieldContribution<$Field extends PlainInputField & { dict: true }> =
  $Field extends { required: true }
    ? Record<$Field['key'], Record<string, string>>
    : Partial<Record<$Field['key'], Record<string, string>>>;

/**
 * Extract the contribution of a list field to the input data. A list
 * field has a `list:true` field. The type for this `key` field in the
 * inputData object will therefore be an array of the value type.
 *
 * @example
 * type result1 = ListFieldContribution<{ key: "a"; list: true, required: true }>;
 * // { a: string[] }
 *
 * type result2 = ListFieldContribution<{ key: "a"; list: true; type: "integer" }>;
 * // { a?: number[] | undefined }
 */
type ListFieldContribution<$Field extends PlainInputField & { list: true }> =
  $Field extends { required: true }
    ? Record<$Field['key'], PrimitiveFieldResultTypesWithChoiceHints<$Field>[]>
    : Partial<
        Record<
          $Field['key'],
          PrimitiveFieldResultTypesWithChoiceHints<$Field>[]
        >
      >;

/**
 * Extract the contribution of a primitive field to the input data. A
 * primitive field is a PlainInputField that is not a parent, dict, or
 * list field. The `type` field MAY be set, but will default to `string`.
 *
 * @example
 * type result1 = PrimitiveFieldContribution<{ key: "a" }>;
 * // { a?: string | undefined }
 *
 * type result2 = PrimitiveFieldContribution<{ key: "a"; type: "integer", required: true }>;
 * // { a: number }
 */
type PrimitiveFieldContribution<$Field extends PlainInputField> =
  PrimitiveFieldResultTypesWithChoiceHints<$Field> extends never
    ? {}
    : $Field extends { required: true }
      ? Record<$Field['key'], PrimitiveFieldResultTypesWithChoiceHints<$Field>>
      : Partial<
          Record<
            $Field['key'],
            PrimitiveFieldResultTypesWithChoiceHints<$Field>
          >
        >;

/**
 * Extract the contribution of multiple plain fields defined in an
 * array (tuple) to the bundle.
 *
 * @example
 * type result = PlainFieldArrayContribution<[
 *   { key: "a"; type: "string", required: true },
 *   { key: "b"; type: "integer", required: false }
 * ]>;
 * // { a: string, b?: number | undefined }
 */
type PlainFieldArrayContribution<$Fields extends PlainInputField[]> = Simplify<
  Merge<{
    [K in keyof $Fields]: PlainFieldContribution<$Fields[K]>;
  }>
>;

/**
 * A function that returns a list of plain fields, sync or async.
 *
 * If the fields this function returns are known, they will be included
 * in the resulting bundle's inputs as optional. Otherwise the
 * contribution  of this function's fields will be `Record<string,
 * unknown>`, because the field IDs can't be known ahead of time.
 */
type FieldFunction<
  $InputData extends InputDataConstraint = InputDataDefault,
  $Output extends PlainInputField[] = PlainInputField[],
> = (z: ZObject, bundle: Bundle<$InputData>) => $Output | Promise<$Output>;

/**
 * Get all possible fields a field function MAY return. This is the
 * union of all possible fields, combined and flattened into tuple form
 * that can be normalised into a bundle contribution object.
 */
type FieldFunctionResult<
  $Func extends (...args: never) => InputField[] | Promise<InputField[]>,
> = Flatten<UnionToTuple<Awaited<ReturnType<$Func>>>>;

/**
 * Get the bundle contribution of a field function that has known field
 * results. All fields returned by functions are considered optional, so
 * each contributed field's presence in the bundle becomes optional.
 *
 * @example
 * const fieldFunction = (async (z, bundle) => {
 *   if (bundle.inputData.isAdmin) {
 *     return [{ key: "number_field", type: "number", required: true }] as const;
 *   }
 *   return [{ key: "text_field", type: "text", required: true }] as const;
 * }) satisfies FieldFunction<{ isAdmin: boolean }>;
 *
 * type result = FieldFunctionContribution<typeof fieldFunction>;
 * // {
 * //   number_field?: number | undefined;
 * //   text_field?: string | undefined;
 * // }
 */
type KnownFieldFunctionContribution<
  $Func extends (...args: never) => InputField[] | Promise<InputField[]>,
> =
  FieldFunctionResult<$Func> extends PlainInputField[]
    ? Partial<PlainFieldArrayContribution<FieldFunctionResult<$Func>>>
    : never;

/**
 * Get the contribution of a field function. If the field function
 * returns known named fields, extract them and their types for the
 * bundle. Otherwise, return a record of unknown key/value pairs because
 * we can't know what the fields will be ahead of time.
 *
 * @example
 * const knownFieldFunction = ((z, bundle) => [
 *   { key: "field_a", type: "text", required: true },
 *   { key: "field_b", type: "integer", required: false },
 * ]) satisfies FieldFunction;
 *
 * type result = FieldFunctionContribution<typeof knownFieldFunction>;
 * // {
 * //   field_a?: string | undefined;
 * //   field_b?: number | undefined;
 * // }
 *
 * const unknownFieldFunction = (async (z, bundle) => {
 *   return [] as SchemaInputField[];
 * }) satisfies FieldFunction;
 *
 * type result = FieldFunctionContribution<typeof unknownFieldFunction>;
 * // Record<string, unknown>
 */
type FieldFunctionContribution<$F> = $F extends (
  ...args: never
) => InputField[] | Promise<InputField[]>
  ? Awaited<ReturnType<$F>> extends InputField[]
    ? JustArray<Awaited<ReturnType<$F>>> extends true
      ? Record<string, unknown> // Unknown fields
      : KnownFieldFunctionContribution<$F> // Known fields
    : never
  : never;

// PUTTING IT ALL TOGETHER
// =======================
//
// These last types are where everything comes together. Note that in
// this module, the more precise term "Contribution" is used to describe
// the fields that a plain field or field function will contribute to
// the bundle, but the more straightforward term "InputData" is
// exposed to for the public API.
//
// TERMINOLOGY
// -----------
// InputField: A PlainInputField or a FieldFunction.
// FieldFunction: A function that returns an array of InputFields.
// KnownFieldFunction: A FieldFunction that returns explicitly named InputFields.
// UnknownFieldFunction: A FieldFunction that returns an array of InputFields that are not explicitly named.
// PlainInputField: From the schema. Object with a key, and possibly `children`, `dict`, `list`, `type`, `required`, fields.
// ParentInputField: A PlainInputField with a `children` field array.
// DictInputField: A PlainInputField with a `dict:true` field.
// ListInputField: A PlainInputField with a `list:true` field.
// PrimitiveInputField: A PlainInputField with a `type` field that is not a parent, dict, or list field.

/**
 * Get the bundle contribution of a single field. This is either a plain
 * field, or a field function.
 */
type InferInputFieldContribution<$Input extends InputField> =
  $Input extends PlainInputField
    ? PlainFieldContribution<$Input>
    : $Input extends (...args: never) => InputField[] | Promise<InputField[]> // Conditional field function
      ? FieldFunctionContribution<$Input>
      : never;

/**
 * Get the shape of bundle.inputData, given the array of input fields.
 * This array can contain plain fields and field functions.
 */
export type InferInputData<$InputFields extends readonly InputField<any>[]> =
  Simplify<
    Merge<
      [
        ...{
          [K in keyof $InputFields]: InferInputFieldContribution<
            $InputFields[K]
          >;
        },
      ]
    >
  >;

/**
 * Helper function to simplify declaring a function with a set of
 * existing input fields.
 */
export type InputFieldFunctionWithInputs<$Inputs extends InputFields = []> = (
  z: ZObject,
  bundle: Bundle<InferInputData<$Inputs>>,
) => InputField[] | Promise<InputField[]>;