/* eslint-disable @typescript-eslint/no-explicit-any */
import * as D from "@effect-ts/core/Collections/Immutable/Dictionary"
import type { None, Some } from "@effect-ts/core/Option"
import type { ComputeFlat, UnionToIntersection } from "@effect-ts/core/Utils"

import { array, prop, props } from "./_schema.js"
import type * as MO from "./_schema.js"
import { positiveInt } from "./custom.js"
import { optProp } from "./ext.js"

type AdaptSchema<Props extends MO.PropertyRecord, Key extends keyof Props> = {
  [K in Key]: Props[K]
}

// TODO: adapt error types too; low prio
const adaptedSchema = <Props extends MO.PropertyRecord>(properties: Props) =>
  <Key extends keyof Props>(keys: readonly Key[]): AdaptSchema<Props, Key> =>
    D.filterWithIndex_(properties, key => keys.includes(key as Key)) as any

// TODO: keep existing fields
export const adaptRes = <Props extends MO.PropertyRecord>(properties: Props) => {
  const adapt = adaptedSchema(properties)
  return <Key extends keyof Props>(keys: readonly Key[]) =>
    props({
      items: prop(array(props(adapt(keys)))),
      // TODO: hide count when not asked for $count. and demand non-opt count, when asked.
      count: optProp(positiveInt)
    })
}

export type Adapted<
  Props extends MO.PropertyRecord,
  Key extends keyof Props
> = /* copy pasted from return type of function */ MO.SchemaProperties<{
  items: MO.Property<
    MO.SchemaDefaultSchema<
      unknown,
      readonly ComputeFlat<
        UnionToIntersection<
          {
            [k in keyof AdaptSchema<Props, Key>]: AdaptSchema<
              Props,
              Key
            >[k] extends MO.AnyProperty ? AdaptSchema<Props, Key>[k]["_optional"] extends "optional" ? {
              readonly [h in k]?:
                | MO.ParsedShapeOf<AdaptSchema<Props, Key>[k]["_schema"]>
                | undefined
            }
            : {
              readonly [h in k]: MO.ParsedShapeOf<
                AdaptSchema<Props, Key>[k]["_schema"]
              >
            }
              : never
          }[Key]
        >
      >[],
      readonly ComputeFlat<
        UnionToIntersection<
          {
            [k in keyof AdaptSchema<Props, Key>]: AdaptSchema<
              Props,
              Key
            >[k] extends MO.AnyProperty ? AdaptSchema<Props, Key>[k]["_optional"] extends "optional" ? {
              readonly [h in k]?:
                | MO.ParsedShapeOf<AdaptSchema<Props, Key>[k]["_schema"]>
                | undefined
            }
            : {
              readonly [h in k]: MO.ParsedShapeOf<
                AdaptSchema<Props, Key>[k]["_schema"]
              >
            }
              : never
          }[Key]
        >
      >[],
      readonly ComputeFlat<
        UnionToIntersection<
          {
            [k in keyof AdaptSchema<Props, Key>]: AdaptSchema<
              Props,
              Key
            >[k] extends MO.AnyProperty ? AdaptSchema<Props, Key>[k]["_optional"] extends "optional" ? {
              readonly [
                h in AdaptSchema<Props, Key>[k]["_as"] extends Some<any> ? AdaptSchema<Props, Key>[k]["_as"]["value"]
                  : k
              ]?:
                | MO.EncodedOf<AdaptSchema<Props, Key>[k]["_schema"]>
                | undefined
            }
            : {
              readonly [
                h in AdaptSchema<Props, Key>[k]["_as"] extends Some<any> ? AdaptSchema<Props, Key>[k]["_as"]["value"]
                  : k
              ]: MO.EncodedOf<AdaptSchema<Props, Key>[k]["_schema"]>
            }
              : never
          }[Key]
        >
      >[],
      // eslint-disable-next-line @typescript-eslint/ban-types
      {}
    >,
    "required",
    None,
    None
  >
}>