From dc383083e522ebb905eda843a236c388d64ded33 Mon Sep 17 00:00:00 2001
From: Ilia Choly <ilia.choly@gmail.com>
Date: Sun, 20 Oct 2024 10:50:05 -0400
Subject: [PATCH] feat(cli): add support for local templates

---
 .../cdktf-cli/src/bin/cmds/helper/init.ts     | 15 +++++---
 packages/cdktf-cli/src/bin/cmds/init.ts       |  2 +-
 .../typescript/init-local-template/cdktf.json |  8 +++++
 test/typescript/init-local-template/test.ts   | 35 +++++++++++++++++++
 4 files changed, 55 insertions(+), 5 deletions(-)
 create mode 100644 test/typescript/init-local-template/cdktf.json
 create mode 100644 test/typescript/init-local-template/test.ts

diff --git a/packages/cdktf-cli/src/bin/cmds/helper/init.ts b/packages/cdktf-cli/src/bin/cmds/helper/init.ts
index ed250e203b..0a42caae00 100644
--- a/packages/cdktf-cli/src/bin/cmds/helper/init.ts
+++ b/packages/cdktf-cli/src/bin/cmds/helper/init.ts
@@ -519,15 +519,22 @@ async function getTemplate(
     }
   }
 
-  // treat as remote url
-  if (!templates.includes(templateName)) {
-    return fetchRemoteTemplate(templateName);
-  } else {
+  if (templates.includes(templateName)) {
     return {
       Name: templateName,
       Path: path.join(templatesDir, templateName),
     };
   }
+  if (
+    templateName.startsWith("http://") ||
+    templateName.startsWith("https://")
+  ) {
+    return fetchRemoteTemplate(templateName);
+  }
+  return {
+    Name: path.basename(templateName),
+    Path: templateName,
+  };
 }
 
 async function fetchRemoteTemplate(templateUrl: string): Promise<Template> {
diff --git a/packages/cdktf-cli/src/bin/cmds/init.ts b/packages/cdktf-cli/src/bin/cmds/init.ts
index b54951c159..bb3d85eb5e 100644
--- a/packages/cdktf-cli/src/bin/cmds/init.ts
+++ b/packages/cdktf-cli/src/bin/cmds/init.ts
@@ -16,7 +16,7 @@ class Command extends BaseCommand {
       .showHelpOnFail(true)
       .option("template", {
         type: "string",
-        desc: `The template to be used to create a new project. Either URL to zip file or one of the built-in templates.`,
+        desc: `The template to be used to create a new project. Either URL to zip file, one of the built-in templates, or a local directory.`,
       })
       .option("project-name", {
         type: "string",
diff --git a/test/typescript/init-local-template/cdktf.json b/test/typescript/init-local-template/cdktf.json
new file mode 100644
index 0000000000..3b1e5ccf9d
--- /dev/null
+++ b/test/typescript/init-local-template/cdktf.json
@@ -0,0 +1,8 @@
+{
+  "language": "typescript",
+  "app": "npx ts-node main.ts",
+  "terraformProviders": [
+    "null@ ~> 3.1.0"
+  ],
+  "context": {}
+}
diff --git a/test/typescript/init-local-template/test.ts b/test/typescript/init-local-template/test.ts
new file mode 100644
index 0000000000..2811993602
--- /dev/null
+++ b/test/typescript/init-local-template/test.ts
@@ -0,0 +1,35 @@
+// Copyright (c) HashiCorp, Inc
+// SPDX-License-Identifier: MPL-2.0
+import { TestDriver } from "../../test-helper";
+const fs = require("fs").promises;
+const path = require("path");
+
+describe("local templates", () => {
+  test("can init", async () => {
+    const driver = new TestDriver(__dirname);
+    const templateDir = path.resolve(
+      __dirname,
+      "..",
+      "..",
+      "..",
+      "packages/@cdktf/cli-core/templates/typescript"
+    );
+    await driver.init(templateDir);
+    const files = await fs.readdir(driver.workingDirectory);
+    expect(files).toEqual(
+      expect.arrayContaining([
+        ".gen",
+        ".gitignore",
+        ".npmrc",
+        "cdktf.json",
+        "help",
+        "jest.config.js",
+        "main.ts",
+        "package.json",
+        "setup.js",
+        "tsconfig.json",
+        "__tests__",
+      ])
+    );
+  });
+});