@argos-ci/core 1.2.0 → 1.2.1

package/dist/index.mjs

@@ -558,9 +558,29 @@ const getConfigFromOptions = ({ parallel, ...options })=>{
     });
     return config;
 };
+async function uploadFilesToS3(files) {
+    debug(`Split files in chunks of ${CHUNK_SIZE}`);
+    const chunks = chunk(files, CHUNK_SIZE);
+    debug(`Starting upload of ${chunks.length} chunks`);
+    for(let i = 0; i < chunks.length; i++){
+        // Upload files
+        debug(`Uploading chunk ${i + 1}/${chunks.length}`);
+        const timeLabel = `Chunk ${i + 1}/${chunks.length}`;
+        debugTime(timeLabel);
+        await Promise.all(chunks[i].map(async ({ url, path, contentType })=>{
+            await upload$1({
+                url,
+                path,
+                contentType
+            });
+        }));
+        debugTimeEnd(timeLabel);
+    }
+}
 /**
  * Upload screenshots to argos-ci.com.
  */ const upload = async (params)=>{
+    var _result_pwTraces;
     debug("Starting upload with params", params);
     // Read config
     const config = getConfigFromOptions(params);
@@ -602,20 +622,21 @@ const getConfigFromOptions = ({ parallel, ...options })=>{
     }));
     // Create build
     debug("Creating build");
-    const screenshotKeys = Array.from(new Set(screenshots.map((screenshot)=>screenshot.hash)));
-    const pwTraces = screenshotKeys.reduce((pwTraces, key)=>{
-        const screenshot = screenshots.find((screenshot)=>screenshot.hash === key);
-        if (!screenshot) {
-            throw new Error(`Invariant: screenshot with hash ${key} not found`);
+    const [pwTraceKeys, screenshotKeys] = screenshots.reduce(([pwTraceKeys, screenshotKeys], screenshot)=>{
+        if (screenshot.pwTrace && !pwTraceKeys.includes(screenshot.pwTrace.hash)) {
+            pwTraceKeys.push(screenshot.pwTrace.hash);
         }
-        if (screenshot.pwTrace) {
-            pwTraces.push({
-                screenshotKey: screenshot.hash,
-                traceKey: screenshot.pwTrace.hash
-            });
+        if (!screenshotKeys.includes(screenshot.hash)) {
+            screenshotKeys.push(screenshot.hash);
         }
-        return pwTraces;
-    }, []);
+        return [
+            pwTraceKeys,
+            screenshotKeys
+        ];
+    }, [
+        [],
+        []
+    ]);
     const result = await apiClient.createBuild({
         commit: config.commit,
         branch: config.branch,
@@ -623,50 +644,38 @@ const getConfigFromOptions = ({ parallel, ...options })=>{
         parallel: config.parallel,
         parallelNonce: config.parallelNonce,
         screenshotKeys,
-        pwTraces,
+        pwTraceKeys,
         prNumber: config.prNumber,
         prHeadCommit: config.prHeadCommit,
         referenceBranch: config.referenceBranch,
         referenceCommit: config.referenceCommit
     });
-    debug("Got screenshots", result);
-    debug(`Split screenshots in chunks of ${CHUNK_SIZE}`);
-    const chunks = chunk(result.screenshots, CHUNK_SIZE);
-    debug(`Starting upload of ${chunks.length} chunks`);
-    for(let i = 0; i < chunks.length; i++){
-        // Upload screenshots
-        debug(`Uploading chunk ${i + 1}/${chunks.length}`);
-        const timeLabel = `Chunk ${i + 1}/${chunks.length}`;
-        debugTime(timeLabel);
-        await Promise.all(chunks[i].map(async ({ key, putUrl, putTraceUrl })=>{
+    debug("Got uploads url", result);
+    const uploadFiles = [
+        ...result.screenshots.map(({ key, putUrl })=>{
             const screenshot = screenshots.find((s)=>s.hash === key);
             if (!screenshot) {
                 throw new Error(`Invariant: screenshot with hash ${key} not found`);
             }
-            await Promise.all([
-                // Upload screenshot
-                upload$1({
-                    url: putUrl,
-                    path: screenshot.optimizedPath,
-                    contentType: "image/png"
-                }),
-                // Upload trace
-                (async ()=>{
-                    if (putTraceUrl) {
-                        if (!screenshot.pwTrace) {
-                            throw new Error(`Invariant: screenshot with hash ${key} has a putTraceUrl but no pwTrace`);
-                        }
-                        await upload$1({
-                            url: putTraceUrl,
-                            path: screenshot.pwTrace.path,
-                            contentType: "application/zip"
-                        });
-                    }
-                })()
-            ]);
-        }));
-        debugTimeEnd(timeLabel);
-    }
+            return {
+                url: putUrl,
+                path: screenshot.optimizedPath,
+                contentType: "image/png"
+            };
+        }),
+        ...((_result_pwTraces = result.pwTraces) === null || _result_pwTraces === void 0 ? void 0 : _result_pwTraces.map(({ key, putUrl })=>{
+            const screenshot = screenshots.find((s)=>s.pwTrace && s.pwTrace.hash === key);
+            if (!screenshot || !screenshot.pwTrace) {
+                throw new Error(`Invariant: trace with ${key} not found`);
+            }
+            return {
+                url: putUrl,
+                path: screenshot.pwTrace.path,
+                contentType: "application/json"
+            };
+        })) ?? []
+    ];
+    await uploadFilesToS3(uploadFiles);
     // Update build
     debug("Updating build");
     await apiClient.updateBuild({

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@argos-ci/core",
   "description": "Visual testing solution to avoid visual regression. The core component of Argos SDK that handles build creation.",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "type": "module",
   "main": "./dist/index.cjs",
   "types": "./dist/index.d.ts",
@@ -59,5 +59,5 @@
     "build": "rollup -c",
     "e2e": "node ./e2e/upload.cjs && node ./e2e/upload.mjs"
   },
-  "gitHead": "47a939474ca0c0d55ca5360dcaa5f8912547ed76"
+  "gitHead": "366dc3bdb4a0a8ed87fc7a2cbf28db1f218567ad"
 }