@zhin.js/core 1.0.24 → 1.0.26

package/src/built/adapter-process.ts

@@ -65,10 +65,109 @@ export class ProcessBot implements Bot<{},{content:string,ts:number}>{
 export class ProcessAdapter extends Adapter<ProcessBot>{
     constructor(plugin: Plugin) {
         super(plugin, 'process', [{}]);
+        
+        // 注册适配器提供的 AI 工具
+        this.registerDefaultTools();
+        this.registerProcessTools();
     }
+    
     createBot(config: Adapter.BotConfig<ProcessBot>): ProcessBot {
         return new ProcessBot(this, config);
     }
+    
+    /**
+     * 注册 Process 适配器特有的工具
+     */
+    private registerProcessTools(): void {
+        // 获取进程信息
+        this.addTool({
+            name: 'process_get_info',
+            description: '获取当前进程的详细信息，包括 PID、内存使用、运行时间等',
+            parameters: {
+                type: 'object',
+                properties: {},
+            },
+            execute: async () => {
+                const memUsage = process.memoryUsage();
+                return {
+                    pid: process.pid,
+                    title: process.title,
+                    uptime: process.uptime(),
+                    platform: process.platform,
+                    arch: process.arch,
+                    nodeVersion: process.version,
+                    memory: {
+                        rss: Math.round(memUsage.rss / 1024 / 1024) + ' MB',
+                        heapTotal: Math.round(memUsage.heapTotal / 1024 / 1024) + ' MB',
+                        heapUsed: Math.round(memUsage.heapUsed / 1024 / 1024) + ' MB',
+                    },
+                    cwd: process.cwd(),
+                };
+            },
+        });
+        
+        // 获取环境变量
+        this.addTool({
+            name: 'process_get_env',
+            description: '获取指定的环境变量值',
+            parameters: {
+                type: 'object',
+                properties: {
+                    name: {
+                        type: 'string',
+                        description: '环境变量名称',
+                    },
+                },
+                required: ['name'],
+            },
+            execute: async (args) => {
+                const { name } = args;
+                // 安全起见，不返回敏感环境变量
+                const sensitive = ['PASSWORD', 'SECRET', 'TOKEN', 'KEY', 'CREDENTIAL'];
+                if (sensitive.some(s => name.toUpperCase().includes(s))) {
+                    return { error: '无法访问敏感环境变量' };
+                }
+                return { name, value: process.env[name] || null };
+            },
+        });
+        
+        // 输出到控制台
+        this.addTool({
+            name: 'process_console_log',
+            description: '向控制台输出信息',
+            parameters: {
+                type: 'object',
+                properties: {
+                    message: {
+                        type: 'string',
+                        description: '要输出的消息',
+                    },
+                    level: {
+                        type: 'string',
+                        description: '日志级别',
+                    },
+                },
+                required: ['message'],
+            },
+            execute: async (args) => {
+                const { message, level = 'info' } = args;
+                switch (level) {
+                    case 'warn':
+                        console.warn(message);
+                        break;
+                    case 'error':
+                        console.error(message);
+                        break;
+                    case 'debug':
+                        console.debug(message);
+                        break;
+                    default:
+                        console.log(message);
+                }
+                return { success: true, message: `已输出: ${message}` };
+            },
+        });
+    }
 }
 declare module '../adapter.js'{
     interface Adapters{

package/src/built/ai-trigger.ts

@@ -0,0 +1,259 @@
+/**
+ * AI Trigger 工具函数
+ * 提供 AI 触发相关的工具函数，可在插件中直接使用
+ * 
+ * 触发方式：
+ * 1. @机器人 - 在群聊中 @ 机器人触发
+ * 2. 前缀触发 - 使用配置的前缀（如 # 或 AI:）
+ * 3. 私聊直接对话 - 私聊时直接对话
+ * 4. 关键词触发 - 包含特定关键词时触发
+ */
+
+import { Message } from "../message.js";
+import type { 
+  MessageElement,
+  ToolScope,
+  ToolPermissionLevel,
+  MaybePromise, 
+} from "../types.js";
+import { segment } from "../utils.js";
+
+// ============================================================================
+// 类型定义
+// ============================================================================
+
+/**
+ * AI 触发配置
+ */
+export interface AITriggerConfig {
+  /** 是否启用（默认 true） */
+  enabled?: boolean;
+  
+  /** 触发前缀列表（默认 ['#', 'AI:']） */
+  prefixes?: string[];
+  
+  /** 是否响应 @ 机器人（默认 true） */
+  respondToAt?: boolean;
+  
+  /** 是否响应私聊（默认 true） */
+  respondToPrivate?: boolean;
+  
+  /** 触发关键词（可选） */
+  keywords?: string[];
+  
+  /** 忽略的前缀（命令前缀，避免与命令冲突） */
+  ignorePrefixes?: string[];
+  
+  /** 超时时间（毫秒，默认 60000） */
+  timeout?: number;
+  
+  /** 思考中提示（可选） */
+  thinkingMessage?: string;
+  
+  /** 错误消息模板 */
+  errorTemplate?: string;
+  
+  /** Zhin 拥有者 ID 列表 */
+  owners?: string[];
+  
+  /** 机器人管理员 ID 列表 */
+  botAdmins?: string[];
+}
+
+/**
+ * AI 触发检查结果
+ */
+export interface AITriggerResult {
+  triggered: boolean;
+  content: string;
+}
+
+/**
+ * 发送者权限信息
+ */
+export interface SenderPermissions {
+  scope: ToolScope;
+  isGroupAdmin: boolean;
+  isGroupOwner: boolean;
+  isBotAdmin: boolean;
+  isOwner: boolean;
+  permissionLevel: ToolPermissionLevel;
+}
+
+// ============================================================================
+// 默认配置
+// ============================================================================
+
+export const DEFAULT_AI_TRIGGER_CONFIG: Required<AITriggerConfig> = {
+  enabled: true,
+  prefixes: ['#', 'AI:', 'ai:'],
+  respondToAt: true,
+  respondToPrivate: true,
+  keywords: [],
+  ignorePrefixes: ['/', '!', '！'],
+  timeout: 60000,
+  thinkingMessage: '',
+  errorTemplate: '❌ AI 处理失败: {error}',
+  owners: [],
+  botAdmins: [],
+};
+
+// ============================================================================
+// 工具函数
+// ============================================================================
+
+/**
+ * 检查消息是否 @ 了机器人
+ */
+export function isAtBot<T extends object>(message: Message<T>): boolean {
+  const botId = String(message.$bot);
+  return message.$content.some(seg =>  {
+    return String(seg.data?.qq) === botId || String(seg.data?.user_id) === botId;
+  });
+}
+
+/**
+ * 提取消息文本内容（保留富媒体信息）
+ * 使用 segment.toString 将 MessageElement 转为 XML 格式
+ */
+export function extractTextContent<T extends object>(message: Message<T>): string {
+  return segment.toString(message.$content);
+}
+
+/**
+ * 解析 AI 回复中的富媒体内容
+ * 将字符串中的 XML-like 标签解析为 MessageElement
+ */
+export function parseRichMediaContent(content: string): MessageElement[] {
+  try {
+    const parsed = segment.from(content);
+    const elements = Array.isArray(parsed) ? parsed : [parsed];
+    return elements.map(el => {
+      if (typeof el === 'string') {
+        return { type: 'text', data: { text: el } };
+      }
+      return el as MessageElement;
+    });
+  } catch (error) {
+    return [{ type: 'text', data: { text: content } }];
+  }
+}
+
+/**
+ * 移除 @ 机器人的部分
+ */
+export function removeAtBot<T extends object>(message: Message<T>): MessageElement[] {
+  const botId = String(message.$bot);
+  return message.$content.filter(seg => {
+    const { type, data } = seg;
+    if (type !== "at") return true; // 非 at 段全部保留
+    const userId = String(data?.user_id || data?.qq);
+    return userId !== botId; // 只移除 @机器人 的段，保留 @其他人
+  });
+}
+
+/**
+ * 从消息中推断发送者的权限级别
+ */
+export function inferSenderPermissions<T extends object>(
+  message: Message<T>,
+  config: AITriggerConfig
+): SenderPermissions {
+  const scope: ToolScope = message.$channel?.type as ToolScope || 'private';
+  const senderId = message.$sender.id;
+  const senderPermissions = message.$sender.permissions || [];
+  
+  const owners = config.owners || [];
+  const isOwner = owners.includes(senderId);
+  
+  const botAdmins = config.botAdmins || [];
+  const isBotAdmin = isOwner || botAdmins.includes(senderId);
+  
+  const isGroupOwner = senderPermissions.includes('owner') || 
+                       senderPermissions.includes('group_owner') ||
+                       (message as any).$sender?.role === 'owner';
+  
+  const isGroupAdmin = isGroupOwner || 
+                       senderPermissions.includes('admin') || 
+                       senderPermissions.includes('group_admin') ||
+                       (message as any).$sender?.role === 'admin';
+  
+  let permissionLevel: ToolPermissionLevel = 'user';
+  if (isOwner) permissionLevel = 'owner';
+  else if (isBotAdmin) permissionLevel = 'bot_admin';
+  else if (isGroupOwner) permissionLevel = 'group_owner';
+  else if (isGroupAdmin) permissionLevel = 'group_admin';
+  
+  return {
+    scope,
+    isGroupAdmin,
+    isGroupOwner,
+    isBotAdmin,
+    isOwner,
+    permissionLevel,
+  };
+}
+
+/**
+ * 检查消息是否应该触发 AI
+ */
+export function shouldTriggerAI<T extends object>(
+  message: Message<T>, 
+  config: AITriggerConfig
+): AITriggerResult {
+  const fullConfig = { ...DEFAULT_AI_TRIGGER_CONFIG, ...config };
+  
+  if (!fullConfig.enabled) {
+    return { triggered: false, content: '' };
+  }
+  
+  const text = extractTextContent(message);
+  
+  // 检查忽略前缀
+  for (const prefix of fullConfig.ignorePrefixes) {
+    if (text.startsWith(prefix)) {
+      return { triggered: false, content: '' };
+    }
+  }
+  
+  // 1. 检查前缀触发
+  for (const prefix of fullConfig.prefixes) {
+    if (text.startsWith(prefix)) {
+      return { triggered: true, content: text.slice(prefix.length).trim() };
+    }
+  }
+  
+  // 2. 检查 @ 触发
+  if (fullConfig.respondToAt && isAtBot(message)) {
+    const content = removeAtBot(message);
+    if (content.length) {
+      return { triggered: true, content: segment.toString(content) };
+    }
+  }
+  
+  // 3. 检查私聊触发
+  if (fullConfig.respondToPrivate && message.$channel?.type === 'private') {
+    if (text.trim()) {
+      return { triggered: true, content: text.trim() };
+    }
+  }
+  
+  // 4. 检查关键词触发
+  if (fullConfig.keywords.length > 0) {
+    const lowerText = text.toLowerCase();
+    for (const keyword of fullConfig.keywords) {
+      if (lowerText.includes(keyword.toLowerCase())) {
+        return { triggered: true, content: text };
+      }
+    }
+  }
+  
+  return { triggered: false, content: '' };
+}
+
+/**
+ * 合并配置
+ */
+export function mergeAITriggerConfig(config: AITriggerConfig): Required<AITriggerConfig> {
+  return { ...DEFAULT_AI_TRIGGER_CONFIG, ...config };
+}

package/src/built/command.ts

@@ -1,10 +1,11 @@
 /**
- * Command Context
- * 管理所有插件注册的命令
+ * CommandFeature
+ * 管理所有插件注册的命令，继承自 Feature 抽象类
  */
+import { Feature, FeatureJSON } from "../feature.js";
 import { MessageCommand } from "../command.js";
 import { Message } from "../message.js";
-import { Context, Plugin, getPlugin } from "../plugin.js";
+import { Plugin, getPlugin } from "../plugin.js";
 import type { RegisteredAdapter, AdapterMessage } from "../types.js";
 
 /**
@@ -20,83 +21,88 @@ declare module "../plugin.js" {
   namespace Plugin {
     interface Extensions extends CommandContextExtensions {}
     interface Contexts {
-      command: CommandService;
+      command: CommandFeature;
     }
   }
 }
 
 /**
- * 命令服务数据
+ * 命令服务 Feature
  */
-export interface CommandService {
-  /** 命令列表（保持顺序） */
-  readonly items: MessageCommand<RegisteredAdapter>[];
+export class CommandFeature extends Feature<MessageCommand<RegisteredAdapter>> {
+  readonly name = 'command' as const;
+  readonly icon = 'Terminal';
+  readonly desc = '命令';
+
   /** 按 pattern 索引 */
-  readonly byName: Map<string, MessageCommand<RegisteredAdapter>>;
-  /** 添加命令 */
-  add(command: MessageCommand<RegisteredAdapter>, pluginName: string): () => void;
-  /** 移除命令 */
-  remove(command: MessageCommand<RegisteredAdapter>): boolean;
-  /** 按名称获取 */
-  get(pattern: string): MessageCommand<RegisteredAdapter> | undefined;
-  /** 处理消息 */
-  handle(message: Message<AdapterMessage<RegisteredAdapter>>, plugin: Plugin): Promise<any>;
-}
+  readonly byName = new Map<string, MessageCommand<RegisteredAdapter>>();
 
-/**
- * 创建命令 Context
- */
-export function createCommandService(): Context<'command', CommandContextExtensions> {
-  const items: MessageCommand<RegisteredAdapter>[] = [];
-  const byName = new Map<string, MessageCommand<RegisteredAdapter>>();
-  const pluginMap = new Map<MessageCommand<RegisteredAdapter>, string>();
-  
-  const value: CommandService = {
-    items,
-    byName,
-    
-    add(command, pluginName) {
-      items.push(command);
-      byName.set(command.pattern, command);
-      pluginMap.set(command, pluginName);
-      return () => value.remove(command);
-    },
-    
-    remove(command) {
-      const index = items.indexOf(command);
-      if (index !== -1) {
-        items.splice(index, 1);
-        byName.delete(command.pattern);
-        pluginMap.delete(command);
-        return true;
-      }
-      return false;
-    },
-    
-    get(pattern) {
-      return byName.get(pattern);
-    },
-    
-    async handle(message, plugin) {
-      for (const command of items) {
-        const result = await command.handle(message, plugin);
-        if (result) return result;
-      }
-      return null;
+  /**
+   * 添加命令
+   */
+  add(command: MessageCommand<RegisteredAdapter>, pluginName: string): () => void {
+    this.byName.set(command.pattern, command);
+    return super.add(command, pluginName);
+  }
+
+  /**
+   * 移除命令
+   */
+  remove(command: MessageCommand<RegisteredAdapter>): boolean {
+    this.byName.delete(command.pattern);
+    return super.remove(command);
+  }
+
+  /**
+   * 按 pattern 获取命令
+   */
+  get(pattern: string): MessageCommand<RegisteredAdapter> | undefined {
+    return this.byName.get(pattern);
+  }
+
+  /**
+   * 处理消息，依次尝试匹配命令
+   */
+  async handle(message: Message<AdapterMessage<RegisteredAdapter>>, plugin: Plugin): Promise<any> {
+    for (const command of this.items) {
+      const result = await command.handle(message, plugin);
+      if (result) return result;
     }
-  };
-  
-  return {
-    name: 'command',
-    description: '命令服务',
-    value,
-    extensions: {
+    return null;
+  }
+
+  /**
+   * 序列化为 JSON
+   */
+  toJSON(pluginName?: string): FeatureJSON {
+    const list = pluginName ? this.getByPlugin(pluginName) : this.items;
+    return {
+      name: this.name,
+      icon: this.icon,
+      desc: this.desc,
+      count: list.length,
+      items: list.map(c => ({
+        name: c.pattern,
+        desc: c.helpInfo?.desc,
+        usage: c.helpInfo?.usage,
+        examples: c.helpInfo?.examples,
+      })),
+    };
+  }
+
+  /**
+   * 提供给 Plugin.prototype 的扩展方法
+   */
+  get extensions() {
+    const feature = this;
+    return {
       addCommand<T extends RegisteredAdapter>(command: MessageCommand<T>) {
         const plugin = getPlugin();
-        const dispose = value.add(command as MessageCommand<RegisteredAdapter>, plugin.name);
+        const dispose = feature.add(command as MessageCommand<RegisteredAdapter>, plugin.name);
+        plugin.recordFeatureContribution(feature.name, command.pattern);
         plugin.onDispose(dispose);
         return dispose;
-      }
-    }
-  };
+      },
+    };
+  }
 }