koishi-plugin-docker-control 0.0.1

package/lib/service/notifier.js

@@ -0,0 +1,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Notifier = void 0;
+const logger_1 = require("../utils/logger");
+/**
+ * 事件消息模板
+ */
+const EVENT_TEMPLATES = {
+    // 容器生命周期
+    'container.start': '容器已启动',
+    'container.stop': '容器已停止',
+    'container.restart': '容器已重启',
+    'container.die': '容器已异常退出',
+    'container.create': '容器已创建',
+    'container.destroy': '容器已销毁',
+    // 健康检查
+    'container.health_status': '容器健康状态变更',
+    'health_status: healthy': '容器健康检查通过',
+    'health_status: unhealthy': '容器健康检查失败',
+    // exec 事件
+    'exec_create': '执行命令',
+    'exec_start': '开始执行',
+    'exec_die': '执行结束',
+    // 节点事件
+    'node.online': '节点已上线',
+    'node.offline': '节点已离线',
+    'node.error': '节点发生错误',
+    // 附加/分离
+    'attach': '容器已附加',
+    'detach': '容器已分离',
+    'kill': '容器已被终止',
+    'oom': '内存不足',
+    'pause': '容器已暂停',
+    'unpause': '容器已恢复',
+};
+/**
+ * 容器状态 Emoji
+ */
+const STATUS_EMOJI = {
+    running: '🟢',
+    stopped: '🔴',
+    restarting: '🟡',
+    paused: '🟣',
+    created: '⚪',
+};
+/**
+ * 事件级别
+ */
+const EVENT_LEVEL = {
+    'container.start': 'info',
+    'container.stop': 'info',
+    'container.restart': 'info',
+    'container.die': 'error',
+    'container.health_status': 'warning',
+    'node.online': 'info',
+    'node.offline': 'warning',
+    'node.error': 'error',
+};
+class Notifier {
+    constructor(ctx, config) {
+        this.ctx = ctx;
+        this.config = config;
+    }
+    /**
+     * 发送通知
+     */
+    async send(eventType, data) {
+        // 检查是否启用
+        if (!this.config?.enabled) {
+            logger_1.notifierLogger.debug(`通知已禁用`);
+            return;
+        }
+        // 检查事件是否需要通知
+        if (!this.config?.events?.includes(eventType)) {
+            logger_1.notifierLogger.debug(`事件 ${eventType} 不在通知列表中`);
+            return;
+        }
+        // 检查通知级别
+        const level = EVENT_LEVEL[eventType];
+        if (this.config.level === 'error' && level !== 'error') {
+            logger_1.notifierLogger.debug(`事件级别 ${level} 被通知级别过滤`);
+            return;
+        }
+        // 构建消息
+        const message = this.buildMessage(eventType, data);
+        logger_1.notifierLogger.debug(`准备发送通知: ${eventType} -> ${message}`);
+        // 发送到所有目标群组
+        const channels = await this.getTargetChannels();
+        logger_1.notifierLogger.debug(`目标群组: ${JSON.stringify(channels)}`);
+        for (const channel of channels) {
+            try {
+                const bot = this.ctx.bots.find(b => b.sid === channel.botId);
+                if (bot) {
+                    await bot.sendMessage(channel.channelId, message);
+                    logger_1.notifierLogger.debug(`通知已发送: ${channel.channelId}`);
+                }
+                else {
+                    logger_1.notifierLogger.debug(`找不到 bot: ${channel.botId}`);
+                }
+            }
+            catch (e) {
+                logger_1.notifierLogger.error(`通知发送失败: ${e}`);
+            }
+        }
+    }
+    /**
+     * 构建消息
+     */
+    buildMessage(eventType, data) {
+        const template = EVENT_TEMPLATES[eventType] || '未知事件';
+        const parts = [];
+        // 节点信息
+        if (data.nodeName) {
+            parts.push(`【${data.nodeName}】`);
+        }
+        // 容器信息
+        if (data.containerName) {
+            const emoji = this.getContainerEmoji(data);
+            parts.push(`${emoji} ${data.containerName}`);
+        }
+        // 事件描述
+        parts.push(template);
+        // 额外信息
+        if (data.action && !template.includes(data.action)) {
+            parts.push(`(${data.action})`);
+        }
+        if (data.attributes?.image) {
+            parts.push(`\n镜像: ${data.attributes.image}`);
+        }
+        if (data.attributes?.exitCode !== undefined) {
+            parts.push(`\n退出码: ${data.attributes.exitCode}`);
+        }
+        // 组合消息
+        return parts.join(' ');
+    }
+    /**
+     * 获取容器状态 Emoji
+     */
+    getContainerEmoji(data) {
+        const status = data.attributes?.status || data.action;
+        return STATUS_EMOJI[status] || '📦';
+    }
+    /**
+     * 获取目标频道
+     */
+    async getTargetChannels() {
+        const channels = [];
+        if (!this.config?.targetGroups?.length) {
+            return channels;
+        }
+        // 获取所有群组频道
+        try {
+            const allChannels = await this.ctx.database.get('channel', {
+                platform: 'onebot',
+            });
+            for (const groupId of this.config.targetGroups) {
+                const channel = allChannels.find((c) => c.id === groupId);
+                if (channel) {
+                    channels.push({
+                        botId: channel.assignee || '',
+                        channelId: groupId,
+                    });
+                }
+            }
+        }
+        catch (e) {
+            logger_1.notifierLogger.warn(`获取频道列表失败: ${e}`);
+        }
+        return channels;
+    }
+    /**
+     * 发送自定义消息
+     */
+    async notifyCustom(content, targets) {
+        const channels = targets || this.config.targetGroups;
+        for (const groupId of channels) {
+            try {
+                // 发送消息给所有机器人的对应群组
+                for (const bot of this.ctx.bots.values()) {
+                    await bot.sendMessage(groupId, content);
+                }
+            }
+            catch (e) {
+                logger_1.notifierLogger.error(`自定义通知发送失败: ${e}`);
+            }
+        }
+    }
+}
+exports.Notifier = Notifier;

package/lib/types.d.ts

@@ -0,0 +1,100 @@
+/**
+ * 类型定义 - 简化版，只支持 SSH 直连模式
+ */
+export type AuthType = 'key' | 'password';
+export type NodeStatusType = 'disconnected' | 'connecting' | 'connected' | 'error';
+export type ContainerStatusType = 'running' | 'stopped' | 'paused' | 'restarting' | 'created';
+export interface CredentialConfig {
+    id: string;
+    name: string;
+    username: string;
+    authType: AuthType;
+    password?: string;
+    privateKey?: string;
+    passphrase?: string;
+}
+export interface NodeConfig {
+    id: string;
+    name: string;
+    tags: string[];
+    host: string;
+    port: number;
+    credentialId: string;
+}
+export interface NotificationConfig {
+    enabled: boolean;
+    level: 'all' | 'error' | 'none';
+    targetGroups: string[];
+    events: NotificationEventType[];
+}
+export type NotificationEventType = 'container.start' | 'container.stop' | 'container.restart' | 'container.die' | 'container.health_status' | 'node.online' | 'node.offline' | 'node.error';
+export interface MonitorConfig {
+    debounceWait: number;
+    flappingWindow: number;
+    flappingThreshold: number;
+}
+export interface DockerControlConfig {
+    requestTimeout: number;
+    debug: boolean;
+    imageOutput: boolean;
+    defaultLogLines: number;
+    monitor: MonitorConfig;
+    credentials: CredentialConfig[];
+    nodes: NodeConfig[];
+    notification: NotificationConfig;
+}
+export interface ContainerInfo {
+    Id: string;
+    Names: string[];
+    Image: string;
+    ImageID: string;
+    Command: string;
+    Created: number;
+    Ports: Array<{
+        PrivatePort: number;
+        PublicPort: number;
+        Type: string;
+    }>;
+    Labels: Record<string, string>;
+    State: ContainerStatusType;
+    Status: string;
+    HostConfig: {
+        NetworkMode: string;
+    };
+    NetworkSettings: {
+        Networks: Record<string, {
+            IPAddress: string;
+            Gateway: string;
+            MacAddress: string;
+        }>;
+    };
+}
+export interface DockerEvent {
+    Type: string;
+    Action: string;
+    Actor: {
+        ID: string;
+        Attributes: Record<string, string>;
+    };
+    scope: 'local' | 'swarm';
+    time: number;
+    timeNano: number;
+}
+export interface SubscriptionConfig {
+    /** 订阅 ID */
+    id?: number;
+    /** 平台 (onebot 等) */
+    platform: string;
+    /** 频道 ID (群组号或用户号) */
+    channelId: string;
+    /** 节点 ID (空表示所有节点) */
+    nodeId?: string;
+    /** 容器名称模式 (空表示所有容器，支持 * 通配符) */
+    containerPattern?: string;
+    /** 推送的事件类型 */
+    eventTypes: string[];
+    /** 是否启用 */
+    enabled: boolean;
+    /** 创建时间 */
+    createdAt?: number;
+}

package/lib/types.js

@@ -0,0 +1,5 @@
+"use strict";
+/**
+ * 类型定义 - 简化版，只支持 SSH 直连模式
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/utils/format.d.ts

@@ -0,0 +1,39 @@
+/**
+ * 格式化工具
+ */
+/**
+ * 格式化字节大小
+ */
+export declare function formatBytes(bytes: number, decimals?: number): string;
+/**
+ * 格式化时间
+ */
+export declare function formatTime(timestamp: number | string | Date, format?: 'iso' | 'local' | 'relative'): string;
+/**
+ * 相对时间格式化
+ */
+export declare function formatRelativeTime(date: Date): string;
+/**
+ * 格式化容器状态
+ */
+export declare function formatContainerStatus(state: string, running: boolean): string;
+/**
+ * 截断字符串
+ */
+export declare function truncate(str: string, maxLength: number, suffix?: string): string;
+/**
+ * 移除 ANSI 颜色码
+ */
+export declare function stripAnsiCodes(str: string): string;
+/**
+ * 格式化表格
+ */
+export declare function formatTable<T>(data: T[], columns: Array<{
+    key: keyof T;
+    header: string;
+    width: number;
+}>): string;
+/**
+ * 颜色状态图标
+ */
+export declare function getStatusEmoji(status: string, running: boolean): string;

package/lib/utils/format.js

@@ -0,0 +1,142 @@
+"use strict";
+/**
+ * 格式化工具
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatBytes = formatBytes;
+exports.formatTime = formatTime;
+exports.formatRelativeTime = formatRelativeTime;
+exports.formatContainerStatus = formatContainerStatus;
+exports.truncate = truncate;
+exports.stripAnsiCodes = stripAnsiCodes;
+exports.formatTable = formatTable;
+exports.getStatusEmoji = getStatusEmoji;
+/**
+ * 格式化字节大小
+ */
+function formatBytes(bytes, decimals = 2) {
+    if (bytes === 0)
+        return '0 B';
+    const k = 1024;
+    const sizes = ['B', 'KB', 'MB', 'GB', 'TB', 'PB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    return parseFloat((bytes / Math.pow(k, i)).toFixed(decimals)) + ' ' + sizes[i];
+}
+/**
+ * 格式化时间
+ */
+function formatTime(timestamp, format = 'local') {
+    const date = timestamp instanceof Date ? timestamp : new Date(timestamp);
+    switch (format) {
+        case 'iso':
+            return date.toISOString();
+        case 'relative':
+            return formatRelativeTime(date);
+        case 'local':
+        default:
+            return date.toLocaleString('zh-CN');
+    }
+}
+/**
+ * 相对时间格式化
+ */
+function formatRelativeTime(date) {
+    const now = new Date();
+    const diff = now.getTime() - date.getTime();
+    const seconds = Math.floor(diff / 1000);
+    const minutes = Math.floor(seconds / 60);
+    const hours = Math.floor(minutes / 60);
+    const days = Math.floor(hours / 24);
+    if (seconds < 60) {
+        return '刚刚';
+    }
+    else if (minutes < 60) {
+        return `${minutes} 分钟前`;
+    }
+    else if (hours < 24) {
+        return `${hours} 小时前`;
+    }
+    else if (days < 7) {
+        return `${days} 天前`;
+    }
+    else {
+        return date.toLocaleDateString('zh-CN');
+    }
+}
+/**
+ * 格式化容器状态
+ */
+function formatContainerStatus(state, running) {
+    if (running) {
+        return `运行中 (${state})`;
+    }
+    const statusMap = {
+        exited: '已停止',
+        stopped: '已停止',
+        created: '已创建',
+        paused: '已暂停',
+        restarting: '重启中',
+        dead: '已失效',
+    };
+    return statusMap[state.toLowerCase()] || state;
+}
+/**
+ * 截断字符串
+ */
+function truncate(str, maxLength, suffix = '...') {
+    if (str.length <= maxLength) {
+        return str;
+    }
+    return str.slice(0, maxLength - suffix.length) + suffix;
+}
+/**
+ * 移除 ANSI 颜色码
+ */
+function stripAnsiCodes(str) {
+    return str.replace(/[\x1b\u001b[0-9;]*[a-zA-Z]/g, '');
+}
+/**
+ * 格式化表格
+ */
+function formatTable(data, columns) {
+    if (data.length === 0) {
+        return '';
+    }
+    const lines = [];
+    // 表头
+    const header = columns
+        .map((col) => col.header.padEnd(col.width))
+        .join(' | ');
+    lines.push(header);
+    // 分隔线
+    const separator = columns.map((col) => '-'.repeat(col.width)).join('-+-');
+    lines.push(separator);
+    // 数据行
+    for (const row of data) {
+        const line = columns
+            .map((col) => {
+            const value = String(row[col.key] ?? '');
+            return value.slice(0, col.width).padEnd(col.width);
+        })
+            .join(' | ');
+        lines.push(line);
+    }
+    return lines.join('\n');
+}
+/**
+ * 颜色状态图标
+ */
+function getStatusEmoji(status, running) {
+    if (running) {
+        return '🟢';
+    }
+    const statusEmoji = {
+        exited: '🔴',
+        stopped: '🔴',
+        created: '⚪',
+        paused: '🟣',
+        restarting: '🟡',
+        dead: '⚫',
+    };
+    return statusEmoji[status.toLowerCase()] || '🔴';
+}

package/lib/utils/logger.d.ts

@@ -0,0 +1,65 @@
+/**
+ * 统一日志模块
+ * 提供带命名空间的日志器，支持调试模式
+ */
+import { Logger } from 'koishi';
+export declare const logger: Logger;
+export declare const nodeLogger: Logger;
+export declare const connectorLogger: Logger;
+export declare const monitorLogger: Logger;
+export declare const notifierLogger: Logger;
+export declare const commandLogger: Logger;
+/**
+ * 创建带命名空间的日志器
+ */
+export declare function createLogger(name: string): Logger;
+/**
+ * 格式化日志消息
+ */
+export declare function formatMessage(message: string, ...args: any[]): string;
+/**
+ * 节点操作日志 - 便捷方法
+ */
+export declare const node: {
+    info: (nodeName: string, message: string, ...args: any[]) => void;
+    warn: (nodeName: string, message: string, ...args: any[]) => void;
+    error: (nodeName: string, message: string, ...args: any[]) => void;
+    debug: (nodeName: string, message: string, ...args: any[]) => void;
+};
+/**
+ * 连接器日志 - 便捷方法
+ */
+export declare const connector: {
+    info: (nodeName: string, message: string, ...args: any[]) => void;
+    warn: (nodeName: string, message: string, ...args: any[]) => void;
+    error: (nodeName: string, message: string, ...args: any[]) => void;
+    debug: (nodeName: string, message: string, ...args: any[]) => void;
+};
+/**
+ * 监控日志 - 便捷方法
+ */
+export declare const monitor: {
+    info: (nodeName: string, message: string, ...args: any[]) => void;
+    warn: (nodeName: string, message: string, ...args: any[]) => void;
+    error: (nodeName: string, message: string, ...args: any[]) => void;
+    debug: (nodeName: string, message: string, ...args: any[]) => void;
+};
+/**
+ * 通知日志 - 便捷方法
+ */
+export declare const notify: {
+    info: (message: string, ...args: any[]) => void;
+    warn: (message: string, ...args: any[]) => void;
+    error: (message: string, ...args: any[]) => void;
+    debug: (message: string, ...args: any[]) => void;
+    success: (message: string, ...args: any[]) => void;
+};
+/**
+ * 指令日志 - 便捷方法
+ */
+export declare const command: {
+    info: (message: string, ...args: any[]) => void;
+    warn: (message: string, ...args: any[]) => void;
+    error: (message: string, ...args: any[]) => void;
+    debug: (message: string, ...args: any[]) => void;
+};

package/lib/utils/logger.js

@@ -0,0 +1,89 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.command = exports.notify = exports.monitor = exports.connector = exports.node = exports.commandLogger = exports.notifierLogger = exports.monitorLogger = exports.connectorLogger = exports.nodeLogger = exports.logger = void 0;
+exports.createLogger = createLogger;
+exports.formatMessage = formatMessage;
+/**
+ * 统一日志模块
+ * 提供带命名空间的日志器，支持调试模式
+ */
+const koishi_1 = require("koishi");
+// 主日志器
+exports.logger = new koishi_1.Logger('docker-control');
+// 子日志器工厂
+function createSubLogger(name) {
+    return exports.logger.extend(name);
+}
+// 子日志器
+exports.nodeLogger = createSubLogger('node');
+exports.connectorLogger = createSubLogger('connector');
+exports.monitorLogger = createSubLogger('monitor');
+exports.notifierLogger = createSubLogger('notifier');
+exports.commandLogger = createSubLogger('command');
+/**
+ * 创建带命名空间的日志器
+ */
+function createLogger(name) {
+    return createSubLogger(name);
+}
+/**
+ * 格式化日志消息
+ */
+function formatMessage(message, ...args) {
+    if (args.length === 0)
+        return message;
+    try {
+        return args.length > 1
+            ? message.replace(/\{(\d+)\}/g, (_, index) => String(args[index] || ''))
+            : String(args[0]);
+    }
+    catch {
+        return message;
+    }
+}
+/**
+ * 节点操作日志 - 便捷方法
+ */
+exports.node = {
+    info: (nodeName, message, ...args) => exports.nodeLogger.info(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    warn: (nodeName, message, ...args) => exports.nodeLogger.warn(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    error: (nodeName, message, ...args) => exports.nodeLogger.error(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    debug: (nodeName, message, ...args) => exports.nodeLogger.debug(`[${nodeName}] ${formatMessage(message, ...args)}`),
+};
+/**
+ * 连接器日志 - 便捷方法
+ */
+exports.connector = {
+    info: (nodeName, message, ...args) => exports.connectorLogger.info(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    warn: (nodeName, message, ...args) => exports.connectorLogger.warn(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    error: (nodeName, message, ...args) => exports.connectorLogger.error(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    debug: (nodeName, message, ...args) => exports.connectorLogger.debug(`[${nodeName}] ${formatMessage(message, ...args)}`),
+};
+/**
+ * 监控日志 - 便捷方法
+ */
+exports.monitor = {
+    info: (nodeName, message, ...args) => exports.monitorLogger.info(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    warn: (nodeName, message, ...args) => exports.monitorLogger.warn(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    error: (nodeName, message, ...args) => exports.monitorLogger.error(`[${nodeName}] ${formatMessage(message, ...args)}`),
+    debug: (nodeName, message, ...args) => exports.monitorLogger.debug(`[${nodeName}] ${formatMessage(message, ...args)}`),
+};
+/**
+ * 通知日志 - 便捷方法
+ */
+exports.notify = {
+    info: (message, ...args) => exports.notifierLogger.info(formatMessage(message, ...args)),
+    warn: (message, ...args) => exports.notifierLogger.warn(formatMessage(message, ...args)),
+    error: (message, ...args) => exports.notifierLogger.error(formatMessage(message, ...args)),
+    debug: (message, ...args) => exports.notifierLogger.debug(formatMessage(message, ...args)),
+    success: (message, ...args) => exports.notifierLogger.info(`✓ ${formatMessage(message, ...args)}`),
+};
+/**
+ * 指令日志 - 便捷方法
+ */
+exports.command = {
+    info: (message, ...args) => exports.commandLogger.info(formatMessage(message, ...args)),
+    warn: (message, ...args) => exports.commandLogger.warn(formatMessage(message, ...args)),
+    error: (message, ...args) => exports.commandLogger.error(formatMessage(message, ...args)),
+    debug: (message, ...args) => exports.commandLogger.debug(formatMessage(message, ...args)),
+};

package/lib/utils/stream.d.ts

@@ -0,0 +1,28 @@
+/**
+ * 流处理工具
+ */
+/**
+ * 将 Stream 转换为字符串
+ */
+export declare function streamToString(stream: NodeJS.ReadableStream): Promise<string>;
+/**
+ * 将字符串转换为 Stream
+ */
+export declare function stringToStream(content: string, encoding?: BufferEncoding): NodeJS.ReadableStream;
+/**
+ * 合并多个 Stream
+ */
+export declare function mergeStreams(streams: NodeJS.ReadableStream[]): NodeJS.ReadableStream;
+/**
+ * 限流 Stream
+ * 控制数据输出的速度
+ */
+export declare function throttleStream(stream: NodeJS.ReadableStream, maxBytesPerSecond: number): NodeJS.ReadableStream;
+/**
+ * 解码 Stream (处理 ANSI 转义码)
+ */
+export declare function decodeStream(stream: NodeJS.ReadableStream): NodeJS.ReadableStream;
+/**
+ * 截取 Stream 的最后 N 行
+ */
+export declare function tailStream(stream: NodeJS.ReadableStream, maxLines: number): NodeJS.ReadableStream;