mioku-service-webui 4.0.9 → 4.3.0

package/index.ts

@@ -8,7 +8,12 @@ import { createNodeWebSocket } from "@hono/node-ws";
 import type { MiokuService } from "mioku";
 import type { WebUISettings } from "./types";
 import { ensureAuthConfig, loginWithToken, requireAuth } from "./auth";
-import { ensureWebUIDist, getWebUISettings, getSystemOverview, getSaying } from "./system";
+import {
+  ensureWebUIDist,
+  getWebUISettings,
+  getSystemOverview,
+  getSaying,
+} from "./system";
 import { CHAT_CONFIG_DIR, CHAT_DATA_DIR, LOGS_DIR, WEBUI_DIST } from "./utils";
 import { initBuiltinDatasources } from "./datasources";
 import {
@@ -21,6 +26,8 @@ import {
   createManageRoutes,
   createStoreRoutes,
   createDataManagementRoutes,
+  createStrangerRoutes,
+  createAccessControlRoutes,
 } from "./routes";
 
 export interface WebUIServiceAPI {
@@ -100,6 +107,8 @@ class WebUIRuntime {
     this.app.route("/api/plugin-config", createPluginConfigRoutes());
     this.app.route("/api/meme", createMemeRoutes());
     this.app.route("/api/data-management", createDataManagementRoutes());
+    this.app.route("/api/stranger", createStrangerRoutes());
+    this.app.route("/api/access-control", createAccessControlRoutes());
 
     this.app.get(
       "/api/ws/logs",
@@ -211,7 +220,7 @@ const runtime = new WebUIRuntime();
 
 const webUIService: MiokuService = {
   name: "webui",
-  version: "2.0.0",
+  version: "4.2.0",
   description: "Mioku WebUI 管理服务",
   api: {
     getSettings: () => getWebUISettings(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mioku-service-webui",
-  "version": "4.0.9",
+  "version": "4.3.0",
   "description": "Mioku WebUI 服务",
   "main": "index.ts",
   "type": "module",

package/routes/access-control.ts

@@ -0,0 +1,162 @@
+import { Hono } from "hono";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { logger } from "mioki";
+import { readJsonFile, writeJsonFile, NODE_MODULES_DIR } from "../utils";
+
+const ACCESS_CONFIG_PATH = path.resolve(
+  process.cwd(),
+  "config/boot/access-control.json",
+);
+
+const DEFAULT_CONFIG = {
+  version: 1,
+  global: { plugins: {}, commands: {} },
+  groups: {},
+  users: {},
+};
+
+export function normalizeAccessConfig(input: any) {
+  return {
+    version: 1,
+    global: {
+      plugins: input?.global?.plugins ?? {},
+      commands: input?.global?.commands ?? {},
+    },
+    groups: input?.groups ?? {},
+    users: input?.users ?? {},
+  };
+}
+
+type AccessItemKind = "plugin" | "command";
+
+interface AccessItem {
+  kind: AccessItemKind;
+  plugin: string;
+  id: string;
+  label: string;
+  desc?: string;
+  match?: string;
+  event?: string;
+  fromHook: boolean;
+}
+
+function readAccessCatalog(): AccessItem[] {
+  if (!fs.existsSync(NODE_MODULES_DIR)) return [];
+  const entries = fs.readdirSync(NODE_MODULES_DIR, { withFileTypes: true });
+  const items: AccessItem[] = [];
+
+  for (const entry of entries) {
+    if (!entry.name.startsWith("mioku-plugin-")) continue;
+    const fullPath = path.join(NODE_MODULES_DIR, entry.name);
+    let stat: fs.Stats;
+    try {
+      stat = fs.lstatSync(fullPath);
+    } catch {
+      continue;
+    }
+    if (!stat.isDirectory() && !stat.isSymbolicLink()) continue;
+
+    const pkgPath = path.join(fullPath, "package.json");
+    if (!fs.existsSync(pkgPath)) continue;
+    let pkg: any;
+    try {
+      pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+    } catch {
+      continue;
+    }
+
+    const name = entry.name.replace(/^mioku-plugin-/, "");
+    const mioku = pkg.mioku || {};
+    const help = mioku.help || { title: name, description: "", commands: [] };
+    const hooks: Array<{ id: string; match?: string; event?: string; description?: string }> =
+      mioku.accessHooks || [];
+    const helpCommands: Array<any> = help.commands || [];
+
+    items.push({
+      kind: "plugin",
+      plugin: name,
+      id: name,
+      label: help.title || name,
+      desc: help.description,
+      fromHook: false,
+    });
+
+    const seen = new Set<string>();
+    for (const hook of hooks) {
+      seen.add(hook.id);
+      items.push({
+        kind: "command",
+        plugin: name,
+        id: hook.id,
+        label: hook.id,
+        desc: hook.description,
+        match: hook.match,
+        event: hook.event,
+        fromHook: true,
+      });
+    }
+    for (const c of helpCommands) {
+      const id = String(c.cmd || "").trim();
+      if (!id || seen.has(id)) continue;
+      items.push({
+        kind: "command",
+        plugin: name,
+        id,
+        label: id,
+        desc: c.desc,
+        fromHook: false,
+      });
+    }
+  }
+
+  return items.sort((a, b) => {
+    if (a.kind !== b.kind) return a.kind === "plugin" ? -1 : 1;
+    if (a.plugin !== b.plugin) return a.plugin.localeCompare(b.plugin);
+    return a.id.localeCompare(b.id);
+  });
+}
+
+function filterCatalog(items: AccessItem[], query?: string): AccessItem[] {
+  if (!query) return items;
+  const q = query.trim().toLowerCase();
+  if (!q) return items;
+  return items.filter((it) => {
+    return (
+      it.id.toLowerCase().includes(q) ||
+      it.label.toLowerCase().includes(q) ||
+      it.plugin.toLowerCase().includes(q) ||
+      (it.desc || "").toLowerCase().includes(q)
+    );
+  });
+}
+
+export function createAccessControlRoutes() {
+  const app = new Hono();
+
+  app.get("/", (c) => {
+    const data = readJsonFile(ACCESS_CONFIG_PATH, DEFAULT_CONFIG);
+    return c.json({ ok: true, data: normalizeAccessConfig(data) });
+  });
+
+  app.put("/", async (c) => {
+    const body = await c.req.json().catch(() => null);
+    if (!body) {
+      return c.json({ ok: false, error: "INVALID_BODY" }, 400);
+    }
+    const data = normalizeAccessConfig(body);
+    writeJsonFile(ACCESS_CONFIG_PATH, data);
+    logger.info(
+      `[webui-action] access-control.update | groups=${Object.keys(data.groups).length} users=${Object.keys(data.users).length}`,
+    );
+    return c.json({ ok: true, data });
+  });
+
+  app.get("/catalog", (c) => {
+    const q = c.req.query("q");
+    const data = filterCatalog(readAccessCatalog(), q);
+    return c.json({ ok: true, data });
+  });
+
+  return app;
+}

package/routes/index.ts

@@ -5,3 +5,5 @@ export { createDBRoutes, createPluginConfigRoutes, createMemeRoutes } from "./db
 export { createManageRoutes } from "./manage";
 export { createStoreRoutes } from "./store";
 export { createDataManagementRoutes } from "./data-management";
+export { createStrangerRoutes } from "./stranger";
+export { createAccessControlRoutes } from "./access-control";

package/routes/stranger.ts

@@ -0,0 +1,145 @@
+import { Hono } from "hono";
+import { connectedBots, logger } from "mioki";
+
+interface StrangerInfoResponse {
+  ok: boolean;
+  data?: { userId: string; nickname: string };
+  error?: string;
+}
+
+interface CacheEntry {
+  nickname: string | null;
+  expiresAt: number;
+}
+
+const CACHE_TTL_MS = 5 * 60 * 1000;
+const FETCH_TIMEOUT_MS = 4000;
+
+const nicknameCache = new Map<string, CacheEntry>();
+const inflight = new Map<string, Promise<string | null>>();
+
+function getConnectedBots(): any[] {
+  return Array.from(connectedBots.values()).filter(Boolean);
+}
+
+function normalizeUserId(input: unknown): string {
+  return String(input ?? "").trim();
+}
+
+function isValidQQNumber(id: string): boolean {
+  return /^\d{5,12}$/.test(id);
+}
+
+function getCached(userId: string): string | null | undefined {
+  const entry = nicknameCache.get(userId);
+  if (!entry) return undefined;
+  if (entry.expiresAt < Date.now()) {
+    nicknameCache.delete(userId);
+    return undefined;
+  }
+  return entry.nickname;
+}
+
+function setCached(userId: string, nickname: string | null): void {
+  nicknameCache.set(userId, {
+    nickname,
+    expiresAt: Date.now() + CACHE_TTL_MS,
+  });
+}
+
+async function fetchNicknameFromBot(
+  bot: any,
+  userId: string,
+): Promise<string | null> {
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
+  try {
+    const res = await bot.api("get_stranger_info", { user_id: userId });
+    const data = res?.data || res;
+    const nickname = String(data?.nickname ?? "").trim();
+    return nickname || null;
+  } catch (error: any) {
+    logger.warn(
+      `stranger-info: bot ${bot?.uin || bot?.bot_id || "?"} 查询 ${userId} 失败: ${error?.message || error}`,
+    );
+    return null;
+  } finally {
+    clearTimeout(timer);
+  }
+}
+
+async function lookupNickname(userId: string): Promise<string | null> {
+  const cached = getCached(userId);
+  if (cached !== undefined) return cached;
+
+  const pending = inflight.get(userId);
+  if (pending) return pending;
+
+  const promise = (async () => {
+    const bots = getConnectedBots();
+    if (bots.length === 0) {
+      return null;
+    }
+    const results = await Promise.all(
+      bots.map((bot) => fetchNicknameFromBot(bot, userId)),
+    );
+    for (const nickname of results) {
+      if (nickname) {
+        setCached(userId, nickname);
+        return nickname;
+      }
+    }
+    setCached(userId, null);
+    return null;
+  })();
+
+  inflight.set(userId, promise);
+  try {
+    return await promise;
+  } finally {
+    inflight.delete(userId);
+  }
+}
+
+export function createStrangerRoutes() {
+  const app = new Hono();
+
+  app.post("/info", async (c) => {
+    const body = await c.req.json().catch(() => ({}));
+    const userId = normalizeUserId(body?.user_id);
+    if (!isValidQQNumber(userId)) {
+      return c.json(
+        { ok: false, error: "INVALID_USER_ID" } satisfies StrangerInfoResponse,
+        400,
+      );
+    }
+
+    try {
+      const nickname = await lookupNickname(userId);
+      if (!nickname) {
+        return c.json(
+          {
+            ok: false,
+            error: "NICKNAME_NOT_FOUND",
+          } satisfies StrangerInfoResponse,
+          404,
+        );
+      }
+      return c.json({
+        ok: true,
+        data: { userId, nickname },
+      } satisfies StrangerInfoResponse);
+    } catch (error: any) {
+      logger.error(`stranger-info 失败: ${error?.message || error}`);
+      return c.json(
+        {
+          ok: false,
+          error: "INTERNAL_ERROR",
+        } satisfies StrangerInfoResponse,
+        500,
+      );
+    }
+  });
+
+  return app;
+}

package/system.ts

@@ -322,6 +322,26 @@ function readPackageJson(dir: string): any {
   }
 }
 
+const WEBUI_PLUGIN_PREFIX = "mioku-plugin-";
+
+function appendToMiokiPlugins(cwd: string, pkgName: string): boolean {
+  if (!pkgName.startsWith(WEBUI_PLUGIN_PREFIX)) return false;
+  const shortName = pkgName.slice(WEBUI_PLUGIN_PREFIX.length);
+
+  const packageJsonPath = path.join(cwd, "package.json");
+  if (!fs.existsSync(packageJsonPath)) return false;
+
+  const pkg = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
+  const mioki = pkg.mioki ?? {};
+  const plugins = Array.isArray(mioki.plugins) ? [...mioki.plugins] : [];
+  if (plugins.includes(shortName)) return false;
+
+  plugins.push(shortName);
+  pkg.mioki = { ...mioki, plugins };
+  fs.writeFileSync(packageJsonPath, `${JSON.stringify(pkg, null, 2)}\n`);
+  return true;
+}
+
 function readRootPackageJson(): any {
   return JSON.parse(fs.readFileSync(ROOT_PACKAGE_PATH, "utf-8"));
 }
@@ -1020,12 +1040,18 @@ export async function installManagedPackage(
     makeManagedUpdateCacheKey(input.target, installedName),
   );
 
+  let enabled = false;
+  if (input.target === "plugin") {
+    enabled = appendToMiokiPlugins(installDir, installedName);
+  }
+
   return {
     ok: true,
     name: installedName,
     missingServices,
     packageManager: "bun",
     restartRequired: true,
+    enabled,
     installOutput: install.stdout || install.stderr,
   };
 }