import type {
  ControlComponentBinding,
  ControlComponentBindingConfig,
  ControlComponentBindingRecordConfig,
  ControlComponentBindingsById,
  ControlComponentBindingsConfig,
  ControlComponentPropsMap,
  ControlComponentPropSource,
} from './controlSchemaTypes'

/** Map control workspace values into component props for direct `v-bind` usage. */
export function controlValuesToComponentProps<TValues extends Record<string, unknown>>(
  values: TValues,
  mapping?: ControlComponentPropsMap<TValues>,
): Record<string, unknown> {
  if (mapping === undefined) return { ...values }

  if (Array.isArray(mapping)) {
    const props: Record<string, unknown> = {}
    for (const key of mapping) {
      props[key] = values[key]
    }
    return props
  }

  const props: Record<string, unknown> = {}
  for (const [prop, source] of Object.entries(mapping) as Array<[string, ControlComponentPropSource<TValues>]>) {
    props[prop] = typeof source === 'function' ? source(values) : values[source]
  }
  return props
}

/** Map control workspace values into named SDK component bindings for direct slot rendering. */
export function controlValuesToComponentBindings<TValues extends Record<string, unknown>>(
  values: TValues,
  bindings?: ControlComponentBindingsConfig<TValues>,
): ControlComponentBinding[] {
  if (bindings === undefined) return []

  return normalizeControlComponentBindingConfigs(bindings).map(binding => ({
    id: binding.id,
    component: binding.component,
    props: controlValuesToComponentProps(values, binding.props),
  }))
}

/** Map control workspace values into SDK component bindings keyed by binding id. */
export function controlValuesToComponentBindingsById<TValues extends Record<string, unknown>>(
  values: TValues,
  bindings?: ControlComponentBindingsConfig<TValues>,
): ControlComponentBindingsById {
  return Object.fromEntries(
    controlValuesToComponentBindings(values, bindings).map(binding => [binding.id, binding]),
  )
}

function normalizeControlComponentBindingConfigs<TValues extends Record<string, unknown>>(
  bindings: ControlComponentBindingsConfig<TValues>,
): Array<Required<Pick<ControlComponentBindingConfig<TValues>, 'id' | 'component'>> & Pick<ControlComponentBindingConfig<TValues>, 'props'>> {
  if (Array.isArray(bindings)) {
    const usedIds = new Map<string, number>()
    return bindings.map(binding => ({
      id: uniqueComponentBindingId(binding.id ?? binding.component, usedIds),
      component: binding.component,
      props: binding.props,
    }))
  }

  return Object.entries(bindings).map(([id, binding]: [string, ControlComponentBindingRecordConfig<TValues>]) => ({
    id,
    component: binding.component,
    props: binding.props,
  }))
}

function uniqueComponentBindingId(id: string, usedIds: Map<string, number>): string {
  const count = usedIds.get(id) ?? 0
  usedIds.set(id, count + 1)
  return count === 0 ? id : `${id}-${count + 1}`
}