import {isRecord} from '../util/isRecord'
import {isArray} from './isArray'
import type {ContentSourceMapParsedPath, WalkMapFn} from './types'

/**
 * generic way to walk a nested object or array and apply a mapping function to each value
 * @internal
 */
export function walkMap(
  value: unknown,
  mappingFn: WalkMapFn,
  path: ContentSourceMapParsedPath = [],
): unknown {
  if (isArray(value)) {
    return value.map((v, idx) => {
      if (isRecord(v)) {
        const _key = v['_key']
        if (typeof _key === 'string') {
          return walkMap(v, mappingFn, path.concat({_key, _index: idx}))
        }
      }

      return walkMap(v, mappingFn, path.concat(idx))
    })
  }

  if (isRecord(value)) {
    // Handle Portable Text in a faster way
    if (value._type === 'block' || value._type === 'span') {
      const result = {...value}
      if (value._type === 'block') {
        result.children = walkMap(value.children, mappingFn, path.concat('children'))
      } else if (value._type === 'span') {
        result.text = walkMap(value.text, mappingFn, path.concat('text'))
      }
      return result
    }

    return Object.fromEntries(
      Object.entries(value).map(([k, v]) => [k, walkMap(v, mappingFn, path.concat(k))]),
    )
  }

  return mappingFn(value, path)
}