poi-plugin-mcp 0.2.1 → 0.2.2

package/lib/poi-http-bridge.js

@@ -2,6 +2,7 @@ const fs = require('fs')
 const http = require('http')
 const os = require('os')
 const path = require('path')
+const packageJson = require('../package.json')
 
 const DEFAULT_PORT = 17777
 const DEFAULT_PORT_FILE = path.join(os.homedir(), '.poi-mcp', 'port')
@@ -228,7 +229,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 +334,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 +358,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 +455,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 +597,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/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,6 +1,6 @@
 {
   "name": "poi-plugin-mcp",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Poi data bridge for local KanColle inventory tools.",
   "main": "index.js",
   "keywords": [