dify/packages/openapi-content-mapper/integration/next.mjs

import assert from 'node:assert/strict'
import { spawnSync } from 'node:child_process'
import { mkdir, mkdtemp, readdir, readFile, rm, symlink, writeFile } from 'node:fs/promises'
import { join, relative, resolve } from 'node:path'
import { test } from 'node:test'
import { loadConsoleDocument } from '../../contracts/scripts/compose-console-openapi.mjs'

const repository = resolve(import.meta.dirname, '../../..')
const web = join(repository, 'web')
const next = join(web, 'node_modules/next/dist/bin/next')

async function javascript(directory) {
  const entries = await readdir(directory, { withFileTypes: true })
  return (
    await Promise.all(
      entries.map(async (entry) => {
        const file = join(directory, entry.name)
        if (entry.isDirectory()) return javascript(file)
        return entry.name.endsWith('.js') ? readFile(file, 'utf8') : ''
      }),
    )
  ).join('\n')
}

for (const bundler of ['turbopack', 'webpack']) {
  test(`Next ${bundler}: Console and Marketplace imports keep separate transports and prune unused operations`, async () => {
    const consoleSpec = loadConsoleDocument()
    const alias = 'enterprise.access.getMetadata'
    consoleSpec.paths['/knowledge-fs/spaces/{control_space_id}/metadata'].get['x-client-path'] =
      alias.split('.')
    const dir = await mkdtemp(join(web, '.openapi-next-'))
    // Match production: contracts owns the document but only Web depends on the SDK.
    const specDir = await mkdtemp(join(repository, 'packages/contracts/.openapi-next-'))
    try {
      await mkdir(join(dir, 'app'))
      await writeFile(join(dir, 'package.json'), JSON.stringify({ private: true, type: 'module' }))
      await writeFile(
        join(dir, 'next.config.mjs'),
        `
const loader = { loader: '@dify/openapi-content-mapper/loader', options: { runtime: ${JSON.stringify(relative(specDir, join(dir, 'runtime.js')))}, sidecars: ${JSON.stringify(relative(specDir, join(dir, 'spec.openapi')))} } };
const marketplaceLoader = { loader: '@dify/openapi-content-mapper/loader', options: { runtime: ${JSON.stringify(relative(specDir, join(dir, 'marketplace-runtime.js')))}, sidecars: ${JSON.stringify(relative(specDir, join(dir, 'marketplace.openapi')))} } };
const namespaceLoader = { loader: '@dify/openapi-content-mapper/loader' };
export default {
  experimental: { cpus: 1, extensionAlias: { '.js': ['.ts', '.js'] } },
  turbopack: { root: ${JSON.stringify(repository)}, rules: { '*console-spec.openapi-source': { loaders: [loader], as: '*.js' }, '*marketplace-spec.json': { loaders: [marketplaceLoader], as: '*.js' }, '*openapi-namespace.js': { loaders: [namespaceLoader], as: '*.js' } } },
  webpack(config) { config.module.rules.push({ test: /console-spec\\.openapi-source$/, type: 'javascript/auto', use: [loader] }, { test: /marketplace-spec\\.json$/, type: 'javascript/auto', use: [marketplaceLoader] }, { test: /openapi-namespace\\.js$/, type: 'javascript/auto', use: [namespaceLoader] }); return config; },
};
`,
      )
      await writeFile(
        join(specDir, 'console-export.mjs'),
        `process.stdout.write(${JSON.stringify(
          JSON.stringify({
            ...consoleSpec,
            info: { title: 'SCHEMA_METADATA_MUST_DISAPPEAR', version: '1' },
            paths: {
              ...consoleSpec.paths,
              '/UNUSED_ENDPOINT_MUST_DISAPPEAR': {
                get: { responses: { 200: { description: 'OK' } } },
              },
            },
          }),
        )});`,
      )
      await writeFile(
        join(specDir, 'console-spec.openapi-source'),
        JSON.stringify({ command: ['node', 'console-export.mjs'], inputs: ['console-export.mjs'] }),
      )
      await symlink(join(specDir, 'console-spec.openapi-source'), join(dir, 'spec.openapi'))
      await writeFile(
        join(specDir, 'marketplace-spec.json'),
        JSON.stringify({
          openapi: '3.1.0',
          info: { title: 'Independent document', version: '1' },
          paths: {
            '/templates/{templateId}': {
              get: {
                'x-client-path': ['templateDetail'],
                parameters: [
                  { name: 'templateId', in: 'path', required: true, schema: { type: 'string' } },
                ],
                responses: { 200: { description: 'OK' } },
              },
            },
            '/template-collections': { get: { responses: { 200: { description: 'Unused' } } } },
          },
        }),
      )
      await symlink(join(specDir, 'marketplace-spec.json'), join(dir, 'marketplace.openapi'))
      await writeFile(
        join(dir, 'marketplace.openapi.queries.ts'),
        "const policy = { queryOptions: { staleTime: 3131 } }; export { policy as 'GET /templates/{templateId}' };",
      )
      await writeFile(
        join(dir, 'marketplace-runtime.js'),
        `
import { createClient } from '@dify/openapi-content-mapper';
export default { queryKeyPrefix: ['marketplace'], createClient: () => createClient({ baseUrl: 'https://marketplace.test', fetch: async url => Response.json({ value: String(url) }) }) };
`,
      )
      const policy = `const selected = { queryOptions: { staleTime: 4242 } }; const unused = { queryOptions: { meta: { marker: 'UNUSED_POLICY_MUST_DISAPPEAR' } } }; export { selected as 'GET /knowledge-fs/spaces/{control_space_id}/metadata', unused as 'GET /UNUSED_ENDPOINT_MUST_DISAPPEAR' };`
      const errors = `const selected = { CONFLICT: { status: 409 } }; const unused = { UNUSED: { status: 400, message: 'UNUSED_ERROR_MUST_DISAPPEAR' } }; export { selected as 'GET /knowledge-fs/spaces/{control_space_id}/metadata', unused as 'GET /UNUSED_ENDPOINT_MUST_DISAPPEAR' };`
      await writeFile(
        join(dir, 'spec.openapi.queries.ts'),
        `export * from ${JSON.stringify(bundler === 'webpack' ? './policies.js' : './policies.ts')};`,
      )
      await writeFile(join(dir, 'policies.ts'), policy)
      await writeFile(
        join(dir, 'spec.openapi.errors.ts'),
        `export * from ${JSON.stringify(bundler === 'webpack' ? './errors.js' : './errors.ts')};`,
      )
      await writeFile(join(dir, 'errors.ts'), errors)

      await writeFile(
        join(dir, 'runtime.js'),
        `
import { createClient } from '@dify/openapi-content-mapper';
export default {
  queryKeyPrefix: ['bound'],
  createClient: ({ context }) => createClient({ baseUrl: 'https://bound.test', fetch: async url => context.user === 'conflict' ? Response.json({ code: 'CONFLICT' }, { status: 409 }) : Response.json({ value: String(url) + ':' + context.user }) }),
};
`,
      )
      await writeFile(
        join(dir, 'app/layout.jsx'),
        'export default function Layout({ children }) { return <html><body>{children}</body></html> }',
      )
      await writeFile(
        join(dir, 'app/button.jsx'),
        `
'use client';
import { api, query } from '../spec.openapi';
import { api as marketplace, query as marketplaceQuery } from '../marketplace.openapi';
export default function Button() {
  return <><button onClick={() => marketplace.templateDetail({ params: { templateId: 'template-1' } })}>{JSON.stringify(marketplaceQuery.templateDetail.key())}</button><button onClick={() => api.${alias}({ params: { control_space_id: 'space-1' } }, { context: { user: 'client' } })}>{JSON.stringify(query.${alias}.key())}</button></>;
}
`,
      )
      await writeFile(
        join(dir, 'app/page.jsx'),
        `
import { api, query } from '../spec.openapi';
import { api as marketplace, query as marketplaceQuery } from '../marketplace.openapi';
import Button from './button';
export default async function Page() {
  const market = await marketplace.templateDetail({ params: { templateId: 'template-1' } });
  const marketOptions = marketplaceQuery.templateDetail.queryOptions({ input: { params: { templateId: 'template-1' } } });
  const response = await api.${alias}({ params: { control_space_id: 'space-1' } }, { context: { user: 'server' } });
  const defined = await api.${alias}({ params: { control_space_id: 'space-1' } }, { context: { user: 'conflict' } }).catch(error => error.defined);
  const options = query.${alias}.queryOptions({ input: { params: { control_space_id: 'space-1' } } });
  return <><p>{market.value};market-stale:{marketOptions.staleTime}</p><p>{response.value}</p><p>defined-error:{String(defined)};stale:{options.staleTime}</p><Button /></>;
}
`,
      )
      const result = spawnSync(process.execPath, [next, 'build', `--${bundler}`], {
        cwd: dir,
        env: {
          ...process.env,
          NEXT_TELEMETRY_DISABLED: '1',
          NODE_OPTIONS: '--max-old-space-size=1024',
          RAYON_NUM_THREADS: '1',
        },
        encoding: 'utf8',
        timeout: 120_000,
        killSignal: 'SIGKILL',
        maxBuffer: 4 * 1024 * 1024,
      })
      assert.equal(result.status, 0, `${result.error ?? ''}\n${result.stdout}\n${result.stderr}`)
      const html = await readFile(join(dir, '.next/server/app/index.html'), 'utf8')
      assert(html.includes('https://bound.test/knowledge-fs/spaces/space-1/metadata:server'))
      assert(html.includes('bound'))
      assert(html.replaceAll(/<!--.*?-->/g, '').includes('defined-error:true;stale:4242'))
      assert(html.includes('4242'))
      assert(
        html
          .replaceAll(/<!--.*?-->/g, '')
          .includes('https://marketplace.test/templates/template-1;market-stale:3131'),
      )
      assert(html.includes('marketplace'))
      const client = await javascript(join(dir, '.next/static/chunks'))
      assert(client.includes('/knowledge-fs/spaces/'))
      assert(!client.includes('/datasets'))
      assert(!client.includes('/template-collections'))
      assert(!client.includes('UNUSED_ENDPOINT_MUST_DISAPPEAR'))
      assert(!client.includes('SCHEMA_METADATA_MUST_DISAPPEAR'))
      assert(!client.includes('UNUSED_POLICY_MUST_DISAPPEAR'))
      assert(!client.includes('UNUSED_ERROR_MUST_DISAPPEAR'))
    } finally {
      await rm(dir, { recursive: true, force: true })
      await rm(specDir, { recursive: true, force: true })
    }
  })
}