koishi-plugin-docker-control 0.0.1

package/lib/commands/list.js

@@ -0,0 +1,269 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerListCommand = registerListCommand;
+/**
+ * 列出容器指令
+ * docker.ls - 支持集群视图和图片渲染
+ */
+const koishi_1 = require("koishi");
+const logger_1 = require("../utils/logger");
+function registerListCommand(ctx, getService, config) {
+    // 检查是否启用了图片输出
+    const useImageOutput = config?.imageOutput === true;
+    ctx
+        .command('docker.ls [selector]', '列出容器')
+        .alias('docker列表', '容器列表', 'dockercs', '容器查看', 'docker查看')
+        .option('all', '-a 列出所有容器，包括已停止', { fallback: false })
+        .option('format', '-f <format> 输出格式: simple|detail|json|image', {
+        fallback: null, // 由 config.imageOutput 决定
+    })
+        .action(async ({ options }, selector) => {
+        logger_1.commandLogger.debug(`docker.ls 被调用: selector=${selector}, all=${options.all}, format=${options.format}`);
+        const service = getService();
+        if (!service) {
+            logger_1.commandLogger.debug('服务未初始化');
+            return 'Docker 服务未初始化';
+        }
+        const all = options.all ?? false;
+        // 如果未指定 format，使用配置的 imageOutput 设置
+        const format = options.format || (useImageOutput ? 'image' : 'simple');
+        logger_1.commandLogger.debug(`列表参数: all=${all}, format=${format}`);
+        // 图片渲染模式
+        if (format === 'image') {
+            logger_1.commandLogger.debug('使用图片渲染模式');
+            if (!ctx.puppeteer) {
+                return '错误: 未安装 koishi-plugin-puppeteer 插件，无法使用图片渲染';
+            }
+            try {
+                // 获取容器数据
+                logger_1.commandLogger.debug('获取容器数据...');
+                const results = await getContainerResults(service, selector, all);
+                logger_1.commandLogger.debug(`获取到 ${results.length} 个节点`);
+                if (results.length === 0) {
+                    return '未发现任何容器';
+                }
+                // 生成 HTML
+                const html = generateHtml(results);
+                // 渲染图片 (puppeteer.render 返回的是 h.image() 元素的字符串)
+                logger_1.commandLogger.debug('渲染图片中...');
+                const imageElement = await ctx.puppeteer.render(html, async (page, next) => {
+                    await page.setViewport({ width: 600, height: 800 });
+                    const body = await page.$('body');
+                    const clip = await body.boundingBox();
+                    const buffer = await page.screenshot({ clip });
+                    return koishi_1.h.image(buffer, 'image/png').toString();
+                });
+                return imageElement;
+            }
+            catch (e) {
+                logger_1.commandLogger.error(`图片渲染失败: ${e.message}`);
+                return `错误: ${e.message}`;
+            }
+        }
+        // 文字模式
+        try {
+            const results = await getContainerResults(service, selector, all);
+            if (results.length === 0) {
+                return selector ? '所有指定节点均未连接' : '未发现任何容器';
+            }
+            const lines = [];
+            for (const { node, containers } of results) {
+                lines.push(`=== ${node.name} ===`);
+                if (containers.length === 0) {
+                    lines.push('  (无容器)');
+                }
+                else {
+                    for (const c of containers) {
+                        lines.push(formatContainerLine(c, format));
+                    }
+                }
+                lines.push('');
+            }
+            return lines.join('\n');
+        }
+        catch (e) {
+            logger_1.commandLogger.error(`列出容器失败: ${e.message}`);
+            return `错误: ${e.message}`;
+        }
+    });
+}
+/**
+ * 获取容器数据
+ */
+async function getContainerResults(service, selector, all) {
+    const results = [];
+    if (selector) {
+        const nodes = service.getNodesBySelector(selector);
+        for (const node of nodes) {
+            if (node.status !== 'connected')
+                continue;
+            const containers = await node.listContainers(all);
+            results.push({ node, containers });
+        }
+    }
+    else {
+        const aggregated = await service.getAggregatedContainers(all);
+        for (const { node, containers } of aggregated) {
+            if (node.status !== 'connected')
+                continue;
+            results.push({ node, containers: containers || [] });
+        }
+    }
+    return results;
+}
+/**
+ * 格式化输出单行容器信息
+ */
+function formatContainerLine(container, format) {
+    const status = container.State;
+    const emoji = status === 'running' ? '🟢' : (status === 'stopped' ? '🔴' : '⚪');
+    const name = container.Names[0]?.replace('/', '') || 'Unknown';
+    const shortId = container.Id.slice(0, 8);
+    let image = container.Image;
+    const parts = image.split('/');
+    if (parts.length > 1) {
+        image = parts[parts.length - 1];
+    }
+    if (format === 'detail') {
+        return `${emoji} **${name}**\n    ID: ${shortId}\n    Image: ${container.Image}\n    State: ${container.Status}`;
+    }
+    // simple 模式：双行显示
+    return `${emoji} ${name}\n    └ ${shortId} | ${image}`;
+}
+/**
+ * 生成 HTML 模板
+ */
+function generateHtml(results) {
+    const styles = `
+    * { margin: 0; padding: 0; box-sizing: border-box; }
+    body {
+      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+      background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
+      min-height: 100vh;
+      padding: 20px;
+      color: #fff;
+    }
+    .container {
+      max-width: 700px;
+      margin: 0 auto;
+    }
+    .node-section {
+      background: rgba(255, 255, 255, 0.1);
+      border-radius: 12px;
+      margin-bottom: 20px;
+      overflow: hidden;
+    }
+    .node-header {
+      background: rgba(79, 172, 254, 0.3);
+      padding: 12px 16px;
+      font-size: 16px;
+      font-weight: 600;
+      border-bottom: 1px solid rgba(255, 255, 255, 0.1);
+    }
+    .table-header {
+      display: grid;
+      grid-template-columns: 40px 1fr 100px 1fr;
+      gap: 10px;
+      padding: 10px 16px;
+      background: rgba(0, 0, 0, 0.2);
+      font-size: 12px;
+      color: rgba(255, 255, 255, 0.6);
+      text-transform: uppercase;
+      letter-spacing: 0.5px;
+    }
+    .row {
+      display: grid;
+      grid-template-columns: 40px 1fr 100px 1fr;
+      gap: 10px;
+      padding: 10px 16px;
+      border-bottom: 1px solid rgba(255, 255, 255, 0.05);
+      align-items: center;
+      transition: background 0.2s;
+    }
+    .row:hover {
+      background: rgba(255, 255, 255, 0.05);
+    }
+    .row:last-child {
+      border-bottom: none;
+    }
+    .status {
+      font-size: 18px;
+      text-align: center;
+    }
+    .name {
+      font-weight: 500;
+      white-space: nowrap;
+      overflow: hidden;
+      text-overflow: ellipsis;
+    }
+    .id {
+      font-family: 'SF Mono', Monaco, monospace;
+      font-size: 12px;
+      color: rgba(255, 255, 255, 0.7);
+    }
+    .image {
+      font-size: 12px;
+      color: rgba(255, 255, 255, 0.7);
+      white-space: nowrap;
+      overflow: hidden;
+      text-overflow: ellipsis;
+    }
+    .running { color: #4ade80; }
+    .stopped { color: #f87171; }
+    .other { color: #94a3b8; }
+    .stats {
+      display: flex;
+      justify-content: center;
+      gap: 20px;
+      padding: 16px;
+      color: rgba(255, 255, 255, 0.6);
+      font-size: 13px;
+    }
+  `;
+    let html = `<!DOCTYPE html><html><head><meta charset="UTF-8"><style>${styles}</style></head><body>`;
+    html += `<div class="container">`;
+    let totalRunning = 0;
+    let totalStopped = 0;
+    for (const { node, containers } of results) {
+        const running = containers.filter(c => c.State === 'running').length;
+        const stopped = containers.length - running;
+        totalRunning += running;
+        totalStopped += stopped;
+        html += `<div class="node-section">`;
+        html += `<div class="node-header">${node.name}</div>`;
+        // 表头
+        html += `<div class="table-header">
+      <span></span>
+      <span>容器</span>
+      <span>ID</span>
+      <span>镜像</span>
+    </div>`;
+        // 容器列表
+        for (const c of containers) {
+            const status = c.State;
+            const emoji = status === 'running' ? '🟢' : (status === 'stopped' ? '🔴' : '⚪');
+            const name = c.Names[0]?.replace('/', '') || 'Unknown';
+            const shortId = c.Id.slice(0, 8);
+            let image = c.Image;
+            const parts = image.split('/');
+            if (parts.length > 1) {
+                image = parts[parts.length - 1];
+            }
+            html += `<div class="row">
+        <span class="status">${emoji}</span>
+        <span class="name" title="${name}">${name}</span>
+        <span class="id">${shortId}</span>
+        <span class="image" title="${image}">${image}</span>
+      </div>`;
+        }
+        // 统计
+        html += `<div class="stats">运行中: ${running} | 已停止: ${stopped}</div>`;
+        html += `</div>`;
+    }
+    // 总体统计
+    html += `<div class="node-section">`;
+    html += `<div class="stats"><strong>总计:</strong> ${totalRunning} 运行中, ${totalStopped} 已停止</div>`;
+    html += `</div>`;
+    html += `</div></body></html>`;
+    return html;
+}

package/lib/commands/logs.d.ts

@@ -0,0 +1,10 @@
+/**
+ * 日志指令
+ * docker.logs <container> [node]
+ */
+import { Context } from 'koishi';
+import type { DockerControlConfig } from '../types';
+/**
+ * 注册日志指令
+ */
+export declare function registerLogsCommand(ctx: Context, getService: () => any, config?: DockerControlConfig): void;

package/lib/commands/logs.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerLogsCommand = registerLogsCommand;
+const logger_1 = require("../utils/logger");
+/**
+ * 获取容器名称
+ */
+function getContainerName(c) {
+    return c.Names[0]?.replace('/', '') || c.Id.slice(0, 8);
+}
+/**
+ * 注册日志指令
+ */
+function registerLogsCommand(ctx, getService, config) {
+    ctx
+        .command('docker.logs <container> [node]', '查看容器日志')
+        .alias('docker日志', '容器日志', 'docker查看日志', '容器查看日志', 'dockerlogs')
+        .option('lines', '-n <lines:number> 显示最后 N 行')
+        .option('timestamp', '-t 显示时间戳')
+        .action(async ({ options }, container, node) => {
+        logger_1.commandLogger.debug(`docker.logs 被调用: container=${container}, node=${node}, lines=${options.lines}, timestamp=${options.timestamp}`);
+        const service = getService();
+        if (!service) {
+            logger_1.commandLogger.debug('服务未初始化');
+            return 'Docker 服务未初始化';
+        }
+        // 参数校验
+        if (!container) {
+            return '请指定容器名或ID\n用法示例:\n  docker.logs my-app\n  docker.logs my-app node-1 -n 50';
+        }
+        // 确定日志行数 (优先级: 命令行参数 > 全局配置 > 默认值)
+        const tail = options.lines || config?.defaultLogLines || 100;
+        const showTimestamp = options.timestamp || false;
+        logger_1.commandLogger.debug(`日志参数: tail=${tail}, showTimestamp=${showTimestamp}`);
+        try {
+            let targetNode = null;
+            let containerInfo = null;
+            // 查找容器逻辑
+            if (node) {
+                // 指定节点查找
+                const nodes = service.getNodesBySelector(node);
+                if (nodes.length === 0) {
+                    logger_1.commandLogger.debug(`找不到节点: ${node}`);
+                    return `❌ 找不到节点: ${node}`;
+                }
+                targetNode = nodes[0];
+                logger_1.commandLogger.debug(`在节点 ${targetNode.name} 中查找容器...`);
+                const result = await service.findContainer(targetNode.id, container);
+                containerInfo = result.container;
+            }
+            else {
+                // 全局模糊查找
+                logger_1.commandLogger.debug(`全局搜索容器: ${container}`);
+                const results = await service.findContainerGlobal(container);
+                if (results.length === 0) {
+                    logger_1.commandLogger.debug(`找不到容器: ${container}`);
+                    return `❌ 找不到容器: ${container}`;
+                }
+                // 优先返回 Running 的，如果没有则返回第一个
+                const running = results.find(r => r.container.State === 'running');
+                const target = running || results[0];
+                targetNode = target.node;
+                containerInfo = target.container;
+            }
+            if (!targetNode || !containerInfo) {
+                return '❌ 未能获取容器信息';
+            }
+            // 获取日志
+            const logs = await targetNode.getContainerLogs(containerInfo.Id, tail);
+            if (!logs || !logs.trim()) {
+                return `${targetNode.name}: ${getContainerName(containerInfo)} - 无日志`;
+            }
+            // 格式化输出
+            const lines = logs.split('\n').filter(l => l.length > 0);
+            const displayLines = lines.length > tail ? lines.slice(-tail) : lines;
+            // 移除 ANSI 颜色代码
+            const cleanLogs = displayLines.map(line => line.replace(/[\u001b\u009b][[()#;?]*(?:[0-9]{1,4}(?:;[0-9]{0,4})*)?[0-9A-ORZcf-nqry=><]/g, '')).join('\n');
+            return `=== ${targetNode.name}: ${getContainerName(containerInfo)} (最后 ${displayLines.length} 行) ===\n${cleanLogs}`;
+        }
+        catch (e) {
+            logger_1.commandLogger.error(e);
+            return `❌ 获取日志失败: ${e.message}`;
+        }
+    });
+}

package/lib/config.d.ts

@@ -0,0 +1,55 @@
+/**
+ * 配置定义 - 简化版，只支持 SSH 直连模式
+ */
+import { Schema } from 'koishi';
+import type { DockerControlConfig, CredentialConfig, NodeConfig } from './types';
+export declare const CredentialSchema: Schema<CredentialConfig>;
+export declare const NotificationSchema: Schema<Schemastery.ObjectS<{
+    enabled: Schema<boolean, boolean>;
+    level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+    targetGroups: Schema<string[], string[]>;
+    events: Schema<string[], string[]>;
+}>, Schemastery.ObjectT<{
+    enabled: Schema<boolean, boolean>;
+    level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+    targetGroups: Schema<string[], string[]>;
+    events: Schema<string[], string[]>;
+}>>;
+export declare const NodeSchema: Schema<NodeConfig>;
+export declare const ConfigSchema: Schema<Schemastery.ObjectS<{
+    requestTimeout: Schema<number, number>;
+    debug: Schema<boolean, boolean>;
+    credentials: Schema<CredentialConfig[], CredentialConfig[]>;
+    nodes: Schema<NodeConfig[], NodeConfig[]>;
+    notification: Schema<Schemastery.ObjectS<{
+        enabled: Schema<boolean, boolean>;
+        level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+        targetGroups: Schema<string[], string[]>;
+        events: Schema<string[], string[]>;
+    }>, Schemastery.ObjectT<{
+        enabled: Schema<boolean, boolean>;
+        level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+        targetGroups: Schema<string[], string[]>;
+        events: Schema<string[], string[]>;
+    }>>;
+}>, Schemastery.ObjectT<{
+    requestTimeout: Schema<number, number>;
+    debug: Schema<boolean, boolean>;
+    credentials: Schema<CredentialConfig[], CredentialConfig[]>;
+    nodes: Schema<NodeConfig[], NodeConfig[]>;
+    notification: Schema<Schemastery.ObjectS<{
+        enabled: Schema<boolean, boolean>;
+        level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+        targetGroups: Schema<string[], string[]>;
+        events: Schema<string[], string[]>;
+    }>, Schemastery.ObjectT<{
+        enabled: Schema<boolean, boolean>;
+        level: Schema<"error" | "all" | "none", "error" | "all" | "none">;
+        targetGroups: Schema<string[], string[]>;
+        events: Schema<string[], string[]>;
+    }>>;
+}>>;
+export declare function getCredentialById(config: DockerControlConfig, id: string): CredentialConfig | undefined;
+export declare function getNodeById(config: DockerControlConfig, id: string): NodeConfig | undefined;
+export declare function getNodesByTag(config: DockerControlConfig, tag: string): NodeConfig[];
+export declare function validateConfig(config: DockerControlConfig): string[];

package/lib/config.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConfigSchema = exports.NodeSchema = exports.NotificationSchema = exports.CredentialSchema = void 0;
+exports.getCredentialById = getCredentialById;
+exports.getNodeById = getNodeById;
+exports.getNodesByTag = getNodesByTag;
+exports.validateConfig = validateConfig;
+/**
+ * 配置定义 - 简化版，只支持 SSH 直连模式
+ */
+const koishi_1 = require("koishi");
+// ==================== 凭证 Schema ====================
+exports.CredentialSchema = koishi_1.Schema.object({
+    id: koishi_1.Schema.string().required().description('凭证 ID (唯一标识)'),
+    name: koishi_1.Schema.string().required().description('凭证名称 (用于显示)'),
+    username: koishi_1.Schema.string().default('root').description('SSH 用户名'),
+    authType: koishi_1.Schema.union(['key', 'password'])
+        .role('radio')
+        .default('key')
+        .description('认证方式'),
+    password: koishi_1.Schema.string().role('secret').description('SSH 密码'),
+    privateKey: koishi_1.Schema.string().role('textarea').description('私钥 (PEM 格式)'),
+    passphrase: koishi_1.Schema.string().role('secret').description('私钥密码'),
+});
+// ==================== 通知 Schema ====================
+const NotificationEventSchema = koishi_1.Schema.array(koishi_1.Schema.string())
+    .default([
+    'container.start',
+    'container.stop',
+    'container.restart',
+    'container.die',
+]);
+exports.NotificationSchema = koishi_1.Schema.object({
+    enabled: koishi_1.Schema.boolean().default(false).description('是否启用通知'),
+    level: koishi_1.Schema.union(['all', 'error', 'none'])
+        .default('all')
+        .description('通知级别'),
+    targetGroups: koishi_1.Schema.array(koishi_1.Schema.string())
+        .default([])
+        .description('通知的群组 ID 列表'),
+    events: NotificationEventSchema
+        .description('通知的事件类型'),
+});
+// ==================== 节点 Schema ====================
+exports.NodeSchema = koishi_1.Schema.object({
+    id: koishi_1.Schema.string().required().description('节点 ID (唯一标识)'),
+    name: koishi_1.Schema.string().required().description('节点名称 (用于显示)'),
+    tags: koishi_1.Schema.array(koishi_1.Schema.string())
+        .default([])
+        .description('标签 (用于集群操作)'),
+    host: koishi_1.Schema.string().required().description('SSH 主机地址'),
+    port: koishi_1.Schema.number().default(22).description('SSH 端口'),
+    credentialId: koishi_1.Schema.string().required().description('SSH 凭证 ID'),
+});
+// ==================== 完整配置 Schema ====================
+exports.ConfigSchema = koishi_1.Schema.object({
+    requestTimeout: koishi_1.Schema.number()
+        .default(30000)
+        .description('全局请求超时 (毫秒)'),
+    debug: koishi_1.Schema.boolean().default(false).description('调试模式'),
+    credentials: koishi_1.Schema.array(exports.CredentialSchema)
+        .default([])
+        .description('SSH 凭证列表'),
+    nodes: koishi_1.Schema.array(exports.NodeSchema)
+        .default([])
+        .description('Docker 节点列表'),
+    notification: exports.NotificationSchema
+        .description('通知配置'),
+});
+// ==================== 辅助函数 ====================
+function getCredentialById(config, id) {
+    return config.credentials?.find((c) => c.id === id);
+}
+function getNodeById(config, id) {
+    return config.nodes?.find((n) => n.id === id);
+}
+function getNodesByTag(config, tag) {
+    return config.nodes?.filter((n) => n.tags.includes(tag)) || [];
+}
+function validateConfig(config) {
+    const errors = [];
+    if (!config.nodes)
+        return errors;
+    for (const node of config.nodes) {
+        if (!getCredentialById(config, node.credentialId)) {
+            errors.push(`节点 ${node.name} 引用的凭证 ${node.credentialId} 不存在`);
+        }
+    }
+    return errors;
+}

package/lib/constants.d.ts

@@ -0,0 +1,26 @@
+/**
+ * 常量定义
+ */
+export declare const DEFAULT_TIMEOUT = 5000;
+export declare const RETRY_INTERVAL = 5000;
+export declare const SSH_TIMEOUT = 5000;
+export declare const MAX_RETRY_COUNT = 3;
+export declare const MONITOR_RETRY_INTERVAL = 30000;
+export declare const EVENTS_POLL_INTERVAL = 60000;
+export declare const CONTAINER_POLL_INTERVAL: number;
+export declare const DEFAULT_LOG_LINES = 100;
+export declare const NodeStatus: {
+    readonly DISCONNECTED: "disconnected";
+    readonly CONNECTING: "connecting";
+    readonly CONNECTED: "connected";
+    readonly ERROR: "error";
+};
+export type NodeStatusType = typeof NodeStatus[keyof typeof NodeStatus];
+export declare const ContainerAction: {
+    readonly START: "start";
+    readonly STOP: "stop";
+    readonly RESTART: "restart";
+    readonly CREATE: "create";
+    readonly DIE: "die";
+};
+export type ContainerActionType = typeof ContainerAction[keyof typeof ContainerAction];

package/lib/constants.js

@@ -0,0 +1,37 @@
+"use strict";
+/**
+ * 常量定义
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContainerAction = exports.NodeStatus = exports.DEFAULT_LOG_LINES = exports.CONTAINER_POLL_INTERVAL = exports.EVENTS_POLL_INTERVAL = exports.MONITOR_RETRY_INTERVAL = exports.MAX_RETRY_COUNT = exports.SSH_TIMEOUT = exports.RETRY_INTERVAL = exports.DEFAULT_TIMEOUT = void 0;
+// 默认请求超时时间 (毫秒)
+exports.DEFAULT_TIMEOUT = 5000;
+// 默认重试间隔 (毫秒)
+exports.RETRY_INTERVAL = 5000;
+// SSH 连接超时 (毫秒)
+exports.SSH_TIMEOUT = 5000;
+// 最大重试次数
+exports.MAX_RETRY_COUNT = 3;
+// 监控重连间隔 (毫秒)
+exports.MONITOR_RETRY_INTERVAL = 30000;
+// Docker Events 监听间隔 (毫秒) - 已改为流式监听，此常量仅作备用
+exports.EVENTS_POLL_INTERVAL = 60000;
+// 容器状态轮询间隔 (毫秒) - 改为5分钟仅作兜底同步
+exports.CONTAINER_POLL_INTERVAL = 5 * 60 * 1000;
+// 日志行数默认
+exports.DEFAULT_LOG_LINES = 100;
+// 节点状态枚举
+exports.NodeStatus = {
+    DISCONNECTED: 'disconnected',
+    CONNECTING: 'connecting',
+    CONNECTED: 'connected',
+    ERROR: 'error',
+};
+// 容器操作类型
+exports.ContainerAction = {
+    START: 'start',
+    STOP: 'stop',
+    RESTART: 'restart',
+    CREATE: 'create',
+    DIE: 'die',
+};