feat: add scripts/generate-e2e-matrix

Author: xiaoyu2er

.github/scripts/generate-e2e-matrix.js

@@ -0,0 +1,75 @@
+#!/usr/bin/env node
+
+/**
+ * Generate E2E test matrix by creating cross-product of locales and shards
+ *
+ * Usage:
+ *   node generate-e2e-matrix.js '<locale-matrix-json>' <shard-total>
+ *
+ * Example:
+ *   node generate-e2e-matrix.js '[{"locale":"zh-hans","secret_project_id":"VERCEL_PROJECT_ID_ZH_HANS"},{"locale":"zh-hant","secret_project_id":"VERCEL_PROJECT_ID_ZH_HANT"}]' 5
+ */
+
+function generateE2EMatrix(localesJson, shardTotal) {
+  try {
+    // Parse the locales input
+    const locales = JSON.parse(localesJson);
+    const shards = Array.from({ length: shardTotal }, (_, i) => i + 1);
+
+    // Generate cross-product of locales and shards
+    const matrix = [];
+
+    for (const locale of locales) {
+      for (const shard of shards) {
+        matrix.push({
+          ...locale,
+          shard: shard,
+        });
+      }
+    }
+
+    const result = {
+      include: matrix,
+    };
+
+    return result;
+  } catch (error) {
+    console.error('Error generating matrix:', error.message);
+    throw error;
+  }
+}
+
+// Main execution
+if (require.main === module) {
+  const args = process.argv.slice(2);
+
+  if (args.length !== 2) {
+    console.error(
+      'Usage: node generate-e2e-matrix.js <locale-matrix-json> <shard-total>',
+    );
+    console.error(
+      'Example: node generate-e2e-matrix.js \'[{"locale":"zh-hans"}]\' 3',
+    );
+    process.exit(1);
+  }
+
+  const [localesJson, shardTotalStr] = args;
+  const shardTotal = Number.parseInt(shardTotalStr, 10);
+
+  if (Number.isNaN(shardTotal) || shardTotal < 1) {
+    console.error('Error: shard-total must be a positive integer');
+    process.exit(1);
+  }
+
+  try {
+    const matrix = generateE2EMatrix(localesJson, shardTotal);
+
+    // Output results in GitHub Actions format (same as prerelease matrix script)
+    console.log(`test-matrix=${JSON.stringify(matrix)}`);
+  } catch (error) {
+    console.error('Failed to generate matrix:', error.message);
+    process.exit(1);
+  }
+}
+
+module.exports = { generateE2EMatrix };

.github/workflows/test-e2e.yml

@@ -14,24 +14,34 @@ on:
         default: 3
 
 jobs:
-  prepare-shards:
+  prepare-matrix:
     runs-on: ubuntu-latest
     outputs:
-      shard-array: ${{ steps.generate-shards.outputs.shard-array }}
+      test-matrix: ${{ steps.generate-matrix.outputs.test-matrix }}
     steps:
-      - name: Generate shard array
-        id: generate-shards
+      - name: Checkout code
+        uses: actions/checkout@v3
+        with:
+          fetch-depth: 1
+      
+      - name: Setup Node
+        uses: actions/[email protected]
+        with:
+          node-version-file: .nvmrc
+      
+      - name: Generate test matrix
+        id: generate-matrix
         run: |
-          shards=$(seq 1 ${{ inputs.shard-total }} | jq -s -c 'map(tonumber)')
-          echo "shard-array=$shards" >> $GITHUB_OUTPUT
+          # Use the Node.js script to generate the matrix
+          output=$(node .github/scripts/generate-e2e-matrix.js '${{ inputs.matrix-include }}' ${{ inputs.shard-total }})
+          echo "$output" >> $GITHUB_OUTPUT
+          echo "Generated matrix: $output"
 
   test:
-    needs: prepare-shards
+    needs: prepare-matrix
     runs-on: ubuntu-latest
     strategy:
-      matrix:
-        include: ${{ fromJson(inputs.matrix-include) }}
-        shard: ${{ fromJson(needs.prepare-shards.outputs.shard-array) }}
+      matrix: ${{ fromJson(needs.prepare-matrix.outputs.test-matrix) }}
     name: Test ${{ matrix.locale }} (Shard ${{ matrix.shard }}/${{ inputs.shard-total }})
     steps:
       - name: Checkout code
@@ -62,6 +72,8 @@ jobs:
           pnpm --filter @next-i18n/docs playwright:install
     
       - name: Run E2E Tests
+        env:
+          LOCALE: ${{ matrix.locale }}
         run: |
           echo "Running tests for ${{ matrix.locale }} (Shard ${{ matrix.shard }}/${{ inputs.shard-total }})"
           pnpm --filter @next-i18n/docs test:e2e --shard ${{ matrix.shard }}/${{ inputs.shard-total }}