Set ghost agent as the current actor for seeds

Author: CarsonF

dbschema/seeds/000.system-agents.ts

@@ -6,7 +6,7 @@ const agents = [
   { name: 'External Mailing Group', roles: ['Leadership'] },
 ];
 
-export default (async function ({ runAndPrint }) {
+export default (async function ({ runAndPrint, e, db, actorId }) {
   await runAndPrint(
     `
       with
@@ -24,4 +24,12 @@ export default (async function ({ runAndPrint }) {
     `,
     { agentsJson: agents },
   );
+
+  const getGhost = e.assert_exists(
+    e.select(e.SystemAgent, () => ({
+      filter_single: { name: 'Ghost' },
+    })),
+  );
+  const ghost = await getGhost.run(db);
+  actorId.next(ghost.id);
 } satisfies SeedFn);

src/core/edgedb/edgedb.module.ts

@@ -40,7 +40,7 @@ import { TransactionContext } from './transaction.context';
           ...connectConfig,
         });
 
-        Object.assign(client, { options: options.currentAsLazyRef });
+        options.attachToClient(client);
 
         if (config.databaseEngine === 'edgedb') {
           await registerCustomScalarCodecs(client, codecs);

src/core/edgedb/options.context.ts

@@ -1,5 +1,6 @@
 import { Injectable, OnModuleDestroy } from '@nestjs/common';
 import { AsyncLocalStorage } from 'async_hooks';
+import type { Client } from 'edgedb';
 import {
   BehaviorSubject,
   combineLatest,
@@ -90,6 +91,9 @@ export class OptionsContext
   get currentAsLazyRef() {
     return lazyRef(() => this.current);
   }
+  attachToClient(client: Client) {
+    Object.assign(client, { options: this.currentAsLazyRef });
+  }
 
   onModuleDestroy() {
     this.disable();

src/core/edgedb/seeds.run.ts

@@ -1,9 +1,14 @@
 import { many, MaybeAsync } from '@seedcompany/common';
 import { BaseContext, Command, runExit } from 'clipanion';
 import { Client, createClient, Executor } from 'edgedb';
+import type { QueryArgs } from 'edgedb/dist/ifaces';
 import { glob } from 'glob';
 import fs from 'node:fs/promises';
+import { BehaviorSubject } from 'rxjs';
 import { inspect } from 'util';
+import { ID } from '~/common';
+import { Options } from './options';
+import { OptionsContext, OptionsFn } from './options.context';
 import { e } from './reexports';
 
 type Query = string | QBQuery;
@@ -17,17 +22,30 @@ export type SeedFn = (
 
 interface SeedParams {
   e: typeof e;
-  runAndPrint: (query: Query) => Promise<void>;
+  runAndPrint: (query: Query, args?: QueryArgs) => Promise<void>;
   db: Client;
   print: (something: unknown) => void;
   context: BaseContext;
+  actorId: BehaviorSubject<ID | undefined>;
 }
 
-const db = createClient().withConfig({
-  // eslint-disable-next-line @typescript-eslint/naming-convention
-  allow_user_specified_id: true,
-  // eslint-disable-next-line @typescript-eslint/naming-convention
-  apply_access_policies: false,
+const optionsContext = new OptionsContext(
+  new Options().withConfig({
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    allow_user_specified_id: true,
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    apply_access_policies: false,
+  }),
+);
+const db = createClient();
+optionsContext.attachToClient(db);
+
+const actor = new BehaviorSubject<ID | undefined>(undefined);
+const actorOptions = new BehaviorSubject<OptionsFn>((opts) => opts);
+actor.subscribe((actor) => {
+  actorOptions.next((options) =>
+    actor ? options.withGlobals({ currentActorId: actor }) : options,
+  );
 });
 
 class SeedCommand extends Command {
@@ -43,11 +61,11 @@ class SeedCommand extends Command {
       }
     };
 
-    const runAndPrint = async (query: Query) => {
+    const runAndPrint = async (query: Query, args?: QueryArgs) => {
       try {
         const rows =
           typeof query === 'string'
-            ? await db.query(query)
+            ? await db.query(query, args)
             : await query.run(db);
         printResult(rows);
       } catch (e) {
@@ -61,6 +79,7 @@ class SeedCommand extends Command {
       print: printResult,
       runAndPrint,
       e,
+      actorId: actor,
     };
 
     for (const file of files) {
@@ -89,7 +108,9 @@ class SeedCommand extends Command {
 
   async validateAndExecute() {
     try {
-      return await super.validateAndExecute();
+      return await optionsContext.usingOptions(actorOptions, async () => {
+        return await super.validateAndExecute();
+      });
     } finally {
       await db.close();
     }