mioku-plugin-agent 0.1.0 → 0.2.0

package/platforms/types.ts

@@ -0,0 +1,53 @@
+import type { Bot } from "mioku";
+
+/** 需要按平台换取下载地址的附件引用 */
+export interface PlatformFileRef {
+  fileId: string;
+  groupId?: string;
+  userId?: string;
+}
+
+export interface PlatformFileLookup {
+  sources: string[];
+  names: string[];
+}
+
+export const EMPTY_FILE_LOOKUP: PlatformFileLookup = { sources: [], names: [] };
+
+/** 把平台返回的原始结果合并成下载来源(各平台实现共用) */
+export const mergeFileLookup = (
+  target: PlatformFileLookup,
+  result: unknown,
+): void => {
+  if (!result || typeof result !== "object") return;
+  const record = result as Record<string, unknown>;
+  for (const key of ["url", "file", "path"]) {
+    const value = record[key];
+    if (typeof value === "string" && value.trim()) {
+      target.sources.push(value.trim());
+    }
+  }
+  const base64 = record.base64 ?? record.data;
+  if (typeof base64 === "string" && base64.trim()) {
+    target.sources.push(`base64://${base64.trim()}`);
+  }
+  for (const key of ["file_name", "name"]) {
+    const value = record[key];
+    if (typeof value === "string" && value.trim()) {
+      target.names.push(value.trim());
+    }
+  }
+};
+
+/**
+ * 一个平台分支：自己订阅 `adapter:message` 路由，自己处理平台专有差异。
+ * 新增平台只要加一个文件并登记到 `platforms/index.ts`。
+ */
+export interface AgentPlatform {
+  /** 适配器名，与 `bot.adapter` 对应 */
+  readonly adapter: string;
+  /** 该分支订阅的事件路由，如 `onebotv11:message` */
+  readonly route: string;
+  /** 平台专有的 file_id → 下载来源；没有该能力时返回空 */
+  resolveFile(bot: Bot, ref: PlatformFileRef): Promise<PlatformFileLookup>;
+}

package/tools/approval.ts

@@ -2,7 +2,7 @@ import type { AgentPermissionLevel } from "../types";
 
 export interface PendingApproval {
   id: string;
-  userId: number;
+  userId: string;
   command: string;
   cwd: string;
   level: AgentPermissionLevel;
@@ -18,7 +18,7 @@ interface PendingEntry extends PendingApproval {
 
 export class ApprovalManager {
   private pending = new Map<string, PendingEntry>();
-  private latestByUser = new Map<number, string>();
+  private latestByUser = new Map<string, string>();
   private seq = 0;
 
   create(
@@ -55,7 +55,7 @@ export class ApprovalManager {
     return true;
   }
 
-  resolveLatest(userId: number, approved: boolean): PendingApproval | null {
+  resolveLatest(userId: string, approved: boolean): PendingApproval | null {
     const id = this.latestByUser.get(userId) ?? this.latestFor(userId);
     if (!id) return null;
     const entry = this.pending.get(id);
@@ -65,13 +65,13 @@ export class ApprovalManager {
     return approval;
   }
 
-  latestByUserId(userId: number): PendingApproval | null {
+  latestByUserId(userId: string): PendingApproval | null {
     const id = this.latestByUser.get(userId) ?? this.latestFor(userId);
     const entry = id ? this.pending.get(id) : undefined;
     return entry ? this.toPending(entry) : null;
   }
 
-  cancelByUser(userId: number): number {
+  cancelByUser(userId: string): number {
     const ids = [...this.pending.values()]
       .filter((entry) => entry.userId === userId)
       .map((entry) => entry.id);
@@ -79,7 +79,7 @@ export class ApprovalManager {
     return ids.length;
   }
 
-  listByUser(userId: number): PendingApproval[] {
+  listByUser(userId: string): PendingApproval[] {
     return [...this.pending.values()]
       .filter((entry) => entry.userId === userId)
       .sort((a, b) => a.createdAt - b.createdAt)
@@ -95,7 +95,7 @@ export class ApprovalManager {
     this.latestByUser.clear();
   }
 
-  private latestFor(userId: number): string | undefined {
+  private latestFor(userId: string): string | undefined {
     let latest: PendingEntry | undefined;
     for (const entry of this.pending.values()) {
       if (entry.userId !== userId) continue;

package/tools/bash.ts

@@ -47,7 +47,7 @@ export interface BashReporter {
 }
 
 interface BashToolDeps {
-  userId: number;
+  userId: string;
   policy: FsPolicy;
   config: BashConfig;
   approvals: ApprovalManager;

package/tools/deliver.ts

@@ -8,7 +8,7 @@ import { sendImageSource, sendLocalFile } from "../core/attachment";
 interface DeliverToolDeps {
   ctx: MiokuContext;
   bot: Bot | undefined;
-  userId: number;
+  userId: string;
   policy: FsPolicy;
 }
 

package/tools/index.ts

@@ -25,7 +25,10 @@ import { assessCommandRisk } from "../core/risk";
 import type { TurnActivity } from "../core/activity";
 
 export interface TurnToolOptions {
-  userId: number;
+  /** 会话/工作区隔离键(适配器 + 平台用户 id) */
+  userId: string;
+  /** 平台原始用户 id,用于发送消息 */
+  sendUserId: string;
   bot: Bot | undefined;
   runId: number;
   reporter: BashReporter;
@@ -120,13 +123,13 @@ export function buildTurnTools(
     createSendFileTool({
       ctx: host.ctx,
       bot: options.bot,
-      userId: options.userId,
+      userId: options.sendUserId,
       policy,
     }),
     createSendImageTool({
       ctx: host.ctx,
       bot: options.bot,
-      userId: options.userId,
+      userId: options.sendUserId,
       policy,
     }),
   ];
@@ -175,6 +178,7 @@ export function buildTurnTools(
     createTodoTool({
       host,
       userId: options.userId,
+      sendUserId: options.sendUserId,
       bot: options.bot,
       quiet: isQuietMode(policy.level),
     }),

package/tools/perm.ts

@@ -19,12 +19,16 @@ export function normalizePermissionLevel(value: unknown): AgentPermissionLevel {
   return "workspace-write";
 }
 
+/** 作用域键里可能含 `:`(适配器前缀),统一替换成目录安全字符 */
+const sanitizeSegment = (value: string): string =>
+  String(value ?? "").replace(/[^A-Za-z0-9._-]+/g, "_") || "unknown";
+
 export interface FsPolicy {
   level: AgentPermissionLevel;
   workspaceRoot: string;
 }
 
-export function workspaceRootFor(baseDir: string, userId: number): string {
+export function workspaceRootFor(baseDir: string, userId: string): string {
   const raw = String(baseDir ?? "").trim();
   const base =
     raw && path.isAbsolute(raw)
@@ -33,7 +37,7 @@ export function workspaceRootFor(baseDir: string, userId: number): string {
           process.cwd(),
           raw || path.join("data", "agent", "workspace"),
         );
-  return path.resolve(base, String(userId));
+  return path.resolve(base, sanitizeSegment(userId));
 }
 
 export function resolveWorkspacePath(policy: FsPolicy, target: string): string {

package/tools/todo.ts

@@ -38,12 +38,15 @@ export function formatPlanText(items: SessionPlanItem[]): string {
 
 export function createTodoTool(options: {
   host: AgentHost;
-  userId: number;
+  /** 会话隔离键 */
+  userId: string;
+  /** 平台原始用户 id,用于推送 */
+  sendUserId: string;
   bot: Bot | undefined;
   /** yolo 模式：不推送清单，用户只看最终回复 */
   quiet?: boolean;
 }): AITool {
-  const { host, userId, bot, quiet } = options;
+  const { host, userId, sendUserId, bot, quiet } = options;
   return {
     name: "todo_write",
     description: DESCRIPTION,
@@ -114,7 +117,7 @@ export function createTodoTool(options: {
       if (bot && !quiet) {
         await bot
           .sendMessage(
-            { type: "private", user_id: userId },
+            { type: "private", user_id: sendUserId },
             [host.ctx.segment.text(formatPlanText(items))],
           )
           .catch((err) =>

package/types.ts

@@ -19,7 +19,7 @@ export type AgentPermissionLevel =
 
 export interface AgentAccessConfig {
   allowAdmins: boolean;
-  users: number[];
+  users: string[];
 }
 
 export interface AgentBaseConfig {
@@ -104,8 +104,8 @@ export interface AgentHost {
   getSettings(): AgentSettingsConfig;
   getChatShared(): Promise<ChatSharedConfig>;
   resolveModel(): ResolvedModel | null;
-  workspaceRoot(userId: number): string;
-  isAllowed(userId: number): Promise<boolean>;
+  workspaceRoot(userId: string): string;
+  isAllowed(userId: string): Promise<boolean>;
   updateBase(patch: Partial<AgentBaseConfig>): Promise<void>;
   logger: MiokuContext["logger"];
 }