@hasna/logs 0.3.9 → 0.3.11

package/src/lib/ingest.ts

@@ -37,7 +37,11 @@ export function ingestLog(db: Database, entry: LogEntry): LogRow {
   return row
 }
 
-export function ingestBatch(db: Database, entries: LogEntry[]): LogRow[] {
+export function ingestBatch(db: Database, entries: LogEntry[], sharedTraceId?: string | null): LogRow[] {
+  // Apply shared trace_id to entries that don't have their own
+  if (sharedTraceId) {
+    entries = entries.map(e => e.trace_id ? e : { ...e, trace_id: sharedTraceId })
+  }
   const insert = db.prepare(`
     INSERT INTO logs (project_id, page_id, level, source, service, message, trace_id, session_id, agent, url, stack_trace, metadata)
     VALUES ($project_id, $page_id, $level, $source, $service, $message, $trace_id, $session_id, $agent, $url, $stack_trace, $metadata)

package/src/mcp/index.ts

@@ -123,9 +123,15 @@ server.tool("log_push_batch", {
     project_id: z.string().optional(), service: z.string().optional(),
     trace_id: z.string().optional(), metadata: z.record(z.unknown()).optional(),
   })),
-}, ({ entries }) => {
-  const rows = ingestBatch(db, entries.map(e => ({ ...e, project_id: rp(e.project_id) })))
-  return { content: [{ type: "text", text: `Logged ${rows.length} entries` }] }
+  trace_id: z.string().optional().describe("Shared trace_id applied to all entries that don't have their own trace_id"),
+  project_id: z.string().optional().describe("Shared project_id applied to all entries (individual entry project_id takes precedence)"),
+}, ({ entries, trace_id, project_id }) => {
+  const mapped = entries.map(e => ({
+    ...e,
+    project_id: rp(e.project_id ?? project_id),
+  }))
+  const rows = ingestBatch(db, mapped, trace_id)
+  return { content: [{ type: "text", text: `Logged ${rows.length} entries${trace_id ? ` (trace: ${trace_id})` : ''}` }] }
 })
 
 server.tool("log_search", {
@@ -300,5 +306,26 @@ server.tool("get_health", {}, () => ({
   content: [{ type: "text", text: JSON.stringify(getHealth(db)) }]
 }))
 
+server.tool("log_stats", {
+  project_id: z.string().optional().describe("Project name or ID (scope stats to a project)"),
+}, (args) => {
+  const projectId = rp(args.project_id)
+  const pFilter = projectId ? `WHERE project_id = ?` : ""
+  const pAnd = projectId ? `AND project_id = ?` : ""
+  const pParam = projectId ? [projectId] : []
+
+  const total = (db.query(`SELECT COUNT(*) as c FROM logs ${pFilter}`).get(...pParam) as { c: number }).c
+  const oldest = (db.query(`SELECT MIN(timestamp) as t FROM logs ${pFilter}`).get(...pParam) as { t: string | null }).t
+  const newest = (db.query(`SELECT MAX(timestamp) as t FROM logs ${pFilter}`).get(...pParam) as { t: string | null }).t
+  const byLevel = db.query(`SELECT level, COUNT(*) as c FROM logs ${pFilter} GROUP BY level ORDER BY c DESC`).all(...pParam) as { level: string; c: number }[]
+  const topServices = db.query(`SELECT COALESCE(service, '-') as service, COUNT(*) as c FROM logs ${pFilter} GROUP BY service ORDER BY c DESC LIMIT 5`).all(...pParam) as { service: string; c: number }[]
+  const days = db.query(`SELECT strftime('%Y-%m-%d', timestamp) as day, COUNT(*) as c FROM logs WHERE timestamp >= datetime('now', '-7 days') ${pAnd} GROUP BY day ORDER BY day`).all(...pParam) as { day: string; c: number }[]
+  const errors = (byLevel.find(r => r.level === "error")?.c ?? 0) + (byLevel.find(r => r.level === "fatal")?.c ?? 0)
+  const error_rate_pct = total > 0 ? parseFloat(((errors / total) * 100).toFixed(2)) : 0
+  return {
+    content: [{ type: "text" as const, text: JSON.stringify({ total, oldest, newest, by_level: Object.fromEntries(byLevel.map(r => [r.level, r.c])), top_services: topServices, last_7_days: days, error_rate_pct }) }]
+  }
+})
+
 const transport = new StdioServerTransport()
 await server.connect(transport)