poi-plugin-mcp 0.2.1 → 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

@@ -2,6 +2,8 @@ const fs = require('fs')
 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')
@@ -19,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
@@ -31,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))
   }
 
@@ -66,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' })
@@ -86,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
 
@@ -228,7 +268,7 @@ function handleMcpMessage(message, readStore, plannerFile) {
           resources: { subscribe: false },
           tools: {},
         },
-        serverInfo: { name: 'poi-plugin-mcp', version: '0.2.0' },
+        serverInfo: { name: 'poi-plugin-mcp', version: packageJson.version },
       })
 
     case 'notifications/initialized':
@@ -333,8 +373,16 @@ const MCP_TOOLS = [
   },
   {
     name: 'get_all',
-    description: 'Get account basics, fleets, ships, equipment, resources, quests, airbase, and names.',
-    inputSchema: { type: 'object', properties: {} },
+    description: 'Get account basics, fleets, ships, equipment, resources, quests, airbase, and names. Optionally include master, event, and planner data.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        include: {
+          type: 'array',
+          items: { type: 'string', enum: ['master', 'event', 'planner'] },
+        },
+      },
+    },
   },
 ]
 
@@ -349,7 +397,7 @@ function callMcpTool(toolName, args, readStore, plannerFile) {
     case 'get_resources':
       return { value: readBridgeData('/resources', readStore, plannerFile) }
     case 'get_all':
-      return { value: readBridgeData('/all', readStore, plannerFile) }
+      return { value: buildAllPayload(args || {}, readStore, plannerFile) }
     default:
       return { error: `Unknown tool: ${toolName}` }
   }
@@ -446,28 +494,69 @@ function buildFleetStatus(args, readStore) {
 
 function searchShips(args, readStore) {
   const store = readStore()
+  const master = extractMasterData(store)
   const ships = Object.values((store.info && store.info.ships) || {}).filter((ship) => {
     if (!ship) return false
     if (args.minLevel != null && ship.api_lv < Number(args.minLevel)) return false
     if (args.maxLevel != null && ship.api_lv > Number(args.maxLevel)) return false
     if (args.minMorale != null && ship.api_cond < Number(args.minMorale)) return false
     return true
-  })
+  }).map((ship) => enrichShip(ship, master))
 
   return { total: ships.length, ships }
 }
 
 function searchEquipment(args, readStore) {
   const store = readStore()
+  const master = extractMasterData(store)
   const equipment = Object.values((store.info && store.info.equips) || {}).filter((equip) => {
     if (!equip) return false
     if (args.minLevel != null && (equip.api_level || 0) < Number(args.minLevel)) return false
     return true
-  })
+  }).map((equip) => enrichEquipment(equip, master))
 
   return { total: equipment.length, equipment }
 }
 
+function buildAllPayload(args, readStore, plannerFile) {
+  const payload = readBridgeData('/all', readStore, plannerFile)
+  const include = Array.isArray(args.include) ? new Set(args.include) : new Set()
+
+  if (include.has('master')) payload.master = readBridgeData('/master', readStore, plannerFile)
+  if (include.has('event')) payload.event = readBridgeData('/event', readStore, plannerFile)
+  if (include.has('planner')) payload.planner = readBridgeData('/planner', readStore, plannerFile)
+
+  return payload
+}
+
+function enrichShip(ship, master) {
+  const masterShip = master.ships && master.ships[ship.api_ship_id]
+  const shipType = masterShip && master.shipTypes && master.shipTypes[masterShip.api_stype]
+
+  return {
+    ...ship,
+    instanceId: ship.api_id,
+    masterId: ship.api_ship_id,
+    name: (masterShip && masterShip.api_name) || '',
+    typeName: (shipType && shipType.api_name) || '',
+  }
+}
+
+function enrichEquipment(equip, master) {
+  const masterEquip = master.equipment && master.equipment[equip.api_slotitem_id]
+  const typeIds = masterEquip && Array.isArray(masterEquip.api_type) ? masterEquip.api_type : []
+  const typeId = typeIds[2] || typeIds[1] || typeIds[0]
+  const equipType = typeId && master.equipmentTypes && master.equipmentTypes[typeId]
+
+  return {
+    ...equip,
+    instanceId: equip.api_id,
+    masterId: equip.api_slotitem_id,
+    name: (masterEquip && masterEquip.api_name) || '',
+    typeName: (equipType && equipType.api_name) || '',
+  }
+}
+
 function jsonRpcResult(id, result) {
   return { jsonrpc: JSONRPC_VERSION, id, result }
 }
@@ -547,6 +636,8 @@ function extractEventData(store) {
     const tag = tags[area - 1] || emptyTag(area)
 
     ships[ship.api_id] = {
+      instanceId: ship.api_id,
+      masterId: ship.api_ship_id,
       shipId: ship.api_id,
       modelId: ship.api_ship_id,
       area,

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/mcp-server.js

@@ -57,6 +57,7 @@ const os = require('os')
 const path = require('path')
 const fs = require('fs')
 const http = require('http')
+const packageJson = require('./package.json')
 
 const PORT_FILE = path.join(os.homedir(), '.poi-mcp', 'port')
 
@@ -139,6 +140,53 @@ function getInjectScript() {
   ].join('\n')
 }
 
+async function fetchMasterData() {
+  try {
+    return await fetchFromPoi('/master')
+  } catch (_) {
+    return { ships: {}, equipment: {}, shipTypes: {}, equipmentTypes: {} }
+  }
+}
+
+function enrichShip(ship, master) {
+  const masterShip = master.ships && master.ships[ship.api_ship_id]
+  const shipType = masterShip && master.shipTypes && master.shipTypes[masterShip.api_stype]
+
+  return {
+    ...ship,
+    instanceId: ship.api_id,
+    masterId: ship.api_ship_id,
+    name: (masterShip && masterShip.api_name) || '',
+    typeName: (shipType && shipType.api_name) || '',
+  }
+}
+
+function enrichEquipment(equip, master) {
+  const masterEquip = master.equipment && master.equipment[equip.api_slotitem_id]
+  const typeIds = masterEquip && Array.isArray(masterEquip.api_type) ? masterEquip.api_type : []
+  const typeId = typeIds[2] || typeIds[1] || typeIds[0]
+  const equipType = typeId && master.equipmentTypes && master.equipmentTypes[typeId]
+
+  return {
+    ...equip,
+    instanceId: equip.api_id,
+    masterId: equip.api_slotitem_id,
+    name: (masterEquip && masterEquip.api_name) || '',
+    typeName: (equipType && equipType.api_name) || '',
+  }
+}
+
+async function fetchAllData(args = {}) {
+  const payload = await fetchFromPoi('/all')
+  const include = Array.isArray(args.include) ? new Set(args.include) : new Set()
+
+  if (include.has('master')) payload.master = await fetchFromPoi('/master')
+  if (include.has('event')) payload.event = await fetchFromPoi('/event')
+  if (include.has('planner')) payload.planner = await fetchFromPoi('/planner')
+
+  return payload
+}
+
 // ─── MCP Protocol ────────────────────────────────────────────────────────────
 
 // Minimum viable MCP stdio server — no external dependencies
@@ -161,7 +209,7 @@ function sendLog(text) {
 // ─── Main ────────────────────────────────────────────────────────────────────
 
 async function main() {
-  sendLog('POI MCP Server v0.1.0')
+  sendLog(`POI MCP Server v${packageJson.version}`)
   sendLog('Checking POI data API...')
 
   const port = getPoiPort()
@@ -226,23 +274,25 @@ async function main() {
 
     search_ships: async (args) => {
       const ships = await fetchFromPoi('/ships')
+      const master = await fetchMasterData()
       const results = Object.values(ships).filter(s => {
         if (!s) return false
         if (args.minLevel != null && s.api_lv < args.minLevel) return false
         if (args.maxLevel != null && s.api_lv > args.maxLevel) return false
         if (args.minMorale != null && s.api_cond < args.minMorale) return false
         return true
-      })
+      }).map(s => enrichShip(s, master))
       return { total: results.length, ships: results }
     },
 
     search_equipment: async (args) => {
       const equips = await fetchFromPoi('/equipment')
+      const master = await fetchMasterData()
       const results = Object.values(equips).filter(e => {
         if (!e) return false
         if (args.minLevel != null && (e.api_level || 0) < args.minLevel) return false
         return true
-      })
+      }).map(e => enrichEquipment(e, master))
       return { total: results.length, equipment: results }
     },
 
@@ -250,14 +300,15 @@ async function main() {
       return await fetchFromPoi('/resources')
     },
 
-    get_all: async () => {
-      return await fetchFromPoi('/all')
+    get_all: async (args) => {
+      return await fetchAllData(args)
     }
   }
 
   const resourceUris = [
     'poi://fleets', 'poi://ships', 'poi://equipment',
-    'poi://resources', 'poi://quests', 'poi://airbase', 'poi://basic', 'poi://all'
+    'poi://resources', 'poi://quests', 'poi://airbase', 'poi://basic',
+    'poi://names', 'poi://master', 'poi://event', 'poi://planner', 'poi://all'
   ]
 
   // ── JSON-RPC over stdio ──────────────────────────────────────────────
@@ -282,7 +333,7 @@ async function main() {
                 resources: { subscribe: false },
                 tools: {}
               },
-              serverInfo: { name: 'poi-mcp', version: '0.1.0' }
+              serverInfo: { name: 'poi-mcp', version: packageJson.version }
             })
             break
 
@@ -357,8 +408,16 @@ async function main() {
                 },
                 {
                   name: 'get_all',
-                  description: '获取所有数据（舰队/舰娘/装备/资源/任务/陆航）',
-                  inputSchema: { type: 'object', properties: {} }
+                  description: '获取所有数据（舰队/舰娘/装备/资源/任务/陆航），可选包含 master/event/planner',
+                  inputSchema: {
+                    type: 'object',
+                    properties: {
+                      include: {
+                        type: 'array',
+                        items: { type: 'string', enum: ['master', 'event', 'planner'] }
+                      }
+                    }
+                  }
                 }
               ]
             })

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "poi-plugin-mcp",
-  "version": "0.2.1",
-  "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
   },