koishi-plugin-docker-control 0.0.7 → 0.1.0

package/lib/commands/audit.d.ts

@@ -0,0 +1,9 @@
+/**
+ * 审计日志指令
+ * 查询和管理审计日志
+ */
+import { Context } from 'koishi';
+/**
+ * 注册审计日志指令
+ */
+export declare function registerAuditCommands(ctx: Context, getService: () => any): void;

package/lib/commands/audit.js

@@ -0,0 +1,136 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerAuditCommands = registerAuditCommands;
+const logger_1 = require("../utils/logger");
+/**
+ * 注册审计日志指令
+ */
+function registerAuditCommands(ctx, getService) {
+    /**
+     * 查询审计日志
+     */
+    ctx
+        .command('docker.audit.log', '查看审计日志')
+        .alias('审计日志', '日志审计')
+        .option('user', '-u <userId> 按用户ID筛选')
+        .option('action', '-a <action> 按操作类型筛选')
+        .option('result', '-r <result> 按结果筛选 (success/failure)')
+        .option('limit', '-l <count> 限制返回条数', { fallback: 20 })
+        .action(async ({ options }) => {
+        const service = getService();
+        if (!service) {
+            return 'Docker 服务未初始化';
+        }
+        if (!service.auditLogger) {
+            return '审计日志功能未启用';
+        }
+        try {
+            const filter = {
+                limit: options.limit || 20
+            };
+            if (options.user)
+                filter.userId = options.user;
+            if (options.action)
+                filter.action = options.action;
+            if (options.result)
+                filter.result = options.result;
+            const logs = await service.auditLogger.query(filter);
+            if (logs.length === 0) {
+                return '没有找到符合条件的审计日志';
+            }
+            const lines = [`=== 审计日志 (最近 ${logs.length} 条) ===`, ''];
+            for (const log of logs) {
+                const status = log.result === 'success' ? '✅' : '❌';
+                const time = new Date(log.timestamp).toLocaleString();
+                const duration = log.duration ? `${log.duration}ms` : '-';
+                lines.push(`${status} ${time}`, `  用户: ${log.userName || log.userId} (${log.platform})`, `  操作: ${log.action}`, `  结果: ${log.result}`, `  耗时: ${duration}`, log.nodeId ? `  节点: ${log.nodeId}` : '', log.containerId ? `  容器: ${log.containerId.slice(0, 12)}` : '', log.errorMessage ? `  错误: ${log.errorMessage}` : '', '');
+            }
+            return lines.filter(Boolean).join('\n');
+        }
+        catch (e) {
+            logger_1.commandLogger.error(`查询审计日志失败: ${e.message}`);
+            return `❌ 查询失败: ${e.message}`;
+        }
+    });
+    /**
+     * 获取审计日志统计
+     */
+    ctx
+        .command('docker.audit.stats', '审计日志统计')
+        .alias('审计统计', '日志统计')
+        .action(async () => {
+        const service = getService();
+        if (!service) {
+            return 'Docker 服务未初始化';
+        }
+        if (!service.auditLogger) {
+            return '审计日志功能未启用';
+        }
+        try {
+            const stats = await service.auditLogger.getStats();
+            return [
+                '=== 审计日志统计 ===',
+                `总操作数: ${stats.total}`,
+                `成功: ${stats.success}`,
+                `失败: ${stats.failure}`,
+                `平均耗时: ${stats.avgDuration}ms`
+            ].join('\n');
+        }
+        catch (e) {
+            logger_1.commandLogger.error(`获取审计统计失败: ${e.message}`);
+            return `❌ 获取失败: ${e.message}`;
+        }
+    });
+    /**
+     * 清理旧日志
+     */
+    ctx
+        .command('docker.audit.cleanup', '清理旧审计日志')
+        .alias('清理审计日志', '清理日志')
+        .option('days', '-d <days> 保留天数', { fallback: null })
+        .action(async ({ options }) => {
+        const service = getService();
+        if (!service) {
+            return 'Docker 服务未初始化';
+        }
+        if (!service.auditLogger) {
+            return '审计日志功能未启用';
+        }
+        try {
+            const deletedCount = await service.auditLogger.cleanup(options.days);
+            return `✅ 已清理 ${deletedCount} 条旧日志`;
+        }
+        catch (e) {
+            logger_1.commandLogger.error(`清理审计日志失败: ${e.message}`);
+            return `❌ 清理失败: ${e.message}`;
+        }
+    });
+    /**
+     * 导出审计日志
+     */
+    ctx
+        .command('docker.audit.export', '导出审计日志')
+        .alias('导出审计日志', '导出日志')
+        .action(async () => {
+        const service = getService();
+        if (!service) {
+            return 'Docker 服务未初始化';
+        }
+        if (!service.auditLogger) {
+            return '审计日志功能未启用';
+        }
+        if (!ctx.assets) {
+            return '❌ 需要assets插件支持才能导出文件';
+        }
+        try {
+            const csv = await service.auditLogger.export();
+            const filename = `audit-logs-${Date.now()}.csv`;
+            const url = await ctx.assets.upload(csv, filename);
+            return `✅ 审计日志已导出: ${url}`;
+        }
+        catch (e) {
+            logger_1.commandLogger.error(`导出审计日志失败: ${e.message}`);
+            return `❌ 导出失败: ${e.message}`;
+        }
+    });
+}

package/lib/commands/cluster.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Docker Swarm 集群管理指令
+ */
+import { Context } from 'koishi';
+export declare function registerClusterCommands(ctx: Context, getService: () => any, config?: any): void;

package/lib/commands/cluster.js

@@ -0,0 +1,294 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerClusterCommands = registerClusterCommands;
+const logger_1 = require("../utils/logger");
+const render_1 = require("../utils/render");
+function registerClusterCommands(ctx, getService, config) {
+    const useImageOutput = config?.imageOutput === true;
+    /**
+     * 查看集群信息
+     */
+    ctx.command('docker.cluster [selector]', '查看 Swarm 集群信息')
+        .alias('集群', 'swarm', 'docker集群')
+        .action(async (_, selector) => {
+        logger_1.commandLogger.debug(`docker.cluster 被调用: selector=${selector}`);
+        const service = getService();
+        if (!service)
+            return '❌ 服务未初始化';
+        const nodes = service.getNodesBySelector(selector || '');
+        if (nodes.length === 0)
+            return `❌ 未找到节点: ${selector}`;
+        const lines = [];
+        for (const node of nodes) {
+            if (node.status !== 'connected') {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (未连接)');
+                lines.push('');
+                continue;
+            }
+            // 检查是否在 Swarm 模式
+            const isSwarm = await node.isSwarmMode();
+            if (!isSwarm) {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (不在 Swarm 模式)');
+                lines.push('');
+                continue;
+            }
+            const swarmInfo = await node.getSwarmInfo();
+            if (!swarmInfo) {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (无法获取集群信息)');
+                lines.push('');
+                continue;
+            }
+            lines.push(`=== ${node.name} ===`);
+            lines.push(`  集群 ID: ${swarmInfo.id}`);
+            lines.push(`  集群名称: ${swarmInfo.name}`);
+            lines.push(`  创建时间: ${swarmInfo.createdAt}`);
+            lines.push(`  更新时间: ${swarmInfo.updatedAt}`);
+            lines.push('');
+        }
+        return lines.join('\n').trim();
+    });
+    /**
+     * 查看集群节点列表
+     */
+    ctx.command('docker.cluster.nodes [selector]', '查看 Swarm 集群节点')
+        .alias('集群节点', 'swarm节点', 'swarm节点')
+        .option('format', '-f <format> 输出格式: simple|image', { fallback: null })
+        .action(async ({ options }, selector) => {
+        logger_1.commandLogger.debug(`docker.cluster.nodes 被调用: selector=${selector}, format=${options.format}`);
+        const service = getService();
+        if (!service)
+            return '❌ 服务未初始化';
+        const nodes = service.getNodesBySelector(selector || '');
+        if (nodes.length === 0)
+            return `❌ 未找到节点: ${selector}`;
+        const format = options.format || (useImageOutput ? 'image' : 'simple');
+        // 图片渲染模式
+        if (format === 'image') {
+            if (!ctx.puppeteer)
+                return '❌ 未安装 puppeteer 插件';
+            try {
+                const results = [];
+                for (const node of nodes) {
+                    if (node.status !== 'connected')
+                        continue;
+                    const isSwarm = await node.isSwarmMode();
+                    if (!isSwarm)
+                        continue;
+                    const swarmNodes = await node.getSwarmNodes();
+                    if (swarmNodes.length > 0) {
+                        results.push({ node, swarmNodes });
+                    }
+                }
+                if (results.length === 0)
+                    return '❌ 未找到任何 Swarm 集群节点';
+                const html = (0, render_1.generateSwarmNodesHtml)(results, '集群节点');
+                return await (0, render_1.renderToImage)(ctx, html);
+            }
+            catch (e) {
+                logger_1.commandLogger.error(`获取集群节点失败: ${e.message}`);
+                return `❌ 错误: ${e.message}`;
+            }
+        }
+        // 文字模式
+        const lines = [];
+        for (const node of nodes) {
+            if (node.status !== 'connected') {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (未连接)');
+                lines.push('');
+                continue;
+            }
+            const isSwarm = await node.isSwarmMode();
+            if (!isSwarm) {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (不在 Swarm 模式)');
+                lines.push('');
+                continue;
+            }
+            const swarmNodes = await node.getSwarmNodes();
+            lines.push(`=== ${node.name} (${swarmNodes.length} 个节点) ===`);
+            if (swarmNodes.length === 0) {
+                lines.push('  (无节点)');
+            }
+            else {
+                for (const n of swarmNodes) {
+                    const shortId = n.ID.slice(0, 12);
+                    const isLeader = n.ManagerStatus?.Leader ? ' 👑' : '';
+                    const statusIcon = n.Status.State === 'ready' ? '🟢' : '🔴';
+                    lines.push(`  ${isLeader}${n.Hostname} (${n.Role})`);
+                    lines.push(`    ID: ${shortId}`);
+                    lines.push(`    状态: ${statusIcon} ${n.Status.State} | 可用性: ${n.Availability}`);
+                    lines.push(`    地址: ${n.Status.Addr}`);
+                    if (n.ManagerStatus?.Reachability) {
+                        lines.push(`    管理可达性: ${n.ManagerStatus.Reachability}`);
+                    }
+                }
+            }
+            lines.push('');
+        }
+        return lines.join('\n').trim();
+    });
+    /**
+     * 查看集群服务列表
+     */
+    ctx.command('docker.cluster.services [selector]', '查看 Swarm 集群服务')
+        .alias('集群服务', 'swarm服务', '集群services')
+        .option('format', '-f <format> 输出格式: simple|image', { fallback: null })
+        .action(async ({ options }, selector) => {
+        logger_1.commandLogger.debug(`docker.cluster.services 被调用: selector=${selector}, format=${options.format}`);
+        const service = getService();
+        if (!service)
+            return '❌ 服务未初始化';
+        const nodes = service.getNodesBySelector(selector || '');
+        if (nodes.length === 0)
+            return `❌ 未找到节点: ${selector}`;
+        const format = options.format || (useImageOutput ? 'image' : 'simple');
+        // 图片渲染模式
+        if (format === 'image') {
+            if (!ctx.puppeteer)
+                return '❌ 未安装 puppeteer 插件';
+            try {
+                const results = [];
+                for (const node of nodes) {
+                    if (node.status !== 'connected')
+                        continue;
+                    const isSwarm = await node.isSwarmMode();
+                    if (!isSwarm)
+                        continue;
+                    const services = await node.getSwarmServices();
+                    if (services.length > 0) {
+                        results.push({ node, services });
+                    }
+                }
+                if (results.length === 0)
+                    return '❌ 未找到任何 Swarm 服务';
+                const html = (0, render_1.generateSwarmServicesHtml)(results, '集群服务');
+                return await (0, render_1.renderToImage)(ctx, html);
+            }
+            catch (e) {
+                logger_1.commandLogger.error(`获取集群服务失败: ${e.message}`);
+                return `❌ 错误: ${e.message}`;
+            }
+        }
+        // 文字模式
+        const lines = [];
+        for (const node of nodes) {
+            if (node.status !== 'connected') {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (未连接)');
+                lines.push('');
+                continue;
+            }
+            const isSwarm = await node.isSwarmMode();
+            if (!isSwarm) {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (不在 Swarm 模式)');
+                lines.push('');
+                continue;
+            }
+            const services = await node.getSwarmServices();
+            lines.push(`=== ${node.name} (${services.length} 个服务) ===`);
+            if (services.length === 0) {
+                lines.push('  (无服务)');
+            }
+            else {
+                for (const s of services) {
+                    const shortId = s.ID.slice(0, 12);
+                    const imageName = s.Image.split('@')[0];
+                    lines.push(`  ${s.Name}`);
+                    lines.push(`    ID: ${shortId} | 副本: ${s.Replicas} | 镜像: ${imageName}`);
+                    if (s.Ports !== '-') {
+                        lines.push(`    端口: ${s.Ports}`);
+                    }
+                }
+            }
+            lines.push('');
+        }
+        return lines.join('\n').trim();
+    });
+    /**
+     * 查看集群服务任务
+     */
+    ctx.command('docker.cluster.ps <selector> <service>', '查看 Swarm 服务任务')
+        .alias('集群任务', 'swarm任务', 'swarmps', '集群ps')
+        .option('format', '-f <format> 输出格式: simple|image', { fallback: null })
+        .action(async ({ options }, selector, serviceName) => {
+        logger_1.commandLogger.debug(`docker.cluster.ps 被调用: selector=${selector}, service=${serviceName}`);
+        const service = getService();
+        if (!service)
+            return '❌ 服务未初始化';
+        if (!serviceName) {
+            return '⚠️ 请指定服务名称\n例如: 集群任务 yun my-service';
+        }
+        const nodes = service.getNodesBySelector(selector || '');
+        if (nodes.length === 0)
+            return `❌ 未找到节点: ${selector}`;
+        const format = options.format || (useImageOutput ? 'image' : 'simple');
+        // 图片渲染模式
+        if (format === 'image') {
+            if (!ctx.puppeteer)
+                return '❌ 未安装 puppeteer 插件';
+            try {
+                const results = [];
+                for (const node of nodes) {
+                    if (node.status !== 'connected')
+                        continue;
+                    const isSwarm = await node.isSwarmMode();
+                    if (!isSwarm)
+                        continue;
+                    const tasks = await node.getSwarmTasks(serviceName);
+                    if (tasks.length > 0) {
+                        results.push({ node, serviceName, tasks });
+                    }
+                }
+                if (results.length === 0)
+                    return `❌ 未找到服务 "${serviceName}" 的任务`;
+                const html = (0, render_1.generateSwarmTasksHtml)(results, `集群任务 - ${serviceName}`);
+                return await (0, render_1.renderToImage)(ctx, html);
+            }
+            catch (e) {
+                logger_1.commandLogger.error(`获取集群任务失败: ${e.message}`);
+                return `❌ 错误: ${e.message}`;
+            }
+        }
+        // 文字模式
+        const lines = [];
+        for (const node of nodes) {
+            if (node.status !== 'connected') {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (未连接)');
+                lines.push('');
+                continue;
+            }
+            const isSwarm = await node.isSwarmMode();
+            if (!isSwarm) {
+                lines.push(`=== ${node.name} ===`);
+                lines.push('  (不在 Swarm 模式)');
+                lines.push('');
+                continue;
+            }
+            const tasks = await node.getSwarmTasks(serviceName);
+            lines.push(`=== ${node.name} (${tasks.length} 个任务) ===`);
+            if (tasks.length === 0) {
+                lines.push(`  (服务 "${serviceName}" 无任务或不存在)`);
+            }
+            else {
+                for (const t of tasks) {
+                    const shortId = t.ID.slice(0, 12);
+                    const statusIcon = t.Status.State === 'running' ? '🟢' :
+                        t.Status.State === 'pending' ? '⏳' :
+                            t.Status.State === 'failed' ? '❌' : '⚪';
+                    lines.push(`  ${statusIcon} Slot ${t.Slot} | ${t.Status.State}`);
+                    lines.push(`    ID: ${shortId}`);
+                    lines.push(`    节点: ${t.NodeID} | 期望状态: ${t.DesiredState}`);
+                    lines.push(`    时间: ${t.Status.Since}`);
+                }
+            }
+            lines.push('');
+        }
+        return lines.join('\n').trim();
+    });
+}