chore(scripts): override smithy partitions

Author: siddsriv

scripts/generate-clients/code-gen.js

@@ -31,7 +31,23 @@ const generateClient = async (clientName) => {
   const filename = `${clientName}.json`;
   copyFileSync(join(DEFAULT_CODE_GEN_INPUT_DIR, filename), join(TEMP_CODE_GEN_INPUT_DIR_SERVICE, filename));
 
-  await spawnProcess("./gradlew", options, { cwd: CODE_GEN_ROOT });
+  await spawnProcess("./gradlew", options, {
+    cwd: CODE_GEN_ROOT,
+    env: {
+      ...process.env,
+      AWS_PARTITIONS_FILE_OVERRIDE: join(
+        __dirname,
+        "..",
+        "..",
+        "packages",
+        "util-endpoints",
+        "src",
+        "lib",
+        "aws",
+        "partitions.json"
+      ),
+    },
+  });
   rmSync(join(__dirname, "..", "..", "codegen", "sdk-codegen", `smithy-build-${clientName}.json`));
 };
 
@@ -51,7 +67,23 @@ const generateClients = async (models, batchSize) => {
       const filename = basename(filepath);
       copyFileSync(filepath, join(TEMP_CODE_GEN_INPUT_DIR, filename));
     }
-    await spawnProcess("./gradlew", options, { cwd: CODE_GEN_ROOT });
+    await spawnProcess("./gradlew", options, {
+      cwd: CODE_GEN_ROOT,
+      env: {
+        ...process.env,
+        AWS_PARTITIONS_FILE_OVERRIDE: join(
+          __dirname,
+          "..",
+          "..",
+          "packages",
+          "util-endpoints",
+          "src",
+          "lib",
+          "aws",
+          "partitions.json"
+        ),
+      },
+    });
     // We're copying generated code to temporary directory as it's cleans
     // codegen directory in every run.
     // Refs: https://github.com/aws/aws-sdk-js-v3/issues/3321
@@ -64,12 +96,40 @@ const generateClients = async (models, batchSize) => {
 const generateProtocolTests = async () => {
   await spawnProcess("./gradlew", [":protocol-test-codegen:clean", ":protocol-test-codegen:build"], {
     cwd: CODE_GEN_ROOT,
+    env: {
+      ...process.env,
+      AWS_PARTITIONS_FILE_OVERRIDE: join(
+        __dirname,
+        "..",
+        "..",
+        "packages",
+        "util-endpoints",
+        "src",
+        "lib",
+        "aws",
+        "partitions.json"
+      ),
+    },
   });
 };
 
 const generateGenericClient = async () => {
   await spawnProcess("./gradlew", [":smithy-aws-typescript-codegen:clean", ":generic-client-test-codegen:build"], {
     cwd: CODE_GEN_ROOT,
+    env: {
+      ...process.env,
+      AWS_PARTITIONS_FILE_OVERRIDE: join(
+        __dirname,
+        "..",
+        "..",
+        "packages",
+        "util-endpoints",
+        "src",
+        "lib",
+        "aws",
+        "partitions.json"
+      ),
+    },
   });
 };
 

scripts/generate-clients/nested-clients/generate-nested-clients.js

@@ -99,7 +99,24 @@ async function generateNestedClient(clientName, operations) {
   });
   writeFileSync(targetModelPath, JSON.stringify(model, null, 2) + "\n");
 
-  await spawnProcess("./gradlew", options, { cwd: CODE_GEN_ROOT });
+  await spawnProcess("./gradlew", options, {
+    cwd: CODE_GEN_ROOT,
+    env: {
+      ...process.env,
+      AWS_PARTITIONS_FILE_OVERRIDE: join(
+        __dirname,
+        "..",
+        "..",
+        "..",
+        "packages",
+        "util-endpoints",
+        "src",
+        "lib",
+        "aws",
+        "partitions.json"
+      ),
+    },
+  });
   rmSync(join(__dirname, "..", "..", "..", "codegen", "sdk-codegen", `smithy-build-${clientName}.json`));
 }
 

scripts/generate-defaults-mode-provider/index.js

@@ -7,6 +7,20 @@ const { CODE_GEN_ROOT } = require("../generate-clients/code-gen-dir");
 const run = async () => {
   await spawnProcess("./gradlew", [":sdk-codegen:generate-default-configs-provider"], {
     cwd: CODE_GEN_ROOT,
+    env: {
+      ...process.env,
+      AWS_PARTITIONS_FILE_OVERRIDE: join(
+        __dirname,
+        "..",
+        "..",
+        "packages",
+        "util-endpoints",
+        "src",
+        "lib",
+        "aws",
+        "partitions.json"
+      ),
+    },
   });
 
   const DEFAULTS_MODE_CONFIG_DIR = normalize(join(__dirname, "..", "..", "packages", "smithy-client", "src"));