@grafana/openapi-to-k6 0.1.2 → 0.2.0

package/.github/workflows/publish.yaml

@@ -19,6 +19,8 @@ jobs:
           node-version: '21'
           registry-url: 'https://registry.npmjs.org'
           scope: '@grafana'
+          cache: 'npm'
+          cache-dependency-path: 'package-lock.json'
       - run: npm ci
       - name: Update version in package.json
         run: |

package/.github/workflows/tests.yaml

@@ -12,7 +12,9 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-node@v4
         with:
-          node-version: '21.5.0'
+          node-version: '21'
+          cache: 'npm'
+          cache-dependency-path: 'package-lock.json'
       - run: npm ci
       - run: npm run lint
       - run: npm test
@@ -22,10 +24,15 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-node@v4
         with:
-          node-version: '21.5.0'
+          node-version: '21'
+          cache: 'npm'
+          cache-dependency-path: 'package-lock.json'
       - run: npm ci
       - name: Generate SDK for E2E tests
-        run: npm run dev -- ./tests/e2e/schema.json ./tests/e2e/sdk.ts --disable-analytics
+        run: |
+          npm run dev -- ./tests/e2e/schema.json ./tests/e2e/single/sdk.ts --disable-analytics --mode single
+          npm run dev -- ./tests/e2e/schema.json ./tests/e2e/split/sdk.ts --disable-analytics --mode split
+          npm run dev -- ./tests/e2e/schema.json ./tests/e2e/tags/sdk.ts --disable-analytics --mode tags
       - name: Start Mockoon CLI
         uses: mockoon/cli-action@v2
         with:
@@ -35,17 +42,18 @@ jobs:
       - uses: grafana/setup-k6-action@v1
       - uses: grafana/run-k6-action@v1
         with:
-          path: './tests/e2e/k6Script.ts'
+          path: './tests/e2e/*/k6Script.ts'
           flags: '--compatibility-mode=experimental_enhanced'
           inspect-flags: '--compatibility-mode=experimental_enhanced'
   run-examples:
     runs-on: ubuntu-latest
-    needs: [test, e2e-test]
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-node@v4
         with:
-          node-version: '21.5.0'
+          node-version: '21'
+          cache: 'npm'
+          cache-dependency-path: 'package-lock.json'
       - run: npm ci
       - name: Update examples
         run: npm run update-examples

package/README.md

@@ -1,10 +1,15 @@
-# openapi-to-k6
-
-<p align="center">⚠️</p>
-
-***This tool is currently in the experimental stage. Expect bugs, incomplete features, and breaking changes as development progresses. Use at your own risk, and please report any issues or feedback to help us improve.***
-
----
+<div align="center">
+
+  <img
+    src="./images/openapi-to-k6.png"
+    width="600"
+    style="pointer-events: none;" />
+  <br />
+  <hr/>
+  <p align="center">⚠️</p>
+<b>This tool is currently in the experimental stage. Expect bugs, incomplete features, and breaking changes as development progresses. Use at your own risk, and please report any issues or feedback to help us improve.</b>
+<hr/>
+</div>
 
 ## Overview
 
@@ -17,16 +22,46 @@ OpenAPI documentation.
 
 Along with the client, it also generates a sample k6 script as an example of how to use the client.
 
-To get started, install the tool with npm via `npm install openapi-to-k6` and run it to convert your
-OpenAPI specification to a TypeScript client for k6.
+The generated client exports a class with methods for each endpoint in the OpenAPI specification. You can create
+a instance of the class and use the methods to call the endpoints.
 
-To take a look at a few examples of how the generated client and sample script looks, check out the [examples](./examples) directory.
+To take a look at a few examples of how the generated client looks and sample script looks, check out the [examples](./examples) directory.
 
-Note: Optional usage analytics are gathered to make the tool better. To disable this, use the option
-`--disable-analytics` or set an environment variable `DISABLE_ANALYTICS=true`.
 
 ## Getting started
 
+1. Install the tool globally via
+
+    ```shell
+    npm install -g @grafana/openapi-to-k6
+    ```
+
+2. To start using the tool either give path to your OpenAPI schema file or provide a URL to your Open
+   API schema and the output path where you want to generate the client files.
+
+    ```shell
+    openapi-to-k6 <path-to-openapi-schema | url-to-openapi-schema> <output path>
+    ```
+
+    This will the generate a TypeScript client and a sample k6 script in the corresponding directory.
+
+### Options
+
+Following are some of the configuration options supported by the tool.
+
+1. `--mode` or ` -m`: Specify the mode to use for generating the client. Following are available options:
+   1. `single`: This is the default mode used is nothing is specified. It generated the TypeScript client as a single file with all the types and implementation in a single file.
+   2. `split`: This mode splits the types and implementation into separate files.
+   3. `tags`: This modes splits your OpenAPI schema based on the tags and generates a separate client for each tag. If a route has no tag set, it will be available in `default.ts` file.
+
+   To check how the output looks for each mode, check out the [examples](./examples) directory.
+2. `--disable-analytics`: Disable anonymous usage analytics reporting which helping making the tool better. You can also set an environment variable `DISABLE_ANALYTICS=true` to disable the analytics.
+3. `--disable-sample-script`: Disable the generation of sample k6 script.
+4. `--verbose` or ` -v` : Enable verbose logging to see more detailed logging output.
+5. `--help` or ` -h` : Show help message.
+
+## Developing locally
+
 1. Clone the repository
 
 ```shell
@@ -69,7 +104,7 @@ cd tests/e2e/
 This will run the mock server in a docker container in background.
 
 ```shell
-docker run -d -v ./schema.json:/tmp/schema.json -p 3000:3000 mockoon/cli:latest -d /tmp/schema.json
+docker run -v ./schema.json:/tmp/schema.json -p 3000:3000 mockoon/cli:latest -d /tmp/schema.json
 ```
 
 3. Assuming you have already followed previous steps and have the environment set up, you can generate the SDK by using
@@ -90,4 +125,4 @@ k6 run --compatibility-mode=experimental_enhanced ./K6Script.ts
 2. Install the compiled package locally by using `npm install .` or `npm install -g .`.
 3. Use the CLI `k6-sdkgen <path-to-openapi-schema> <output path>`
 
-Special thanks for the the open-source library [Orval](https://orval.dev/) to facilitate the generation of these SDK.
+Special mention for the the open-source library [Orval](https://orval.dev/) which is used for the generation of the TypeScript client.

package/dist/cli.js

@@ -16,13 +16,26 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const chalk_1 = __importDefault(require("chalk"));
 const commander_1 = require("commander");
 const analytics_1 = require("./analytics");
+const constants_1 = require("./constants");
 const generator_1 = __importDefault(require("./generator"));
 const helper_1 = require("./helper");
 const logger_1 = require("./logger");
 const program = new commander_1.Command();
 const packageDetails = (0, helper_1.getPackageDetails)();
+/**
+ * Validate that the mode argument is one of the supported modes.
+ *
+ * @param {string} value - The mode value to validate
+ * @return {Mode} - The validated mode value
+ */
+function validateMode(value) {
+    if (!Object.values(constants_1.Mode).includes(value)) {
+        throw new commander_1.InvalidArgumentError(`Supported modes: ${Object.values(constants_1.Mode).join(', ')}`);
+    }
+    return value;
+}
 function generateSDK(_a) {
-    return __awaiter(this, arguments, void 0, function* ({ openApiPath, outputDir, shouldGenerateSampleK6Script, analyticsData, }) {
+    return __awaiter(this, arguments, void 0, function* ({ openApiPath, outputDir, shouldGenerateSampleK6Script, analyticsData, mode, }) {
         logger_1.logger.logMessage('Generating TypeScript client for k6...\n');
         logger_1.logger.logMessage(`OpenAPI schema: ${openApiPath}`);
         logger_1.logger.logMessage(`Output: ${outputDir}\n`);
@@ -31,6 +44,7 @@ function generateSDK(_a) {
             outputDir,
             shouldGenerateSampleK6Script,
             analyticsData,
+            mode,
         });
         if (shouldGenerateSampleK6Script) {
             logger_1.logger.logMessage(`TypeScript client and sample k6 script generated successfully.`, chalk_1.default.green);
@@ -46,8 +60,9 @@ program
     .version(packageDetails.version)
     .argument('<openApiPath>', 'Path or URL for the OpenAPI schema file')
     .argument('<outputDir>', 'Directory where the SDK should be generated')
+    .option('-m, --mode <string>', `mode to use for generating the client. Valid values - ${Object.values(constants_1.Mode).join(', ')}`, validateMode, constants_1.Mode.SINGLE)
     .option('-v, --verbose', 'enable verbose mode to show debug logs')
-    .option('--no-sample-script', 'disable generating sample k6 script')
+    .option('--disable-sample-script', 'disable generating sample k6 script')
     .option('--disable-analytics', 'disable anonymous usage data collection')
     .action((openApiPath, outputDir, options) => __awaiter(void 0, void 0, void 0, function* () {
     let analyticsData;
@@ -75,6 +90,7 @@ program
             outputDir,
             shouldGenerateSampleK6Script: !shouldDisableSampleScript,
             analyticsData,
+            mode: options.mode,
         });
     }
     catch (error) {

package/dist/constants.js

@@ -1,13 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.K6_SCRIPT_TEMPLATE = exports.SAMPLE_K6_SCRIPT_FILE_NAME = exports.DEFAULT_SCHEMA_TITLE = void 0;
+exports.Mode = exports.K6_SCRIPT_TEMPLATE = exports.SAMPLE_K6_SCRIPT_FILE_NAME = exports.DEFAULT_SCHEMA_TITLE = void 0;
 exports.DEFAULT_SCHEMA_TITLE = 'K6Client';
 exports.SAMPLE_K6_SCRIPT_FILE_NAME = 'k6-script.sample.ts';
 exports.K6_SCRIPT_TEMPLATE = `
 import { {{clientFunctionName}} } from '{{clientPath}}'
 
 const baseUrl = '<BASE_URL>';
-const client = {{clientFunctionName}}({ baseUrl })
+const client = new {{clientFunctionName}}({ baseUrl })
 
 export default function () {
     {{#each clientFunctionsList}}
@@ -19,3 +19,9 @@ export default function () {
     {{/each}}
 }
 `;
+var Mode;
+(function (Mode) {
+    Mode["SINGLE"] = "single";
+    Mode["SPLIT"] = "split";
+    Mode["TAGS"] = "tags";
+})(Mode || (exports.Mode = Mode = {}));

package/dist/{generator.js → generator/index.js}

@@ -15,10 +15,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const fs_1 = __importDefault(require("fs"));
 const orval_1 = __importDefault(require("orval"));
 const path_1 = __importDefault(require("path"));
-const constants_1 = require("./constants");
-const helper_1 = require("./helper");
-const k6SdkClient_1 = require("./k6SdkClient");
-const logger_1 = require("./logger");
+const constants_1 = require("../constants");
+const helper_1 = require("../helper");
+const logger_1 = require("../logger");
+const k6Client_1 = require("./k6Client");
 const outputOverrider = helper_1.OutputOverrider.getInstance();
 const packageDetails = (0, helper_1.getPackageDetails)();
 const generatedFileHeaderGenerator = (info) => {
@@ -44,10 +44,7 @@ const afterAllFilesWriteHandler = (filePaths) => __awaiter(void 0, void 0, void
         }
     }
 });
-exports.default = (_a) => __awaiter(void 0, [_a], void 0, function* ({ openApiPath, outputDir, shouldGenerateSampleK6Script, analyticsData, }) {
-    const schemaDetails = {
-        title: '',
-    };
+exports.default = (_a) => __awaiter(void 0, [_a], void 0, function* ({ openApiPath, outputDir, shouldGenerateSampleK6Script, analyticsData, mode, }) {
     /**
      * Note!
      * 1. override.requestOptions is not supported for the custom K6 client
@@ -58,8 +55,8 @@ exports.default = (_a) => __awaiter(void 0, [_a], void 0, function* ({ openApiPa
             input: openApiPath,
             output: {
                 target: outputDir,
-                mode: 'single',
-                client: () => (0, k6SdkClient_1.getK6ClientBuilder)(schemaDetails, shouldGenerateSampleK6Script, analyticsData),
+                mode: mode,
+                client: () => (0, k6Client_1.getK6ClientBuilder)(shouldGenerateSampleK6Script, analyticsData),
                 override: {
                     header: generatedFileHeaderGenerator,
                 },
@@ -70,7 +67,4 @@ exports.default = (_a) => __awaiter(void 0, [_a], void 0, function* ({ openApiPa
             },
         });
     }));
-    if (!schemaDetails.title) {
-        logger_1.logger.warning('Could not find schema title in the OpenAPI spec. Please provide a `title` in the schema in `info` block to generate proper file names');
-    }
 });

package/dist/{k6SdkClient.js → generator/k6Client.js}

@@ -12,17 +12,119 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateFooter = exports.generateK6Header = exports.generateTitle = exports.getK6Dependencies = void 0;
 exports.getK6ClientBuilder = getK6ClientBuilder;
 const core_1 = require("@orval/core");
 const handlebars_1 = __importDefault(require("handlebars"));
 const path_1 = __importDefault(require("path"));
-const constants_1 = require("./constants");
-const helper_1 = require("./helper");
-const logger_1 = require("./logger");
-// A map to store the operationNames for which a return type is to be written at the end to export
-// and the return type definition
-const returnTypesToWrite = new Map();
+const constants_1 = require("../constants");
+const helper_1 = require("../helper");
+const logger_1 = require("../logger");
+/**
+ * In case the supplied schema does not have a title set, it will set the default title to ensure
+ * proper client generation
+ *
+ * @param context - The context object containing the schema details
+ */
+function _setDefaultSchemaTitle(context) {
+    const schemaDetails = context.specs[context.specKey];
+    if (schemaDetails && !schemaDetails.info.title) {
+        schemaDetails.info.title = constants_1.DEFAULT_SCHEMA_TITLE;
+    }
+}
+function _generateResponseTypeDefinition(response) {
+    let responseDataType = '';
+    if (response.definition.success &&
+        !['any', 'unknown'].includes(response.definition.success)) {
+        responseDataType += response.definition.success;
+    }
+    else {
+        responseDataType += 'ResponseBody';
+    }
+    return `{
+    response: Response
+    data: ${responseDataType}
+}`;
+}
+function _getRequestParametersMergerFunctionImplementation() {
+    return `/**
+ * Merges the provided request parameters with default parameters for the client.
+ *
+ * @param {Params} requestParameters - The parameters provided specifically for the request
+ * @param {Params} commonRequestParameters - Common parameters for all requests
+ * @returns {Params} - The merged parameters
+ */
+  private _mergeRequestParameters (requestParameters?: Params, commonRequestParameters?: Params): Params {
+    return {
+        ...commonRequestParameters,  // Default to common parameters
+        ...requestParameters,        // Override with request-specific parameters
+        headers: {
+            ...commonRequestParameters?.headers || {},  // Ensure headers are defined
+            ...requestParameters?.headers || {},
+        },
+        cookies: {
+            ...commonRequestParameters?.cookies || {},  // Ensure cookies are defined
+            ...requestParameters?.cookies || {},
+        },
+        tags: {
+            ...commonRequestParameters?.tags || {},     // Ensure tags are defined
+            ...requestParameters?.tags || {},
+        },
+    };
+};`;
+}
+const _getRequestParamsValue = ({ response, queryParams, headers, body, }) => {
+    if (!queryParams && !headers && !response.isBlob && !body.contentType) {
+        // No parameters to merge, return the request parameters directly
+        return 'mergedRequestParameters';
+    }
+    let value = '\n    ...mergedRequestParameters,';
+    if (response.isBlob) {
+        value += `\n        responseType: 'binary',`;
+    }
+    // Expand the headers
+    if (body.contentType || headers) {
+        let headersValue = `\n       headers: {`;
+        if (body.contentType) {
+            if (body.formData) {
+                headersValue += `\n'Content-Type': '${body.contentType}; boundary=' + formData.boundary,`;
+            }
+            else {
+                headersValue += `\n'Content-Type': '${body.contentType}',`;
+            }
+        }
+        if (headers) {
+            headersValue += `\n// In the schema, headers can be of any type like number but k6 accepts only strings as headers, hence converting all headers to string`;
+            headersValue += `\n...Object.fromEntries(Object.entries(headers || {}).map(([key, value]) => [key, String(value)])),`;
+        }
+        headersValue += `\n...mergedRequestParameters?.headers},`;
+        value += headersValue;
+    }
+    return `{${value}}`;
+};
+const _getK6RequestOptions = (verbOptions) => {
+    const { body, headers, queryParams, response, verb } = verbOptions;
+    let fetchBodyOption = 'undefined';
+    if (body.formData) {
+        // Use the FormData.body() method to get the body of the request
+        fetchBodyOption = 'formData.body()';
+    }
+    else if (body.formUrlEncoded || body.implementation) {
+        fetchBodyOption = `JSON.stringify(${body.implementation})`;
+    }
+    // Generate the params input for the call
+    const requestParametersValue = _getRequestParamsValue({
+        response,
+        body,
+        headers: headers === null || headers === void 0 ? void 0 : headers.schema,
+        queryParams: queryParams === null || queryParams === void 0 ? void 0 : queryParams.schema,
+    });
+    // Sample output
+    // 'GET', 'http://test.com/route', <body>, <options>
+    return `"${verb.toUpperCase()}",
+        url.toString(),
+        ${fetchBodyOption},
+        ${requestParametersValue}`;
+};
 const getK6Dependencies = () => [
     {
         exports: [
@@ -67,32 +169,8 @@ const getK6Dependencies = () => [
         dependency: 'https://jslib.k6.io/formdata/0.0.2/index.js',
     },
 ];
-exports.getK6Dependencies = getK6Dependencies;
-function getSchemaTitleFromContext(context) {
-    const specData = Object.values(context.specs);
-    let schemaTitle;
-    if (specData[0]) {
-        schemaTitle = specData[0].info.title;
-    }
-    schemaTitle !== null && schemaTitle !== void 0 ? schemaTitle : (schemaTitle = constants_1.DEFAULT_SCHEMA_TITLE);
-    return schemaTitle;
-}
-function _generateResponseTypeName(operationName) {
-    return `${(0, core_1.pascal)(operationName)}Response`;
-}
-function _generateResponseTypeDefinition(operationName, response) {
-    const typeName = _generateResponseTypeName(operationName);
-    let responseDataType = '';
-    if (response.definition.success) {
-        responseDataType += response.definition.success + ' | ';
-    }
-    responseDataType += 'ResponseBody';
-    return `export type ${typeName} = {
-    response: Response
-    data: ${responseDataType}
-};`;
-}
-const generateK6Implementation = ({ headers, queryParams, operationName, response, body, props, verb, override, formData, formUrlEncoded, paramsSerializer, }, { route }, analyticsData) => {
+const generateK6Implementation = (verbOptions, { route }, analyticsData) => {
+    const { queryParams, operationName, response, body, props, verb, formData, formUrlEncoded, } = verbOptions;
     if (analyticsData) {
         analyticsData.generatedRequestsCount[verb] += 1;
     }
@@ -103,33 +181,21 @@ const generateK6Implementation = ({ headers, queryParams, operationName, respons
         isFormData: true,
         isFormUrlEncoded: false,
     });
-    // Generate response return types
-    returnTypesToWrite.set(operationName, _generateResponseTypeDefinition(operationName, response));
-    let url = `cleanBaseUrl + \`${route}\``;
+    let url = `this.cleanBaseUrl + \`${route}\``;
     if (queryParams) {
         url += '+`?${new URLSearchParams(params).toString()}`';
     }
     const urlGeneration = `const url = new URL(${url});`;
-    const options = getK6RequestOptions({
-        route,
-        body,
-        headers,
-        queryParams,
-        response,
-        verb,
-        requestOptions: override === null || override === void 0 ? void 0 : override.requestOptions,
-        paramsSerializer,
-        paramsSerializerOptions: override === null || override === void 0 ? void 0 : override.paramsSerializerOptions,
-    });
-    return `const ${operationName} = (\n    ${(0, core_1.toObjectString)(props, 'implementation')} requestParameters?: Params): ${_generateResponseTypeName(operationName)} => {${bodyForm}
+    const options = _getK6RequestOptions(verbOptions);
+    return `${operationName}(\n    ${(0, core_1.toObjectString)(props, 'implementation')} requestParameters?: Params): ${_generateResponseTypeDefinition(response)} {\n${bodyForm}
         ${urlGeneration}
-        const mergedRequestParameters = _mergeRequestParameters(requestParameters || {}, clientOptions.commonRequestParameters);
+        const mergedRequestParameters = this._mergeRequestParameters(requestParameters || {}, this.commonRequestParameters);
         const response = http.request(${options});
         let data;
 
         try {
             data = response.json();
-        } catch (error) {
+        } catch {
             data = response.body;
         }
       return {
@@ -139,120 +205,41 @@ const generateK6Implementation = ({ headers, queryParams, operationName, respons
     }
   `;
 };
-const getParamsInputValue = ({ response, queryParams, headers, body, }) => {
-    if (!queryParams && !headers && !response.isBlob && !body.contentType) {
-        // No parameters to merge, return the request parameters directly
-        return 'mergedRequestParameters';
-    }
-    let value = '\n    ...mergedRequestParameters,';
-    if (response.isBlob) {
-        value += `\n        responseType: 'binary',`;
-    }
-    // Expand the headers
-    if (body.contentType || headers) {
-        let headersValue = `\n       headers: {`;
-        if (body.contentType) {
-            if (body.formData) {
-                headersValue += `\n'Content-Type': '${body.contentType}; boundary=' + formData.boundary,`;
-            }
-            else {
-                headersValue += `\n'Content-Type': '${body.contentType}',`;
-            }
-        }
-        if (headers) {
-            headersValue += `\n...headers,`;
-        }
-        headersValue += `\n...mergedRequestParameters?.headers},`;
-        value += headersValue;
-    }
-    return `{${value}}`;
-};
-const getK6RequestOptions = (options) => {
-    const { body, headers, queryParams, response, verb } = options;
-    let fetchBodyOption = 'undefined';
-    if (body.formData) {
-        // Use the FormData.body() method to get the body of the request
-        fetchBodyOption = 'formData.body()';
-    }
-    else if (body.formUrlEncoded || body.implementation) {
-        fetchBodyOption = `JSON.stringify(${body.implementation})`;
-    }
-    // Generate the params input for the call
-    const paramsValue = getParamsInputValue({
-        response,
-        body,
-        headers: headers === null || headers === void 0 ? void 0 : headers.schema,
-        queryParams: queryParams === null || queryParams === void 0 ? void 0 : queryParams.schema,
-    });
-    // Sample output
-    // 'GET', 'http://test.com/route', <body>, <options>
-    return `"${verb.toUpperCase()}",
-        url.toString(),
-        ${fetchBodyOption},
-        ${paramsValue}`;
-};
-function _getRequestParametersMergerFunctionImplementation() {
-    return `/**
- * Merges the provided request parameters with default parameters for the client.
- *
- * @param {Params} requestParameters - The parameters provided specifically for the request
- * @param {Params} commonRequestParameters - Common parameters for all requests
- * @returns {Params} - The merged parameters
- */
-  const _mergeRequestParameters = (requestParameters?: Params, commonRequestParameters?: Params): Params => {
-    return {
-        ...commonRequestParameters,  // Default to common parameters
-        ...requestParameters,        // Override with request-specific parameters
-        headers: {
-            ...commonRequestParameters?.headers || {},  // Ensure headers are defined
-            ...requestParameters?.headers || {},
-        },
-        cookies: {
-            ...commonRequestParameters?.cookies || {},  // Ensure cookies are defined
-            ...requestParameters?.cookies || {},
-        },
-        tags: {
-            ...commonRequestParameters?.tags || {},     // Ensure tags are defined
-            ...requestParameters?.tags || {},
-        },
-    };
-};`;
-}
 const generateTitle = (title) => {
     const sanTitle = (0, core_1.sanitize)(title || constants_1.DEFAULT_SCHEMA_TITLE);
-    return `create${(0, core_1.pascal)(sanTitle)}`;
+    return `${(0, core_1.pascal)(sanTitle)}Client`;
 };
-exports.generateTitle = generateTitle;
 const generateK6Header = ({ title }) => {
-    const clientOptionsTypeName = `${(0, core_1.pascal)(title)}Options`;
     return `
-  export type ${clientOptionsTypeName} = {
+  /**
+   * This is the base client to use for interacting with the API.
+   */
+  export class ${title} {
+      private cleanBaseUrl: string;
+      private commonRequestParameters: Params;
+
+      constructor (clientOptions: {
     baseUrl: string,
     commonRequestParameters?: Params
-  }
-
-  /**
- * This is the base client to use for interacting with the API.
- */
-  export const ${title} = (clientOptions: ${clientOptionsTypeName}) => {\n
-        const cleanBaseUrl = clientOptions.baseUrl.replace(/\\/+$/, '');\n`;
+}) {
+       this.cleanBaseUrl = clientOptions.baseUrl.replace(/\\/+$/, '');\n
+      }\n
+`;
 };
-exports.generateK6Header = generateK6Header;
-const generateFooter = ({ operationNames }) => {
-    let footer = '';
-    footer += `return {${operationNames.join(',')}}};\n\n`;
-    operationNames.forEach((operationName) => {
-        if (returnTypesToWrite.has(operationName)) {
-            footer += returnTypesToWrite.get(operationName) + '\n';
-        }
-    });
+const generateFooter = () => {
     // Add function definition for merging request parameters
-    footer += `\n\n${_getRequestParametersMergerFunctionImplementation()}\n`;
+    const footer = `
+
+  ${_getRequestParametersMergerFunctionImplementation()}
+
+}
+
+  `;
     return footer;
 };
-exports.generateFooter = generateFooter;
 const k6ScriptBuilder = (verbOptions, output, context) => __awaiter(void 0, void 0, void 0, function* () {
-    const schemaTitle = getSchemaTitleFromContext(context);
+    var _a;
+    const schemaTitle = ((_a = context.specs[context.specKey]) === null || _a === void 0 ? void 0 : _a.info.title) || constants_1.DEFAULT_SCHEMA_TITLE;
     const { path: pathOfGeneratedClient, filename, extension, } = yield (0, helper_1.getGeneratedClientPath)(output.target, schemaTitle);
     const directoryPath = (0, helper_1.getDirectoryForPath)(pathOfGeneratedClient);
     const generateScriptPath = path_1.default.join(directoryPath, constants_1.SAMPLE_K6_SCRIPT_FILE_NAME);
@@ -275,7 +262,7 @@ const k6ScriptBuilder = (verbOptions, output, context) => __awaiter(void 0, void
         });
     }
     const scriptContentData = {
-        clientFunctionName: (0, exports.generateTitle)(schemaTitle),
+        clientFunctionName: generateTitle(schemaTitle),
         clientPath: `./${filename}${extension}`,
         clientFunctionsList,
     };
@@ -287,11 +274,11 @@ const k6ScriptBuilder = (verbOptions, output, context) => __awaiter(void 0, void
         },
     ];
 });
-function getK6Client(schemaDetails, analyticsData) {
+function getK6Client(analyticsData) {
     return function (verbOptions, options) {
+        _setDefaultSchemaTitle(options.context);
         const imports = (0, core_1.generateVerbImports)(verbOptions);
         const implementation = generateK6Implementation(verbOptions, options, analyticsData);
-        schemaDetails.title = getSchemaTitleFromContext(options.context);
         const specData = Object.values(options.context.specs);
         if (specData[0]) {
             if (analyticsData) {
@@ -301,13 +288,13 @@ function getK6Client(schemaDetails, analyticsData) {
         return { implementation, imports };
     };
 }
-function getK6ClientBuilder(schemaDetails, shouldGenerateSampleK6Script, analyticsData) {
+function getK6ClientBuilder(shouldGenerateSampleK6Script, analyticsData) {
     return {
-        client: getK6Client(schemaDetails, analyticsData),
-        header: exports.generateK6Header,
-        dependencies: exports.getK6Dependencies,
-        footer: exports.generateFooter,
-        title: exports.generateTitle,
+        client: getK6Client(analyticsData),
+        header: generateK6Header,
+        dependencies: getK6Dependencies,
+        footer: generateFooter,
+        title: generateTitle,
         extraFiles: shouldGenerateSampleK6Script ? k6ScriptBuilder : undefined,
     };
 }