import { JSONSchema7, JSONSchema7Definition } from 'json-schema'
import { EnumType } from '../core/create-enum'
import { FormatName } from 'ajv-formats'

export type NoUndefinedField<T> = { [P in keyof T]-?: NoUndefinedField<NonNullable<T[P]>> }

type Default = string | number | boolean | null | Date | ModelType
interface BaseProperty {
  type: string | ModelNotGeneric | { enum: EnumType }
  isArray?: boolean
  isRequired?: boolean
  default?: Default | Default[]
}
interface StringProperty<IsArray extends boolean> extends BaseProperty {
  type: 'string'
  default?: IsArray extends true ? string[] : string
  isArray?: IsArray
  format?: FormatName
  minLength?: number
  maxLength?: number
  /**
   * Regex expression as string
   */
  pattern?: string
}

interface NumberProperty<IsArray extends boolean> extends BaseProperty {
  type: 'number'
  default?: IsArray extends true ? number[] : number
  isArray?: IsArray
  maximum?: number
  minimum?: number
}

interface EnumProperty<IsArray extends boolean> extends BaseProperty {
  type: { enum: EnumType }
  isArray?: IsArray
}
interface BooleanProperty<IsArray extends boolean> extends BaseProperty {
  type: 'boolean'
  default?: IsArray extends true ? boolean[] : boolean
  isArray?: IsArray
}

interface DateTimeProperty<IsArray extends boolean> extends BaseProperty {
  type: 'datetime' | 'date' | 'time'
  default?: IsArray extends true ? string[] : string
  isArray?: IsArray
}

interface ModelChildrenProperty<IsArray extends boolean> extends BaseProperty {
  type: ModelNotGeneric
  isArray?: IsArray
  default?: IsArray extends true ? ModelType[] : ModelType
}

export type ModelProperty =
  | StringProperty<true>
  | StringProperty<false>
  | NumberProperty<true>
  | NumberProperty<false>
  | EnumProperty<true>
  | EnumProperty<false>
  | BooleanProperty<true>
  | BooleanProperty<false>
  | DateTimeProperty<true>
  | DateTimeProperty<false>
  | ModelChildrenProperty<true>
  | ModelChildrenProperty<false>
export type ModelDefault = Record<string, ModelProperty>

export type GenerateAtomType<T> = T extends 'string'
  ? string
  : T extends 'boolean'
  ? boolean
  : T extends 'number'
  ? number
  : T extends 'datetime'
  ? Date
  : T extends 'date'
  ? Date
  : T extends 'time'
  ? string
  : T extends { enum: EnumType }
  ? T['enum']
  : T extends { model: ModelType }
  ? T['model']
  : T

type GetArray<T extends ModelProperty> = T['isArray'] extends true
  ? Array<GenerateAtomType<T['type']>>
  : GenerateAtomType<T['type']>

type GetRequiredKeys<T extends ModelDefault> = {
  [K in keyof T]: T[K]['isRequired'] extends true ? K : never
}[keyof T]

type GetNonRequiredKeys<T extends ModelDefault> = {
  [K in keyof T]: T[K]['isRequired'] extends true ? never : K
}[keyof T]

type ConvertRequiredNonRequired<T extends ModelDefault> = Pick<
  ModelTypeBase<T>,
  GetRequiredKeys<T>
> &
  Partial<Pick<ModelTypeBase<T>, GetNonRequiredKeys<T>>>

type ModelTypeBase<T extends ModelDefault = ModelDefault> = {
  [P in keyof T]: GetArray<T[P]>
}
export type ModelType<T extends ModelDefault = ModelDefault> = ConvertRequiredNonRequired<T>

export interface ModelNotGeneric {
  new (data?: any): ModelType
  model: ModelType
  getSchema: (options?: { returnFullSchema?: boolean }) => JSONSchema7
  getBody: (parameterName: string) => string
  properties: Record<string, ModelProperty>
  usedReferences: string[]
}

export interface Model<T extends ModelDefault = ModelDefault> {
  new <D extends ModelType<T>>(data?: D): ModelType<T>
  model: ModelType<T>
  getSchema: (options?: { returnFullSchema?: boolean }) => JSONSchema7
  getBody: (parameterName: string) => string
  properties: T
  usedReferences: string[]
}

export type GetString = (propertyName?: string) => string
export interface Property {
  schema: JSONSchema7Definition
  getString: GetString
}