@aglyn/plugins-ai 1.0.0-beta.266 → 1.0.0-beta.268

package/src/lib/jobs/ai-jobs.js

@@ -47,6 +47,7 @@ import { AI_SITE_GUIDED_BUILD_FAILED_COPY, aiCustomerSafeCopy, aiCustomerSafeIte
 import { aiSiteStartInputsOf } from "../model/ai-site-start.js";
 import { addAdminAudit } from "@aglyn/tenant-data-admin/server/admin-audit-write";
 import { AI_JOB_CANCELED_ITEM_NOTE, aiJobCancelPending, isAiJobCanceledError, watchAiJobCancel } from "./ai-job-cancel.js";
+import { clearAiEmptySiteSweep, scheduleAiEmptySiteSweep } from "./ai-empty-site-sweep.js";
 /**
  * AI generation jobs — the Firestore state machine (AGL-2904).
  *
@@ -256,6 +257,16 @@ let transitionListener = null;
 /** Whether a transition listener is registered: what a surface that runs jobs must have. */ export function aiJobTransitionListenerRegistered() {
     return transitionListener !== null;
 }
+/**
+ * A job just ENDED `canceled` or `failed`, in the write the caller made
+ * (AGL-3936): a site start that built nothing schedules the removal of the
+ * empty site it was for, an hour out (`ai-empty-site-sweep.ts`). Every write
+ * that ends a job calls this once, after its transaction commits. Never
+ * throws.
+ */ async function noteAiJobEnded(firestore, job, now) {
+    if (!job || job.status !== 'canceled' && job.status !== 'failed') return;
+    await scheduleAiEmptySiteSweep(firestore, job, now);
+}
 /** Tells the listener; a listener that throws never fails the write it follows. */ async function announceAiJobTransition(job, to, failure) {
     if (!transitionListener) return;
     try {
@@ -408,6 +419,8 @@ function jobFrom(snapshot) {
         creditsConfirmed: input.creditsConfirmed
     } : {});
     await ref.set(data);
+    // A new job on the site — a guided start begun again — keeps it (AGL-3936).
+    await clearAiEmptySiteSweep(firestore, input.hostId);
     await logAiJobCreated(input.orgId, {
         uid: input.createdBy,
         email: (_input_createdByEmail = input.createdByEmail) != null ? _input_createdByEmail : null
@@ -527,6 +540,8 @@ export async function getAiJob(firestore, orgId, jobId) {
         publishAsked: true
     } : {}, aiJobCancelPending(job) ? {
         cancelRequested: true
+    } : {}, job.siteRemoved ? {
+        siteRemoved: true
     } : {});
 }
 function leaseIsLive(job, owner, nowMs) {
@@ -607,16 +622,20 @@ function nextStepIndex(job) {
  * stale lease is released by being taken, not by a sweep of its own.
  */ export async function claimNextStep(firestore, orgId, jobId, owner, now = new Date()) {
     const ref = jobsCollection(firestore, orgId).doc(jobId);
-    return firestore.runTransaction(async (tx)=>{
+    let ended = null;
+    const claimed = await firestore.runTransaction(async (tx)=>{
+        ended = null;
         const job = jobFrom(await tx.get(ref));
         if (!job) return null;
         // A cancel asked while a step ran, whose runner never came back to end
         // it (AGL-3616): the lease ran out, so the job ends here, unclaimed.
         if (aiJobCancelPending(job)) {
             if (!leaseIsLive(job, owner, now.getTime())) {
-                tx.set(ref, aiJobCanceledPatch(job, now, cancelRequestOf(job, now)), {
+                const patch = aiJobCanceledPatch(job, now, cancelRequestOf(job, now));
+                tx.set(ref, patch, {
                     merge: true
                 });
+                ended = _extends({}, job, patch);
             }
             return null;
         }
@@ -652,6 +671,8 @@ function nextStepIndex(job) {
             stepIndex
         };
     });
+    await noteAiJobEnded(firestore, ended, now);
+    return claimed;
 }
 /**
  * Extend the lease while a long step is still running. `false` when the
@@ -689,15 +710,19 @@ function nextStepIndex(job) {
  * parked, no step left, or another owner's live lease.
  */ export async function holdPausedAiJob(firestore, orgId, jobId, owner, now = new Date()) {
     const ref = jobsCollection(firestore, orgId).doc(jobId);
-    return firestore.runTransaction(async (tx)=>{
+    let ended = null;
+    const held = await firestore.runTransaction(async (tx)=>{
+        ended = null;
         const job = jobFrom(await tx.get(ref));
         if (!job) return null;
         // A paused workspace's job a person canceled ends canceled (AGL-3616).
         if (aiJobCancelPending(job)) {
             if (!leaseIsLive(job, owner, now.getTime())) {
-                tx.set(ref, aiJobCanceledPatch(job, now, cancelRequestOf(job, now)), {
+                const patch = aiJobCanceledPatch(job, now, cancelRequestOf(job, now));
+                tx.set(ref, patch, {
                     merge: true
                 });
+                ended = _extends({}, job, patch);
             }
             return null;
         }
@@ -711,6 +736,8 @@ function nextStepIndex(job) {
             updatedAt: now
         });
     });
+    await noteAiJobEnded(firestore, ended, now);
+    return held;
 }
 /**
  * A build's ledger with one pass applied (AGL-3616): the pass's credits on
@@ -874,9 +901,12 @@ function nextStepIndex(job) {
             job: _extends({}, job, patch),
             remaining,
             // Entered `needs_review` in this write, rather than parked again.
-            parked: parksForReview && job.status !== 'needs_review'
+            parked: parksForReview && job.status !== 'needs_review',
+            // Ended `canceled` in this write (AGL-3936).
+            ended: canceling && !isAiJobTerminal(job.status)
         };
     });
+    if (recorded.ended) await noteAiJobEnded(firestore, recorded.job, now);
     if (recorded.parked && !input.quiet) await announceAiJobTransition(recorded.job, 'needs-review');
     return {
         job: recorded.job,
@@ -912,6 +942,7 @@ async function transition(firestore, orgId, jobId, apply) {
             updatedAt: now
         });
     if (changed && job.status === 'done') await announceAiJobTransition(job, 'done');
+    if (changed) await noteAiJobEnded(firestore, job, now);
     return job;
 }
 /**
@@ -946,6 +977,7 @@ async function transition(firestore, orgId, jobId, apply) {
             error: (detail == null ? void 0 : detail.error) instanceof Error ? detail.error.message : (detail == null ? void 0 : detail.error) == null ? null : String(detail.error)
         });
     }
+    if (changed) await noteAiJobEnded(firestore, job, now);
     return job;
 }
 /**
@@ -953,7 +985,7 @@ async function transition(firestore, orgId, jobId, apply) {
  * step goes back to `pending` so the same step is what runs when the beat
  * tries again; its attempt is not counted, because nothing was attempted.
  */ export async function markAiJobNeedsInput(firestore, orgId, jobId, reason, now = new Date()) {
-    const { job } = await transition(firestore, orgId, jobId, (current)=>isAiJobTerminal(current.status) ? null : aiJobCancelPending(current) ? aiJobCanceledPatch(current, now, cancelRequestOf(current, now)) : {
+    const { job, changed } = await transition(firestore, orgId, jobId, (current)=>isAiJobTerminal(current.status) ? null : aiJobCancelPending(current) ? aiJobCanceledPatch(current, now, cancelRequestOf(current, now)) : {
             status: 'needs_input',
             error: reason,
             lease: null,
@@ -966,6 +998,7 @@ async function transition(firestore, orgId, jobId, apply) {
             }),
             updatedAt: now
         });
+    if (changed) await noteAiJobEnded(firestore, job, now);
     return job;
 }
 /**
@@ -1016,6 +1049,7 @@ async function transition(firestore, orgId, jobId, apply) {
             kind: result.job.kind
         });
     }
+    if (result.changed) await noteAiJobEnded(firestore, result.job, now);
     return result;
 }
 /**
@@ -1372,6 +1406,8 @@ async function transition(firestore, orgId, jobId, apply) {
             updatedAt: now
         });
     });
+    // A site start tried again keeps its site (AGL-3936).
+    if (result.changed) await clearAiEmptySiteSweep(firestore, result.job.hostId);
     return _extends({}, result, {
         retried: result.changed ? retried : []
     });
@@ -1388,7 +1424,7 @@ async function transition(firestore, orgId, jobId, apply) {
  * and a confirmed plan is held at what the whole job is estimated to cost
  * (AGL-3031), creations included, where that is more.
  */ export async function resumeAiJob(firestore, orgId, jobId, actor, now = new Date(), options = {}) {
-    return transition(firestore, orgId, jobId, (current)=>{
+    const resumed = await transition(firestore, orgId, jobId, (current)=>{
         var _options_reducePlan;
         var _current_review, _current_inputs;
         // A job the meter paused (AGL-3660) carries on from the step it paused
@@ -1435,6 +1471,9 @@ async function transition(firestore, orgId, jobId, apply) {
             updatedAt: now
         });
     });
+    // A site start resumed keeps its site (AGL-3936).
+    if (resumed.changed) await clearAiEmptySiteSweep(firestore, resumed.job.hostId);
+    return resumed;
 }
 /**
  * Record what a person applied from a finished job's outputs (AGL-2910): the