import { useCoverable, Coverable } from '@fexd/pro-utils'
import { useMemoizedFn } from 'ahooks'

import { ProFormProps } from './types'
import defineCoverableProps, { CoverableProFormProps, CoverableProFormConfig } from './defineCoverableProps'

const REF_KEYS = ['ref', 'formRef', 'tableRef'] as const

export default function useCoverableProps<T extends CoverableProFormProps>(
  value: T & CoverableProFormProps,
): Coverable<
  T &
    CoverableProFormProps & {
      getProps: () => ProFormProps
    },
  CoverableProFormConfig<T>
> & {
  getProps: () => ProFormProps
} {
  const extractedRefs: Record<string, any> = {}
  const safeValue = { ...value } as any
  for (const key of REF_KEYS) {
    if (key in safeValue) {
      extractedRefs[key] = safeValue[key]
      delete safeValue[key]
    }
  }

  const config = defineCoverableProps(safeValue)

  const coverableConfig = useCoverable(config) as any
  const raw_getConfig = coverableConfig.getConfig.bind(coverableConfig)

  const getProps = useMemoizedFn(() => {
    const finalConfig = raw_getConfig()
    const props = finalConfig?.getProps?.() ?? {}

    for (const key of REF_KEYS) {
      if (key in extractedRefs && !(key in props)) {
        props[key] = extractedRefs[key]
      }
    }

    return props
  })

  return Object.assign(coverableConfig, {
    getProps,
  })
}