screenright-client 0.9.1 → 0.9.3

package/dist/index.js

@@ -7,46 +7,50 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import * as fs from 'fs';
 import fetch from 'node-fetch';
 import process from 'node:process';
-const tmpDir = 'screenright/tmp';
+import FormData from 'form-data';
 const result = { screenshotItemAttributes: [] };
 let deploymentId = null;
+const deploymentToken = process.env.SCREENRIGHT_DEPLOYMENT_TOKEN || '';
 const baseUrl = () => {
     return `${process.env.SCREENRIGHT_ENDPOINT}/client_api`;
 };
+const errorOccurred = (message) => {
+    console.error('[ScreenRight] Error occurred', message);
+    deploymentId = null;
+};
 export const initializeScreenwright = () => __awaiter(void 0, void 0, void 0, function* () {
     const diagramId = process.env.SCREENRIGHT_DIAGRAM_ID;
-    const deploymentToken = process.env.SCREENRIGHT_DEPLOYMENT_TOKEN;
-    console.log("#", deploymentToken);
+    if (!diagramId || !deploymentToken) {
+        errorOccurred('Not set require environments.');
+        return;
+    }
     try {
         const response = yield fetch(`${baseUrl()}/diagrams/${diagramId}/deployments`, {
             method: 'POST',
             body: JSON.stringify({ deployment_token: deploymentToken }),
             headers: { 'Content-Type': 'application/json' }
         });
+        if (!response.ok) {
+            errorOccurred('Failed create deployment.');
+        }
         const body = yield response.text();
         const json = JSON.parse(body);
         deploymentId = json.id;
     }
     catch (e) {
-        console.error('[ScreenRight] Error catch', e.message);
+        errorOccurred(e.message);
     }
 });
 export const finalize = () => __awaiter(void 0, void 0, void 0, function* () {
-    console.log("################## -1");
     if (!deploymentId) {
         return;
     }
-    fs.writeFileSync(`${tmpDir}/result.json`, JSON.stringify({
-        screenshotItemAttributes: result.screenshotItemAttributes,
-    }));
     const diagramId = process.env.SCREENRIGHT_DIAGRAM_ID;
-    const deploymentToken = process.env.SCREENRIGHT_DEPLOYMENT_TOKEN;
     yield fetch(`${baseUrl()}/diagrams/${diagramId}/deployments/${deploymentId}/done_upload`, {
         method: 'PUT',
-        body: JSON.stringify({ deployment_token: deploymentToken, screenshotItemAttributes: result.screenshotItemAttributes }),
+        body: JSON.stringify({ deployment_token: deploymentToken, blueprint: JSON.stringify({ screenshotItemAttributes: result.screenshotItemAttributes }) }),
         headers: { 'Content-Type': 'application/json' }
     });
     deploymentId = null;
@@ -55,13 +59,33 @@ export const capture = (page, key, title, parentKey) => __awaiter(void 0, void 0
     if (deploymentId === null) {
         return;
     }
-    fs.mkdirSync(tmpDir, { recursive: true });
-    const path = `${tmpDir}/${key}.png`;
-    yield page.screenshot({ path, fullPage: true });
+    const fileName = `${key}.jpg`;
+    try {
+        const buffer = yield page.screenshot({ fullPage: true, type: 'jpeg' });
+        const formData = new FormData();
+        formData.append('file', buffer, fileName);
+        const diagramId = process.env.SCREENRIGHT_DIAGRAM_ID;
+        const response = yield fetch(`${baseUrl()}/diagrams/${diagramId}/deployments/${deploymentId}/screenshot`, {
+            method: 'POST',
+            headers: {
+                'X-File-Key': key,
+                'X-Deployment-Token': deploymentToken,
+            },
+            body: formData
+        });
+        if (!response.ok) {
+            errorOccurred('Faild screenshot upload');
+            return;
+        }
+    }
+    catch (e) {
+        errorOccurred(`capture: ${key}, ${e.message}`);
+        return;
+    }
     const attribute = {
         key,
         title,
-        src: path,
+        url: page.url(),
         childrens: [],
     };
     if (parentKey) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "screenright-client",
-  "version": "0.9.1",
+  "version": "0.9.3",
   "description": "screenright's nodejs client library",
   "type": "module",
   "main": "dist/index.js",

package/src/index.ts

@@ -7,7 +7,7 @@ import FormData from 'form-data'
 type ScreenshotItemAttribute = {
   key: string
   title: string
-  src: string
+  url: string
   childrens: ScreenshotItemAttribute[]
 }
 
@@ -15,8 +15,6 @@ type Result = {
   screenshotItemAttributes: ScreenshotItemAttribute[]
 }
 
-const tmpDir = 'screenright/tmp'
-
 const result: Result = { screenshotItemAttributes: [] }
 
 let deploymentId: string | null = null
@@ -63,17 +61,10 @@ export const finalize = async () => {
     return
   }
 
-  fs.writeFileSync(
-    `${tmpDir}/result.json`,
-    JSON.stringify({
-      screenshotItemAttributes: result.screenshotItemAttributes,
-    })
-  )
-
   const diagramId = process.env.SCREENRIGHT_DIAGRAM_ID
   await fetch(`${baseUrl()}/diagrams/${diagramId}/deployments/${deploymentId}/done_upload`, {
     method: 'PUT',
-    body: JSON.stringify({ deployment_token: deploymentToken, blueprint: { screenshotItemAttributes: result.screenshotItemAttributes } }),
+    body: JSON.stringify({ deployment_token: deploymentToken, blueprint: JSON.stringify({ screenshotItemAttributes: result.screenshotItemAttributes}) }),
     headers: { 'Content-Type': 'application/json' }
   })
 
@@ -90,21 +81,19 @@ export const capture = async (
     return
   }
 
-  const fileName = `${key}.png`
-  const outPath = `${tmpDir}/${fileName}`
+  const fileName = `${key}.jpg`
   try {
-    fs.mkdirSync(tmpDir, { recursive: true })
-    const buffer = new Uint8Array(await page.screenshot({ path: outPath, fullPage: true }))
+    const buffer = await page.screenshot({ fullPage: true, type: 'jpeg' })
     const formData = new FormData()
 
-    formData.append('file', fs.createReadStream(outPath))
+    formData.append('file', buffer, fileName)
 
     const diagramId = process.env.SCREENRIGHT_DIAGRAM_ID
     const response = await fetch(`${baseUrl()}/diagrams/${diagramId}/deployments/${deploymentId}/screenshot`, {
       method: 'POST',
       headers: {
         'X-File-Key': key,
-        'X-Deployment-Token': deploymentToken
+        'X-Deployment-Token': deploymentToken,
       },
       body: formData
     })
@@ -114,14 +103,14 @@ export const capture = async (
       return
     }
   } catch(e: any) {
-    errorOccurred(`capture: ${key}, ${outPath}, ${e.message}`)
+    errorOccurred(`capture: ${key}, ${e.message}`)
     return
   }
 
   const attribute: ScreenshotItemAttribute = {
     key,
     title,
-    src: outPath,
+    url: page.url(),
     childrens: [],
   }