feat: Add ability to add/insert type parameters with a default type n…

…ode.
dsherret · Jun 9, 2018 · cd05c3f · cd05c3f
1 parent 68fa2b1
commit cd05c3f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/src/structurePrinters/types/TypeParameterDeclarationStructurePrinter.ts b/src/structurePrinters/types/TypeParameterDeclarationStructurePrinter.ts
@@ -22,5 +22,7 @@ export class TypeParameterDeclarationStructurePrinter extends FactoryStructurePr
         writer.write(structure.name);
         if (structure.constraint != null && structure.constraint.length > 0)
             writer.write(` extends ${structure.constraint}`);
+        if (structure.default != null && structure.default.length > 0)
+            writer.write(` = ${structure.default}`);
     }
 }
diff --git a/src/structures/type/TypeParameterDeclarationStructure.ts b/src/structures/type/TypeParameterDeclarationStructure.ts
@@ -2,4 +2,5 @@
 
 export interface TypeParameterDeclarationStructure extends NamedNodeStructure {
     constraint?: string;
+    default?: string;
 }
diff --git a/src/tests/compiler/base/typeParameteredNodeTests.ts b/src/tests/compiler/base/typeParameteredNodeTests.ts
@@ -124,12 +124,17 @@ describe(nameof(TypeParameteredNode), () => {
             doTest("function identifier<T, U>() {}", 1, { name: "V", constraint: "string" }, "function identifier<T, V extends string, U>() {}");
         });
 
+        it("should insert with default", () => {
+            doTest("function identifier<T, U>() {}", 1, { name: "V", default: "string" }, "function identifier<T, V = string, U>() {}");
+        });
+
         it("should insert all the properties of the structure", () => {
             const structure: MakeRequired<TypeParameterDeclarationStructure> = {
                 name: "V",
-                constraint: "string"
+                constraint: "string",
+                default: "number"
             };
-            doTest("function identifier() {}", 0, structure, "function identifier<V extends string>() {}");
+            doTest("function identifier() {}", 0, structure, "function identifier<V extends string = number>() {}");
         });
     });