Add AgentOS daemon control plane and maintenance tooling

packages/backend/convex/daemonRuntime.ts

@@ -0,0 +1,107 @@
+import { v } from "convex/values";
+
+import { mutation, query } from "./_generated/server";
+
+export const connect = mutation({
+  args: {
+    daemonId: v.string(),
+    sessionId: v.string(),
+    hostname: v.string(),
+    platform: v.string(),
+    architecture: v.string(),
+    version: v.string(),
+  },
+  handler: async (ctx, args) => {
+    const existing = await ctx.db
+      .query("daemonPresence")
+      .withIndex("by_sessionId", (q) => q.eq("sessionId", args.sessionId))
+      .unique();
+    const timestamp = Date.now();
+    if (existing) {
+      await ctx.db.patch("daemonPresence", existing._id, {
+        status: "online",
+        lastHeartbeatAt: timestamp,
+      });
+      return existing._id;
+    }
+    return await ctx.db.insert("daemonPresence", {
+      ...args,
+      status: "online",
+      startedAt: timestamp,
+      lastHeartbeatAt: timestamp,
+    });
+  },
+});
+
+export const heartbeat = mutation({
+  args: {
+    daemonId: v.string(),
+    sessionId: v.string(),
+  },
+  handler: async (ctx, args) => {
+    const presence = await ctx.db
+      .query("daemonPresence")
+      .withIndex("by_sessionId", (q) => q.eq("sessionId", args.sessionId))
+      .unique();
+    if (!presence || presence.daemonId !== args.daemonId) {
+      return false;
+    }
+    await ctx.db.patch("daemonPresence", presence._id, {
+      status: "online",
+      lastHeartbeatAt: Date.now(),
+    });
+    return true;
+  },
+});
+
+export const disconnect = mutation({
+  args: {
+    daemonId: v.string(),
+    sessionId: v.string(),
+  },
+  handler: async (ctx, args) => {
+    const presence = await ctx.db
+      .query("daemonPresence")
+      .withIndex("by_sessionId", (q) => q.eq("sessionId", args.sessionId))
+      .unique();
+    if (!presence || presence.daemonId !== args.daemonId) {
+      return false;
+    }
+    await ctx.db.patch("daemonPresence", presence._id, {
+      status: "offline",
+      lastHeartbeatAt: Date.now(),
+    });
+    return true;
+  },
+});
+
+export const recordEvent = mutation({
+  args: {
+    daemonId: v.string(),
+    commandId: v.optional(v.id("daemonCommands")),
+    kind: v.string(),
+    data: v.any(),
+  },
+  handler: async (ctx, args) => {
+    return await ctx.db.insert("daemonCommandEvents", {
+      ...args,
+      createdAt: Date.now(),
+    });
+  },
+});
+
+export const listEvents = query({
+  args: {
+    daemonId: v.string(),
+    limit: v.optional(v.number()),
+  },
+  handler: async (ctx, args) => {
+    return await ctx.db
+      .query("daemonCommandEvents")
+      .withIndex("by_daemonId_and_createdAt", (q) =>
+        q.eq("daemonId", args.daemonId)
+      )
+      .order("desc")
+      .take(Math.min(args.limit ?? 100, 500));
+  },
+});