import type {ContentSourceMap} from '@sanity/client'
import type {ResolveStudioUrl, StudioPathLike, StudioUrl} from '@sanity/client/csm'

import {
  defineEncodeDataAttribute,
  type EncodeDataAttributeFunction,
} from '@sanity/core-loader/encode-data-attribute'
import {useMemo} from 'react'

/** @public */
export type EncodeDataAttributeCallback = (path: StudioPathLike) => string | undefined

/** @public */
export function useEncodeDataAttribute<QueryResponseResult = unknown>(
  result: QueryResponseResult,
  sourceMap: ContentSourceMap | undefined,
  studioUrl: StudioUrl | ResolveStudioUrl | undefined,
): EncodeDataAttributeFunction {
  return useMemo(
    () => defineEncodeDataAttribute(result, sourceMap, studioUrl),
    [result, sourceMap, studioUrl],
  )
}

export type {ContentSourceMap, ResolveStudioUrl, StudioPathLike, StudioUrl}