poi-plugin-mcp 0.2.2 → 0.2.3

package/README.md

@@ -61,8 +61,15 @@ The default port is `17777`.
 | `/master` | Master ship, equipment, ship type, equipment type, and mission data |
 | `/event` | Event ship tag definitions plus owned ships' current sally area |
 | `/planner` | Ship Info deck planner areas and ship assignments |
+| `/screenshot` | In-memory PNG capture of the game WebView |
 | `/all` | Combined basic runtime data |
 
+`/screenshot` accepts `GET` only. It uses Poi's existing
+`screenshot::get` WebContents capture path and returns PNG base64 in JSON. It
+does not save a file, write the clipboard, capture the desktop, or appear in
+MCP resources and tools. Its response disables CORS and uses
+`Cache-Control: no-store`.
+
 ## MCP Endpoint
 
 The same local server also exposes a JSON-RPC MCP endpoint:

package/lib/bridge-controller.js

@@ -23,6 +23,7 @@ function createBridgeController(options = {}) {
   function createCurrentBridge() {
     return createBridge({
       getStore: options.getStore,
+      captureScreenshot: options.captureScreenshot,
       port: settings.port,
       portFile: options.portFile,
       logger,

package/lib/poi-http-bridge.js

@@ -3,6 +3,7 @@ const http = require('http')
 const os = require('os')
 const path = require('path')
 const packageJson = require('../package.json')
+const { createPoiScreenshotProvider } = require('./poi-screenshot')
 
 const DEFAULT_PORT = 17777
 const DEFAULT_PORT_FILE = path.join(os.homedir(), '.poi-mcp', 'port')
@@ -20,6 +21,7 @@ function createPoiDataBridge(options = {}) {
   const portFile = options.portFile || DEFAULT_PORT_FILE
   const plannerFile = options.plannerFile || DEFAULT_PLANNER_FILE
   const logger = options.logger || console
+  let captureScreenshot = options.captureScreenshot || null
 
   let server = null
   let actualPort = 0
@@ -32,11 +34,17 @@ function createPoiDataBridge(options = {}) {
     return store
   }
 
-  function sendJson(res, statusCode, data) {
-    res.writeHead(statusCode, {
-      'Access-Control-Allow-Origin': '*',
+  function sendJson(res, statusCode, data, options = {}) {
+    const headers = {
       'Content-Type': 'application/json',
-    })
+    }
+    if (options.allowCors !== false) {
+      headers['Access-Control-Allow-Origin'] = '*'
+    }
+    if (options.noStore) {
+      headers['Cache-Control'] = 'no-store'
+    }
+    res.writeHead(statusCode, headers)
     res.end(JSON.stringify(data))
   }
 
@@ -67,7 +75,7 @@ function createPoiDataBridge(options = {}) {
       })
   }
 
-  function handleRequest(req, res) {
+  async function handleRequest(req, res) {
     try {
       if (req.url === '/shutdown') {
         sendJson(res, 200, { status: 'shutting down' })
@@ -87,6 +95,37 @@ function createPoiDataBridge(options = {}) {
         return
       }
 
+      if (endpoint === '/screenshot') {
+        if (req.method !== 'GET') {
+          sendJson(
+            res,
+            405,
+            { error: 'Screenshot endpoint only accepts GET requests.' },
+            { allowCors: false, noStore: true },
+          )
+          return
+        }
+        try {
+          if (!captureScreenshot) {
+            captureScreenshot = createPoiScreenshotProvider()
+          }
+          sendJson(
+            res,
+            200,
+            await captureScreenshot(),
+            { allowCors: false, noStore: true },
+          )
+        } catch (error) {
+          sendJson(
+            res,
+            503,
+            { error: error.message },
+            { allowCors: false, noStore: true },
+          )
+        }
+        return
+      }
+
       const store = readStore()
       const info = store.info
 

package/lib/poi-screenshot.js

@@ -0,0 +1,104 @@
+const DEFAULT_MAX_BASE64_LENGTH = 16 * 1024 * 1024
+
+function createPoiScreenshotProvider(options = {}) {
+  const getStore = options.getStore || defaultGetStore
+  const ipcRenderer = options.ipcRenderer || require('electron').ipcRenderer
+  const devicePixelRatio = options.devicePixelRatio == null
+    ? defaultDevicePixelRatio()
+    : options.devicePixelRatio
+  const now = options.now || (() => new Date())
+  const maxBase64Length = options.maxBase64Length == null
+    ? DEFAULT_MAX_BASE64_LENGTH
+    : options.maxBase64Length
+
+  if (!Number.isFinite(devicePixelRatio) || devicePixelRatio <= 0) {
+    throw new Error('devicePixelRatio must be a positive finite number')
+  }
+  if (!Number.isInteger(maxBase64Length) || maxBase64Length <= 0) {
+    throw new Error('maxBase64Length must be a positive integer')
+  }
+
+  return async function capturePoiScreenshot() {
+    const layout = getStore('layout.webview')
+    if (
+      !layout ||
+      !layout.ref ||
+      typeof layout.ref.getWebContentsId !== 'function'
+    ) {
+      throw new Error('Poi game WebView is not ready')
+    }
+    if (
+      !Number.isInteger(layout.width) ||
+      layout.width <= 0 ||
+      !Number.isInteger(layout.height) ||
+      layout.height <= 0
+    ) {
+      throw new Error('Poi game WebView dimensions must be positive integers')
+    }
+
+    const webContentsId = layout.ref.getWebContentsId()
+    if (!Number.isInteger(webContentsId) || webContentsId <= 0) {
+      throw new Error('Poi game WebContents id is invalid')
+    }
+
+    const rect = {
+      x: 0,
+      y: 0,
+      width: Math.floor(layout.width * devicePixelRatio),
+      height: Math.floor(layout.height * devicePixelRatio),
+    }
+    const actualSize = {
+      width: layout.width,
+      height: layout.height,
+    }
+    const dataUrl = await ipcRenderer.invoke(
+      'screenshot::get',
+      webContentsId,
+      rect,
+      actualSize,
+    )
+    const prefix = 'data:image/png;base64,'
+    if (typeof dataUrl !== 'string' || !dataUrl.startsWith(prefix)) {
+      throw new Error('Poi screenshot did not return a PNG data URL')
+    }
+
+    const dataBase64 = dataUrl.slice(prefix.length)
+    if (
+      dataBase64.length === 0 ||
+      !/^[A-Za-z0-9+/]+={0,2}$/.test(dataBase64)
+    ) {
+      throw new Error('Poi screenshot returned invalid base64 data')
+    }
+    if (dataBase64.length > maxBase64Length) {
+      throw new Error(`Poi screenshot exceeds ${maxBase64Length} base64 characters`)
+    }
+
+    return {
+      capturedAt: now().toISOString(),
+      mimeType: 'image/png',
+      dataBase64,
+    }
+  }
+}
+
+function defaultGetStore(path) {
+  if (typeof window !== 'undefined' && typeof window.getStore === 'function') {
+    return window.getStore(path)
+  }
+  return null
+}
+
+function defaultDevicePixelRatio() {
+  if (
+    typeof window !== 'undefined' &&
+    Number.isFinite(window.devicePixelRatio) &&
+    window.devicePixelRatio > 0
+  ) {
+    return window.devicePixelRatio
+  }
+  return 1
+}
+
+module.exports = {
+  createPoiScreenshotProvider,
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "poi-plugin-mcp",
-  "version": "0.2.2",
-  "description": "Poi data bridge for local KanColle inventory tools.",
+  "version": "0.2.3",
+  "description": "Poi read-only data and game WebView capture bridge for local KanColle tools.",
   "main": "index.js",
   "keywords": [
     "poi-plugin",
@@ -24,7 +24,7 @@
   "poiPlugin": {
     "title": "MCP 数据桥",
     "id": "mcp_data_bridge",
-    "description": "Expose Poi fleet, equipment, master, event tag, and planner data on local HTTP for CLI/MCP agents.",
+    "description": "Expose Poi state and in-memory game WebView captures on a local read-only bridge.",
     "icon": "fa/database",
     "priority": 53
   },