Function applyCustomOpenAPIOperation

packages/openapi/src/openapi-custom.ts:31–67

(operation: OpenAPI.OperationObject, contract: AnyContractProcedure)

}

export function applyCustomOpenAPIOperation(operation: OpenAPI.OperationObject, contract: AnyContractProcedure): OpenAPI.OperationObject {
  const operationCustoms: OverrideOperationValue[] = []

  for (const errorItem of Object.values(contract['~orpc'].errorMap) as ErrorMap[keyof ErrorMap][]) {
    const maybeExtender = errorItem ? getCustomOpenAPIOperation(errorItem) : undefined

    if (maybeExtender) {
      operationCustoms.push(maybeExtender)
    }
  }

  if (isProcedure(contract)) {
    for (const middleware of contract['~orpc'].middlewares) {
      const maybeExtender = getCustomOpenAPIOperation(middleware)

      if (maybeExtender) {
        operationCustoms.push(maybeExtender)
      }
    }
  }

  let currentOperation = operation

  for (const custom of operationCustoms) {
    if (typeof custom === 'function') {
      currentOperation = custom(currentOperation, contract)
    }
    else {
      currentOperation = {
        ...currentOperation,
        ...custom,
      }
    }
  }

  return currentOperation
}