import Fuse from 'fuse.js'
import type { Ref } from 'vue'
import { isRef, watch } from 'vue'

type Path<T> = T extends object
  ? {
      [K in keyof T]: `${Exclude<K, symbol>}${'' | `.${Path<T[K]>}`}`
    }[keyof T]
  : never

export const useFuzzySearch = <T>(items: T[] | Ref<T[]>, keys: Path<T>[] | string[]) => {
  const fuse = new Fuse<T>([], { keys })
  let documents = items

  const setDocuments = (newDocuments: T[]) => {
    documents = newDocuments
    fuse.setCollection(documents)
  }

  if (isRef<T[]>(items)) {
    fuse.setCollection(items.value)
    watch(items, () => setDocuments(items.value))
  } else {
    setDocuments(items)
  }

  const search = (query: string | null) => {
    query = query?.trim() ?? null

    return query ? fuse.search(query).map(result => result.item) : isRef(documents) ? documents.value : documents
  }

  return {
    search,
    setDocuments,
  }
}