import type { CSSProperties } from 'react'
import type { PreferredContents } from '../core'

// these fields are shared among all RegularControlDescription. the helper function getControlSharedFields makes sure the types line up
// Ensure that the fields are also added to the object within `getControlSharedFields` for that typechecking.
interface ControlBaseFields {
  control: RegularControlType
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export const PropertyControlIcons = ['arrow-left', 'arrow-right', 'arrow-up', 'arrow-down'] as const
export type PropertyControlIcon = (typeof PropertyControlIcons)[number]

// Base Level Controls

export type BaseControlType =
  | 'checkbox'
  | 'color'
  | 'euler'
  | 'expression-input'
  | 'expression-popuplist'
  | 'matrix3'
  | 'matrix4'
  | 'none'
  | 'number-input'
  | 'popuplist'
  | 'radio'
  | 'string-input'
  | 'html-input'
  | 'style-controls'
  | 'vector2'
  | 'vector3'
  | 'vector4'
  | 'jsx'

export interface CheckboxControlDescription {
  control: 'checkbox'
  label?: string
  visibleByDefault?: boolean
  disabledTitle?: string
  enabledTitle?: string
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface ColorControlDescription {
  control: 'color'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export type AllowedEnumType = string | boolean | number | undefined | null
export interface BasicControlOption<T> {
  value: T
  label: string
}

export interface BasicControlOptionWithIcon<T> {
  value: T
  label: string
  icon?: PropertyControlIcon
}

export type BasicControlOptions<T> = AllowedEnumType[] | BasicControlOption<T>[]

export interface PopUpListControlDescription {
  control: 'popuplist'
  label?: string
  visibleByDefault?: boolean
  options: BasicControlOptions<unknown>
  required?: boolean
  defaultValue?: AllowedEnumType | BasicControlOption<unknown>
  folder?: string
}

export interface ImportType {
  source: string // importSource
  name: string | null
  type: 'star' | 'default' | null
}

export interface ExpressionControlOption<T> {
  value: T
  expression: string
  label?: string
  requiredImport?: ImportType
}

export interface ExpressionPopUpListControlDescription {
  control: 'expression-popuplist'
  label?: string
  visibleByDefault?: boolean
  options: ExpressionControlOption<unknown>[]
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface EulerControlDescription {
  control: 'euler'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: [number, number, number, string]
  folder?: string
}

export interface NoneControlDescription {
  control: 'none'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

// prettier-ignore
export type Matrix3 = [
  number, number, number,
  number, number, number,
  number, number, number,
]

export interface Matrix3ControlDescription {
  control: 'matrix3'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: Matrix3
  folder?: string
}

// prettier-ignore
export type Matrix4 = [
  number, number, number, number,
  number, number, number, number,
  number, number, number, number,
  number, number, number, number,
]

export interface Matrix4ControlDescription {
  control: 'matrix4'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: Matrix4
  folder?: string
}

export interface NumberInputControlDescription {
  control: 'number-input'
  label?: string
  visibleByDefault?: boolean
  max?: number
  min?: number
  unit?: string
  step?: number
  displayStepper?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface RadioControlDescription {
  control: 'radio'
  label?: string
  visibleByDefault?: boolean
  options: AllowedEnumType[] | BasicControlOptionWithIcon<unknown>[]
  required?: boolean
  defaultValue?: AllowedEnumType | BasicControlOptionWithIcon<unknown>
  folder?: string
}

export interface ExpressionInputControlDescription {
  control: 'expression-input'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface StringInputControlDescription {
  control: 'string-input'
  label?: string
  visibleByDefault?: boolean
  placeholder?: string
  obscured?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface HtmlInputControlDescription {
  control: 'html-input'
  label?: string
  visibleByDefault?: boolean
  placeholder?: string
  obscured?: boolean
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface StyleControlsControlDescription {
  control: 'style-controls'
  label?: string
  visibleByDefault?: boolean
  placeholder?: CSSProperties
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export type Vector2 = [number, number]

export interface Vector2ControlDescription {
  control: 'vector2'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: Vector2
  folder?: string
}

export type Vector3 = [number, number, number]

export interface Vector3ControlDescription {
  control: 'vector3'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: Vector3
  folder?: string
}

export type Vector4 = [number, number, number, number]

export interface Vector4ControlDescription {
  control: 'vector4'
  label?: string
  visibleByDefault?: boolean
  required?: boolean
  defaultValue?: Vector4
  folder?: string
}

export interface JSXControlDescription {
  control: 'jsx'
  label?: string
  visibleByDefault?: boolean
  preferredContents?: PreferredContents | PreferredContents[]
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export type BaseControlDescription =
  | CheckboxControlDescription
  | ColorControlDescription
  | ExpressionInputControlDescription
  | ExpressionPopUpListControlDescription
  | EulerControlDescription
  | NoneControlDescription
  | Matrix3ControlDescription
  | Matrix4ControlDescription
  | NumberInputControlDescription
  | RadioControlDescription
  | PopUpListControlDescription
  | StringInputControlDescription
  | HtmlInputControlDescription
  | StyleControlsControlDescription
  | Vector2ControlDescription
  | Vector3ControlDescription
  | Vector4ControlDescription
  | JSXControlDescription

export type PropertyControlsOptions<T> = Omit<ControlBaseFields, 'control'> & { defaultValue?: T }

// Higher Level Controls

export type HigherLevelControlType = 'array' | 'tuple' | 'object' | 'union'
export type RegularControlType = BaseControlType | HigherLevelControlType
export type ControlType = RegularControlType | 'folder'

export interface ArrayControlDescription {
  control: 'array'
  label?: string
  visibleByDefault?: boolean
  propertyControl: RegularControlDescription
  maxCount?: number
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface ObjectControlDescription {
  control: 'object'
  label?: string
  visibleByDefault?: boolean
  object: { [prop: string]: RegularControlDescription }
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export interface UnionControlDescription {
  control: 'union'
  label?: string
  visibleByDefault?: boolean
  controls: Array<RegularControlDescription>
  required?: boolean
  defaultValue?: unknown
  folder?: string
}
export interface TupleControlDescription {
  control: 'tuple'
  label?: string
  visibleByDefault?: boolean
  propertyControls: RegularControlDescription[]
  required?: boolean
  defaultValue?: unknown
  folder?: string
}

export type HigherLevelControlDescription =
  | ArrayControlDescription
  | ObjectControlDescription
  | TupleControlDescription
  | UnionControlDescription

export type RegularControlDescription = BaseControlDescription | HigherLevelControlDescription

// Please ensure that `property-controls-utils.ts` is kept up to date
// with any changes to this or the component types.
export type ControlDescription = RegularControlDescription

export function isBaseControlDescription(
  control: ControlDescription,
): control is BaseControlDescription {
  switch (control.control) {
    case 'checkbox':
    case 'color':
    case 'euler':
    case 'expression-input':
    case 'expression-popuplist':
    case 'matrix3':
    case 'matrix4':
    case 'none':
    case 'number-input':
    case 'popuplist':
    case 'radio':
    case 'string-input':
    case 'html-input':
    case 'style-controls':
    case 'vector2':
    case 'vector3':
    case 'vector4':
    case 'jsx':
      return true
    case 'array':
    case 'object':
    case 'tuple':
    case 'union':
      return false
    default:
      const _exhaustiveCheck: never = control
      throw new Error(`Unhandled controls ${JSON.stringify(control)}`)
  }
}

export function isHigherLevelControlDescription(
  control: ControlDescription,
): control is HigherLevelControlDescription {
  return !isBaseControlDescription(control)
}

export type PropertyControls = {
  [key: string]: ControlDescription
}

export function addPropertyControls(component: unknown, propertyControls: PropertyControls): void {
  if (component != null) {
    ;(component as any)['propertyControls'] = propertyControls
  }
}