updated param

added second write for also printing type
fern-api · Dec 18, 2024 · 57c9091 · 57c9091
1 parent 22e6bfc
commit 57c9091
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/generators/typescript-v2/ast/src/ast/Func.ts b/generators/typescript-v2/ast/src/ast/Func.ts
@@ -56,7 +56,7 @@ export class Func extends AstNode {
         writer.write("const ");
         writer.write(`${this.name}`);
         writer.write(" = ");
-        writer.delimit(this._parameters, ", ", (parameter) => parameter.write(writer), "(", ")");
+        writer.delimit(this._parameters, ", ", (parameter) => parameter.writeWithType(writer), "(", ")");
         if (this._return_.length > 0) {
             writer.write(": ");
             if (this._return_.length === 1) {

diff --git a/generators/typescript-v2/ast/src/ast/Method.ts b/generators/typescript-v2/ast/src/ast/Method.ts
@@ -31,7 +31,7 @@ export class Method extends Func {
         // TODO: Add visibility, make this nest in an object eventually
         writer.write(`function `);
         writer.write(`${this._name}`);
-        writer.delimit(this._parameters, ", ", (parameter) => parameter.write(writer), "(", ")");
+        writer.delimit(this._parameters, ", ", (parameter) => parameter.writeWithType(writer), "(", ")");
         if (this._return_.length > 0) {
             writer.write(": ");
             if (this._return_.length === 1) {

diff --git a/generators/typescript-v2/ast/src/ast/Parameter.ts b/generators/typescript-v2/ast/src/ast/Parameter.ts
@@ -24,6 +24,13 @@ export class Parameter extends AstNode {
     }
 
     public write(writer: Writer): void {
+        writer.write(`${this.name}`);
+    }
+
+    public writeWithType(writer: Writer): void {
+        if (this.docs != null) {
+            writer.writeNode(new Comment({ docs: this.docs }));
+        }
         writer.write(`${this.name}: `);
         this.type.write(writer);
     }