@autorest/typescript 6.0.0-beta.14 → 6.0.0-beta.17

package/src/utils/autorestOptions.ts

@@ -1,6 +1,6 @@
-import { Channel, Host } from "@autorest/extension-base";
+import { Channel, AutorestExtensionHost } from "@autorest/extension-base";
 import { AutorestOptions, getHost, getSession } from "../autorestSession";
-import { TracingInfo } from "../models/clientDetails";
+import { DependencyInfo, TracingInfo } from "../models/clientDetails";
 import { PackageDetails } from "../models/packageDetails";
 import { NameType, normalizeName } from "./nameUtils";
 
@@ -11,6 +11,7 @@ export async function extractAutorestOptions(): Promise<AutorestOptions> {
   const host = getHost();
   const useCoreV2 = await getUseCoreV2(host);
   const restLevelClient = await getRestLevelClient(host);
+  const rlcShortcut = await getHasShortcutMethods(host);
   const azureArm = await getIsAzureArm(host);
   const addCredentials = await getAddCredentials(host);
   const credentialKeyHeaderName = await getKeyCredentialHeaderName(host);
@@ -31,7 +32,14 @@ export async function extractAutorestOptions(): Promise<AutorestOptions> {
   const azureOutputDirectory = await getAzureOutputDirectoryPath(host);
   const headAsBoolean = await getHeadAsBoolean(host);
   const isTestPackage = await getIsTestPackage(host);
-  const generateTest = await getGenerateTest(host)
+  const generateTest = await getGenerateTest(host);
+  const batch = await getBatch(host);
+  const multiClient = await getMultiClient(host);
+  const generateSample = await getGenerateSample(host);
+  const productDocLink = await getProductDocLink(host);
+  const coreHttpCompatMode = await getCoreHttpCompatMode(host);
+  const azureSdkForJs = await getAzureSdkForJs(host);
+  const dependencyInfo = await getDependencyInfo(host);
 
   return {
     azureArm,
@@ -39,6 +47,7 @@ export async function extractAutorestOptions(): Promise<AutorestOptions> {
     credentialKeyHeaderName,
     credentialScopes,
     restLevelClient,
+    rlcShortcut,
     srcPath,
     outputPath,
     packageDetails,
@@ -56,77 +65,118 @@ export async function extractAutorestOptions(): Promise<AutorestOptions> {
     headAsBoolean,
     isTestPackage,
     generateTest,
+    batch,
+    multiClient,
+    generateSample,
+    azureSdkForJs,
+    productDocLink,
+    coreHttpCompatMode,
+    dependencyInfo
   };
 }
 
-async function getHeadAsBoolean(host: Host): Promise<boolean> {
-  const headAsBoolean = await host.GetValue("head-as-boolean");
+async function getHasShortcutMethods(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  const headAsBoolean = await host.getValue("rlc-shortcut");
 
   return Boolean(headAsBoolean);
 }
 
-async function getIsTestPackage(host: Host): Promise<boolean> {
-  const isTestPackage: boolean = await host.GetValue("is-test-package");
+async function getHeadAsBoolean(host: AutorestExtensionHost): Promise<boolean> {
+  const headAsBoolean = await host.getValue("head-as-boolean");
+
+  return Boolean(headAsBoolean);
+}
+
+async function getIsTestPackage(host: AutorestExtensionHost): Promise<boolean> {
+  const isTestPackage = await host.getValue("is-test-package");
   return isTestPackage === null ? false : Boolean(isTestPackage);
 }
 
-async function getGenerateTest(host: Host): Promise<boolean> {
-  const generateTest: boolean = await host.GetValue("generate-test");
+async function getGenerateTest(host: AutorestExtensionHost): Promise<boolean> {
+  const generateTest = await host.getValue("generate-test");
   return generateTest === null ? false : Boolean(generateTest);
 }
 
-async function getSkipEnumValidation(host: Host): Promise<boolean> {
-  const skipEnumValidation = await host.GetValue("skip-enum-validation");
+async function getGenerateSample(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  const generateSample = await host.getValue("generate-sample");
+  return generateSample === null ? false : Boolean(generateSample);
+}
+
+async function getAzureSdkForJs(host: AutorestExtensionHost): Promise<boolean> {
+  const azureSdkForJs = await host.getValue("azure-sdk-for-js");
+  return azureSdkForJs === null ? true : Boolean(azureSdkForJs);
+}
+
+async function getSkipEnumValidation(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  const skipEnumValidation = await host.getValue("skip-enum-validation");
 
   return skipEnumValidation === true;
 }
 
-async function getAllowInsecureConnection(host: Host): Promise<boolean> {
-  return (await host.GetValue("allow-insecure-connection")) || false;
+async function getAllowInsecureConnection(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  return (await host.getValue("allow-insecure-connection")) || false;
 }
 
-async function getIgnoreNullableOnOptional(host: Host): Promise<boolean> {
+async function getIgnoreNullableOnOptional(
+  host: AutorestExtensionHost
+): Promise<boolean> {
   const isAzureArm = await getIsAzureArm(host);
-  return (await host.GetValue("ignore-nullable-on-optional"))
+  return (await host.getValue("ignore-nullable-on-optional"))
     ? true
     : isAzureArm;
 }
 
-async function getDisableAsyncOperators(host: Host): Promise<boolean> {
-  return (await host.GetValue("disable-async-iterators")) === true;
+async function getDisableAsyncOperators(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  return (await host.getValue("disable-async-iterators")) === true;
 }
 
-async function getHideClients(host: Host): Promise<boolean> {
-  return (await host.GetValue("hide-clients")) || false;
+async function getHideClients(host: AutorestExtensionHost): Promise<boolean> {
+  return (await host.getValue("hide-clients")) || false;
 }
-async function getGenerateMetadata(host: Host) {
-  return (await host.GetValue("generate-metadata")) !== false;
+async function getGenerateMetadata(host: AutorestExtensionHost) {
+  return (await host.getValue("generate-metadata")) !== false;
 }
 
-async function getLicenseHeader(host: Host): Promise<boolean> {
-  return (await host.GetValue("license-header")) || false;
+async function getLicenseHeader(host: AutorestExtensionHost): Promise<boolean> {
+  return (await host.getValue("license-header")) || false;
 }
 
-async function getTitle(host: Host): Promise<string | undefined> {
-  return (await host.GetValue("title")) || undefined;
+async function getTitle(
+  host: AutorestExtensionHost
+): Promise<string | undefined> {
+  return (await host.getValue("title")) || undefined;
 }
 
-async function getSrcPath(host: Host): Promise<string> {
-  return ((await host.GetValue("source-code-folder-path")) as string) || "src";
+async function getSrcPath(host: AutorestExtensionHost): Promise<string> {
+  return ((await host.getValue("source-code-folder-path")) as string) || "src";
 }
 
-async function getOutputPath(host: Host): Promise<string | undefined> {
-  return (await host.GetValue("output-folder")) || undefined;
+async function getOutputPath(
+  host: AutorestExtensionHost
+): Promise<string | undefined> {
+  return (await host.getValue("output-folder")) || undefined;
 }
 
 async function getKeyCredentialHeaderName(
-  host: Host
+  host: AutorestExtensionHost
 ): Promise<string | undefined> {
-  return (await host.GetValue("credential-key-header-name")) || undefined;
+  return (await host.getValue("credential-key-header-name")) || undefined;
 }
 
-async function getAddCredentials(host: Host): Promise<boolean> {
-  const addCredentials = await host.GetValue("add-credentials");
+async function getAddCredentials(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  const addCredentials = await host.getValue("add-credentials");
 
   // Only set addCredentials to false if explicitly set to false
   // otherwise default to true
@@ -136,34 +186,38 @@ async function getAddCredentials(host: Host): Promise<boolean> {
     return true;
   }
 }
-async function getIsAzureArm(host: Host): Promise<boolean> {
-  const flag = (await host.GetValue("azure-arm")) === true;
-  const openapi = (await host.GetValue("openapi-type")) === "arm";
+async function getIsAzureArm(host: AutorestExtensionHost): Promise<boolean> {
+  const flag = (await host.getValue("azure-arm")) === true;
+  const openapi = (await host.getValue("openapi-type")) === "arm";
 
   return flag || openapi;
 }
 
-async function getRestLevelClient(host: Host): Promise<boolean> {
-  return (await host.GetValue("rest-level-client")) === true;
+async function getRestLevelClient(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  return (await host.getValue("rest-level-client")) === true;
 }
 
-async function getUseCoreV2(host: Host): Promise<boolean> {
-  const useCoreV2Option: boolean = await host.GetValue("use-core-v2");
+async function getUseCoreV2(host: AutorestExtensionHost): Promise<boolean> {
+  const useCoreV2Option = await host.getValue("use-core-v2");
   return useCoreV2Option === null ? true : Boolean(useCoreV2Option);
 }
 
-async function getTracingInfo(host: Host): Promise<TracingInfo | undefined> {
+async function getTracingInfo(
+  host: AutorestExtensionHost
+): Promise<TracingInfo | undefined> {
   const tracing: TracingInfo | undefined =
-    (await host.GetValue("tracing-info")) || undefined;
+    (await host.getValue("tracing-info")) || undefined;
 
   if (tracing && tracing.namespace && tracing.packagePrefix) {
     return tracing;
   }
 
-  const namespace =
-    (await host.GetValue("tracing-info.namespace")) || undefined;
-  const packagePrefix =
-    (await host.GetValue("tracing-info.packagePrefix")) || undefined;
+  const namespace: string | undefined =
+    (await host.getValue<string>("tracing-info.namespace")) || undefined;
+  const packagePrefix: string | undefined =
+    (await host.getValue("tracing-info.packagePrefix")) || undefined;
 
   if (packagePrefix && namespace) {
     return {
@@ -181,13 +235,17 @@ async function getTracingInfo(host: Host): Promise<TracingInfo | undefined> {
   );
 }
 
-async function getPackageDetails(host: Host): Promise<PackageDetails> {
+async function getPackageDetails(
+  host: AutorestExtensionHost
+): Promise<PackageDetails> {
   const { model } = getSession();
   const name = normalizeName(model.language.default.name, NameType.File);
   // TODO: Look for an existing package.json and
-  const packageName = (await host.GetValue("package-name")) || name;
-  const packageNameParts = packageName.match(/(^@(.*)\/)?(.*)/);
-  const version = (await host.GetValue("package-version")) || "1.0.0-beta.1";
+  const packageName: string = (await host.getValue("package-name")) || name;
+  const packageNameParts: RegExpMatchArray =
+    packageName.match(/(^@(.*)\/)?(.*)/) ?? [];
+  const version: string =
+    (await host.getValue("package-version")) || "1.0.0-beta.1";
 
   return {
     name: packageName,
@@ -199,11 +257,11 @@ async function getPackageDetails(host: Host): Promise<PackageDetails> {
 }
 
 export async function getCredentialScopes(
-  host: Host
+  host: AutorestExtensionHost
 ): Promise<string[] | undefined> {
-  const addCredentials = await host.GetValue("add-credentials");
-  const credentialScopes = await host.GetValue("credential-scopes");
-  const azureArm = await host.GetValue("azure-arm");
+  const addCredentials = await host.getValue("add-credentials");
+  const credentialScopes = await host.getValue("credential-scopes");
+  const azureArm = await host.getValue("azure-arm");
 
   if (credentialScopes && !addCredentials) {
     throw new Error(
@@ -215,7 +273,7 @@ export async function getCredentialScopes(
     if (azureArm) {
       return ["https://management.azure.com/.default"];
     } else if (addCredentials) {
-      host.Message({
+      host.message({
         Channel: Channel.Warning,
         Text: `You have default credential policy BearerTokenCredentialPolicy
         but not the --credential-scopes flag set while generating non-management plane code.
@@ -233,11 +291,9 @@ export async function getCredentialScopes(
 }
 
 async function getAzureOutputDirectoryPath(
-  host: Host
+  host: AutorestExtensionHost
 ): Promise<string | undefined> {
-  const outputDirectoryPath: string | null = await host.GetValue(
-    "outputFolderUri"
-  );
+  const outputDirectoryPath = await host.getValue<string>("outputFolderUri");
   const outputDirectoryRelativePath: string | undefined = outputDirectoryPath
     ?.replace(/\/$/, "")
     .split("/")
@@ -247,3 +303,58 @@ async function getAzureOutputDirectoryPath(
     ? outputDirectoryRelativePath
     : undefined;
 }
+
+async function getBatch(
+  host: AutorestExtensionHost
+): Promise<[string, any][] | undefined> {
+  const batch = await host.getValue<[string, any][]>("batch");
+  return batch;
+}
+
+async function getProductDocLink(
+  host: AutorestExtensionHost
+): Promise<string | undefined> {
+  return (await host.getValue("product-doc-link")) || undefined;
+}
+
+async function getMultiClient(host: AutorestExtensionHost): Promise<boolean> {
+  const multiClient = (await host.getValue("multi-client")) || undefined;
+  return !!multiClient;
+}
+
+async function getCoreHttpCompatMode(
+  host: AutorestExtensionHost
+): Promise<boolean> {
+  return (await host.getValue("core-http-compat-mode")) || false;
+}
+
+async function getDependencyInfo(
+  host: AutorestExtensionHost
+): Promise<DependencyInfo | undefined> {
+  const dependency: DependencyInfo | undefined =
+    (await host.getValue("dependency-info")) || undefined;
+
+  if (dependency && dependency.description && dependency.link) {
+    return dependency;
+  }
+
+  const link: string | undefined =
+    (await host.getValue<string>("dependency-info.link")) || undefined;
+  const description: string | undefined =
+    (await host.getValue("dependency-info.description")) || undefined;
+
+  if (description && link) {
+    return {
+      link,
+      description
+    };
+  }
+
+  if (!dependency && !description && !link) {
+    return undefined;
+  }
+
+  throw new Error(
+    "Invalid dependency-info. Make sure that link and description are defined"
+  );
+}

package/src/utils/nameUtils.ts

@@ -4,7 +4,7 @@ import { Operation, OperationGroup } from "@autorest/codemodel";
 import { getLanguageMetadata } from "./languageHelpers";
 import { TypeDetails, PropertyKind } from "../models/modelDetails";
 
-interface ReservedName {
+export interface ReservedName {
   name: string;
   reservedFor: NameType[];
 }
@@ -93,11 +93,15 @@ export enum CasingConvention {
   Camel
 }
 
-export function guardReservedNames(name: string, nameType: NameType): string {
+export function guardReservedNames(
+  name: string,
+  nameType: NameType,
+  customReservedNames: ReservedName[] = []
+): string {
   const suffix = getSuffix(nameType);
-  return ReservedModelNames.filter(r => r.reservedFor.includes(nameType)).find(
-    r => r.name === name.toLowerCase()
-  )
+  return [...ReservedModelNames, ...customReservedNames]
+    .filter(r => r.reservedFor.includes(nameType))
+    .find(r => r.name === name.toLowerCase())
     ? `${name}${suffix}`
     : name;
 }
@@ -136,12 +140,14 @@ export function normalizeTypeName({ kind, typeName }: TypeDetails) {
 export function normalizeName(
   name: string,
   nameType: NameType,
-  shouldGuard?: boolean
+  shouldGuard?: boolean,
+  customReservedNames: ReservedName[] = [],
+  casingOverride?: CasingConvention
 ): string {
   if (name.startsWith("$DO_NOT_NORMALIZE$")) {
     return name.replace("$DO_NOT_NORMALIZE$", "");
   }
-  const casingConvention = getCasingConvention(nameType);
+  const casingConvention = casingOverride ?? getCasingConvention(nameType);
   const sanitizedName = sanitizeName(name);
   const parts = getNameParts(sanitizedName);
   const [firstPart, ...otherParts] = parts;
@@ -153,7 +159,9 @@ export function normalizeName(
     .join("");
 
   const normalized = checkBeginning(`${normalizedFirstPart}${normalizedParts}`);
-  return shouldGuard ? guardReservedNames(normalized, nameType) : normalized;
+  return shouldGuard
+    ? guardReservedNames(normalized, nameType, customReservedNames)
+    : normalized;
 }
 
 function checkBeginning(name: string): string {

package/src/utils/schemaHelpers.ts

@@ -68,7 +68,7 @@ export function getTypeForSchema(
 ): TypeDetails {
   let typeName: string = "";
   let usedModels: string[] = [];
-  let defaultValue: string = "";
+  let defaultValue = undefined;
   let kind: PropertyKind = PropertyKind.Primitive;
   switch (schema.type) {
     case SchemaType.Any:
@@ -194,7 +194,7 @@ export function getTypeForSchema(
     usedModels,
     isConstant: schema.type === SchemaType.Constant,
     nullable: isNullable,
-    ...(defaultValue && { defaultValue })
+    defaultValue
   };
 }
 

package/dist/src/generators/clientContextFileGenerator.d.ts

@@ -1,5 +0,0 @@
-import { Project } from "ts-morph";
-import { ClientDetails } from "../models/clientDetails";
-import { PackageDetails } from "../models/packageDetails";
-export declare function generateClientContext(clientDetails: ClientDetails, packageDetails: PackageDetails, project: Project): void;
-//# sourceMappingURL=clientContextFileGenerator.d.ts.map

package/dist/src/generators/clientContextFileGenerator.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"clientContextFileGenerator.d.ts","sourceRoot":"","sources":["../../../src/generators/clientContextFileGenerator.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,OAAO,EAMR,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAO1D,wBAAgB,qBAAqB,CACnC,aAAa,EAAE,aAAa,EAC5B,cAAc,EAAE,cAAc,EAC9B,OAAO,EAAE,OAAO,QAmEjB"}