Something went wrong. Try again.
fork of hey-api/openapi-ts because I need some additional things
Something went wrong. Try again.
8.0 kB · 259 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260import fs from 'node:fs';import path from 'node:path';import { fileURLToPath } from 'node:url';
import type { IProject, ProjectRenderMeta } from '@hey-api/codegen-core';import type { DefinePlugin } from '@hey-api/shared';import { ensureDirSync } from '@hey-api/shared';
import type { Config } from '../config/types';import type { Client } from '../plugins/@hey-api/client-core/types';import { getClientPlugin } from '../plugins/@hey-api/client-core/utils';
const __filename = fileURLToPath(import.meta.url);const __dirname = path.dirname(__filename);
/** * Dev mode: 'src' appears after 'dist' (or dist doesn't exist), and 'generate' follows 'src' */function isDevMode(): boolean { const normalized = __dirname.split(path.sep); const srcIndex = normalized.lastIndexOf('src'); const distIndex = normalized.lastIndexOf('dist'); return ( srcIndex !== -1 && srcIndex > distIndex && srcIndex === normalized.length - 2 && normalized[srcIndex + 1] === 'generate' );}
/** * Returns paths to client bundle files based on execution context */function getClientBundlePaths(pluginName: string): { clientPath: string; corePath: string;} { const clientName = pluginName.slice('@hey-api/client-'.length);
if (isDevMode()) { // Dev: source bundle folders at src/plugins/@hey-api/{client}/bundle const pluginsDir = path.resolve(__dirname, '..', 'plugins', '@hey-api'); return { clientPath: path.resolve(pluginsDir, `client-${clientName}`, 'bundle'), corePath: path.resolve(pluginsDir, 'client-core', 'bundle'), }; }
// Prod: copied to dist/clients/{clientName} return { clientPath: path.resolve(__dirname, 'clients', clientName), corePath: path.resolve(__dirname, 'clients', 'core'), };}
/** * Returns absolute path to the client folder. This is hard-coded for now. */export function clientFolderAbsolutePath(config: Config): string { const client = getClientPlugin(config);
if ('bundle' in client.config && client.config.bundle) { // not proud of this one const renamed: Map<string, string> | undefined = // @ts-expect-error config._FRAGILE_CLIENT_BUNDLE_RENAMED; return path.resolve(config.output.path, 'client', `${renamed?.get('index') ?? 'index'}.py`); }
return client.name;}
/** * Recursively copies files and directories. * This is a PnP-compatible alternative to fs.cpSync that works with Yarn PnP's * virtualized filesystem. */function copyRecursivePnP(src: string, dest: string): void { const stat = fs.statSync(src);
if (stat.isDirectory()) { if (!fs.existsSync(dest)) { fs.mkdirSync(dest, { recursive: true }); }
const files = fs.readdirSync(src); for (const file of files) { copyRecursivePnP(path.join(src, file), path.join(dest, file)); } } else { const content = fs.readFileSync(src); fs.writeFileSync(dest, content); }}
function renameFile({ filePath, project, renamed,}: { filePath: string; project: IProject; renamed: Map<string, string>;}): void { const extension = path.extname(filePath); const name = path.basename(filePath, extension); const renamedName = project.fileName?.(name) || name; if (renamedName !== name) { const outputPath = path.dirname(filePath); fs.renameSync(filePath, path.resolve(outputPath, `${renamedName}${extension}`)); renamed.set(name, renamedName); }}
function replaceImports({ filePath, isDevMode, meta, renamed,}: { filePath: string; isDevMode?: boolean; meta: ProjectRenderMeta; renamed: Map<string, string>;}): void { let content = fs.readFileSync(filePath, 'utf8');
// Dev mode: rewrite source bundle imports to match output structure if (isDevMode) { // ../../client-core/bundle/foo -> ../core/foo content = content.replace(/from\s+['"]\.\.\/\.\.\/client-core\/bundle\//g, "from '../core/"); // ../../client-core/bundle' (index import) content = content.replace(/from\s+['"]\.\.\/\.\.\/client-core\/bundle['"]/g, "from '../core'"); }
content = content.replace(/from\s+['"](\.\.?\/[^'"]*?)['"]/g, (match, importPath) => { const importIndex = match.indexOf(importPath); const extension = path.extname(importPath); const fileName = path.basename(importPath, extension); const importDir = path.dirname(importPath); const replacedName = (renamed.get(fileName) ?? fileName) + (meta.importFileExtension ? meta.importFileExtension : extension); const replacedMatch = match.slice(0, importIndex) + [importDir, replacedName].filter(Boolean).join('/') + match.slice(importIndex + importPath.length); return replacedMatch; });
const header = '# This file is auto-generated by @hey-api/openapi-python\n\n';
content = `${header}${content}`;
fs.writeFileSync(filePath, content, 'utf8');}
/** * Creates a `client` folder containing the same modules as the client package. */export function generateClientBundle({ meta, outputPath, plugin, project,}: { meta: ProjectRenderMeta; outputPath: string; plugin: DefinePlugin<Client.Config & { name: string }>['Config']; project?: IProject;}): Map<string, string> | undefined { const renamed = new Map<string, string>(); const devMode = isDevMode();
// copy Hey API clients to output const isHeyApiClientPlugin = plugin.name.startsWith('@hey-api/client-'); if (isHeyApiClientPlugin) { const { clientPath } = getClientBundlePaths(plugin.name); // const { clientPath, corePath } = getClientBundlePaths(plugin.name);
// copy client core // const coreOutputPath = path.resolve(outputPath, 'core'); // ensureDirSync(coreOutputPath); // copyRecursivePnP(corePath, coreOutputPath);
// copy client bundle const clientOutputPath = path.resolve(outputPath, 'client'); ensureDirSync(clientOutputPath); copyRecursivePnP(clientPath, clientOutputPath);
if (project) { // const copiedCoreFiles = fs.readdirSync(coreOutputPath); // for (const file of copiedCoreFiles) { // renameFile({ // filePath: path.resolve(coreOutputPath, file), // project, // renamed, // }); // }
const copiedClientFiles = fs.readdirSync(clientOutputPath); for (const file of copiedClientFiles) { renameFile({ filePath: path.resolve(clientOutputPath, file), project, renamed, }); } }
// const coreFiles = fs.readdirSync(coreOutputPath); // for (const file of coreFiles) { // replaceImports({ // filePath: path.resolve(coreOutputPath, file), // isDevMode: devMode, // meta, // renamed, // }); // }
const clientFiles = fs.readdirSync(clientOutputPath); for (const file of clientFiles) { replaceImports({ filePath: path.resolve(clientOutputPath, file), isDevMode: devMode, meta, renamed, }); } return renamed; }
const clientSrcPath = path.isAbsolute(plugin.name) ? path.dirname(plugin.name) : undefined;
// copy custom local client to output if (clientSrcPath) { const dirPath = path.resolve(outputPath, 'client'); ensureDirSync(dirPath); copyRecursivePnP(clientSrcPath, dirPath); return; }
// copy third-party client to output const clientModulePath = path.normalize(require.resolve(plugin.name)); const clientModulePathComponents = clientModulePath.split(path.sep); const clientDistPath = clientModulePathComponents .slice(0, clientModulePathComponents.indexOf('dist') + 1) .join(path.sep);
const indexJsFile = clientModulePathComponents[clientModulePathComponents.length - 1]; const distFiles = [indexJsFile!, 'index.d.mts', 'index.d.cts']; const dirPath = path.resolve(outputPath, 'client'); ensureDirSync(dirPath); for (const file of distFiles) { fs.copyFileSync(path.resolve(clientDistPath, file), path.resolve(dirPath, file)); }
return;}