Fix the operation comment

Author: mitchell-merry

src/index.ts

@@ -182,8 +182,9 @@ async function openapiTS(
   // 3. operations (only get fully built after all external schemas transformed)
   if (Object.keys(ctx.operations).length) {
     output.push(options.exportType ? "export type operations = {" : "export interface operations {", "");
-    for (const k of Object.keys(ctx.operations)) {
-      output.push(indent(`${escObjKey(k)}: ${ctx.operations[k]};`, 1));
+    for (const [ key, { operationType, comment }] of Object.entries(ctx.operations)) {
+      if (comment) output.push(indent(comment, 1));
+      output.push(indent(`${escObjKey(key)}: ${operationType};`, 1));
     }
     output.push(`}${options.exportType ? ";" : ""}`, "");
   } else {

src/transform/operation-object.ts

@@ -30,8 +30,6 @@ export default function transformOperationObject(
   let { indentLv } = ctx;
   const output: string[] = wrapObject ? ["{"] : [];
   indentLv++;
-  const c = getSchemaObjectComment(operationObject, indentLv);
-  if (c) output.push(indent(c, indentLv));
 
   // parameters
   {

src/transform/path-item-object.ts

@@ -30,7 +30,10 @@ export default function transformPathItemObject(
     // if operationId exists, move into an `operations` export and pass the reference in here
     else if (operationObject.operationId) {
       const operationType = transformOperationObject(operationObject, { path, ctx: { ...ctx, indentLv: 1 } });
-      ctx.operations[operationObject.operationId] = operationType;
+      ctx.operations[operationObject.operationId] = { 
+        operationType,
+        comment: c,
+      };
       output.push(indent(`${method}: operations[${escStr(operationObject.operationId)}];`, indentLv));
     } else {
       const operationType = transformOperationObject(operationObject, { path, ctx: { ...ctx, indentLv } });

src/types.ts

@@ -638,7 +638,10 @@ export interface GlobalContext {
   postTransform: OpenAPITSOptions["postTransform"];
   immutableTypes: boolean;
   indentLv: number;
-  operations: Record<string, string>;
+  operations: Record<string, {
+    comment?: string;
+    operationType: string;
+  }>;
   pathParamsAsTypes: boolean;
   silent: boolean;
   supportArrayLength: boolean;

test/path-item-object.test.ts

@@ -1,4 +1,4 @@
-import type { PathItemObject } from "../src/types";
+import type { GlobalContext, PathItemObject } from "../src/types";
 import transformPathItemObject, { TransformPathItemObjectOptions } from "../src/transform/path-item-object.js";
 
 const options: TransformPathItemObjectOptions = {
@@ -23,6 +23,7 @@ describe("Path Item Object", () => {
   test("basic", () => {
     const schema: PathItemObject = {
       get: {
+        description: "Basic GET",
         responses: {
           200: {
             description: "OK",
@@ -57,6 +58,7 @@ describe("Path Item Object", () => {
         },
       },
       post: {
+        description: "Basic POST",
         requestBody: {
           content: {
             "application/json": { $ref: 'components["schemas"]["User"]' },
@@ -70,6 +72,7 @@ describe("Path Item Object", () => {
     };
     const generated = transformPathItemObject(schema, options);
     expect(generated).toBe(`{
+  /** @description Basic GET */
   get: {
     responses: {
       /** @description OK */
@@ -92,6 +95,7 @@ describe("Path Item Object", () => {
       };
     };
   };
+  /** @description Basic POST */
   post: {
     requestBody?: {
       content: {
@@ -107,9 +111,10 @@ describe("Path Item Object", () => {
   });
 
   test("operations", () => {
-    const operations: Record<string, string> = {};
+    const operations: GlobalContext['operations'] = {};
     const schema: PathItemObject = {
       get: {
+        description: "Get a user",
         operationId: "getUser",
         responses: {
           200: {
@@ -132,10 +137,13 @@ describe("Path Item Object", () => {
     };
     const generated = transformPathItemObject(schema, { ...options, ctx: { ...options.ctx, operations } });
     expect(generated).toBe(`{
+  /** @description Get a user */
   get: operations["getUser"];
 }`);
     expect(operations).toEqual({
-      getUser: `{
+      getUser: {
+        comment: "/** @description Get a user */",
+        operationType: `{
     responses: {
       /** @description Get User */
       200: {
@@ -148,6 +156,7 @@ describe("Path Item Object", () => {
       };
     };
   }`,
+      }        
     });
   });
 });