import { DeepPartial } from '../../../types'
import { ValueSpec, ValueSpecUnion } from '../inputSpecTypes'
import {
  LazyBuild,
  InputSpec,
  ExtractInputSpecType,
  ExtractInputSpecStaticValidatedAs,
} from './inputSpec'
import { Parser, any, anyOf, literal, object } from 'ts-matches'

export type UnionRes<
  VariantValues extends {
    [K in string]: {
      name: string
      spec: InputSpec<any>
    }
  },
  K extends keyof VariantValues & string = keyof VariantValues & string,
> = {
  [key in keyof VariantValues]: {
    selection: key
    value: ExtractInputSpecType<VariantValues[key]['spec']>
    other?: {
      [key2 in Exclude<keyof VariantValues & string, key>]?: DeepPartial<
        ExtractInputSpecType<VariantValues[key2]['spec']>
      >
    }
  }
}[K]

export type UnionResStaticValidatedAs<
  VariantValues extends {
    [K in string]: {
      name: string
      spec: InputSpec<any>
    }
  },
  K extends keyof VariantValues & string = keyof VariantValues & string,
> = {
  [key in keyof VariantValues]: {
    selection: key
    value: ExtractInputSpecStaticValidatedAs<VariantValues[key]['spec']>
    other?: {
      [key2 in Exclude<keyof VariantValues & string, key>]?: DeepPartial<
        ExtractInputSpecStaticValidatedAs<VariantValues[key2]['spec']>
      >
    }
  }
}[K]

/**
 * Used in the the Value.select { @link './value.ts' }
 * to indicate the type of select variants that are available. The key for the record passed in will be the
 * key to the tag.id in the Value.select
```ts
 
export const disabled = InputSpec.of({});
export const size = Value.number({
  name: "Max Chain Size",
  default: 550,
  description: "Limit of blockchain size on disk.",
  warning: "Increasing this value will require re-syncing your node.",
  required: true,
  range: "[550,1000000)",
  integral: true,
  units: "MiB",
  placeholder: null,
});
export const automatic = InputSpec.of({ size: size });
export const size1 = Value.number({
  name: "Failsafe Chain Size",
  default: 65536,
  description: "Prune blockchain if size expands beyond this.",
  warning: null,
  required: true,
  range: "[550,1000000)",
  integral: true,
  units: "MiB",
  placeholder: null,
});
export const manual = InputSpec.of({ size: size1 });
export const pruningSettingsVariants = Variants.of({
  disabled: { name: "Disabled", spec: disabled },
  automatic: { name: "Automatic", spec: automatic },
  manual: { name: "Manual", spec: manual },
});
export const pruning = Value.union(
  {
    name: "Pruning Settings",
    description:
      '- Disabled: Disable pruning\n- Automatic: Limit blockchain size on disk to a certain number of megabytes\n- Manual: Prune blockchain with the "pruneblockchain" RPC\n',
    warning: null,
    default: "disabled",
  },
  pruningSettingsVariants
);
```
 */
export class Variants<
  VariantValues extends {
    [K in string]: {
      name: string
      spec: InputSpec<any, any>
    }
  },
> {
  private constructor(
    public build: LazyBuild<{
      spec: ValueSpecUnion['variants']
      validator: Parser<unknown, UnionRes<VariantValues>>
    }>,
    public readonly validator: Parser<
      unknown,
      UnionResStaticValidatedAs<VariantValues>
    >,
  ) {}
  readonly _TYPE: UnionRes<VariantValues> = null as any
  static of<
    VariantValues extends {
      [K in string]: {
        name: string
        spec: InputSpec<any>
      }
    },
  >(a: VariantValues) {
    const staticValidators = {} as {
      [K in keyof VariantValues]: Parser<
        unknown,
        ExtractInputSpecStaticValidatedAs<VariantValues[K]['spec']>
      >
    }
    for (const key in a) {
      const value = a[key]
      staticValidators[key] = value.spec.validator
    }
    const other = object(
      Object.fromEntries(
        Object.entries(staticValidators).map(([k, v]) => [k, any.optional()]),
      ),
    ).optional()
    return new Variants<VariantValues>(
      async (options) => {
        const validators = {} as {
          [K in keyof VariantValues]: Parser<
            unknown,
            ExtractInputSpecType<VariantValues[K]['spec']>
          >
        }
        const variants = {} as {
          [K in keyof VariantValues]: {
            name: string
            spec: Record<string, ValueSpec>
          }
        }
        for (const key in a) {
          const value = a[key]
          const built = await value.spec.build(options as any)
          variants[key] = {
            name: value.name,
            spec: built.spec,
          }
          validators[key] = built.validator
        }
        const other = object(
          Object.fromEntries(
            Object.entries(validators).map(([k, v]) => [k, any.optional()]),
          ),
        ).optional()
        return {
          spec: variants,
          validator: anyOf(
            ...Object.entries(validators).map(([k, v]) =>
              object({
                selection: literal(k),
                value: v,
                other,
              }),
            ),
          ) as any,
        }
      },
      anyOf(
        ...Object.entries(staticValidators).map(([k, v]) =>
          object({
            selection: literal(k),
            value: v,
            other,
          }),
        ),
      ) as any,
    )
  }
}