@azure-tools/typespec-ts 0.54.0 → 0.54.2

package/src/framework/load-static-helpers.ts

@@ -1,4 +1,5 @@
-import { readdir, stat, readFile } from "fs/promises";
+import { NoTarget, Program } from "@typespec/compiler";
+import { readdir, readFile, stat } from "fs/promises";
 import * as path from "path";
 import {
   ClassDeclaration,
@@ -7,14 +8,13 @@ import {
   InterfaceDeclaration,
   Project,
   SourceFile,
-  TypeAliasDeclaration
+  TypeAliasDeclaration,
 } from "ts-morph";
-import { refkey } from "./refkey.js";
-import { resolveProjectRoot } from "../utils/resolve-project-root.js";
-import { isAzurePackage } from "../rlc-common/index.js";
-import { ModularEmitterOptions } from "../modular/interfaces.js";
-import { NoTarget, Program } from "@typespec/compiler";
 import { reportDiagnostic } from "../lib.js";
+import { ModularEmitterOptions } from "../modular/interfaces.js";
+import { isAzurePackage } from "../rlc-common/index.js";
+import { resolveProjectRoot } from "../utils/resolve-project-root.js";
+import { refkey } from "./refkey.js";
 export const SourceFileSymbol = Symbol("SourceFile");
 export interface StaticHelperMetadata {
   name: string;
@@ -23,15 +23,9 @@ export interface StaticHelperMetadata {
   [SourceFileSymbol]?: SourceFile;
 }
 
-export function isStaticHelperMetadata(
-  metadata: any
-): metadata is StaticHelperMetadata {
+export function isStaticHelperMetadata(metadata: any): metadata is StaticHelperMetadata {
   return Boolean(
-    metadata &&
-    metadata.name &&
-    metadata.kind &&
-    metadata.location &&
-    metadata[SourceFileSymbol]
+    metadata && metadata.name && metadata.kind && metadata.location && metadata[SourceFileSymbol],
   );
 }
 
@@ -57,35 +51,34 @@ interface FileMetadata {
 export async function loadStaticHelpers(
   project: Project,
   helpers: StaticHelpers,
-  options: LoadStaticHelpersOptions = {}
+  options: LoadStaticHelpersOptions = {},
 ): Promise<Map<string, StaticHelperMetadata>> {
   const helpersMap = new Map<string, StaticHelperMetadata>();
   // Load static helpers used in sources code
   const defaultStaticHelpersPath = path.join(
     resolveProjectRoot(),
-    DEFAULT_SOURCES_STATIC_HELPERS_PATH
+    DEFAULT_SOURCES_STATIC_HELPERS_PATH,
   );
   const filesInSources = await traverseDirectory(
     options.helpersAssetDirectory ?? defaultStaticHelpersPath,
-    options.program
+    options.program,
   );
   await loadFiles(filesInSources, options.sourcesDir ?? "");
   // Load static helpers used in testing code (only when loadTestHelpers is enabled)
   if (
     options.loadTestHelpers ??
-    (options.options?.generateTest &&
-      isAzurePackage({ options: options.options }))
+    (options.options?.generateTest && isAzurePackage({ options: options.options }))
   ) {
     const defaultTestingHelpersPath = path.join(
       resolveProjectRoot(),
-      DEFAULT_SOURCES_TESTING_HELPERS_PATH
+      DEFAULT_SOURCES_TESTING_HELPERS_PATH,
     );
     const filesInTestings = await traverseDirectory(
       defaultTestingHelpersPath,
       options.program,
       [],
       "",
-      "test/generated/util"
+      "test/generated/util",
     );
     await loadFiles(filesInTestings, options.rootDir ?? "");
   }
@@ -96,24 +89,14 @@ export async function loadStaticHelpers(
       const targetPath = path.join(generateDir, file.target);
       const contents = await readFile(file.source, "utf-8");
       const addedFile = project.createSourceFile(targetPath, contents, {
-        overwrite: true
+        overwrite: true,
       });
       addedFile.getImportDeclarations().map((i) => {
         if (!isAzurePackage({ options: options.options })) {
-          if (
-            i
-              .getModuleSpecifier()
-              .getFullText()
-              .includes("@azure/core-rest-pipeline")
-          ) {
+          if (i.getModuleSpecifier().getFullText().includes("@azure/core-rest-pipeline")) {
             i.setModuleSpecifier("@typespec/ts-http-runtime");
           }
-          if (
-            i
-              .getModuleSpecifier()
-              .getFullText()
-              .includes("@azure-rest/core-client")
-          ) {
+          if (i.getModuleSpecifier().getFullText().includes("@azure-rest/core-client")) {
             i.setModuleSpecifier("@typespec/ts-http-runtime");
           }
         }
@@ -144,7 +127,7 @@ export async function loadStaticHelpers(
           throw new Error(
             `Declaration ${
               entry.name
-            } not found in file ${addedFile.getFilePath()}\n This is an Emitter bug, make sure that the map of static helpers passed to loadStaticHelpers matches what is in the file.`
+            } not found in file ${addedFile.getFilePath()}\n This is an Emitter bug, make sure that the map of static helpers passed to loadStaticHelpers matches what is in the file.`,
           );
         }
 
@@ -155,9 +138,7 @@ export async function loadStaticHelpers(
   }
 }
 
-function assertAllHelpersLoadedPresent(
-  helpers: Map<string, StaticHelperMetadata>
-) {
+function assertAllHelpersLoadedPresent(helpers: Map<string, StaticHelperMetadata>) {
   const missingHelpers = [];
   for (const helper of helpers.values()) {
     if (!helper[SourceFileSymbol]) {
@@ -171,7 +152,7 @@ function assertAllHelpersLoadedPresent(
       .join("\n");
 
     throw new Error(
-      `The following helpers were not found in the project, make sure they are defined in the expected static helper file: ${missingHelpersString}`
+      `The following helpers were not found in the project, make sure they are defined in the expected static helper file: ${missingHelpersString}`,
     );
   }
 
@@ -180,7 +161,7 @@ function assertAllHelpersLoadedPresent(
 
 function getDeclarationByMetadata(
   file: SourceFile,
-  declaration: StaticHelperMetadata
+  declaration: StaticHelperMetadata,
 ):
   | ClassDeclaration
   | FunctionDeclaration
@@ -201,7 +182,7 @@ function getDeclarationByMetadata(
       return file.getEnum(declaration.name);
     default:
       throw new Error(
-        `invalid helper kind ${declaration.kind}\nAll helpers provided to loadStaticHelpers are of kind: function, interface, typeAlias, class`
+        `invalid helper kind ${declaration.kind}\nAll helpers provided to loadStaticHelpers are of kind: function, interface, typeAlias, class`,
       );
   }
 }
@@ -212,7 +193,7 @@ async function traverseDirectory(
   program?: Program,
   result: { source: string; target: string }[] = [],
   relativePath: string = "",
-  targetBaseDir: string = _targetStaticHelpersBaseDir
+  targetBaseDir: string = _targetStaticHelpersBaseDir,
 ): Promise<{ source: string; target: string }[]> {
   try {
     const files = await readdir(directory);
@@ -228,17 +209,13 @@ async function traverseDirectory(
             program,
             result,
             path.join(relativePath, file),
-            targetBaseDir
+            targetBaseDir,
           );
-        } else if (
-          fileStat.isFile() &&
-          !file.endsWith(".d.ts") &&
-          /.*\..?ts$/.test(file)
-        ) {
+        } else if (fileStat.isFile() && !file.endsWith(".d.ts") && /.*\..?ts$/.test(file)) {
           const target = path.join(targetBaseDir, relativePath, file);
           result.push({ source: filePath, target });
         }
-      })
+      }),
     );
 
     return result;
@@ -247,7 +224,7 @@ async function traverseDirectory(
       reportDiagnostic(program, {
         code: "directory-traversal-error",
         format: { directory, error: String(error) },
-        target: NoTarget
+        target: NoTarget,
       });
     }
     throw error;

package/src/framework/reference.ts

@@ -1,10 +1,7 @@
+import { ReferenceableSymbol } from "./dependency.js";
 import { useBinder } from "./hooks/binder.js";
+import { SourceFileSymbol, StaticHelperMetadata } from "./load-static-helpers.js";
 import { refkey as getRefkey } from "./refkey.js";
-import { ReferenceableSymbol } from "./dependency.js";
-import {
-  SourceFileSymbol,
-  StaticHelperMetadata
-} from "./load-static-helpers.js";
 
 export function resolveReference(refkey: unknown): string {
   const binder = useBinder();

package/src/framework/sample.ts

@@ -1,11 +1,11 @@
 import {
+  FunctionDeclarationStructure,
+  InterfaceDeclarationStructure,
   Project,
   StructureKind,
-  InterfaceDeclarationStructure,
-  FunctionDeclarationStructure
 } from "ts-morph";
-import { useBinder } from "./hooks/binder.js";
 import { addDeclaration } from "./declaration.js";
+import { useBinder } from "./hooks/binder.js";
 import { resolveReference } from "./reference.js";
 
 // Create a new ts-morph project
@@ -21,14 +21,14 @@ const binder = useBinder();
 // At this framework level, we're just using a simple object, there is no coupling with the actual type system, but this is flexible so any object can be used.
 const modelType = {
   name: "MyInterface",
-  properties: [{ name: "id", type: "number" }]
+  properties: [{ name: "id", type: "number" }],
 };
 
 // Define a function model
 const functionType = {
   name: "MyFunction",
   returnType: "void",
-  body: `console.log("Hello World");`
+  body: `console.log("Hello World");`,
 };
 
 // Create an interface declaration structure. This illustrates a similar pattern to the one used in the emitter. Transorming a model into a structure for ts-morph.
@@ -37,8 +37,8 @@ const interfaceDeclaration: InterfaceDeclarationStructure = {
   name: modelType.name,
   properties: modelType.properties.map((p) => ({
     name: p.name,
-    type: p.type
-  }))
+    type: p.type,
+  })),
 };
 
 // Create a function declaration structure
@@ -46,7 +46,7 @@ const functionDeclaration: FunctionDeclarationStructure = {
   kind: StructureKind.Function,
   name: functionType.name,
   returnType: functionType.returnType,
-  statements: functionType.body
+  statements: functionType.body,
 };
 
 // Helper functions to add the declarations to the source file. These are needed to be able to leverage the binder to track the declarations.
@@ -56,7 +56,7 @@ addDeclaration(sourceFile, functionDeclaration, functionType);
 
 // Create another source file
 const sourceFile2 = project.createSourceFile("test2.ts", "", {
-  overwrite: true
+  overwrite: true,
 });
 
 // Add statements referencing the tracked declarations