import { camelCase } from 'change-case'
import type { Model } from '../core/model'
import { getColumns } from '../decorators/column'
import { QueryBuilder } from './query-builder'
import { safeParseJSON } from '../utils/model-helper'

export default class ModelQueryBuilder<M> extends QueryBuilder<M> {
  constructor(m: Model, table: string) {
    super()
    this.modelInstance = m
    this.table(table)
  }

  override hydrate(instance: Model, data: Record<string, any>) {
    // Get column metadata to map database column names to property names
    const columns = getColumns(instance.constructor)
    const columnMap = new Map(columns.map((col: any) => [col.name, col.propertyKey]))
    const typeMap = new Map(columns.map((col: any) => [col.name, col.type]))

    // Map database column names to property names
    const mappedData: Record<string, any> = {}
    for (const [dbColumn, value] of Object.entries(data)) {
      const propertyName = columnMap.get(dbColumn) ?? camelCase(dbColumn)
      const propertyType = typeMap.get(dbColumn)

      let parsedValue = value
      if (parsedValue === null || parsedValue === undefined) {
        parsedValue = null
      } else if (propertyType === 'Date') {
        parsedValue = new Date(value)
      } else if (propertyType === 'Boolean') {
        parsedValue = Boolean(value)
      } else if (propertyType === 'Number') {
        parsedValue = Number(value)
      } else if (propertyType === 'String') {
        parsedValue = String(value)
      } else if (propertyType === 'Object') {
        parsedValue = safeParseJSON(value)
      }
      mappedData[propertyName as string] = parsedValue
    }
    Object.assign(instance, mappedData)
    return instance
  }
}