rl-rockcli 0.0.1

package/src/core/utils/time.js

@@ -0,0 +1,52 @@
+/**
+ * 时间解析工具
+ * 支持时间戳、相对时间、日期字符串等多种格式
+ */
+
+/**
+ * 解析时间字符串为时间戳（毫秒）
+ * @param {string} timeStr - 时间字符串
+ * @returns {number|null} 时间戳（毫秒）
+ * @throws {Error} 无法解析的时间格式
+ */
+function parseTime(timeStr) {
+    if (!timeStr) return null;
+
+    // 如果是纯数字，智能判断是秒级还是毫秒级时间戳
+    if (/^\d+$/.test(timeStr)) {
+        const timestamp = parseInt(timeStr);
+        // Unix 时间戳标准是秒级的（10位数字），毫秒级通常是13位数字
+        // 如果数字长度 <= 10，认为是秒级时间戳，需要转换为毫秒
+        // 如果数字长度 > 10，认为是毫秒级时间戳，直接使用
+        if (timestamp <= 9999999999) {
+            return timestamp * 1000; // 秒级 -> 毫秒级
+        } else {
+            return timestamp; // 已经是毫秒级
+        }
+    }
+
+    // 解析相对时间格式（如 "15m", "1h", "2d"）
+    const relativeMatch = timeStr.match(/^(\d+)([mhd])$/);
+    if (relativeMatch) {
+        const value = parseInt(relativeMatch[1]);
+        const unit = relativeMatch[2];
+        const now = Date.now();
+        switch (unit) {
+            case 'm': return now - value * 60 * 1000; // 分钟
+            case 'h': return now - value * 60 * 60 * 1000; // 小时
+            case 'd': return now - value * 24 * 60 * 60 * 1000; // 天
+        }
+    }
+
+    // 尝试解析为日期字符串
+    const date = new Date(timeStr);
+    if (!isNaN(date.getTime())) {
+        return date.getTime();
+    }
+
+    throw new Error(`无法解析时间格式: ${timeStr}`);
+}
+
+module.exports = {
+    parseTime,
+};

package/src/plugins/oss-file-log/file-client.js

@@ -0,0 +1,186 @@
+/**
+ * 本地文件日志客户端
+ * 实现 LogProvider 接口
+ * 
+ * 默认读取路径：~/.rock/logs/rock-cli-YYYY-MM-DD.log
+ * 每行日志建议为 JSON 字符串，至少包含 timestamp(ms) 或 time_iso8601、level、message 等字段。
+ */
+
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const readline = require('readline');
+
+/**
+ * 根据时间范围推导需要读取的日志文件列表。
+ * @param {Object} params
+ * @param {string} [params.logDir]  日志目录，默认 ~/.rock/logs
+ * @param {number} params.startTime 毫秒时间戳
+ * @param {number} params.endTime   毫秒时间戳
+ * @returns {string[]}
+ */
+function getLogFilesInRange({ logDir, startTime, endTime }) {
+    const dir = logDir || path.join(os.homedir(), '.rock', 'logs');
+    const files = [];
+
+    const start = new Date(startTime);
+    const end = new Date(endTime);
+
+    const current = new Date(start.getFullYear(), start.getMonth(), start.getDate());
+    const last = new Date(end.getFullYear(), end.getMonth(), end.getDate());
+
+    while (current <= last) {
+        const yyyy = current.getFullYear();
+        const mm = String(current.getMonth() + 1).padStart(2, '0');
+        const dd = String(current.getDate()).padStart(2, '0');
+        const fileName = `rock-cli-${yyyy}-${mm}-${dd}.log`;
+        files.push(path.join(dir, fileName));
+        current.setDate(current.getDate() + 1);
+    }
+
+    return files;
+}
+
+/**
+ * 简单的 filter 函数构造器：把 filter 当作空格分隔的关键字 AND 匹配。
+ * @param {string | null | undefined} filter
+ */
+function buildSimpleFilterFn(filter) {
+    if (!filter) {
+        return () => true;
+    }
+
+    const keywords = String(filter)
+        .split(/\s+/)
+        .map((s) => s.trim())
+        .filter(Boolean)
+        .map((s) => s.toLowerCase());
+
+    return (logObj, rawLine) => {
+        const haystack = rawLine.toLowerCase();
+        return keywords.every((kw) => haystack.includes(kw));
+    };
+}
+
+/**
+ * 创建本地文件日志客户端。
+ * 其接口尽量与内部版的 mysqlClient 保持一致：connect/query/close。
+ *
+ * @param {Object} config
+ * @param {string} [config.logDir] 日志目录，默认 ~/.rock/logs
+ */
+function createFileLogClient(config = {}) {
+    return {
+        async connect() {
+            // 与数据库客户端接口对齐，这里不需要实际连接
+            return this;
+        },
+
+        /**
+         * 查询日志。
+         * @param {Object} queryOptions
+         * @param {number} queryOptions.startTime - 毫秒时间戳
+         * @param {number} queryOptions.endTime   - 毫秒时间戳
+         * @param {string} [queryOptions.filter]  - 简单查询字符串（关键字匹配）
+         * @param {number} [queryOptions.limit]   - 返回条数限制
+         * @param {number} [queryOptions.offset]  - 偏移量
+         * @param {boolean} [queryOptions.count]  - 是否只返回数量
+         * @param {string} [queryOptions.table]   - 表名（内网专有字段，开源版忽略）
+         * @param {string} [queryOptions.orderBy] - 排序字段（开源版忽略）
+         * @param {string} [queryOptions.orderDirection] - 排序方向（开源版忽略）
+         */
+        async query(queryOptions) {
+            const {
+                startTime,
+                endTime,
+                filter,
+                limit = 1000,
+                offset = 0,
+                count = false,
+                // table, orderBy, orderDirection 字段被命令层传递，但开源版忽略
+            } = queryOptions;
+
+            if (typeof startTime !== 'number' || typeof endTime !== 'number') {
+                throw new Error('startTime and endTime are required (number, ms)');
+            }
+
+            const logFiles = getLogFilesInRange({
+                logDir: config.logDir,
+                startTime,
+                endTime,
+            });
+
+            const filterFn = buildSimpleFilterFn(filter);
+
+            let matched = [];
+            let total = 0;
+
+            for (const file of logFiles) {
+                if (!fs.existsSync(file)) continue;
+
+                const stream = fs.createReadStream(file, { encoding: 'utf8' });
+                const rl = readline.createInterface({ input: stream, crlfDelay: Infinity });
+
+                // eslint-disable-next-line no-restricted-syntax
+                for await (const line of rl) {
+                    if (!line.trim()) continue;
+
+                    let obj;
+                    try {
+                        obj = JSON.parse(line);
+                    } catch (e) {
+                        obj = { message: line };
+                    }
+
+                    let ts = obj.timestamp;
+                    if (!ts && obj.time_iso8601) {
+                        const parsed = Date.parse(obj.time_iso8601);
+                        if (!Number.isNaN(parsed)) {
+                            ts = parsed;
+                        }
+                    }
+                    if (!ts) continue;
+
+                    if (ts < startTime || ts > endTime) {
+                        continue;
+                    }
+
+                    if (!filterFn(obj, line)) {
+                        continue;
+                    }
+
+                    total += 1;
+
+                    if (!count) {
+                        matched.push({
+                            ...obj,
+                            '@timestamp': String(ts),
+                            _file: file,
+                        });
+                    }
+                }
+            }
+
+            if (count) {
+                // 与内部版约定保持一致：返回带 total 字段的对象数组
+                return [{ total }];
+            }
+
+            matched.sort((a, b) => {
+                const ta = Number(a['@timestamp']) || 0;
+                const tb = Number(b['@timestamp']) || 0;
+                return ta - tb;
+            });
+
+            const sliced = matched.slice(offset, offset + limit);
+            return sliced;
+        },
+
+        async close() {
+            // 文件读取无需显式关闭资源（readline 已处理），此处保留接口
+            return undefined;
+        },
+    };
+}
+
+module.exports = createFileLogClient;

package/src/plugins/oss-file-log/index.js

@@ -0,0 +1,18 @@
+/**
+ * oss-file-log 插件入口
+ * 提供本地文件日志后端
+ */
+
+const createFileLogClient = require('./file-client');
+
+module.exports = {
+    name: 'oss-file-log',
+    description: 'Local file log backend for rock-cli',
+    version: '0.1.0',
+
+    // 导出客户端创建函数
+    createClient: createFileLogClient,
+
+    // Provider 工厂
+    createProvider: (config = {}) => createFileLogClient(config),
+};