Add agent workflows & stuff

2026-06-21 21:15:15 -05:00
parent cf7ff2ee4e
commit 2dfa97ee4f
102 changed files with 8488 additions and 161 deletions
@@ -0,0 +1,97 @@
+import { v } from 'convex/values';
+
+import { internalMutation, query } from './_generated/server';
+import { getOwnedSpoon, getRequiredUserId } from './model';
+
+const scope = v.union(
+  v.literal('fork'),
+  v.literal('upstream'),
+  v.literal('from_fork_to_upstream'),
+);
+
+const state = v.union(
+  v.literal('open'),
+  v.literal('closed'),
+  v.literal('merged'),
+);
+
+export const listForSpoon = query({
+  args: {
+    spoonId: v.id('spoons'),
+    scope: v.optional(scope),
+    state: v.optional(state),
+    limit: v.optional(v.number()),
+  },
+  handler: async (ctx, args) => {
+    const ownerId = await getRequiredUserId(ctx);
+    await getOwnedSpoon(ctx, args.spoonId, ownerId);
+    const requestedScope = args.scope;
+    const rows = requestedScope
+      ? await ctx.db
+          .query('spoonPullRequests')
+          .withIndex('by_spoon_scope', (q) =>
+            q.eq('spoonId', args.spoonId).eq('scope', requestedScope),
+          )
+          .order('desc')
+          .collect()
+      : await ctx.db
+          .query('spoonPullRequests')
+          .withIndex('by_spoon', (q) => q.eq('spoonId', args.spoonId))
+          .order('desc')
+          .collect();
+    return rows
+      .filter((row) => !args.state || row.state === args.state)
+      .slice(0, args.limit ?? 100);
+  },
+});
+
+export const replaceForSpoon = internalMutation({
+  args: {
+    spoonId: v.id('spoons'),
+    ownerId: v.id('users'),
+    scope,
+    pullRequests: v.array(
+      v.object({
+        githubId: v.number(),
+        number: v.number(),
+        repoFullName: v.string(),
+        title: v.string(),
+        state,
+        draft: v.boolean(),
+        authorLogin: v.optional(v.string()),
+        baseRef: v.string(),
+        headRef: v.string(),
+        headRepoFullName: v.optional(v.string()),
+        htmlUrl: v.string(),
+        createdAtGithub: v.optional(v.number()),
+        updatedAtGithub: v.optional(v.number()),
+        mergedAtGithub: v.optional(v.number()),
+      }),
+    ),
+  },
+  handler: async (ctx, { spoonId, ownerId, scope, pullRequests }) => {
+    const existing = await ctx.db
+      .query('spoonPullRequests')
+      .withIndex('by_spoon_scope', (q) =>
+        q.eq('spoonId', spoonId).eq('scope', scope),
+      )
+      .collect();
+    await Promise.all(
+      existing.map((pullRequest) => ctx.db.delete(pullRequest._id)),
+    );
+    const now = Date.now();
+    await Promise.all(
+      pullRequests.map((pullRequest) =>
+        ctx.db.insert('spoonPullRequests', {
+          spoonId,
+          ownerId,
+          scope,
+          ...pullRequest,
+          createdAt: now,
+          updatedAt: now,
+        }),
+      ),
+    );
+    return { success: true };
+  },
+});