@aiscene/shared 8.0.1 → 8.0.2

package/dist/es/img/info.mjs

@@ -1,18 +1,27 @@
 import node_assert from "node:assert";
 import get_photon from "./get-photon.mjs";
 async function imageInfoOfBase64(imageBase64) {
-    const { PhotonImage } = await get_photon();
+    if (!imageBase64 || 'string' != typeof imageBase64) throw new Error('Invalid image base64: data is empty or not a string');
     const base64Data = imageBase64.replace(/^data:image\/\w+;base64,/, '');
-    const result = PhotonImage.new_from_base64(base64Data);
-    const image = result instanceof Promise ? await result : result;
-    const width = image.get_width();
-    const height = image.get_height();
-    image.free();
-    node_assert(width && height, 'Invalid image: cannot get width or height');
-    return {
-        width,
-        height
-    };
+    if (!base64Data || 0 === base64Data.length) throw new Error('Invalid image base64: no data after stripping prefix');
+    if (!/^[A-Za-z0-9+/]*={0,2}$/.test(base64Data)) throw new Error('Invalid image base64: contains invalid characters');
+    if (base64Data.length < 4) throw new Error(`Invalid image base64: data too short (${base64Data.length} characters)`);
+    try {
+        const { PhotonImage } = await get_photon();
+        const result = PhotonImage.new_from_base64(base64Data);
+        const image = result instanceof Promise ? await result : result;
+        const width = image.get_width();
+        const height = image.get_height();
+        image.free();
+        node_assert(width && height, 'Invalid image: cannot get width or height');
+        return {
+            width,
+            height
+        };
+    } catch (error) {
+        if (error instanceof Error && error.message.includes('unreachable')) throw new Error(`Failed to decode image: invalid or corrupted image data. Original error: ${error.message}`);
+        throw error;
+    }
 }
 function isValidPNGImageBuffer(buffer) {
     if (!buffer || buffer.length < 8) return false;

package/dist/es/mcp/base-server.mjs

@@ -1,9 +1,9 @@
 import { randomUUID } from "node:crypto";
-import { setIsMcp } from "../utils.mjs";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
 import express from "express";
+import { setIsMcp } from "../utils.mjs";
 function _define_property(obj, key, value) {
     if (key in obj) Object.defineProperty(obj, key, {
         value: value,

package/dist/es/mcp/tool-generator.mjs

@@ -1,5 +1,5 @@
-import { parseBase64 } from "../img/index.mjs";
 import { z } from "zod";
+import { parseBase64 } from "../img/index.mjs";
 import { getZodDescription, getZodTypeName, isMidsceneLocatorField, unwrapZodField } from "../zod-schema-utils.mjs";
 function getErrorMessage(error) {
     return error instanceof Error ? error.message : String(error);

package/dist/lib/img/info.js

@@ -43,18 +43,27 @@ var external_node_assert_default = /*#__PURE__*/ __webpack_require__.n(external_
 const external_get_photon_js_namespaceObject = require("./get-photon.js");
 var external_get_photon_js_default = /*#__PURE__*/ __webpack_require__.n(external_get_photon_js_namespaceObject);
 async function imageInfoOfBase64(imageBase64) {
-    const { PhotonImage } = await external_get_photon_js_default()();
+    if (!imageBase64 || 'string' != typeof imageBase64) throw new Error('Invalid image base64: data is empty or not a string');
     const base64Data = imageBase64.replace(/^data:image\/\w+;base64,/, '');
-    const result = PhotonImage.new_from_base64(base64Data);
-    const image = result instanceof Promise ? await result : result;
-    const width = image.get_width();
-    const height = image.get_height();
-    image.free();
-    external_node_assert_default()(width && height, 'Invalid image: cannot get width or height');
-    return {
-        width,
-        height
-    };
+    if (!base64Data || 0 === base64Data.length) throw new Error('Invalid image base64: no data after stripping prefix');
+    if (!/^[A-Za-z0-9+/]*={0,2}$/.test(base64Data)) throw new Error('Invalid image base64: contains invalid characters');
+    if (base64Data.length < 4) throw new Error(`Invalid image base64: data too short (${base64Data.length} characters)`);
+    try {
+        const { PhotonImage } = await external_get_photon_js_default()();
+        const result = PhotonImage.new_from_base64(base64Data);
+        const image = result instanceof Promise ? await result : result;
+        const width = image.get_width();
+        const height = image.get_height();
+        image.free();
+        external_node_assert_default()(width && height, 'Invalid image: cannot get width or height');
+        return {
+            width,
+            height
+        };
+    } catch (error) {
+        if (error instanceof Error && error.message.includes('unreachable')) throw new Error(`Failed to decode image: invalid or corrupted image data. Original error: ${error.message}`);
+        throw error;
+    }
 }
 function isValidPNGImageBuffer(buffer) {
     if (!buffer || buffer.length < 8) return false;

package/dist/lib/mcp/base-server.js

@@ -38,12 +38,12 @@ __webpack_require__.d(__webpack_exports__, {
     BaseMCPServer: ()=>BaseMCPServer
 });
 const external_node_crypto_namespaceObject = require("node:crypto");
-const external_utils_js_namespaceObject = require("../utils.js");
 const mcp_js_namespaceObject = require("@modelcontextprotocol/sdk/server/mcp.js");
 const stdio_js_namespaceObject = require("@modelcontextprotocol/sdk/server/stdio.js");
 const streamableHttp_js_namespaceObject = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
 const external_express_namespaceObject = require("express");
 var external_express_default = /*#__PURE__*/ __webpack_require__.n(external_express_namespaceObject);
+const external_utils_js_namespaceObject = require("../utils.js");
 function _define_property(obj, key, value) {
     if (key in obj) Object.defineProperty(obj, key, {
         value: value,

package/dist/lib/mcp/tool-generator.js

@@ -27,8 +27,8 @@ __webpack_require__.d(__webpack_exports__, {
     generateToolsFromActionSpace: ()=>generateToolsFromActionSpace,
     generateCommonTools: ()=>generateCommonTools
 });
-const index_js_namespaceObject = require("../img/index.js");
 const external_zod_namespaceObject = require("zod");
+const index_js_namespaceObject = require("../img/index.js");
 const external_zod_schema_utils_js_namespaceObject = require("../zod-schema-utils.js");
 function getErrorMessage(error) {
     return error instanceof Error ? error.message : String(error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aiscene/shared",
-  "version": "8.0.1",
+  "version": "8.0.2",
   "repository": "https://github.com/web-infra-dev/midscene",
   "homepage": "https://midscenejs.com/",
   "types": "./dist/types/index.d.ts",

package/src/img/info.ts

@@ -14,16 +14,52 @@ export interface ImageInfo extends Size {}
 export async function imageInfoOfBase64(
   imageBase64: string,
 ): Promise<ImageInfo> {
-  const { PhotonImage } = await getPhoton();
+  // Validate input before passing to Photon
+  if (!imageBase64 || typeof imageBase64 !== 'string') {
+    throw new Error('Invalid image base64: data is empty or not a string');
+  }
+
   const base64Data = imageBase64.replace(/^data:image\/\w+;base64,/, '');
-  // Support both sync (Photon) and async (Canvas fallback) versions
-  const result = PhotonImage.new_from_base64(base64Data);
-  const image = result instanceof Promise ? await result : result;
-  const width = image.get_width();
-  const height = image.get_height();
-  image.free();
-  assert(width && height, 'Invalid image: cannot get width or height');
-  return { width, height };
+
+  // Check if base64 data is valid after stripping data URL prefix
+  if (!base64Data || base64Data.length === 0) {
+    throw new Error('Invalid image base64: no data after stripping prefix');
+  }
+
+  // Basic validation: base64 should only contain valid characters
+  // (A-Z, a-z, 0-9, +, /, and = for padding)
+  if (!/^[A-Za-z0-9+/]*={0,2}$/.test(base64Data)) {
+    throw new Error(
+      'Invalid image base64: contains invalid characters',
+    );
+  }
+
+  // Check minimum length for a valid image (at least a few bytes)
+  if (base64Data.length < 4) {
+    throw new Error(
+      `Invalid image base64: data too short (${base64Data.length} characters)`,
+    );
+  }
+
+  try {
+    const { PhotonImage } = await getPhoton();
+    // Support both sync (Photon) and async (Canvas fallback) versions
+    const result = PhotonImage.new_from_base64(base64Data);
+    const image = result instanceof Promise ? await result : result;
+    const width = image.get_width();
+    const height = image.get_height();
+    image.free();
+    assert(width && height, 'Invalid image: cannot get width or height');
+    return { width, height };
+  } catch (error) {
+    // Provide more helpful error message for Photon failures
+    if (error instanceof Error && error.message.includes('unreachable')) {
+      throw new Error(
+        `Failed to decode image: invalid or corrupted image data. Original error: ${error.message}`,
+      );
+    }
+    throw error;
+  }
 }
 
 /**

package/src/mcp/base-server.ts

@@ -1,6 +1,5 @@
 import { randomUUID } from 'node:crypto';
 import type { ParseArgsConfig } from 'node:util';
-import { setIsMcp } from '../utils';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
@@ -9,6 +8,7 @@ import express, {
   type Request,
   type Response,
 } from 'express';
+import { setIsMcp } from '../utils';
 import type { IMidsceneTools } from './types';
 
 export interface BaseMCPServerConfig {

package/src/mcp/base-tools.ts

@@ -1,6 +1,6 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { parseBase64 } from '../img';
 import { getDebug } from '../logger';
-import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import {
   generateCommonTools,
   generateToolsFromActionSpace,

package/src/mcp/tool-generator.ts

@@ -1,5 +1,5 @@
-import { parseBase64 } from '../img';
 import { z } from 'zod';
+import { parseBase64 } from '../img';
 import {
   getZodDescription,
   getZodTypeName,