make it an option

Author: mb21

README.md

@@ -1,13 +1,16 @@
 # OpenAPI Typescript Codegen
 
 Fork ([diff](https://github.com/ferdikoomen/openapi-typescript-codegen/compare/master...mb21:openapi-typescript-codegen:generate-custom))
-which exports a `generateCustom` function that can be used like:
+which adds a `--serviceTemplate` option.
 
-```js
-require('openapi-typescript-codegen').generateCustom('api.yaml', 'outputDir/', 'serviceTemplate.hbs')
+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.
+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.
 
 ---
 
@@ -61,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);

src/generateCustom.ts

@@ -8,8 +8,8 @@ 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 { generateCustom } from './generateCustom';
 export { HttpClient } from './HttpClient';
 export { Indent } from './Indent';
 
@@ -27,6 +27,7 @@ export type Options = {
     indent?: Indent;
     postfix?: string;
     request?: string;
+    serviceTemplate?: string;
     write?: boolean;
 };
 
@@ -63,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;
@@ -73,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,
@@ -99,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,
@@ -120,6 +127,19 @@ export const generate = async ({
             break;
         }
     }
+
+    if (serviceTemplate) {
+        await writeClientServicesCustomTemplate(
+            clientFinal,
+            output,
+            httpClient,
+            useOptions,
+            useUnionTypes,
+            indent,
+            postfix,
+            serviceTemplate
+        );
+    }
 };
 
 export default {

src/index.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));
+    }
+};