Add option to pass in a custom handlebars template to generate the service files

.gitignore

@@ -8,7 +8,6 @@ junit.xml
 .idea
 .vscode
 *.iml
-dist
 coverage
 test/generated
 test/e2e/generated

README.md

@@ -1,5 +1,19 @@
 # OpenAPI Typescript Codegen
 
+Fork ([diff](https://github.com/ferdikoomen/openapi-typescript-codegen/compare/master...mb21:openapi-typescript-codegen:generate-custom))
+which adds a `--serviceTemplate` option.
+
+Can be used in another project by adding to `package.json`::
+
+```json
+"openapi-typescript-codegen": "https://github.com/mb21/openapi-typescript-codegen.git#generate-custom",
+```
+
+To release a new version, run `npm run release` and push everything, including the `dist/index.js` file to GitHub. Then, in the project using it, delete the
+`openapi-typescript-codegen` entry in the `package-lock.json` and run `npm install` to install the new version.
+
+---
+
 [![NPM][npm-image]][npm-url]
 [![License][license-image]][license-url]
 [![Coverage][coverage-image]][coverage-url]
@@ -50,6 +64,7 @@ $ openapi --help
     --indent <value>          Indentation options [4, 2, tab] (default: "4")
     --postfix <value>         Service name postfix (default: "Service")
     --request <value>         Path to custom request file
+    --serviceTemplate         Path to custom service handlebars template to generate the service files
     -h, --help                display help for command
 
   Examples

bin/index.js

@@ -23,6 +23,7 @@ const params = program
     .option('--indent <value>', 'Indentation options [4, 2, tabs]', '4')
     .option('--postfix <value>', 'Service name postfix', 'Service')
     .option('--request <value>', 'Path to custom request file')
+    .option('--serviceTemplate <value>', 'Path to custom service handlebars template to generate the service files')
     .parse(process.argv)
     .opts();
 
@@ -43,6 +44,7 @@ if (OpenAPI) {
         indent: params.indent,
         postfix: params.postfix,
         request: params.request,
+        serviceTemplate: params.serviceTemplate,
     })
         .then(() => {
             process.exit(0);

package.json

@@ -121,5 +121,8 @@
     },
     "resolutions": {
         "node-fetch": "2.6.7"
+    },
+    "volta": {
+        "node": "16.17.0"
     }
 }

src/index.ts

@@ -8,6 +8,7 @@ import { isString } from './utils/isString';
 import { postProcessClient } from './utils/postProcessClient';
 import { registerHandlebarTemplates } from './utils/registerHandlebarTemplates';
 import { writeClient } from './utils/writeClient';
+import { writeClientServicesCustomTemplate } from './utils/writeClientServicesCustomTemplate';
 
 export { HttpClient } from './HttpClient';
 export { Indent } from './Indent';
@@ -26,6 +27,7 @@ export type Options = {
     indent?: Indent;
     postfix?: string;
     request?: string;
+    serviceTemplate?: string;
     write?: boolean;
 };
 
@@ -62,6 +64,7 @@ export const generate = async ({
     indent = Indent.SPACE_4,
     postfix = 'Service',
     request,
+    serviceTemplate,
     write = true,
 }: Options): Promise<void> => {
     const openApi = isString(input) ? await getOpenApiSpec(input) : input;
@@ -72,10 +75,15 @@ export const generate = async ({
         useOptions,
     });
 
+    if (serviceTemplate) {
+        exportServices = false;
+    }
+
+    let clientFinal;
     switch (openApiVersion) {
         case OpenApiVersion.V2: {
             const client = parseV2(openApi);
-            const clientFinal = postProcessClient(client);
+            clientFinal = postProcessClient(client);
             if (!write) break;
             await writeClient(
                 clientFinal,
@@ -98,7 +106,7 @@ export const generate = async ({
 
         case OpenApiVersion.V3: {
             const client = parseV3(openApi);
-            const clientFinal = postProcessClient(client);
+            clientFinal = postProcessClient(client);
             if (!write) break;
             await writeClient(
                 clientFinal,
@@ -119,6 +127,19 @@ export const generate = async ({
             break;
         }
     }
+
+    if (serviceTemplate) {
+        await writeClientServicesCustomTemplate(
+            clientFinal,
+            output,
+            httpClient,
+            useOptions,
+            useUnionTypes,
+            indent,
+            postfix,
+            serviceTemplate
+        );
+    }
 };
 
 export default {

src/utils/registerHandlebarHelpers.ts

@@ -1,5 +1,5 @@
 import camelCase from 'camelcase';
-import Handlebars from 'handlebars/runtime';
+import HandlebarsRuntime from 'handlebars/runtime';
 import { EOL } from 'os';
 
 import type { Enum } from '../client/interfaces/Enum';
@@ -11,7 +11,10 @@ export const registerHandlebarHelpers = (root: {
     httpClient: HttpClient;
     useOptions: boolean;
     useUnionTypes: boolean;
+    handlebars?: typeof HandlebarsRuntime;
 }): void => {
+    const Handlebars = root.handlebars || HandlebarsRuntime;
+
     Handlebars.registerHelper('ifdef', function (this: any, ...args): string {
         const options = args.pop();
         if (!args.every(value => !value)) {

src/utils/registerHandlebarTemplates.ts

@@ -1,4 +1,4 @@
-import Handlebars from 'handlebars/runtime';
+import HandlebarsRuntime from 'handlebars/runtime';
 
 import { HttpClient } from '../HttpClient';
 import templateClient from '../templates/client.hbs';
@@ -113,7 +113,10 @@ export const registerHandlebarTemplates = (root: {
     httpClient: HttpClient;
     useOptions: boolean;
     useUnionTypes: boolean;
+    handlebars?: typeof HandlebarsRuntime;
 }): Templates => {
+    const Handlebars = root.handlebars || HandlebarsRuntime;
+
     registerHandlebarHelpers(root);
 
     // Main templates (entry points for the files we write to disk)

src/utils/writeClientServicesCustomTemplate.ts

@@ -0,0 +1,51 @@
+import { mkdir, readFile, remove } from 'fs-extra';
+import Handlebars from 'handlebars';
+import { resolve } from 'path';
+
+import { Client } from '../client/interfaces/Client';
+import { HttpClient } from '../HttpClient';
+import { Indent } from '../Indent';
+import { writeFile } from './fileSystem';
+import { formatCode } from './formatCode';
+import { formatIndentation } from './formatIndentation';
+import { registerHandlebarTemplates } from './registerHandlebarTemplates';
+
+export const writeClientServicesCustomTemplate = async (
+    client: Client,
+    outputPath: string,
+    httpClient: HttpClient,
+    useOptions: boolean,
+    useUnionTypes: boolean,
+    indent: Indent,
+    postfix: string,
+    templatePath: string
+) => {
+    registerHandlebarTemplates({
+        httpClient,
+        useUnionTypes,
+        useOptions,
+        handlebars: Handlebars, // since we're not using precompiled templates, we need a different object here
+    });
+    Handlebars.registerHelper('capitalize', str => {
+        return str.charAt(0).toUpperCase() + str.slice(1);
+    });
+
+    const serviceTemplate = Handlebars.compile(await readFile(templatePath, 'utf8'));
+
+    const servicesDir = resolve(outputPath, 'services');
+    await remove(servicesDir);
+    await mkdir(servicesDir);
+
+    for (const service of client.services) {
+        const file = resolve(outputPath, `services/${service.name}${postfix}.ts`);
+        const templateResult = serviceTemplate({
+            ...service,
+            serviceBaseUrl: client.server,
+            httpClient,
+            useUnionTypes,
+            useOptions,
+            postfix,
+        });
+        await writeFile(file, formatIndentation(formatCode(templateResult), indent));
+    }
+};