@longshine-aimanager/ai-manager-my-ai 0.1.0

package/README.md

@@ -0,0 +1,63 @@
+# ai-manager-my-ai MCP
+
+AI Manager“我的 AI”个人只读 MCP。它查询当前 Key 所属用户及其“我的部门”授权范围，不提供写入、上报或管理能力。
+
+## 要求
+
+- Node.js 18+
+- AI Manager“我的 AI 授权 Key”
+
+登录 `https://token.longshine.com/ai-manage/employee/my-ai`，在右上角头像菜单中生成 Key。Key 只应写入 MCP 配置，不要粘贴到对话、日志或代码中。
+
+## Codex 配置
+
+```toml
+[mcp_servers.ai-manager-my-ai]
+command = "npx"
+args = ["-y", "@longshine-aimanager/ai-manager-my-ai@latest"]
+
+[mcp_servers.ai-manager-my-ai.env]
+AI_MANAGER_MY_AI_KEY = "aimq_xxx"
+```
+
+## Claude Desktop 配置
+
+```json
+{
+  "mcpServers": {
+    "ai-manager-my-ai": {
+      "command": "npx",
+      "args": ["-y", "@longshine-aimanager/ai-manager-my-ai@latest"],
+      "env": {
+        "AI_MANAGER_MY_AI_KEY": "aimq_xxx"
+      }
+    }
+  }
+}
+```
+
+## 工具
+
+| 工具 | 能力 |
+| --- | --- |
+| `status` | 检查 Key 是否已配置，不返回 Key。 |
+| `get_my_ai_summary` | 查询个人费用、Token、交互及渠道汇总。 |
+| `get_my_ai_details` | 查询个人完整日级明细。 |
+| `get_my_ai_code_output` | 查询个人代码产出与代码活动量/元。 |
+| `get_my_ai_autonomous_details` | 查询个人实时自主上报完整明细。 |
+| `get_my_department_summary` | 查询授权部门费用、Token、交互及渠道汇总。 |
+| `get_my_department_details` | 查询授权部门完整日级明细。 |
+| `get_my_department_code_output` | 查询授权部门代码产出与代码活动量/元。 |
+
+默认查询上海时区最近 30 个自然日。自定义日期为闭区间，最长 90 天，结束日期不能晚于上海当天。明细工具会自动拉取完整分页；任一分页失败时整个工具调用失败。
+
+MCP 固定连接 `https://token.longshine.com/ai-manage`。Key 被撤销、轮换、所属用户停用或部门权限被撤销后，下一次请求立即按最新状态生效。
+
+## 发布
+
+```bash
+npm run publish:dry-run
+npm run publish:npm
+```
+
+发布脚本默认使用 npm staged approval；新包首次发布需使用 `../../script/publish-ai-manager-my-ai-mcp-npm.sh --direct`。发布结果记录在本目录的 `PUBLISH_LOG.jsonl`。

package/bin/ai-manager-my-ai.mjs

@@ -0,0 +1,4 @@
+#!/usr/bin/env node
+import { startServer } from "../src/server.mjs";
+
+startServer();

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "@longshine-aimanager/ai-manager-my-ai",
+  "version": "0.1.0",
+  "description": "Read-only AI Manager My AI MCP server.",
+  "type": "module",
+  "bin": {
+    "ai-manager-my-ai": "bin/ai-manager-my-ai.mjs"
+  },
+  "files": ["bin/**/*.mjs", "src/**/*.mjs", "README.md"],
+  "scripts": {
+    "test": "node --test tests/*.test.mjs",
+    "pack:dry-run": "npm pack --dry-run",
+    "publish:dry-run": "../../script/publish-ai-manager-my-ai-mcp-npm.sh --dry-run",
+    "publish:npm": "../../script/publish-ai-manager-my-ai-mcp-npm.sh"
+  },
+  "engines": {"node": ">=18"},
+  "license": "UNLICENSED"
+}

package/src/client.mjs

@@ -0,0 +1,126 @@
+export const PRODUCTION_BASE_URL = "https://token.longshine.com/ai-manage";
+
+const ISO_DATE = /^\d{4}-\d{2}-\d{2}$/;
+const DAY_MS = 24 * 60 * 60 * 1000;
+
+function parseDate(value, label) {
+  if (!ISO_DATE.test(value)) throw new Error(`${label} 必须为 YYYY-MM-DD`);
+  const parsed = new Date(`${value}T00:00:00.000Z`);
+  if (Number.isNaN(parsed.valueOf()) || parsed.toISOString().slice(0, 10) !== value) {
+    throw new Error(`${label} 日期无效`);
+  }
+  return parsed;
+}
+
+function shiftDate(value, days) {
+  const parsed = parseDate(value, "日期");
+  parsed.setUTCDate(parsed.getUTCDate() + days);
+  return parsed.toISOString().slice(0, 10);
+}
+
+export function shanghaiToday(now = new Date()) {
+  const parts = Object.fromEntries(
+    new Intl.DateTimeFormat("en-US", {
+      timeZone: "Asia/Shanghai",
+      year: "numeric",
+      month: "2-digit",
+      day: "2-digit",
+    })
+      .formatToParts(now)
+      .map(({ type, value }) => [type, value]),
+  );
+  return `${parts.year}-${parts.month}-${parts.day}`;
+}
+
+export function normalizeRange(args = {}, today = shanghaiToday()) {
+  if (Boolean(args.from) !== Boolean(args.to)) {
+    throw new Error("from 和 to 必须同时提供");
+  }
+  const from = args.from || shiftDate(today, -29);
+  const to = args.to || today;
+  const start = parseDate(from, "from");
+  const end = parseDate(to, "to");
+  const current = parseDate(today, "上海当天");
+  if (start > end) throw new Error("开始日期不能晚于结束日期");
+  if (end > current) throw new Error("结束日期不能晚于上海当天");
+  if ((end - start) / DAY_MS >= 90) {
+    throw new Error("查询范围不能超过 90 天");
+  }
+  return { range: "custom", from, to };
+}
+
+export function configuredKey(env = process.env) {
+  const key = String(env.AI_MANAGER_MY_AI_KEY || "").trim();
+  if (!key) throw new Error("未配置 AI_MANAGER_MY_AI_KEY");
+  if (!key.startsWith("aimq_")) throw new Error("AI_MANAGER_MY_AI_KEY 格式无效");
+  return key;
+}
+
+export function createMyAiClient({
+  key,
+  fetchImpl = globalThis.fetch,
+  baseUrl = PRODUCTION_BASE_URL,
+}) {
+  return {
+    async get(path, params = {}) {
+      const url = new URL(`${baseUrl}${path}`);
+      for (const [name, value] of Object.entries(params)) {
+        if (value !== undefined && value !== null) {
+          url.searchParams.set(name, String(value));
+        }
+      }
+      const response = await fetchImpl(url.toString(), {
+        method: "GET",
+        headers: { Authorization: `Bearer ${key}`, Accept: "application/json" },
+        signal: AbortSignal.timeout(30_000),
+      });
+      if (!response.ok) throw await responseError(response, path);
+      try {
+        return await response.json();
+      } catch {
+        throw new Error(`接口 ${path} 返回非 JSON 响应`);
+      }
+    },
+  };
+}
+
+export async function allRows(fetchPage, path, params) {
+  const rows = [];
+  let page = 1;
+  let expectedTotal = null;
+  while (expectedTotal === null || rows.length < expectedTotal) {
+    const payload = await fetchPage(path, { ...params, page, page_size: 100 });
+    if (
+      !Array.isArray(payload?.rows) ||
+      !Number.isInteger(payload?.total) ||
+      payload.total < 0
+    ) {
+      throw new Error(`接口 ${path} 分页结构无效`);
+    }
+    if (expectedTotal !== null && payload.total !== expectedTotal) {
+      throw new Error(`接口 ${path} 分页总数发生变化`);
+    }
+    expectedTotal = payload.total;
+    if (payload.rows.length === 0 && rows.length < expectedTotal) {
+      throw new Error(`接口 ${path} 分页提前结束`);
+    }
+    rows.push(...payload.rows);
+    page += 1;
+  }
+  return rows;
+}
+
+async function responseError(response, path) {
+  if (response.status === 401) {
+    return new Error("授权 Key 无效、已撤销、已轮换或所属用户已停用");
+  }
+  if (response.status === 403) {
+    return new Error("授权 Key 无权访问该数据");
+  }
+  if (response.status === 429) {
+    return new Error(
+      `请求过于频繁，请在 ${response.headers.get("retry-after") || "稍后"} 秒后重试`,
+    );
+  }
+  return new Error(`接口 ${path} 返回 HTTP ${response.status}`);
+}

package/src/server.mjs

@@ -0,0 +1,83 @@
+import fs from "node:fs";
+import readline from "node:readline";
+
+import { createToolRuntime, tools } from "./tools.mjs";
+
+const packageJson = JSON.parse(
+  fs.readFileSync(new URL("../package.json", import.meta.url), "utf8"),
+);
+
+function toolResult(payload) {
+  return {
+    content: [{ type: "text", text: JSON.stringify(payload, null, 2) }],
+  };
+}
+
+function toolError(error) {
+  return {
+    isError: true,
+    content: [
+      {
+        type: "text",
+        text: error instanceof Error ? error.message : String(error),
+      },
+    ],
+  };
+}
+
+export function startServer({
+  input = process.stdin,
+  output = process.stdout,
+  env = process.env,
+} = {}) {
+  const runtime = createToolRuntime({
+    key: String(env.AI_MANAGER_MY_AI_KEY || "").trim(),
+  });
+  const write = (value) => output.write(`${JSON.stringify(value)}\n`);
+  const result = (id, value) => write({ jsonrpc: "2.0", id, result: value });
+  const fail = (id, code, message) =>
+    write({ jsonrpc: "2.0", id, error: { code, message } });
+  const lines = readline.createInterface({ input, terminal: false });
+
+  lines.on("line", async (line) => {
+    if (!line.trim()) return;
+    let request;
+    try {
+      request = JSON.parse(line);
+    } catch {
+      fail(null, -32700, "Parse error");
+      return;
+    }
+    const { id, method, params } = request;
+    if (method === "notifications/initialized") return;
+    if (method === "initialize") {
+      result(id, {
+        protocolVersion: "2024-11-05",
+        capabilities: { tools: {} },
+        serverInfo: {
+          name: packageJson.name,
+          version: packageJson.version,
+        },
+      });
+      return;
+    }
+    if (method === "tools/list") {
+      result(id, { tools });
+      return;
+    }
+    if (method === "tools/call") {
+      try {
+        result(
+          id,
+          toolResult(await runtime.call(params?.name, params?.arguments || {})),
+        );
+      } catch (error) {
+        result(id, toolError(error));
+      }
+      return;
+    }
+    fail(id, -32601, `Method not found: ${method}`);
+  });
+
+  return lines;
+}

package/src/tools.mjs

@@ -0,0 +1,147 @@
+import {
+  allRows,
+  configuredKey,
+  createMyAiClient,
+  normalizeRange,
+} from "./client.mjs";
+
+const dateProperties = {
+  from: {
+    type: "string",
+    pattern: "^\\d{4}-\\d{2}-\\d{2}$",
+    description: "开始日期，YYYY-MM-DD。",
+  },
+  to: {
+    type: "string",
+    pattern: "^\\d{4}-\\d{2}-\\d{2}$",
+    description: "结束日期，YYYY-MM-DD。",
+  },
+};
+
+function inputSchema({ scope = false } = {}) {
+  return {
+    type: "object",
+    properties: {
+      ...dateProperties,
+      ...(scope
+        ? {
+            scope: {
+              type: "string",
+              enum: ["all", "platform", "self_reporting"],
+              default: "all",
+            },
+          }
+        : {}),
+    },
+    additionalProperties: false,
+  };
+}
+
+export const tools = [
+  {
+    name: "status",
+    description: "检查个人只读 Key 是否已配置，不返回 Key 内容。",
+    inputSchema: inputSchema(),
+  },
+  {
+    name: "get_my_ai_summary",
+    description: "查询本人 AI 使用汇总。",
+    inputSchema: inputSchema({ scope: true }),
+  },
+  {
+    name: "get_my_ai_details",
+    description: "查询本人完整 AI 使用明细。",
+    inputSchema: inputSchema({ scope: true }),
+  },
+  {
+    name: "get_my_ai_code_output",
+    description: "查询本人代码产出与代码活动量/元。",
+    inputSchema: inputSchema(),
+  },
+  {
+    name: "get_my_ai_autonomous_details",
+    description: "查询本人实时自主上报完整明细。",
+    inputSchema: inputSchema(),
+  },
+  {
+    name: "get_my_department_summary",
+    description: "查询当前授权部门范围的 AI 使用汇总。",
+    inputSchema: inputSchema({ scope: true }),
+  },
+  {
+    name: "get_my_department_details",
+    description: "查询当前授权部门范围的完整 AI 使用明细。",
+    inputSchema: inputSchema({ scope: true }),
+  },
+  {
+    name: "get_my_department_code_output",
+    description: "查询当前授权部门范围的代码产出与代码活动量/元。",
+    inputSchema: inputSchema(),
+  },
+];
+
+const routes = {
+  get_my_ai_summary: {
+    path: "/api/my-ai/summary",
+    paged: false,
+    scope: true,
+  },
+  get_my_ai_details: {
+    path: "/api/my-ai/details",
+    paged: true,
+    scope: true,
+  },
+  get_my_ai_code_output: {
+    path: "/api/my-ai/code-output",
+    paged: false,
+  },
+  get_my_ai_autonomous_details: {
+    path: "/api/my-ai/autonomous/details",
+    paged: true,
+  },
+  get_my_department_summary: {
+    path: "/api/my-ai/department/summary",
+    paged: false,
+    scope: true,
+  },
+  get_my_department_details: {
+    path: "/api/my-ai/department/details",
+    paged: true,
+    scope: true,
+  },
+  get_my_department_code_output: {
+    path: "/api/my-ai/department/code-output",
+    paged: false,
+  },
+};
+
+export function createToolRuntime({ key = "", client, today } = {}) {
+  return {
+    async call(name, args = {}) {
+      if (name === "status") {
+        return {
+          configured: key.startsWith("aimq_"),
+          environment: "production",
+        };
+      }
+      const route = routes[name];
+      if (!route) throw new Error(`未知工具: ${name}`);
+      const resolvedKey = configuredKey({ AI_MANAGER_MY_AI_KEY: key });
+      const scope = args.scope || "all";
+      if (
+        route.scope &&
+        !["all", "platform", "self_reporting"].includes(scope)
+      ) {
+        throw new Error("scope 必须为 all、platform 或 self_reporting");
+      }
+      const params = {
+        ...normalizeRange(args, today),
+        ...(route.scope ? { scope } : {}),
+      };
+      const api = client || createMyAiClient({ key: resolvedKey });
+      return route.paged
+        ? { ...params, rows: await allRows(api.get, route.path, params) }
+        : api.get(route.path, params);
+    },
+  };
+}