Function enhanceRouter

packages/server/src/router-utils.ts:115–169

(
  router: T,
  options: EnhanceRouterOptions<TErrorMap>,
)

}

export function enhanceRouter<
  T extends Lazyable<AnyRouter>,
  TInitialContext extends Context,
  TCurrentContext extends Context,
  TErrorMap extends ErrorMap,
>(
  router: T,
  options: EnhanceRouterOptions<TErrorMap>,
): EnhancedRouter<T, TInitialContext, TCurrentContext, TErrorMap> {
  if (isLazy(router)) {
    const laziedMeta = getLazyMeta(router)
    const enhancedPrefix = laziedMeta?.prefix ? mergePrefix(options.prefix, laziedMeta?.prefix) : options.prefix

    const enhanced = lazy(async () => {
      const { default: unlaziedRouter } = await unlazy(router)
      const enhanced = enhanceRouter(unlaziedRouter, options)
      return unlazy(enhanced)
    }, {
      ...laziedMeta,
      prefix: enhancedPrefix,
    })

    const accessible = createAccessibleLazyRouter(enhanced)

    return accessible as any
  }

  if (isProcedure(router)) {
    const newMiddlewares = mergeMiddlewares(options.middlewares, router['~orpc'].middlewares, { dedupeLeading: options.dedupeLeadingMiddlewares })
    const newMiddlewareAdded = newMiddlewares.length - router['~orpc'].middlewares.length

    const enhanced = new Procedure({
      ...router['~orpc'],
      route: enhanceRoute(router['~orpc'].route, options),
      errorMap: mergeErrorMap(options.errorMap, router['~orpc'].errorMap),
      middlewares: newMiddlewares,
      inputValidationIndex: router['~orpc'].inputValidationIndex + newMiddlewareAdded,
      outputValidationIndex: router['~orpc'].outputValidationIndex + newMiddlewareAdded,
    })

    return enhanced as any
  }

  if (typeof router !== 'object' || router === null) {
    return router as any
  }

  const enhanced = {} as Record<string, any>

  for (const key in router) {
    enhanced[key] = enhanceRouter(router[key]!, options)
  }

  return enhanced as any
}

export interface TraverseContractProceduresOptions {
  router: AnyContractRouter | AnyRouter