import { resolvePhpCallable } from '../_helpers/_callbackResolver.ts'
import type { PhpCallableDescriptor, PhpList, PhpRuntimeValue } from '../_helpers/_phpTypes.ts'

type ArrayMapValue = PhpRuntimeValue
type ArrayMapInputs = [firstArray: PhpList<ArrayMapValue>, ...restArrays: Array<PhpList<ArrayMapValue>>]
type ArrayMapTupleArgs<TInputs extends readonly PhpList<ArrayMapValue>[]> = [
  ...{
    [K in keyof TInputs]: TInputs[K] extends PhpList<infer TValue> ? TValue | undefined : ArrayMapValue | undefined
  },
]
type ArrayMapCallbackArgsVariadic = Array<ArrayMapValue | undefined>

export function array_map<TInputs extends ArrayMapInputs, TResult>(
  callback: PhpCallableDescriptor<ArrayMapTupleArgs<TInputs>, TResult>,
  ...inputArrays: TInputs
): PhpList<TResult>

export function array_map<TInputs extends ArrayMapInputs>(
  callback: null | undefined,
  ...inputArrays: TInputs
): PhpList<ArrayMapTupleArgs<TInputs>>

export function array_map<TResult = ArrayMapValue>(
  callback: PhpCallableDescriptor<ArrayMapCallbackArgsVariadic, TResult> | null | undefined,
  ...inputArrays: ArrayMapInputs
): PhpList<TResult> | PhpList<ArrayMapCallbackArgsVariadic> {
  //  discuss at: https://locutus.io/php/array_map/
  // original by: Andrea Giammarchi (https://webreflection.blogspot.com)
  // improved by: Kevin van Zonneveld (https://kvz.io)
  // improved by: Brett Zamir (https://brett-zamir.me)
  //    input by: thekid
  //      note 1: If the callback is a string (or object, if an array is supplied),
  //      note 1: it can only work if the function name is in the global context
  //   example 1: array_map( function (a){return (a * a * a)}, [1, 2, 3, 4, 5] )
  //   returns 1: [ 1, 8, 27, 64, 125 ]

  const argc = inputArrays.length + 1
  const itemCount = inputArrays[0]?.length ?? 0

  const resolved =
    callback === null || typeof callback === 'undefined'
      ? null
      : resolvePhpCallable<ArrayMapCallbackArgsVariadic, TResult>(callback, {
          invalidMessage: 'array_map(): Invalid callback',
          missingScopeMessage: (scopeName: string) => 'Object not found: ' + scopeName,
        })

  if (resolved) {
    const mapped: PhpList<TResult> = []
    for (let itemIndex = 0; itemIndex < itemCount; itemIndex++) {
      const args: ArrayMapCallbackArgsVariadic = []
      for (let arrayIndex = 0; arrayIndex < argc - 1; arrayIndex++) {
        args.push(inputArrays[arrayIndex]?.[itemIndex])
      }
      mapped[itemIndex] = resolved.fn.apply(resolved.scope, args)
    }
    return mapped
  }

  const mapped: PhpList<ArrayMapCallbackArgsVariadic> = []
  for (let itemIndex = 0; itemIndex < itemCount; itemIndex++) {
    const args: ArrayMapCallbackArgsVariadic = []
    for (let arrayIndex = 0; arrayIndex < argc - 1; arrayIndex++) {
      args.push(inputArrays[arrayIndex]?.[itemIndex])
    }
    mapped[itemIndex] = args
  }

  return mapped
}