feat: Ability to set a SourceFileStructure's bodyText using a writer …

…function.
dsherret · Apr 30, 2018 · 792c530 · 792c530
1 parent 585793c
commit 792c530
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/src/structurePrinters/decorator/DecoratorStructurePrinter.ts b/src/structurePrinters/decorator/DecoratorStructurePrinter.ts
@@ -24,7 +24,7 @@ export class DecoratorStructurePrinter extends FactoryStructurePrinter<Decorator
         writer.write("(");
         for (let i = 0; i < structure.arguments.length; i++) {
             writer.conditionalWrite(i > 0, ", ");
-            printTextFromStringOrWriter(writer, structure.arguments[0]);
+            printTextFromStringOrWriter(writer, structure.arguments[i]);
         }
         writer.write(")");
     }

diff --git a/src/structurePrinters/statement/BodyTextStructurePrinter.ts b/src/structurePrinters/statement/BodyTextStructurePrinter.ts
@@ -4,7 +4,7 @@ import { ArrayUtils } from "../../utils";
 import { FactoryStructurePrinter } from "../FactoryStructurePrinter";
 import { StructurePrinterFactory } from "../../factories";
 
-export type BodyTextStructures = StatementedNodeStructure | { bodyText?: string; };
+export type BodyTextStructures = StatementedNodeStructure | { bodyText?: string | ((writer: CodeBlockWriter) => void); };
 
 export class BodyTextStructurePrinter extends FactoryStructurePrinter<BodyTextStructures> {
     constructor(factory: StructurePrinterFactory, private readonly options: { isAmbient: boolean; }) {

diff --git a/src/structures/file/SourceFileStructure.ts b/src/structures/file/SourceFileStructure.ts
@@ -1,9 +1,10 @@
-import { StatementedNodeStructure } from "../statement";
+import { CodeBlockWriter } from "../../codeBlockWriter";
+import { StatementedNodeStructure } from "../statement";
 import { ImportDeclarationStructure } from "./ImportDeclarationStructure";
 import { ExportDeclarationStructure } from "./ExportDeclarationStructure";
 
 export interface SourceFileStructure extends SourceFileSpecificStructure, StatementedNodeStructure {
-    bodyText?: string;
+    bodyText?: string | ((writer: CodeBlockWriter) => void);
 }
 
 export interface SourceFileSpecificStructure {

diff --git a/src/tests/projectTests.ts b/src/tests/projectTests.ts
@@ -478,9 +478,10 @@ describe(nameof(Project), () => {
                     name: "MyEnum"
                 }],
                 imports: [{ moduleSpecifier: "./test" }],
-                exports: [{ moduleSpecifier: "./test" }]
+                exports: [{ moduleSpecifier: "./test" }],
+                bodyText: writer => writer.write("print('test');")
             });
-            expect(sourceFile.getFullText()).to.equal(`import "./test";\n\nenum MyEnum {\n}\n\nexport * from "./test";\n`);
+            expect(sourceFile.getFullText()).to.equal(`import "./test";\n\nenum MyEnum {\n}\n\nprint('test');\n\nexport * from "./test";\n`);
         });
 
         it("should add for everything in the structure", () => {