@samebase/alchemy-convex 0.10.0 → 0.12.0

package/src/store/convex/state.ts

@@ -2,17 +2,21 @@
 // Convex team, pushed by Convex.state() from this directory. These raw state
 // functions are internal. reader.ts exposes authenticated, redacted reads.
 //
-// A stage holds one row for each resource of its stack, so the reads of one
-// stage collect all rows. Samebase reads `stacks`, `getOutput`, and
-// `readStack` by name: a renamed function is a version bump of the store and
-// of Samebase.
+// The rows are in the alchemyState component (src/store/component). Each
+// function here forwards to the component function of the same name, with the
+// same results; an absent `run` or `stage` goes to the component as null.
+// These names and arguments are the wire contract of STORE_VERSION. Samebase
+// reads `stacks`, `getOutput`, and `readStack` by name: a renamed function is
+// a version bump of the store and of Samebase.
 import { v } from "convex/values";
-import type { MutationCtx, QueryCtx } from "./_generated/server.js";
+import { runRecord } from "../component/runRecord.ts";
+import { stackReply } from "../component/state.ts";
+import { components } from "./_generated/api.js";
 import { internalMutation, internalQuery } from "./_generated/server.js";
-import { runRecord } from "./runRecord.ts";
-import type { RunRecord } from "../../RunRecord.ts";
 import { STORE_VERSION } from "./version.ts";
 
+const state = components.alchemyState.state;
+
 /** The wire version of this store, compared with the client's STORE_VERSION. */
 export const version = internalQuery({
   args: {},
@@ -20,272 +24,100 @@ export const version = internalQuery({
   handler: async () => STORE_VERSION,
 });
 
-const stage = { stack: v.string(), stage: v.string() };
-const resource = { ...stage, fqn: v.string() };
-
-const rowOf = (ctx: QueryCtx, args: { stack: string; stage: string; fqn: string }) =>
-  ctx.db
-    .query("resources")
-    .withIndex("by_fqn", (q) =>
-      q.eq("stack", args.stack).eq("stage", args.stage).eq("fqn", args.fqn),
-    )
-    .unique();
-
-const rowsOf = (ctx: QueryCtx, args: { stack: string; stage: string }) =>
-  ctx.db
-    .query("resources")
-    .withIndex("by_fqn", (q) => q.eq("stack", args.stack).eq("stage", args.stage))
-    .collect();
-
-const outputOf = (ctx: QueryCtx, args: { stack: string; stage: string }) =>
-  ctx.db
-    .query("outputs")
-    .withIndex("by_stage", (q) => q.eq("stack", args.stack).eq("stage", args.stage))
-    .unique();
-
-/** A resource write and its current run become visible in one transaction. */
-const writeCurrentRun = async (
-  ctx: MutationCtx,
-  args: { stack: string; stage: string; run?: RunRecord | undefined },
-) => {
-  const output = await outputOf(ctx, args);
-  // A write without a run record clears the record of the previous run.
-  if (output !== null) await ctx.db.patch("outputs", output._id, { currentRun: args.run });
-  else if (args.run !== undefined)
-    await ctx.db.insert("outputs", { stack: args.stack, stage: args.stage, currentRun: args.run });
-};
-
-/** The sorted union of two lists of names. */
-const union = (a: readonly string[], b: readonly string[]) => [...new Set([...a, ...b])].sort();
-
-/**
- * The distinct names of one table in index order, with one index read for
- * each name: `first(after)` is the first row whose name is above `after`.
- */
-const distinct = async (first: (after: string | undefined) => Promise<string | null>) => {
-  const names: string[] = [];
-  for (;;) {
-    const next = await first(names.at(-1));
-    if (next === null) return names;
-    names.push(next);
-  }
-};
-
 /** Stacks with at least one resource or one output, sorted. */
 export const stacks = internalQuery({
   args: {},
   returns: v.array(v.string()),
-  handler: async (ctx) =>
-    union(
-      await distinct(
-        async (after) =>
-          (
-            await ctx.db
-              .query("resources")
-              .withIndex("by_fqn", (q) => (after === undefined ? q : q.gt("stack", after)))
-              .first()
-          )?.stack ?? null,
-      ),
-      await distinct(
-        async (after) =>
-          (
-            await ctx.db
-              .query("outputs")
-              .withIndex("by_stage", (q) => (after === undefined ? q : q.gt("stack", after)))
-              .first()
-          )?.stack ?? null,
-      ),
-    ),
+  handler: (ctx) => ctx.runQuery(state.stacks, {}),
 });
 
 /** Stages of a stack with at least one resource or one output, sorted. */
 export const stages = internalQuery({
   args: { stack: v.string() },
   returns: v.array(v.string()),
-  handler: async (ctx, args) =>
-    union(
-      await distinct(
-        async (after) =>
-          (
-            await ctx.db
-              .query("resources")
-              .withIndex("by_fqn", (q) =>
-                after === undefined
-                  ? q.eq("stack", args.stack)
-                  : q.eq("stack", args.stack).gt("stage", after),
-              )
-              .first()
-          )?.stage ?? null,
-      ),
-      await distinct(
-        async (after) =>
-          (
-            await ctx.db
-              .query("outputs")
-              .withIndex("by_stage", (q) =>
-                after === undefined
-                  ? q.eq("stack", args.stack)
-                  : q.eq("stack", args.stack).gt("stage", after),
-              )
-              .first()
-          )?.stage ?? null,
-      ),
-    ),
+  handler: (ctx, args) => ctx.runQuery(state.stages, { stack: args.stack }),
 });
 
 export const list = internalQuery({
-  args: stage,
+  args: { stack: v.string(), stage: v.string() },
   returns: v.array(v.string()),
-  handler: async (ctx, args) => (await rowsOf(ctx, args)).map((row) => row.fqn),
+  handler: (ctx, args) => ctx.runQuery(state.list, { stack: args.stack, stage: args.stage }),
 });
 
 export const get = internalQuery({
-  args: resource,
+  args: { stack: v.string(), stage: v.string(), fqn: v.string() },
   returns: v.union(v.string(), v.null()),
-  handler: async (ctx, args) => (await rowOf(ctx, args))?.value ?? null,
+  handler: (ctx, args) =>
+    ctx.runQuery(state.get, { stack: args.stack, stage: args.stage, fqn: args.fqn }),
 });
 
 /** Resources whose state has `status: "replaced"`. */
 export const replaced = internalQuery({
-  args: stage,
+  args: { stack: v.string(), stage: v.string() },
   returns: v.array(v.string()),
-  handler: async (ctx, args) =>
-    (await rowsOf(ctx, args))
-      .map((row) => row.value)
-      .filter((value) => {
-        const parsed: unknown = JSON.parse(value);
-        return (
-          typeof parsed === "object" &&
-          parsed !== null &&
-          "status" in parsed &&
-          parsed.status === "replaced"
-        );
-      }),
+  handler: (ctx, args) => ctx.runQuery(state.replaced, { stack: args.stack, stage: args.stage }),
 });
 
 export const set = internalMutation({
-  args: { ...resource, value: v.string(), run: v.optional(runRecord) },
-  returns: v.null(),
-  handler: async (ctx, args) => {
-    const row = await rowOf(ctx, args);
-    if (row === null)
-      await ctx.db.insert("resources", {
-        stack: args.stack,
-        stage: args.stage,
-        fqn: args.fqn,
-        value: args.value,
-      });
-    else await ctx.db.patch("resources", row._id, { value: args.value });
-    await writeCurrentRun(ctx, args);
-    return null;
+  args: {
+    stack: v.string(),
+    stage: v.string(),
+    fqn: v.string(),
+    value: v.string(),
+    run: v.optional(runRecord),
   },
+  returns: v.null(),
+  handler: (ctx, args) =>
+    ctx.runMutation(state.set, {
+      stack: args.stack,
+      stage: args.stage,
+      fqn: args.fqn,
+      value: args.value,
+      run: args.run ?? null,
+    }),
 });
 
 export const remove = internalMutation({
-  args: { ...resource, run: v.optional(runRecord) },
+  args: { stack: v.string(), stage: v.string(), fqn: v.string(), run: v.optional(runRecord) },
   returns: v.null(),
-  handler: async (ctx, args) => {
-    const row = await rowOf(ctx, args);
-    if (row !== null) await ctx.db.delete("resources", row._id);
-    await writeCurrentRun(ctx, args);
-    return null;
-  },
+  handler: (ctx, args) =>
+    ctx.runMutation(state.remove, {
+      stack: args.stack,
+      stage: args.stage,
+      fqn: args.fqn,
+      run: args.run ?? null,
+    }),
 });
 
 /** Deletes the resources and outputs of one stage, or of every stage of the stack. */
 export const removeStack = internalMutation({
   args: { stack: v.string(), stage: v.optional(v.string()) },
   returns: v.null(),
-  handler: async (ctx, args) => {
-    const { stack } = args;
-    const resources = await ctx.db
-      .query("resources")
-      .withIndex("by_fqn", (q) =>
-        args.stage === undefined
-          ? q.eq("stack", stack)
-          : q.eq("stack", stack).eq("stage", args.stage),
-      )
-      .collect();
-    const outputs = await ctx.db
-      .query("outputs")
-      .withIndex("by_stage", (q) =>
-        args.stage === undefined
-          ? q.eq("stack", stack)
-          : q.eq("stack", stack).eq("stage", args.stage),
-      )
-      .collect();
-    for (const row of resources) await ctx.db.delete("resources", row._id);
-    for (const row of outputs) await ctx.db.delete("outputs", row._id);
-    return null;
-  },
+  handler: (ctx, args) =>
+    ctx.runMutation(state.removeStack, { stack: args.stack, stage: args.stage ?? null }),
 });
 
 export const getOutput = internalQuery({
-  args: stage,
+  args: { stack: v.string(), stage: v.string() },
   returns: v.union(v.string(), v.null()),
-  handler: async (ctx, args) => (await outputOf(ctx, args))?.value ?? null,
+  handler: (ctx, args) => ctx.runQuery(state.getOutput, { stack: args.stack, stage: args.stage }),
 });
 
 export const setOutput = internalMutation({
-  args: { ...stage, value: v.string(), run: v.optional(runRecord) },
+  args: { stack: v.string(), stage: v.string(), value: v.string(), run: v.optional(runRecord) },
   returns: v.null(),
-  handler: async (ctx, args) => {
-    const row = await outputOf(ctx, args);
-    if (row === null)
-      await ctx.db.insert("outputs", { stack: args.stack, stage: args.stage, value: args.value });
-    else await ctx.db.patch("outputs", row._id, { value: args.value });
-    await writeCurrentRun(ctx, args);
-    return null;
-  },
+  handler: (ctx, args) =>
+    ctx.runMutation(state.setOutput, {
+      stack: args.stack,
+      stage: args.stage,
+      value: args.value,
+      run: args.run ?? null,
+    }),
 });
 
-/**
- * One call for a reader: every stage of a stack in name order, each with its
- * resources in fqn order and its output, as the JSON texts that the rows
- * hold. A stage with an output and no resource is listed with an empty
- * resource list; a stage with resources and no output with `output: null`.
- * One function reads at most 16 MiB (Convex limit): a stack whose rows go
- * beyond that fails here and is read with `stages`, `list`, `get`, and
- * `getOutput` instead. Since store version 2.
- */
+/** Every stage of a stack with its resources, output, and current record, in one call. Since store version 2. */
 export const readStack = internalQuery({
   args: { stack: v.string() },
-  returns: v.object({
-    stages: v.array(
-      v.object({
-        stage: v.string(),
-        resources: v.array(v.object({ fqn: v.string(), value: v.string() })),
-        output: v.union(v.string(), v.null()),
-        record: v.union(runRecord, v.null()),
-      }),
-    ),
-  }),
-  handler: async (ctx, args) => {
-    // Both indexes order by stack, then stage, then (for resources) fqn.
-    const resources = await ctx.db
-      .query("resources")
-      .withIndex("by_fqn", (q) => q.eq("stack", args.stack))
-      .collect();
-    const outputs = await ctx.db
-      .query("outputs")
-      .withIndex("by_stage", (q) => q.eq("stack", args.stack))
-      .collect();
-    const stages = union(
-      [...new Set(resources.map((row) => row.stage))],
-      [...new Set(outputs.map((row) => row.stage))],
-    );
-    return {
-      stages: stages.map((stage) => {
-        const output = outputs.find((row) => row.stage === stage);
-        return {
-          stage,
-          resources: resources
-            .filter((row) => row.stage === stage)
-            .map((row) => ({ fqn: row.fqn, value: row.value })),
-          output: output?.value ?? null,
-          record: output?.currentRun ?? null,
-        };
-      }),
-    };
-  },
+  returns: stackReply,
+  handler: (ctx, args) => ctx.runQuery(state.readStack, { stack: args.stack }),
 });

package/src/store/convex/version.ts

@@ -1,7 +1,8 @@
 // The wire version of the store: the `state:version` query answers it, and
 // Convex.state() pushes this directory when the store of the team answers an
 // older one or has no `version` yet, and fails the run on a newer one. Raise
-// it with each change of a function. The store and its readers (Samebase) are
+// it with each change of a function of this app or of the alchemyState
+// component (src/store/component). The store and its readers (Samebase) are
 // released and upgraded together: a renamed function is a version bump on
 // both sides.
 //
@@ -9,4 +10,5 @@
 //    `set`, `remove`, `removeStack`, `setOutput` (0.6.0).
 // 2: `readStack` (0.7.0).
 // 3: authenticated subscriptions and current stage run records (0.9.0).
-export const STORE_VERSION = 3;
+// 4: the store is a component (same functions).
+export const STORE_VERSION = 4;