dify/web/service/console-router-loader.ts

import type { AnyContractRouter } from '@orpc/contract'
import { contractLoaders } from '@dify/contracts/api/console/orpc.gen'

const generatedConsoleContractLoaders: Partial<Record<string, () => Promise<AnyContractRouter>>> = contractLoaders

async function loadGeneratedConsoleContract(segment: string) {
  const loader = generatedConsoleContractLoaders[segment]
  if (!loader)
    return null

  return loader()
}

async function loadEnterpriseContract(): Promise<AnyContractRouter> {
  const { contract } = await import('@dify/contracts/enterprise/orpc.gen')
  return { enterprise: contract }
}

export async function loadConsoleContractForSegment(segment: string) {
  if (segment === 'enterprise')
    return loadEnterpriseContract()

  const generatedContract = await loadGeneratedConsoleContract(segment)
  if (generatedContract)
    return generatedContract

  throw new Error(`Console contract segment "${segment}" is not configured.`)
}