Load templates

packages/compiler/src/internals/init.ts

@@ -1,3 +1,2 @@
 export { NodeSystemHost } from "../core/node-system-host.js";
-export { getTypeSpecCoreTemplates } from "../init/core-templates.js";
 export { scaffoldNewProject } from "../init/scaffold.js";

packages/typespec-vscode/.gitignore

@@ -0,0 +1 @@
+templates/

packages/typespec-vscode/package.json

@@ -225,10 +225,11 @@
   },
   "scripts": {
     "clean": "rimraf ./dist ./temp",
-    "build": "npm run compile && npm run copy-tmlanguage && npm run generate-language-configuration && npm run generate-third-party-notices && npm run package-vsix",
+    "build": "npm run compile && npm run copy-templates && npm run copy-tmlanguage && npm run generate-language-configuration && npm run generate-third-party-notices && npm run package-vsix",
     "compile": "rollup --config rollup.config.ts --configPlugin typescript --failAfterWarnings 2>&1",
     "watch": "rollup --config rollup.config.ts --configPlugin typescript --watch",
     "dogfood": "node scripts/dogfood.js",
+    "copy-templates": "node scripts/copy-templates.js",
     "copy-tmlanguage": "node scripts/copy-tmlanguage.js",
     "generate-language-configuration": "node scripts/generate-language-configuration.js",
     "generate-third-party-notices": "typespec-build-tool generate-third-party-notices",

packages/typespec-vscode/scripts/copy-templates.js

@@ -0,0 +1,5 @@
+import { cp } from "fs/promises";
+
+await cp("node_modules/@typespec/compiler/templates", "../typespec-vscode/templates", {
+  recursive: true,
+});

packages/typespec-vscode/src/vscode-cmd/create-tsp-project.ts

@@ -1,20 +1,24 @@
 import {
+  SystemHost,
   type InitProjectConfig,
   type InitProjectTemplate,
   type InitProjectTemplateEmitterTemplate,
   type InitProjectTemplateLibrarySpec,
 } from "@typespec/compiler";
-import {
-  getTypeSpecCoreTemplates,
-  NodeSystemHost,
-  scaffoldNewProject,
-} from "@typespec/compiler/internals/init";
+import { NodeSystemHost, scaffoldNewProject } from "@typespec/compiler/internals/init";
 import { readdir } from "fs/promises";
 import * as semver from "semver";
+import { fileURLToPath } from "url";
 import vscode, { OpenDialogOptions, QuickPickItem } from "vscode";
 import { ExtensionStateManager } from "../extension-state-manager.js";
 import logger from "../log/logger.js";
-import { getBaseFileName, getDirectoryPath, joinPaths, normalizePath } from "../path-utils.js";
+import {
+  getBaseFileName,
+  getDirectoryPath,
+  joinPaths,
+  normalizePath,
+  resolvePath,
+} from "../path-utils.js";
 import { ResultCode, SettingName } from "../types.js";
 import {
   createPromiseWithCancelAndTimeout,
@@ -559,6 +563,37 @@ async function selectTemplate(
   return selected?.info;
 }
 
+async function findProjectRoot(host: SystemHost) {
+  let current = getDirectoryPath(fileURLToPath(import.meta.url));
+  while (true) {
+    const pkgPath = joinPaths(current, "package.json");
+    try {
+      const stat = await host.stat(pkgPath);
+      if (stat?.isFile()) {
+        return current;
+      }
+    } catch (e) {}
+    const parent = getDirectoryPath(current);
+    if (parent === current) {
+      throw new Error("Unexpected: Failed to resolve project root.");
+    }
+    current = parent;
+  }
+}
+
+async function getTypeSpecCoreTemplates(
+  host: SystemHost,
+): Promise<{ readonly baseUri: string; readonly templates: Record<string, any> }> {
+  const projectRoot = await findProjectRoot(host);
+  const templatesDir = resolvePath(projectRoot, "templates");
+  const file = await host.readFile(resolvePath(templatesDir, "scaffolding.json"));
+  const content = JSON.parse(file.text);
+  return {
+    baseUri: templatesDir,
+    templates: content,
+  };
+}
+
 async function loadInitTemplates(): Promise<Map<string, InitTemplateInfo[]>> {
   logger.info("Loading init templates from compiler...");
   const templateInfoMap: Map<string, InitTemplateInfo[]> = new Map();