fix(project generator): Add join and CfnInclude imports only when…

… needed
guardian · Oct 5, 2023 · d43919a · d43919a
1 parent 16404a7
commit d43919a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/src/bin/commands/new-project/index.ts b/src/bin/commands/new-project/index.ts
@@ -117,7 +117,7 @@ export const newCdkProject = async (props: NewProjectProps): CliCommandResponse
 
   // lib directory
   await constructStack({
-    imports: Imports.newStackImports(),
+    imports: Imports.newStackImports(!!config.yamlTemplateLocation),
     appName: config.appName,
     outputFile: basename(config.stackPath),
     outputDir: dirname(config.stackPath),

diff --git a/src/bin/commands/new-project/utils/imports.ts b/src/bin/commands/new-project/utils/imports.ts
@@ -14,16 +14,18 @@ export class Imports {
     this.imports = imports;
   }
 
-  public static newStackImports(): Imports {
+  public static newStackImports(addCfnIncludeImports: boolean): Imports {
     return new Imports({
-      path: {
-        types: [],
-        components: ["join"],
-      },
-      "aws-cdk-lib/cloudformation-include": {
-        types: [],
-        components: ["CfnInclude"],
-      },
+      ...(addCfnIncludeImports && {
+        path: {
+          types: [],
+          components: ["join"],
+        },
+        "aws-cdk-lib/cloudformation-include": {
+          types: [],
+          components: ["CfnInclude"],
+        },
+      }),
       "aws-cdk-lib": {
         types: ["App"],
         components: [],