cubeapm-mcp 1.3.0 → 1.3.1

package/dist/index.js

@@ -25,6 +25,90 @@ const server = new McpServer({
 // ============================================
 // LOGS APIs
 // ============================================
+// Convert RFC3339 string or Unix timestamp (s/ms/μs/ns) to nanoseconds.
+// CubeAPM's prod UI embeds time in queries as `_time:[ns_start, ns_end)`,
+// so we mirror that wire format rather than relying on start/end form params.
+function toNanoseconds(t) {
+    if (/^\d+$/.test(t)) {
+        const n = BigInt(t);
+        if (n < 10000000000n)
+            return (n * 1000000000n).toString(); // seconds
+        if (n < 10000000000000n)
+            return (n * 1000000n).toString(); // milliseconds
+        if (n < 10000000000000000n)
+            return (n * 1000n).toString(); // microseconds
+        return n.toString(); // nanoseconds
+    }
+    const ms = new Date(t).getTime();
+    if (Number.isNaN(ms))
+        throw new Error(`Invalid timestamp: ${t}`);
+    return (BigInt(ms) * 1000000n).toString();
+}
+// Find the index of the first top-level pipe operator in a LogsQL query,
+// ignoring `|` inside quoted strings (regex alternation like `=~"a|b"`),
+// brace groups `{...}`, or parens `(...)`. Returns -1 if no pipe found.
+function findFirstTopLevelPipe(query) {
+    let inDouble = false;
+    let inSingle = false;
+    let braceDepth = 0;
+    let parenDepth = 0;
+    for (let i = 0; i < query.length; i++) {
+        const c = query[i];
+        const prev = i > 0 ? query[i - 1] : "";
+        // Honour backslash escapes inside strings (e.g. `"shopify\\-prod"`).
+        if (prev === "\\")
+            continue;
+        if (inDouble) {
+            if (c === '"')
+                inDouble = false;
+            continue;
+        }
+        if (inSingle) {
+            if (c === "'")
+                inSingle = false;
+            continue;
+        }
+        if (c === '"') {
+            inDouble = true;
+            continue;
+        }
+        if (c === "'") {
+            inSingle = true;
+            continue;
+        }
+        if (c === "{") {
+            braceDepth++;
+            continue;
+        }
+        if (c === "}") {
+            braceDepth--;
+            continue;
+        }
+        if (c === "(") {
+            parenDepth++;
+            continue;
+        }
+        if (c === ")") {
+            parenDepth--;
+            continue;
+        }
+        if (c === "|" && braceDepth === 0 && parenDepth === 0)
+            return i;
+    }
+    return -1;
+}
+// Inject `_time:[start_ns,end_ns)` before the first top-level pipe operator
+// (or at end if none). Quote/brace/paren aware to handle regex alternations
+// like `=~"shopify-latest|shopify-prod"` without splitting mid-string.
+function injectTimeFilter(query, startNs, endNs) {
+    const timeFilter = `_time:[${startNs},${endNs})`;
+    const pipeIdx = findFirstTopLevelPipe(query);
+    if (pipeIdx === -1)
+        return `${query.trim()} ${timeFilter}`.trim();
+    const head = query.slice(0, pipeIdx).trim();
+    const tail = query.slice(pipeIdx);
+    return `${head} ${timeFilter} ${tail}`;
+}
 server.tool("query_logs", `Query logs from CubeAPM using LogsQL syntax (VictoriaLogs compatible). Returns log entries matching the query.
 
 IMPORTANT - Log Label Discovery:
@@ -79,10 +163,11 @@ Example queries:
     end: z.string().describe("End time in RFC3339 format or Unix timestamp in seconds"),
     limit: z.number().optional().default(100).describe("Maximum log entries to return (default: 100)"),
 }, async ({ query, start, end, limit }) => {
+    const startNs = toNanoseconds(start);
+    const endNs = toNanoseconds(end);
+    const finalQuery = injectTimeFilter(query, startNs, endNs);
     const params = new URLSearchParams({
-        query,
-        start,
-        end,
+        query: finalQuery,
         limit: String(limit),
     });
     const response = await fetch(`${queryBaseUrl}/api/logs/select/logsql/query`, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cubeapm-mcp",
-  "version": "1.3.0",
+  "version": "1.3.1",
   "description": "MCP server for CubeAPM - Query traces, metrics, and logs from your observability platform using AI assistants",
   "type": "module",
   "main": "dist/index.js",