@vantageos/agent-engine 0.1.0-alpha.8 → 0.3.0-alpha.1

package/dist/component/engine/world.js

@@ -0,0 +1,689 @@
+import { v } from "convex/values";
+import { mutation, query, } from "../_generated/server.js";
+import { assertSameOrg, requireOrgId } from "../lib/auth.js";
+function pad26(n) {
+    return String(n).padStart(26, "0");
+}
+function slotToEventId(seq) {
+    return `evnt_${pad26(seq)}`;
+}
+function newRunId() {
+    return `${Date.now().toString(36)}${Math.random().toString(36).slice(2, 14)}`;
+}
+const eventResult = v.object({
+    eventId: v.string(),
+    seq: v.number(),
+    runId: v.string(),
+    eventType: v.string(),
+    stepId: v.optional(v.string()),
+    hookId: v.optional(v.string()),
+    correlationId: v.optional(v.string()),
+    dataJson: v.string(),
+    createdAt: v.number(),
+});
+const runResult = v.object({
+    runId: v.string(),
+    status: v.union(v.literal("running"), v.literal("completed"), v.literal("failed"), v.literal("cancelled")),
+    eventCount: v.number(),
+    createdAt: v.number(),
+    updatedAt: v.number(),
+});
+const stepResult = v.object({
+    runId: v.string(),
+    stepId: v.string(),
+    status: v.union(v.literal("running"), v.literal("completed"), v.literal("failed")),
+    dataJson: v.optional(v.string()),
+    createdAt: v.number(),
+    updatedAt: v.number(),
+});
+const hookResult = v.object({
+    runId: v.string(),
+    hookId: v.string(),
+    token: v.string(),
+    status: v.union(v.literal("pending"), v.literal("disposed")),
+    createdAt: v.number(),
+    updatedAt: v.number(),
+});
+export async function appendEventCore(ctx, args) {
+    const orgId = requireOrgId(args.orgId);
+    const now = Date.now();
+    let runId = args.runId;
+    let runDoc;
+    if (runId === null) {
+        if (args.eventType !== "run_created") {
+            throw new Error("Validation: a null runId is only valid for a run_created event");
+        }
+        if (args.correlationId !== undefined) {
+            const correlationId = args.correlationId;
+            const prior = await ctx.db
+                .query("worldEvents")
+                .withIndex("by_org_correlation", (q) => q.eq("orgId", orgId).eq("correlationId", correlationId))
+                .collect();
+            const priorCreated = prior.find((e) => e.eventType === "run_created");
+            if (priorCreated) {
+                const existingRun = await ctx.db
+                    .query("worldRuns")
+                    .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", priorCreated.runId))
+                    .unique();
+                assertSameOrg(orgId, existingRun, "Run not found");
+                return {
+                    event: {
+                        eventId: slotToEventId(priorCreated.seq),
+                        seq: priorCreated.seq,
+                        runId: priorCreated.runId,
+                        eventType: priorCreated.eventType,
+                        stepId: priorCreated.stepId,
+                        hookId: priorCreated.hookId,
+                        correlationId: priorCreated.correlationId,
+                        dataJson: priorCreated.dataJson,
+                        createdAt: priorCreated.createdAt,
+                    },
+                    run: {
+                        runId: existingRun.runId,
+                        status: existingRun.status,
+                        eventCount: existingRun.eventCount,
+                        createdAt: existingRun.createdAt,
+                        updatedAt: existingRun.updatedAt,
+                    },
+                    step: null,
+                    hook: null,
+                };
+            }
+        }
+        runId = newRunId();
+        const runDocId = await ctx.db.insert("worldRuns", {
+            orgId,
+            runId,
+            status: "running",
+            eventCount: 0,
+            createdAt: now,
+            updatedAt: now,
+        });
+        runDoc = await ctx.db.get(runDocId);
+    }
+    else {
+        runDoc = await ctx.db
+            .query("worldRuns")
+            .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", runId))
+            .unique();
+        assertSameOrg(orgId, runDoc, "Run not found");
+    }
+    if (!runDoc) {
+        throw new Error("Run not found");
+    }
+    let stepAlreadyExists = false;
+    if (args.eventType === "step_created") {
+        if (!args.stepId) {
+            throw new Error("Validation: step_created/step_started requires stepId");
+        }
+        const existingStep = await ctx.db
+            .query("worldSteps")
+            .withIndex("by_org_run_step", (q) => q.eq("orgId", orgId).eq("runId", runId).eq("stepId", args.stepId))
+            .unique();
+        if (existingStep) {
+            assertSameOrg(orgId, existingStep, "Step not found");
+            stepAlreadyExists = true;
+            const journal = await ctx.db
+                .query("worldEvents")
+                .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", runId))
+                .collect();
+            const prior = journal.find((e) => e.eventType === "step_created" && e.stepId === args.stepId);
+            if (prior) {
+                return {
+                    event: {
+                        eventId: slotToEventId(prior.seq),
+                        seq: prior.seq,
+                        runId: prior.runId,
+                        eventType: prior.eventType,
+                        stepId: prior.stepId,
+                        hookId: prior.hookId,
+                        correlationId: prior.correlationId,
+                        dataJson: prior.dataJson,
+                        createdAt: prior.createdAt,
+                    },
+                    run: {
+                        runId: runDoc.runId,
+                        status: runDoc.status,
+                        eventCount: runDoc.eventCount,
+                        createdAt: runDoc.createdAt,
+                        updatedAt: runDoc.updatedAt,
+                    },
+                    step: {
+                        runId: existingStep.runId,
+                        stepId: existingStep.stepId,
+                        status: existingStep.status,
+                        dataJson: existingStep.dataJson,
+                        createdAt: existingStep.createdAt,
+                        updatedAt: existingStep.updatedAt,
+                    },
+                    hook: null,
+                };
+            }
+        }
+    }
+    const seq = runDoc.eventCount + 1;
+    const eventId = slotToEventId(seq);
+    await ctx.db.insert("worldEvents", {
+        orgId,
+        runId,
+        seq,
+        eventType: args.eventType,
+        stepId: args.stepId,
+        hookId: args.hookId,
+        correlationId: args.correlationId,
+        dataJson: args.dataJson,
+        createdAt: now,
+    });
+    await ctx.db.patch(runDoc._id, { eventCount: seq, updatedAt: now });
+    let stepDoc = null;
+    let hookDoc = null;
+    let finalRunDoc = { ...runDoc, eventCount: seq, updatedAt: now };
+    switch (args.eventType) {
+        case "step_created":
+        case "step_started": {
+            if (!args.stepId) {
+                throw new Error("Validation: step_created/step_started requires stepId");
+            }
+            const existingStep = await ctx.db
+                .query("worldSteps")
+                .withIndex("by_org_run_step", (q) => q.eq("orgId", orgId).eq("runId", runId).eq("stepId", args.stepId))
+                .unique();
+            if (existingStep && args.eventType === "step_started") {
+                assertSameOrg(orgId, existingStep, "Step not found");
+                await ctx.db.patch(existingStep._id, {
+                    status: "running",
+                    dataJson: args.dataJson,
+                    updatedAt: now,
+                });
+                stepDoc = await ctx.db.get(existingStep._id);
+                break;
+            }
+            if (stepAlreadyExists) {
+                stepDoc = await ctx.db
+                    .query("worldSteps")
+                    .withIndex("by_org_run_step", (q) => q.eq("orgId", orgId).eq("runId", runId).eq("stepId", args.stepId))
+                    .unique();
+                break;
+            }
+            const stepId = await ctx.db.insert("worldSteps", {
+                orgId,
+                runId,
+                stepId: args.stepId,
+                status: "running",
+                dataJson: args.dataJson,
+                createdAt: now,
+                updatedAt: now,
+            });
+            stepDoc = await ctx.db.get(stepId);
+            break;
+        }
+        case "step_completed":
+        case "step_failed": {
+            if (!args.stepId) {
+                throw new Error("Validation: step_completed/step_failed requires stepId");
+            }
+            const existing = await ctx.db
+                .query("worldSteps")
+                .withIndex("by_org_run_step", (q) => q.eq("orgId", orgId).eq("runId", runId).eq("stepId", args.stepId))
+                .unique();
+            assertSameOrg(orgId, existing, "Step not found");
+            await ctx.db.patch(existing._id, {
+                status: args.eventType === "step_completed" ? "completed" : "failed",
+                dataJson: args.dataJson,
+                updatedAt: now,
+            });
+            stepDoc = await ctx.db.get(existing._id);
+            break;
+        }
+        case "hook_created": {
+            if (!args.hookId) {
+                throw new Error("Validation: hook_created requires hookId");
+            }
+            const parsed = JSON.parse(args.dataJson);
+            if (!parsed.token) {
+                throw new Error("Validation: hook_created payload requires token");
+            }
+            const hookDocId = await ctx.db.insert("worldHooks", {
+                orgId,
+                runId,
+                hookId: args.hookId,
+                token: parsed.token,
+                status: "pending",
+                createdAt: now,
+                updatedAt: now,
+            });
+            hookDoc = await ctx.db.get(hookDocId);
+            break;
+        }
+        case "hook_disposed": {
+            if (!args.hookId) {
+                throw new Error("Validation: hook_disposed requires hookId");
+            }
+            const existing = await ctx.db
+                .query("worldHooks")
+                .withIndex("by_org_hook", (q) => q.eq("orgId", orgId).eq("hookId", args.hookId))
+                .unique();
+            assertSameOrg(orgId, existing, "Hook not found");
+            await ctx.db.delete(existing._id);
+            hookDoc = null;
+            break;
+        }
+        case "run_cancelled": {
+            await ctx.db.patch(runDoc._id, { status: "cancelled", updatedAt: now });
+            finalRunDoc = { ...finalRunDoc, status: "cancelled" };
+            break;
+        }
+        case "run_completed": {
+            await ctx.db.patch(runDoc._id, { status: "completed", updatedAt: now });
+            finalRunDoc = { ...finalRunDoc, status: "completed" };
+            break;
+        }
+        case "run_failed": {
+            await ctx.db.patch(runDoc._id, { status: "failed", updatedAt: now });
+            finalRunDoc = { ...finalRunDoc, status: "failed" };
+            break;
+        }
+        case "run_created":
+            break;
+        default:
+            break;
+    }
+    return {
+        event: {
+            eventId,
+            seq,
+            runId,
+            eventType: args.eventType,
+            stepId: args.stepId,
+            hookId: args.hookId,
+            correlationId: args.correlationId,
+            dataJson: args.dataJson,
+            createdAt: now,
+        },
+        run: {
+            runId: finalRunDoc.runId,
+            status: finalRunDoc.status,
+            eventCount: finalRunDoc.eventCount,
+            createdAt: finalRunDoc.createdAt,
+            updatedAt: finalRunDoc.updatedAt,
+        },
+        step: stepDoc
+            ? {
+                runId: stepDoc.runId,
+                stepId: stepDoc.stepId,
+                status: stepDoc.status,
+                dataJson: stepDoc.dataJson,
+                createdAt: stepDoc.createdAt,
+                updatedAt: stepDoc.updatedAt,
+            }
+            : null,
+        hook: hookDoc
+            ? {
+                runId: hookDoc.runId,
+                hookId: hookDoc.hookId,
+                token: hookDoc.token,
+                status: hookDoc.status,
+                createdAt: hookDoc.createdAt,
+                updatedAt: hookDoc.updatedAt,
+            }
+            : null,
+    };
+}
+export const appendEvent = mutation({
+    args: {
+        orgId: v.string(),
+        runId: v.union(v.string(), v.null()),
+        eventType: v.string(),
+        stepId: v.optional(v.string()),
+        hookId: v.optional(v.string()),
+        correlationId: v.optional(v.string()),
+        dataJson: v.string(),
+    },
+    returns: v.object({
+        event: eventResult,
+        run: v.union(runResult, v.null()),
+        step: v.union(stepResult, v.null()),
+        hook: v.union(hookResult, v.null()),
+    }),
+    handler: async (ctx, args) => {
+        if (args.correlationId?.startsWith(RUN_KIND_PREFIX) ||
+            args.correlationId?.startsWith(LEGACY_KIND_PREFIX)) {
+            throw new Error("Validation: invalid correlationId");
+        }
+        if (args.eventType === RUN_WORKFLOW_BOUND) {
+            throw new Error("Validation: invalid eventType");
+        }
+        return appendEventCore(ctx, args);
+    },
+});
+export const RUN_KIND_PREFIX = "run:";
+export const LEGACY_KIND_PREFIX = "legacy:";
+const LEGACY_KIND_NAME = {
+    session: "sessions",
+    job: "durableJobs",
+};
+function runKindPrefix(kind) {
+    return `${RUN_KIND_PREFIX}${kind}:`;
+}
+function legacyKindPrefix(kind) {
+    return `${LEGACY_KIND_PREFIX}${LEGACY_KIND_NAME[kind]}:`;
+}
+export function legacyCorrelationId(kind, legacyId) {
+    return `${legacyKindPrefix(kind)}${legacyId}`;
+}
+export async function findRunIdByCorrelation(ctx, orgId, correlationId) {
+    const events = await ctx.db
+        .query("worldEvents")
+        .withIndex("by_org_correlation", (q) => q.eq("orgId", orgId).eq("correlationId", correlationId))
+        .collect();
+    return events.find((e) => e.eventType === "run_created")?.runId ?? null;
+}
+export async function openRun(ctx, args) {
+    const result = await appendEventCore(ctx, {
+        orgId: args.orgId,
+        runId: null,
+        eventType: "run_created",
+        correlationId: `${runKindPrefix(args.kind)}${crypto.randomUUID()}`,
+        dataJson: JSON.stringify(args.data),
+    });
+    return result.event.runId;
+}
+export const RUN_WORKFLOW_BOUND = "run_workflow_bound";
+function workflowCorrelationId(runId) {
+    return `workflow:${runId}`;
+}
+export async function bindWorkflowToRun(ctx, args) {
+    await appendEventCore(ctx, {
+        orgId: args.orgId,
+        runId: args.runId,
+        eventType: RUN_WORKFLOW_BOUND,
+        correlationId: workflowCorrelationId(args.runId),
+        dataJson: JSON.stringify({
+            workflowId: args.workflowId,
+            workflowVersion: args.workflowVersion,
+        }),
+    });
+}
+export async function workflowOfRun(ctx, orgId, runId) {
+    const events = await ctx.db
+        .query("worldEvents")
+        .withIndex("by_org_correlation", (q) => q.eq("orgId", orgId).eq("correlationId", workflowCorrelationId(runId)))
+        .collect();
+    let latest = null;
+    for (const e of events) {
+        if (e.eventType !== RUN_WORKFLOW_BOUND || e.runId !== runId)
+            continue;
+        if (latest === null || e.seq > latest.seq)
+            latest = e;
+    }
+    if (latest === null)
+        return null;
+    try {
+        const data = JSON.parse(latest.dataJson);
+        if (typeof data.workflowId !== "string" || typeof data.workflowVersion !== "string") {
+            return null;
+        }
+        return { workflowId: data.workflowId, workflowVersion: data.workflowVersion };
+    }
+    catch {
+        return null;
+    }
+}
+export async function loadRun(ctx, orgId, runId, kind, notFoundMessage) {
+    const run = await ctx.db
+        .query("worldRuns")
+        .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", runId))
+        .unique();
+    assertSameOrg(orgId, run, notFoundMessage);
+    const first = await ctx.db
+        .query("worldEvents")
+        .withIndex("by_org_run_seq", (q) => q.eq("orgId", orgId).eq("runId", runId).eq("seq", 1))
+        .unique();
+    if (!first ||
+        first.eventType !== "run_created" ||
+        first.correlationId === undefined ||
+        !(first.correlationId.startsWith(runKindPrefix(kind)) ||
+            first.correlationId.startsWith(legacyKindPrefix(kind)))) {
+        throw new Error(notFoundMessage);
+    }
+    let created = {};
+    try {
+        const parsed = JSON.parse(first.dataJson);
+        if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) {
+            created = parsed;
+        }
+    }
+    catch {
+        created = {};
+    }
+    return {
+        runId: run.runId,
+        status: run.status,
+        eventCount: run.eventCount,
+        created,
+    };
+}
+export function agentIdOfRun(ctx, run) {
+    const raw = run.created.agentId;
+    const id = typeof raw === "string" ? ctx.db.normalizeId("agents", raw) : null;
+    if (id === null)
+        throw new Error("Agent not found");
+    return id;
+}
+export function payloadOrUndefined(dataJson) {
+    return dataJson === "{}" ? undefined : dataJson;
+}
+export { slotToEventId };
+export const getRun = query({
+    args: { orgId: v.string(), runId: v.string() },
+    returns: v.union(runResult, v.null()),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const doc = await ctx.db
+            .query("worldRuns")
+            .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", args.runId))
+            .unique();
+        if (!doc || doc.orgId !== orgId) {
+            return null;
+        }
+        return {
+            runId: doc.runId,
+            status: doc.status,
+            eventCount: doc.eventCount,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        };
+    },
+});
+export const listRuns = query({
+    args: { orgId: v.string() },
+    returns: v.array(runResult),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldRuns")
+            .withIndex("by_org", (q) => q.eq("orgId", orgId))
+            .collect();
+        return docs.map((doc) => ({
+            runId: doc.runId,
+            status: doc.status,
+            eventCount: doc.eventCount,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        }));
+    },
+});
+export const getStep = query({
+    args: { orgId: v.string(), runId: v.string(), stepId: v.string() },
+    returns: v.union(stepResult, v.null()),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const doc = await ctx.db
+            .query("worldSteps")
+            .withIndex("by_org_run_step", (q) => q.eq("orgId", orgId).eq("runId", args.runId).eq("stepId", args.stepId))
+            .unique();
+        if (!doc || doc.orgId !== orgId) {
+            return null;
+        }
+        return {
+            runId: doc.runId,
+            stepId: doc.stepId,
+            status: doc.status,
+            dataJson: doc.dataJson,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        };
+    },
+});
+export const listSteps = query({
+    args: { orgId: v.string(), runId: v.string() },
+    returns: v.array(stepResult),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldSteps")
+            .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", args.runId))
+            .collect();
+        return docs.map((doc) => ({
+            runId: doc.runId,
+            stepId: doc.stepId,
+            status: doc.status,
+            dataJson: doc.dataJson,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        }));
+    },
+});
+export const getEvent = query({
+    args: { orgId: v.string(), runId: v.string(), eventId: v.string() },
+    returns: v.union(eventResult, v.null()),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldEvents")
+            .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", args.runId))
+            .collect();
+        const doc = docs.find((d) => slotToEventId(d.seq) === args.eventId);
+        if (!doc || doc.orgId !== orgId) {
+            return null;
+        }
+        return {
+            eventId: slotToEventId(doc.seq),
+            seq: doc.seq,
+            runId: doc.runId,
+            eventType: doc.eventType,
+            stepId: doc.stepId,
+            hookId: doc.hookId,
+            correlationId: doc.correlationId,
+            dataJson: doc.dataJson,
+            createdAt: doc.createdAt,
+        };
+    },
+});
+export const listEvents = query({
+    args: { orgId: v.string(), runId: v.string() },
+    returns: v.array(eventResult),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldEvents")
+            .withIndex("by_org_run_seq", (q) => q.eq("orgId", orgId).eq("runId", args.runId))
+            .collect();
+        return docs.map((doc) => ({
+            eventId: slotToEventId(doc.seq),
+            seq: doc.seq,
+            runId: doc.runId,
+            eventType: doc.eventType,
+            stepId: doc.stepId,
+            hookId: doc.hookId,
+            correlationId: doc.correlationId,
+            dataJson: doc.dataJson,
+            createdAt: doc.createdAt,
+        }));
+    },
+});
+export const listEventsByCorrelationId = query({
+    args: { orgId: v.string(), correlationId: v.string() },
+    returns: v.array(eventResult),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldEvents")
+            .withIndex("by_org_correlation", (q) => q.eq("orgId", orgId).eq("correlationId", args.correlationId))
+            .collect();
+        return docs.map((doc) => ({
+            eventId: slotToEventId(doc.seq),
+            seq: doc.seq,
+            runId: doc.runId,
+            eventType: doc.eventType,
+            stepId: doc.stepId,
+            hookId: doc.hookId,
+            correlationId: doc.correlationId,
+            dataJson: doc.dataJson,
+            createdAt: doc.createdAt,
+        }));
+    },
+});
+export const getHook = query({
+    args: { orgId: v.string(), hookId: v.string() },
+    returns: v.union(hookResult, v.null()),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const doc = await ctx.db
+            .query("worldHooks")
+            .withIndex("by_org_hook", (q) => q.eq("orgId", orgId).eq("hookId", args.hookId))
+            .unique();
+        if (!doc || doc.orgId !== orgId) {
+            return null;
+        }
+        return {
+            runId: doc.runId,
+            hookId: doc.hookId,
+            token: doc.token,
+            status: doc.status,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        };
+    },
+});
+export const getHookByToken = query({
+    args: { orgId: v.string(), token: v.string() },
+    returns: v.union(hookResult, v.null()),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const doc = await ctx.db
+            .query("worldHooks")
+            .withIndex("by_org_token", (q) => q.eq("orgId", orgId).eq("token", args.token))
+            .unique();
+        if (!doc || doc.orgId !== orgId) {
+            return null;
+        }
+        return {
+            runId: doc.runId,
+            hookId: doc.hookId,
+            token: doc.token,
+            status: doc.status,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        };
+    },
+});
+export const listHooks = query({
+    args: { orgId: v.string(), runId: v.string() },
+    returns: v.array(hookResult),
+    handler: async (ctx, args) => {
+        const orgId = requireOrgId(args.orgId);
+        const docs = await ctx.db
+            .query("worldHooks")
+            .withIndex("by_org_run", (q) => q.eq("orgId", orgId).eq("runId", args.runId))
+            .collect();
+        return docs.map((doc) => ({
+            runId: doc.runId,
+            hookId: doc.hookId,
+            token: doc.token,
+            status: doc.status,
+            createdAt: doc.createdAt,
+            updatedAt: doc.updatedAt,
+        }));
+    },
+});