import type { ErrorType } from '../errors/utils.js'

type FormatFn<parameters = any, returnType = any> = (
  parameters: parameters,
) => returnType

type OverrideFormatFn<parameters, returnType, baseFn extends FormatFn> = (
  parameters: parameters,
  baseFn: baseFn,
) => returnType

export type Formatter<rpcType = any, type = any> = {
  fromRpc: FormatFn<rpcType, type>
  toRpc: FormatFn<type, rpcType>
}

export type DefineFormatterParameters<
  fromRpc extends FormatFn = FormatFn,
  toRpc extends FormatFn = FormatFn,
> = {
  fromRpc: fromRpc
  toRpc: toRpc
}

export type DefineFormatterErrorType = ErrorType

export function defineFormatter<
  fromRpc extends FormatFn,
  toRpc extends FormatFn,
>(parameters: DefineFormatterParameters<fromRpc, toRpc>) {
  const { fromRpc: fromRpc_base, toRpc: toRpc_base } = parameters

  return <type = ReturnType<fromRpc>, rpcType = ReturnType<toRpc>>(overrides: {
    fromRpc: OverrideFormatFn<rpcType, type, fromRpc>
    toRpc: OverrideFormatFn<type, rpcType, toRpc>
  }): Formatter<rpcType, type> => {
    const { fromRpc, toRpc } = overrides

    return {
      fromRpc(parameters) {
        return fromRpc(parameters, fromRpc_base)
      },
      toRpc(parameters) {
        return toRpc(parameters, toRpc_base)
      },
    }
  }
}