@azure-tools/typespec-ts 0.54.1 → 0.54.2

package/src/modular/build-classical-client.ts

@@ -0,0 +1,434 @@
+import {
+  ClassDeclaration,
+  MethodDeclarationStructure,
+  Scope,
+  SourceFile,
+  StructureKind,
+} from "ts-morph";
+import { NameType, normalizeName } from "../rlc-common/index.js";
+import { buildUserAgentOptions, getClientParametersDeclaration } from "./helpers/client-helpers.js";
+import { getClassicalClientName, getClientName } from "./helpers/naming-helpers.js";
+import { ModularEmitterOptions } from "./interfaces.js";
+
+import {
+  InitializedByFlags,
+  SdkClientType,
+  SdkServiceMethod,
+  SdkServiceOperation,
+} from "@azure-tools/typespec-client-generator-core";
+import { useContext } from "../context-manager.js";
+import { useDependencies } from "../framework/hooks/use-dependencies.js";
+import { resolveReference } from "../framework/reference.js";
+import { refkey } from "../framework/refkey.js";
+import { getModularClientOptions, isRLCMultiEndpoint } from "../utils/client-utils.js";
+import { SdkContext } from "../utils/interfaces.js";
+import { getMethodHierarchiesMap, isTenantLevelOperation } from "../utils/operation-util.js";
+import { AzurePollingDependencies } from "./external-dependencies.js";
+import { getPagingLROMethodName } from "./helpers/classical-operation-helpers.js";
+import { getDocsFromDescription } from "./helpers/docs-helpers.js";
+import { getOperationFunction } from "./helpers/operation-helpers.js";
+import { PagingHelpers, SimplePollerHelpers } from "./static-helpers-metadata.js";
+
+export function buildClassicalClient(
+  dpgContext: SdkContext,
+  clientMap: [string[], SdkClientType<SdkServiceOperation>],
+  emitterOptions: ModularEmitterOptions,
+) {
+  const project = useContext("outputProject");
+  const [_hierarchy, client] = clientMap;
+  const dependencies = useDependencies();
+  const modularClientName = getClientName(client);
+  const classicalClientName = `${getClassicalClientName(client)}`;
+  const classicalParams = getClientParametersDeclaration(client, dpgContext, {
+    requiredOnly: true,
+  });
+  const contextParams = getClientParametersDeclaration(client, dpgContext, {
+    onClientOnly: false,
+    requiredOnly: true,
+  });
+  const srcPath = emitterOptions.modularOptions.sourceRoot;
+  const { subfolder, rlcClientName } = getModularClientOptions(clientMap);
+
+  const clientFile = project.createSourceFile(
+    `${srcPath}/${subfolder && subfolder !== "" ? subfolder + "/" : ""}${normalizeName(
+      classicalClientName,
+      NameType.File,
+    )}.ts`,
+  );
+
+  clientFile.addExportDeclaration({
+    isTypeOnly: true,
+    namedExports: [`${classicalClientName}OptionalParams`],
+    moduleSpecifier: `./api/${normalizeName(modularClientName, NameType.File)}Context.js`,
+  });
+
+  const clientClass = clientFile.addClass({
+    isExported: true,
+    name: `${classicalClientName}`,
+  });
+
+  // Add the private client member. This will be the client context from /api
+  if (isRLCMultiEndpoint(dpgContext)) {
+    clientClass.addProperty({
+      name: "_client",
+      type: `Client.${rlcClientName}`,
+      scope: Scope.Private,
+    });
+  } else {
+    clientClass.addProperty({
+      name: "_client",
+      type: `${rlcClientName}`,
+      scope: Scope.Private,
+    });
+  }
+  // Add the pipeline member. This will be the pipeline from /api
+  clientClass.addProperty({
+    name: "pipeline",
+    type: resolveReference(dependencies.Pipeline),
+    scope: Scope.Public,
+    isReadonly: true,
+    docs: ["The pipeline used by this client to make requests"],
+  });
+
+  const hasChildClient =
+    client.children &&
+    client.children.some((childClient) => {
+      return childClient.clientInitialization.initializedBy & InitializedByFlags.Parent;
+    });
+  if (hasChildClient) {
+    clientClass.addProperty({
+      name: "_clientParams",
+      type: `{${classicalParams.map((p) => {
+        return `${p.name}: ${p.type}`;
+      })}}`,
+      scope: Scope.Private,
+      docs: ["The parent client parameters that are used in the constructors."],
+    });
+  }
+
+  // Check if constructor overloads for subscriptionId is needed
+  const hasSubscriptionIdParam = classicalParams.some(
+    (param) => param.name.toLowerCase() === "subscriptionid",
+  );
+  const shouldSubscriptionIdOptional =
+    dpgContext.arm && hasSubscriptionIdParam && hasTenantLevelOperations(client, dpgContext);
+
+  let constructor;
+  if (shouldSubscriptionIdOptional) {
+    constructor = generateConstructorWithOverloads(clientClass, classicalParams, client);
+  } else {
+    constructor = clientClass.addConstructor({
+      docs: getDocsFromDescription(client.doc),
+      parameters: classicalParams,
+    });
+  }
+
+  const paramNames = (contextParams ?? [])
+    .map((p) => p.name)
+    .map((x) => {
+      if (x === "options") {
+        return `{...options, userAgentOptions: ${buildUserAgentOptions(
+          constructor,
+          emitterOptions,
+          "azsdk-js-client",
+        )}}`;
+      } else if (x.toLowerCase() === "subscriptionid" && shouldSubscriptionIdOptional) {
+        return `subscriptionId ?? ""`;
+      } else {
+        return x;
+      }
+    });
+
+  constructor.addStatements([
+    `this._client = create${modularClientName}(${paramNames.join(",")});`,
+  ]);
+  constructor.addStatements(`this.pipeline = this._client.pipeline;`);
+
+  if (hasChildClient && client.children) {
+    constructor.addStatements(
+      `this._clientParams = {${classicalParams.map((p) => p.name).join(",")}};`,
+    );
+    for (const childClient of client.children) {
+      const subfolder = normalizeName(childClient.name.replace("Client", ""), NameType.File);
+      clientFile.addImportDeclaration({
+        moduleSpecifier: `./${subfolder}/${normalizeName(childClient.name, NameType.File)}.js`,
+        namedImports: [
+          `${getClassicalClientName(childClient)}`,
+          `${getClassicalClientName(childClient)}OptionalParams`,
+        ],
+      });
+    }
+  }
+  buildClientOperationGroups(clientMap, dpgContext, clientClass);
+  if (client.children) {
+    client.children
+      .filter((childClient) => {
+        return childClient.clientInitialization.initializedBy & InitializedByFlags.Parent;
+      })
+      .forEach((childClient) => {
+        addChildClient(dpgContext, clientClass, client, childClient);
+      });
+  }
+  importAllApis(clientFile, srcPath, subfolder ?? "");
+  clientFile.fixUnusedIdentifiers();
+  return clientFile;
+}
+
+function importAllApis(clientFile: SourceFile, srcPath: string, subfolder: string) {
+  const project = clientFile.getProject();
+  const apiModels = project.getSourceFile(
+    `${srcPath}/${subfolder && subfolder !== "" ? subfolder + "/" : ""}api/index.ts`,
+  );
+
+  if (!apiModels) {
+    return;
+  }
+
+  const exported = [...apiModels.getExportedDeclarations().keys()];
+
+  clientFile.addImportDeclaration({
+    moduleSpecifier: `./api/index.js`,
+    namedImports: exported,
+  });
+}
+
+function generateMethod(
+  context: SdkContext,
+  clientType: string,
+  method: [string[], SdkServiceMethod<SdkServiceOperation>],
+): MethodDeclarationStructure[] {
+  const res: MethodDeclarationStructure[] = [];
+  const declaration = getOperationFunction(context, method, clientType);
+  const methodName = declaration.propertyName ?? declaration.name ?? "FIXME";
+  const methodParams = declaration.parameters?.filter((p) => p.name !== "context") ?? [];
+  const declarationRefKey = resolveReference(refkey(method[1], "api"));
+  const methodParamStr = ["this._client", ...methodParams.map((p) => p.name)].join(", ");
+
+  res.push({
+    docs: declaration.docs,
+    name: methodName,
+    kind: StructureKind.Method,
+    returnType: declaration.returnType,
+    parameters: methodParams,
+    statements: `return ${declarationRefKey}(${methodParamStr})`,
+  });
+
+  // add LRO helper methods if applicable
+  if (context.rlcOptions?.compatibilityLro && declaration?.isLro && !declaration?.isLroPaging) {
+    const operationStateReference = resolveReference(AzurePollingDependencies.OperationState);
+    const simplePollerLikeReference = resolveReference(SimplePollerHelpers.SimplePollerLike);
+    const getSimplePollerReference = resolveReference(SimplePollerHelpers.getSimplePoller);
+    const returnType = declaration?.lroFinalReturnType ?? "void";
+    const beginName = normalizeName(`begin_${methodName}`, NameType.Method);
+    const beginAndWaitName = normalizeName(`${beginName}_andWait`, NameType.Method);
+    // add begin method
+    res.push({
+      isAsync: true,
+      docs: [`@deprecated use ${methodName} instead`],
+      name: beginName,
+      kind: StructureKind.Method,
+      returnType: `Promise<${simplePollerLikeReference}<${operationStateReference}<${returnType}>, ${returnType}>>`,
+      parameters: methodParams,
+      statements: `const poller = ${declarationRefKey}(${methodParamStr});
+                  await poller.submitted();
+                  return ${getSimplePollerReference}(poller);`,
+    });
+    // add begin and wait method
+    res.push({
+      isAsync: true,
+      docs: [`@deprecated use ${methodName} instead`],
+      name: beginAndWaitName,
+      kind: StructureKind.Method,
+      returnType: `Promise<${returnType}>`,
+      parameters: methodParams,
+      statements: `return await ${declarationRefKey}(${methodParamStr});`,
+    });
+  } // For LRO+Paging operations, use different return types and implementation
+  else if (context.rlcOptions?.compatibilityLro && declaration?.isLroPaging) {
+    const returnType = declaration?.lropagingFinalReturnType ?? "void";
+    const pagedAsyncIterableIteratorReference = resolveReference(
+      PagingHelpers.PagedAsyncIterableIterator,
+    );
+    const beginListAndWaitName = normalizeName(
+      `${getPagingLROMethodName(methodName)}`,
+      NameType.Method,
+    );
+    // add begin and wait method for LRO+Paging - directly returns paged iterator
+    res.push({
+      isAsync: false,
+      docs: [`@deprecated use ${methodName} instead`],
+      name: beginListAndWaitName,
+      kind: StructureKind.Method,
+      returnType: `${pagedAsyncIterableIteratorReference}<${returnType}>`,
+      parameters: methodParams,
+      statements: `return ${declarationRefKey}(${methodParamStr});`,
+    });
+  }
+
+  return res;
+}
+function buildClientOperationGroups(
+  clientMap: [string[], SdkClientType<SdkServiceOperation>],
+  dpgContext: SdkContext,
+  clientClass: ClassDeclaration,
+) {
+  let clientType = "Client";
+  const [_hierarchy, client] = clientMap;
+  const { subfolder } = getModularClientOptions(clientMap);
+  if (subfolder && subfolder !== "") {
+    clientType = `Client.${clientClass.getName()}`;
+  }
+  const methodMap = getMethodHierarchiesMap(dpgContext, client);
+  for (const [prefixKey, operations] of methodMap) {
+    const prefixes = prefixKey.split("/");
+    const layer = 0;
+    if (prefixKey === "") {
+      operations.forEach((op) => {
+        const methods = generateMethod(dpgContext, clientType, [prefixes, op]);
+        clientClass.addMethods(methods);
+      });
+    } else {
+      // The `rawGroupName` is used to any places where we need normalized name twice so we need to keep the raw as PascalCase.
+      const rawGroupName = normalizeName(prefixes[0] ?? "", NameType.Interface);
+      const operationName = `_get${rawGroupName}Operations`;
+      const propertyType = `${rawGroupName}Operations`;
+      // The `groupName` is used to any places where we don't need normalized name again
+      const groupName = normalizeName(rawGroupName, NameType.Property);
+      const existProperty = clientClass.getProperties().filter((p) => {
+        return p.getName() === normalizeName(groupName, NameType.Property);
+      });
+      if (!existProperty || existProperty.length === 0) {
+        clientClass.addProperty({
+          name: groupName,
+          type: resolveReference(refkey(propertyType, layer, "classicOperations")),
+          scope: Scope.Public,
+          isReadonly: true,
+          docs: ["The operation groups for " + groupName],
+        });
+        clientClass
+          .getConstructors()[0]
+          ?.addStatements(
+            `this.${groupName} = ${resolveReference(refkey(operationName, layer, "getClassicOperations"))}(this._client)`,
+          );
+      }
+    }
+  }
+}
+
+function addChildClient(
+  context: SdkContext,
+  clientClass: ClassDeclaration,
+  parentClient: SdkClientType<SdkServiceOperation>,
+  client: SdkClientType<SdkServiceOperation>,
+) {
+  const parentParams = getClientParametersDeclaration(parentClient, context, {
+    requiredOnly: true,
+  });
+  const clientParams = getClientParametersDeclaration(client, context, {
+    requiredOnly: true,
+  });
+  const diffParams = clientParams.filter((p) => {
+    return !parentParams.some((pp) => pp.name === p.name && pp.name !== "options");
+  });
+  const name = getClassicalClientName(client);
+  const getChildClientFunction = clientClass.addMethod({
+    docs: getDocsFromDescription(client.doc),
+    name: `get${name}`,
+    returnType: `${getClassicalClientName(client)}`,
+    parameters: diffParams,
+  });
+  getChildClientFunction.addStatements(
+    `return new ${getClassicalClientName(client)}(
+      ${parentParams
+        .filter((p) => !p.name.includes("options"))
+        .map((p) => `this._clientParams.${p.name}`)
+        .join(",")}${parentParams.filter((p) => !p.name.includes("options")).length > 0 ? "," : ""}
+      ${diffParams
+        .filter((p) => p.name !== "options")
+        .map((p) => `${p.name}`)
+        .join(",")}${diffParams.filter((p) => p.name !== "options").length > 0 ? "," : ""}
+      { ...this._clientParams.options, ...options }
+    )`,
+  );
+}
+
+function hasTenantLevelOperations(
+  client: SdkClientType<SdkServiceOperation>,
+  dpgContext: SdkContext,
+): boolean {
+  const methodMap = getMethodHierarchiesMap(dpgContext, client);
+
+  for (const [_, operations] of methodMap) {
+    for (const op of operations) {
+      if (isTenantLevelOperation(op, client)) {
+        // Found a tenant-level operation
+        return true;
+      }
+    }
+  }
+
+  return false;
+}
+
+function generateConstructorWithOverloads(
+  clientClass: ClassDeclaration,
+  classicalParams: any[],
+  client: SdkClientType<SdkServiceOperation>,
+) {
+  const filteredParams = classicalParams.filter(
+    (p) => p.name.toLowerCase() !== "subscriptionid" && p.name.toLowerCase() !== "options",
+  );
+
+  const clientConstructor = clientClass.addConstructor({
+    docs: getDocsFromDescription(client.doc),
+    parameters: [
+      ...filteredParams,
+      {
+        name: "subscriptionIdOrOptions",
+        type: `string | ${getClassicalClientName(client)}OptionalParams`,
+        hasQuestionToken: true,
+      },
+      {
+        name: "options",
+        type: `${getClassicalClientName(client)}OptionalParams`,
+        hasQuestionToken: true,
+      },
+    ],
+  });
+
+  clientConstructor.addOverload({
+    parameters: [
+      ...filteredParams,
+      {
+        name: "options",
+        type: `${getClassicalClientName(client)}OptionalParams`,
+        hasQuestionToken: true,
+      },
+    ],
+  });
+
+  clientConstructor.addOverload({
+    parameters: [
+      ...filteredParams,
+      ...classicalParams.filter((p) => p.name.toLowerCase() === "subscriptionid"),
+      {
+        name: "options",
+        type: `${getClassicalClientName(client)}OptionalParams`,
+        hasQuestionToken: true,
+      },
+    ],
+  });
+
+  clientConstructor.addStatements([
+    `let subscriptionId: string | undefined;`,
+    ``,
+    `if (typeof subscriptionIdOrOptions === "string") {`,
+    `  subscriptionId = subscriptionIdOrOptions;`,
+    `} else if (typeof subscriptionIdOrOptions === "object") {`,
+    `  options = subscriptionIdOrOptions;`,
+    `}`,
+    `options = options ?? {};`,
+  ]);
+
+  return clientConstructor;
+}

package/src/modular/build-classical-operation-groups.ts

@@ -0,0 +1,69 @@
+import { SdkClientType, SdkServiceOperation } from "@azure-tools/typespec-client-generator-core";
+import { SourceFile } from "ts-morph";
+import { useContext } from "../context-manager.js";
+import { NameType } from "../rlc-common/index.js";
+import { getModularClientOptions } from "../utils/client-utils.js";
+import { SdkContext } from "../utils/interfaces.js";
+import { getMethodHierarchiesMap } from "../utils/operation-util.js";
+import { getClassicalOperation } from "./helpers/classical-operation-helpers.js";
+import { getClassicalLayerPrefix } from "./helpers/naming-helpers.js";
+import { ModularEmitterOptions } from "./interfaces.js";
+
+export function buildClassicOperationFiles(
+  dpgContext: SdkContext,
+  clientMap: [string[], SdkClientType<SdkServiceOperation>],
+  emitterOptions: ModularEmitterOptions,
+) {
+  // const sdkPackage = dpgContext.sdkPackage;
+  const project = useContext("outputProject");
+  const [_hierarchy, client] = clientMap;
+  const { subfolder } = getModularClientOptions(clientMap);
+  const classicOperationFiles: Map<string, SourceFile> = new Map<string, SourceFile>();
+  const methodMap = getMethodHierarchiesMap(dpgContext, client);
+  for (const [prefixKey, operations] of methodMap) {
+    const prefixes = prefixKey.split("/");
+    if (prefixes.length > 0 && prefixKey !== "") {
+      const classicOperationFileName =
+        prefixes.length > 0
+          ? `${getClassicalLayerPrefix(prefixes, NameType.File, "/", prefixes.length - 1)}/index`
+          : // When the program has no operation groups defined all operations are put
+            // into a nameless operation group. We'll call this operations.
+            "index";
+
+      const srcPath = emitterOptions.modularOptions.sourceRoot;
+      const classicFile =
+        classicOperationFiles.get(classicOperationFileName) ??
+        project.createSourceFile(
+          `${srcPath}/${
+            subfolder && subfolder !== "" ? subfolder + "/" : ""
+          }classic/${classicOperationFileName}.ts`,
+        );
+      getClassicalOperation(dpgContext, clientMap, classicFile, [prefixes, operations]);
+      classicOperationFiles.set(classicOperationFileName, classicFile);
+    }
+  }
+  for (const [prefixKey, operations] of methodMap) {
+    const prefixes = prefixKey.split("/");
+    if (prefixes.length > 0 && prefixKey !== "") {
+      for (let layer = 0; layer < prefixes.length - 1; layer++) {
+        const classicOperationFileName =
+          prefixes.length > 0
+            ? `${getClassicalLayerPrefix(prefixes, NameType.File, "/", layer)}/index`
+            : // When the program has no operation groups defined all operations are put
+              // into a nameless operation group. We'll call this operations.
+              "index";
+        const srcPath = emitterOptions.modularOptions.sourceRoot;
+        const classicFile =
+          classicOperationFiles.get(classicOperationFileName) ??
+          project.createSourceFile(
+            `${srcPath}/${
+              subfolder && subfolder !== "" ? subfolder + "/" : ""
+            }classic/${classicOperationFileName}.ts`,
+          );
+        getClassicalOperation(dpgContext, clientMap, classicFile, [prefixes, operations], layer);
+        classicOperationFiles.set(classicOperationFileName, classicFile);
+      }
+    }
+  }
+  return classicOperationFiles;
+}