dm-bot 1.0.0

package/.env.example

@@ -0,0 +1,3 @@
+DISCORD_TOKEN=你的机器人令牌
+# 在这里粘贴你的 Discord 机器人令牌
+# 不要将 .env 文件提交到版本控制

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,139 @@
+# Discord 交互式机器人
+
+[![Node.js Version](https://img.shields.io/badge/node-%3E%3D18.0.0-brightgreen.svg)](https://nodejs.org/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![Discord.js](https://img.shields.io/badge/discord.js-v14.14.1-blue.svg)](https://discord.js.org/)
+
+一个基于 Discord.js 的交互式机器人，集成 OpenCode 功能，支持私聊对话和消息队列管理。
+
+## 功能特性
+
+- 私聊消息处理
+- 消息队列管理（防止并发处理）
+- 用户 Session 持久化
+- OpenCode 集成
+- SQLite 数据存储
+- 支持以下命令：
+  - `!help` - 显示帮助信息
+  - `!reset` - 清空消息队列
+  - `!status` - 查看机器人状态
+
+## 环境要求
+
+- Node.js >= 18.0.0
+- npm 或 yarn
+- OpenCode CLI（用于 AI 功能）
+
+## 快速开始
+
+### 1. 克隆仓库
+
+```bash
+git clone https://github.com/958877748/skills.git
+cd discord-bot
+```
+
+### 2. 安装依赖
+
+```bash
+npm install
+```
+
+### 3. 配置环境变量
+
+```bash
+cp .env.example .env
+# 编辑 .env 文件，填入你的 Discord 机器人令牌
+```
+
+### 4. 获取 Discord 机器人令牌
+
+1. 访问 [Discord Developer Portal](https://discord.com/developers/applications)
+2. 点击 "New Application" 创建新应用
+3. 进入 "Bot" 标签页，点击 "Add Bot"
+4. 在 "Privileged Gateway Intents" 中启用:
+   - MESSAGE CONTENT INTENT
+5. 复制 Bot Token 到 `.env` 文件
+
+### 5. 邀请机器人到服务器
+
+1. 在 Developer Portal 中，进入 "OAuth2" -> "URL Generator"
+2. 在 SCOPES 中选择 `bot`
+3. 在 BOT PERMISSIONS 中选择:
+   - Send Messages
+   - Read Message History
+   - Read Messages/View Channels
+4. 复制生成的 URL 并在浏览器中打开，选择要添加的服务器
+
+### 6. 启动机器人
+
+```bash
+# 生产环境
+npm start
+
+# 开发环境（带自动重启）
+npm run dev
+```
+
+或者使用脚本:
+
+```bash
+# Linux/Mac
+chmod +x start.sh
+./start.sh
+
+# Windows
+start.bat
+```
+
+## 使用方法
+
+1. 向机器人发送私信开始对话
+2. 使用以下命令:
+   - `!help` - 显示帮助信息
+   - `!reset` - 清空当前队列
+   - `!status` - 查看机器人运行状态
+
+## 项目结构
+
+```
+discord-bot/
+├── index.js          # 主入口文件
+├── db.js             # 数据库操作模块
+├── .env.example      # 环境变量示例
+├── .gitignore        # Git 忽略配置
+├── package.json      # 项目配置
+├── start.sh          # Linux/Mac 启动脚本
+├── start.bat         # Windows 启动脚本
+└── README.md         # 项目说明
+```
+
+## 数据库
+
+使用 SQLite 存储数据:
+- `message_queue` - 消息队列表
+- `user_sessions` - 用户 Session 表
+
+数据库文件会自动创建，无需手动初始化。
+
+## 注意事项
+
+- 机器人只处理私聊消息，服务器中的消息将被忽略
+- 确保 OpenCode CLI 已安装并可用
+- 不要将 `.env` 文件提交到版本控制
+- 数据库文件 (`*.db`) 不会被 Git 追踪
+
+## 故障排除
+
+### 机器人不响应消息
+- 检查 Bot Token 是否正确
+- 确认 MESSAGE CONTENT INTENT 已启用
+- 查看控制台是否有错误信息
+
+### OpenCode 执行失败
+- 确保 OpenCode CLI 已安装: `opencode --version`
+- 检查 opencode 是否在系统 PATH 中
+
+## 许可证
+
+[MIT](LICENSE)

package/bin/cli.js

@@ -0,0 +1,120 @@
+#!/usr/bin/env node
+
+const { Command } = require('commander');
+const path = require('path');
+const fs = require('fs');
+const os = require('os');
+const { startBot } = require('../index');
+const { resetDatabase } = require('../db');
+
+// 标准配置目录: ~/.config/discord-bot/
+const configDir = path.join(os.homedir(), '.config', 'discord-bot');
+const configPath = path.join(configDir, 'config.json');
+
+// 确保配置目录存在
+function ensureConfigDir() {
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+  }
+}
+
+// 读取配置
+function readConfig() {
+  if (fs.existsSync(configPath)) {
+    return JSON.parse(fs.readFileSync(configPath, 'utf8'));
+  }
+  return {};
+}
+
+// 保存配置
+function saveConfig(config) {
+  ensureConfigDir();
+  fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
+}
+
+const program = new Command();
+
+program
+  .name('dm-bot')
+  .description('Discord DM Bot CLI with OpenCode integration')
+  .version('1.0.0');
+
+program
+  .command('start')
+  .description('启动 Discord 机器人')
+  .option('-t, --token <token>', 'Discord Bot Token')
+  .action(async (options) => {
+    let token = options.token;
+    
+    // 优先级: 1. 命令行参数 2. 环境变量 3. 配置文件
+    if (!token) {
+      token = process.env.DISCORD_TOKEN;
+    }
+    if (!token) {
+      const config = readConfig();
+      token = config.token;
+    }
+    
+    if (!token) {
+      ensureConfigDir();
+      
+      // 生成配置文件模板
+      const configTemplate = {
+        token: "YOUR_DISCORD_BOT_TOKEN_HERE",
+        description: "请将上面的 token 替换为你的 Discord Bot Token"
+      };
+      saveConfig(configTemplate);
+      
+      console.error('错误: 未设置 Discord Token');
+      console.log('');
+      console.log('已为你生成配置文件模板:');
+      console.log(`  ${configPath}`);
+      console.log('');
+      console.log('请按以下步骤操作:');
+      console.log('  1. 打开上述文件');
+      console.log('  2. 将 YOUR_DISCORD_BOT_TOKEN_HERE 替换为你的 Discord Bot Token');
+      console.log('  3. 再次运行 dm-bot start');
+      console.log('');
+      console.log('或者你可以使用以下方式直接设置:');
+      console.log('  dm-bot start --token <your-token>');
+      process.exit(1);
+    }
+    
+    process.env.DISCORD_TOKEN = token;
+    console.log('正在启动 Discord Bot...');
+    await startBot();
+  });
+
+program
+  .command('config')
+  .description('配置 Discord Bot')
+  .option('-t, --token <token>', '设置 Discord Bot Token', '')
+  .action((options) => {
+    if (options.token) {
+      const config = readConfig();
+      config.token = options.token;
+      saveConfig(config);
+      console.log(`Token 已保存到 ${configPath}`);
+    } else {
+      const config = readConfig();
+      if (config.token) {
+        console.log('当前配置:');
+        console.log(`配置文件位置: ${configPath}`);
+        console.log(`Token: ${config.token.substring(0, 10)}...`);
+      } else {
+        console.log('未找到配置');
+        console.log('使用: dm-bot config --token <your-token>');
+      }
+    }
+  });
+
+program
+  .command('reset')
+  .description('重置数据库（清空消息队列和会话）')
+  .action(() => {
+    console.log('正在重置数据库...');
+    resetDatabase();
+    console.log('数据库已重置');
+  });
+
+program.parse();

package/db.js

@@ -0,0 +1,93 @@
+const Database = require('better-sqlite3');
+const path = require('path');
+
+const dbPath = path.join(process.cwd(), '.discord-bot.db');
+const db = new Database(dbPath);
+
+// 启用 WAL 模式避免锁定问题
+db.pragma('journal_mode = WAL');
+
+// 创建表
+db.exec(`
+  CREATE TABLE IF NOT EXISTS message_queue (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_id TEXT NOT NULL,
+    channel_id TEXT NOT NULL,
+    content TEXT NOT NULL,
+    status TEXT DEFAULT 'pending',
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    processed_at DATETIME
+  )
+`);
+
+function addMessage(userId, channelId, content) {
+  const stmt = db.prepare('INSERT INTO message_queue (user_id, channel_id, content) VALUES (?, ?, ?)');
+  const result = stmt.run(userId, channelId, content);
+  return result.lastInsertRowid;
+}
+
+function getPendingMessage() {
+  const stmt = db.prepare("SELECT * FROM message_queue WHERE status = 'pending' ORDER BY id ASC LIMIT 1");
+  return stmt.get();
+}
+
+function hasProcessingMessage() {
+  const stmt = db.prepare("SELECT COUNT(*) as count FROM message_queue WHERE status = 'processing'");
+  const result = stmt.get();
+  return result.count > 0;
+}
+
+function markAsProcessing(id) {
+  const stmt = db.prepare("UPDATE message_queue SET status = 'processing' WHERE id = ?");
+  stmt.run(id);
+}
+
+function markAsCompleted(id) {
+  const stmt = db.prepare("UPDATE message_queue SET status = 'completed', processed_at = CURRENT_TIMESTAMP WHERE id = ?");
+  stmt.run(id);
+}
+
+function clearPendingMessages() {
+  db.prepare("DELETE FROM message_queue WHERE status IN ('pending', 'processing')").run();
+}
+
+// 用户 session 管理
+db.exec(`
+  CREATE TABLE IF NOT EXISTS user_sessions (
+    user_id TEXT PRIMARY KEY,
+    session_id TEXT NOT NULL,
+    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+  )
+`);
+
+function getUserSession(userId) {
+  const stmt = db.prepare("SELECT session_id FROM user_sessions WHERE user_id = ?");
+  const result = stmt.get(userId);
+  return result ? result.session_id : null;
+}
+
+function setUserSession(userId, sessionId) {
+  const stmt = db.prepare(`
+    INSERT INTO user_sessions (user_id, session_id) VALUES (?, ?)
+    ON CONFLICT(user_id) DO UPDATE SET session_id = ?, updated_at = CURRENT_TIMESTAMP
+  `);
+  stmt.run(userId, sessionId, sessionId);
+}
+
+function resetDatabase() {
+  db.prepare("DELETE FROM message_queue").run();
+  db.prepare("DELETE FROM user_sessions").run();
+  db.prepare("DELETE FROM sqlite_sequence WHERE name='message_queue'").run();
+}
+
+module.exports = {
+  addMessage,
+  getPendingMessage,
+  hasProcessingMessage,
+  markAsProcessing,
+  markAsCompleted,
+  clearPendingMessages,
+  getUserSession,
+  setUserSession,
+  resetDatabase
+};

package/index.js

@@ -0,0 +1,196 @@
+require('dotenv').config();
+const { Client, GatewayIntentBits, Partials } = require('discord.js');
+const { spawn } = require('child_process');
+const db = require('./db');
+
+const client = new Client({
+  intents: [
+    GatewayIntentBits.Guilds,
+    GatewayIntentBits.GuildMessages,
+    GatewayIntentBits.MessageContent,
+    GatewayIntentBits.DirectMessages
+  ],
+  partials: [Partials.Channel]
+});
+
+let isProcessing = false;
+
+client.once('clientReady', async () => {
+  console.log(`机器人已上线！登录身份：${client.user.tag}`);
+  
+  // 清空未完成的消息
+  db.clearPendingMessages();
+  isProcessing = false;
+  
+  // 启动消息队列轮询
+  startPolling();
+});
+
+client.on('messageCreate', async message => {
+  // 忽略机器人消息
+  if (message.author.bot) return;
+
+  // 只处理私聊消息
+  if (message.guild) return;
+
+  // 处理命令
+  if (message.content.startsWith('!')) {
+    const command = message.content.slice(1).toLowerCase();
+    
+    if (command === 'reset') {
+      db.clearPendingMessages();
+      await message.reply('队列已清空。');
+    } else if (command === 'status') {
+      const processing = db.hasProcessingMessage();
+      await message.reply(`状态: ${isProcessing ? '处理中' : '空闲'}\n队列中有处理中消息: ${processing ? '是' : '否'}`);
+    } else if (command === 'help') {
+      await message.reply('可用命令：\n!reset - 清空队列\n!status - 查看状态\n!help - 显示帮助\n\n直接发送消息，我会帮你处理。');
+    } else {
+      await message.reply('未知命令。输入 !help 查看可用命令。');
+    }
+    return;
+  }
+  
+  // 将消息存入数据库队列
+  try {
+    const msgId = db.addMessage(message.author.id, message.channel.id, message.content);
+    console.log(`消息已存入队列 ID: ${msgId}, 来自用户: ${message.author.tag}`);
+  } catch (error) {
+    console.error('存入消息失败:', error);
+    await message.reply('抱歉，处理消息时出错，请稍后重试。');
+  }
+});
+
+// 轮询处理消息队列
+function startPolling() {
+  setInterval(async () => {
+    if (isProcessing || db.hasProcessingMessage()) return;
+    
+    const message = db.getPendingMessage();
+    if (message) await processMessage(message);
+  }, 2000);
+}
+
+// 处理单条消息
+async function processMessage(message) {
+  isProcessing = true;
+  console.log(`[开始处理] 消息 ID: ${message.id}, 用户: ${message.user_id}`);
+  
+  try {
+    db.markAsProcessing(message.id);
+    
+    const channel = await client.channels.fetch(message.channel_id);
+    if (!channel) throw new Error('无法找到频道');
+
+    // 获取用户 session
+    const userSessionId = db.getUserSession(message.user_id);
+    if (userSessionId) {
+      console.log(`[使用 Session] ${userSessionId}`);
+    }
+
+    // 执行 opencode run
+    console.log(`[执行 opencode] 消息 ID: ${message.id}`);
+    const result = await runOpencode(message.content, userSessionId);
+    console.log(`[执行完成] 结果长度: ${result.text.length}`);
+    
+    // 保存 session
+    if (result.sessionId) {
+      db.setUserSession(message.user_id, result.sessionId);
+      console.log(`[保存 Session] ${result.sessionId}`);
+    }
+    
+    await channel.send(result.text);
+    db.markAsCompleted(message.id);
+    console.log(`[处理完成] 消息 ID: ${message.id}`);
+    
+  } catch (error) {
+    console.error(`[处理失败] 消息 ID: ${message.id}:`, error);
+    try {
+      const channel = await client.channels.fetch(message.channel_id);
+      if (channel) await channel.send(`处理消息时出错: ${error.message}`);
+    } catch (e) {}
+    db.markAsCompleted(message.id);
+  } finally {
+    isProcessing = false;
+  }
+}
+
+// 调用 opencode run
+function runOpencode(prompt, sessionId = null) {
+  return new Promise((resolve, reject) => {
+    const timeout = 120000;
+    let finished = false;
+    
+    const args = ['run', '--format', 'json', prompt];
+    if (sessionId) args.push('--session', sessionId);
+    
+    const childProcess = spawn('opencode', args, {
+      cwd: process.cwd(),
+      env: process.env,
+      stdio: ['ignore', 'pipe', 'pipe']
+    });
+
+    let stdout = '';
+    let stderr = '';
+    let newSessionId = sessionId;
+
+    childProcess.stdout.on('data', (data) => { stdout += data.toString(); });
+    childProcess.stderr.on('data', (data) => { stderr += data.toString(); });
+
+    childProcess.on('close', (code) => {
+      if (finished) return;
+      finished = true;
+      
+      if (code === 0) {
+        let text = '';
+        for (const line of stdout.trim().split('\n')) {
+          try {
+            const json = JSON.parse(line);
+            if (!newSessionId && json.sessionID) newSessionId = json.sessionID;
+            if (json.type === 'text' && json.part?.text) text += json.part.text;
+          } catch (e) {}
+        }
+        resolve({ text: text || '处理完成', sessionId: newSessionId });
+      } else {
+        reject(new Error(stderr.trim() || `进程退出码: ${code}`));
+      }
+    });
+
+    childProcess.on('error', (error) => {
+      if (!finished) {
+        finished = true;
+        reject(new Error(`执行opencode失败: ${error.message}`));
+      }
+    });
+
+    setTimeout(() => {
+      if (!finished) {
+        finished = true;
+        childProcess.kill('SIGTERM');
+        reject(new Error('处理超时'));
+      }
+    }, timeout);
+  });
+}
+
+client.on('error', error => console.error('机器人发生错误：', error));
+
+// 启动机器人
+function startBot() {
+  return client.login(process.env.DISCORD_TOKEN)
+    .then(() => {
+      console.log('正在登录 Discord...');
+      return client;
+    })
+    .catch(error => {
+      console.error('登录失败：', error);
+      throw error;
+    });
+}
+
+module.exports = { startBot };
+
+// 如果直接运行此文件，则启动机器人
+if (require.main === module) {
+  startBot();
+}

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "dm-bot",
+  "version": "1.0.0",
+  "description": "Interactive Discord bot with OpenCode integration",
+  "main": "index.js",
+  "bin": {
+    "dm-bot": "./bin/cli.js"
+  },
+  "scripts": {
+    "start": "node index.js",
+    "dev": "nodemon index.js",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/958877748/skills.git"
+  },
+  "keywords": [
+    "discord",
+    "bot",
+    "interactive",
+    "discord-bot",
+    "opencode",
+    "chatbot"
+  ],
+  "author": "",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "better-sqlite3": "^12.6.2",
+    "commander": "^14.0.3",
+    "discord.js": "^14.14.1",
+    "dotenv": "^16.3.1"
+  },
+  "devDependencies": {
+    "nodemon": "^3.0.2"
+  }
+}