@midscene/core 0.1.0 → 0.1.2

package/dist/lib/image.js

@@ -45,7 +45,7 @@ module.exports = __toCommonJS(image_exports);
 
 // src/image/info.ts
 var import_node_assert = __toESM(require("assert"));
-var import_buffer = require("buffer");
+var import_node_buffer = require("buffer");
 var import_node_fs = require("fs");
 var import_sharp = __toESM(require("sharp"));
 async function imageInfo(image) {
@@ -55,7 +55,7 @@ async function imageInfo(image) {
 }
 async function imageInfoOfBase64(imageBase64) {
   const base64Data = imageBase64.replace(/^data:image\/\w+;base64,/, "");
-  return imageInfo(import_buffer.Buffer.from(base64Data, "base64"));
+  return imageInfo(import_node_buffer.Buffer.from(base64Data, "base64"));
 }
 function base64Encoded(image, withHeader = true) {
   const imageBuffer = (0, import_node_fs.readFileSync)(image);
@@ -71,12 +71,12 @@ function base64Encoded(image, withHeader = true) {
 }
 
 // src/image/transform.ts
-var import_node_buffer = require("buffer");
+var import_node_buffer2 = require("buffer");
 var import_sharp2 = __toESM(require("sharp"));
 async function saveBase64Image(options) {
   const { base64Data, outputPath } = options;
   const base64Image = base64Data.split(";base64,").pop() || base64Data;
-  const imageBuffer = import_node_buffer.Buffer.from(base64Image, "base64");
+  const imageBuffer = import_node_buffer2.Buffer.from(base64Image, "base64");
   await (0, import_sharp2.default)(imageBuffer).toFile(outputPath);
   console.log("Image successfully written to file.");
 }
@@ -88,7 +88,7 @@ async function transformImgPathToBase64(inputPath) {
 }
 async function resizeImg(base64Data) {
   const base64Image = base64Data.split(";base64,").pop() || base64Data;
-  const imageBuffer = import_node_buffer.Buffer.from(base64Image, "base64");
+  const imageBuffer = import_node_buffer2.Buffer.from(base64Image, "base64");
   const metadata = await (0, import_sharp2.default)(imageBuffer).metadata();
   const { width, height } = metadata;
   if (!width || !height) {
@@ -120,7 +120,12 @@ function calculateNewDimensions(originalWidth, originalHeight) {
   };
 }
 async function trimImage(image) {
-  const { info } = await (0, import_sharp2.default)(image).trim().toBuffer({
+  const imgInstance = (0, import_sharp2.default)(image);
+  const instanceInfo = await imgInstance.metadata();
+  if (!instanceInfo.width || instanceInfo.width <= 3 || !instanceInfo.height || instanceInfo.height <= 3) {
+    return null;
+  }
+  const { info } = await imgInstance.trim().toBuffer({
     resolveWithObject: true
   });
   if (typeof info.trimOffsetLeft === "undefined" || typeof info.trimOffsetTop === "undefined") {
@@ -133,22 +138,31 @@ async function trimImage(image) {
     height: info.height
   };
 }
-async function alignCoordByTrim(image, center) {
-  const img = await (0, import_sharp2.default)(image).extract(center).toBuffer();
-  const trimInfo = await trimImage(img);
-  if (!trimInfo) {
-    return center;
+async function alignCoordByTrim(image, centerRect) {
+  const imgInfo = await (0, import_sharp2.default)(image).metadata();
+  if (!(imgInfo == null ? void 0 : imgInfo.width) || !imgInfo.height || imgInfo.width <= 3 || imgInfo.height <= 3) {
+    return centerRect;
+  }
+  try {
+    const img = await (0, import_sharp2.default)(image).extract(centerRect).toBuffer();
+    const trimInfo = await trimImage(img);
+    if (!trimInfo) {
+      return centerRect;
+    }
+    return {
+      left: centerRect.left - trimInfo.trimOffsetLeft,
+      top: centerRect.top - trimInfo.trimOffsetTop,
+      width: trimInfo.width,
+      height: trimInfo.height
+    };
+  } catch (e) {
+    console.log(imgInfo);
+    throw e;
   }
-  return {
-    left: center.left - trimInfo.trimOffsetLeft,
-    top: center.top - trimInfo.trimOffsetTop,
-    width: trimInfo.width,
-    height: trimInfo.height
-  };
 }
 
 // src/image/visualization.ts
-var import_buffer2 = require("buffer");
+var import_buffer = require("buffer");
 var import_sharp3 = __toESM(require("sharp"));
 
 // src/utils.ts
@@ -277,7 +291,7 @@ async function composeSectionDiagram(sections, context) {
         ${rects.join("\n")}
         </svg>
     `;
-  const svgBuffer = import_buffer2.Buffer.from(rectangles);
+  const svgBuffer = import_buffer.Buffer.from(rectangles);
   const file = getTmpFile("png");
   await (0, import_sharp3.default)({
     create: {