@highstate/backend 0.9.33 → 0.9.35

package/src/business/unit-extra.test.ts

@@ -1,4 +1,5 @@
 import type { UnitPage, UnitTerminal, UnitTrigger } from "@highstate/contract"
+import { createId } from "@paralleldrive/cuid2"
 import { describe, expect } from "vitest"
 import { test } from "../test-utils"
 import { UnitExtraService } from "./unit-extra"
@@ -537,3 +538,115 @@ To login, use the following token:`,
     })
   })
 })
+
+describe("pruneInstanceArtifacts", () => {
+  test("removes artifacts not included in the keep list", async ({
+    database,
+    project,
+    createInstanceState,
+    projectDatabase,
+  }) => {
+    const service = new UnitExtraService(database)
+    const instance = await createInstanceState(project.id)
+
+    const artifactToKeep = await projectDatabase.artifact.create({
+      data: {
+        hash: createId(),
+        size: 1024,
+        chunkSize: 1024,
+        meta: { title: "keep" },
+        instances: {
+          connect: { id: instance.id },
+        },
+      },
+      select: { id: true },
+    })
+
+    const artifactToRemove = await projectDatabase.artifact.create({
+      data: {
+        hash: createId(),
+        size: 512,
+        chunkSize: 1024,
+        meta: { title: "remove" },
+        instances: {
+          connect: { id: instance.id },
+        },
+      },
+      select: { id: true },
+    })
+
+    await projectDatabase.$transaction(async tx => {
+      await service.pruneInstanceArtifacts(tx, instance.id, [artifactToKeep.id])
+    })
+
+    const remaining = await projectDatabase.artifact.findMany({
+      where: {
+        instances: {
+          some: {
+            id: instance.id,
+          },
+        },
+      },
+      select: { id: true },
+    })
+
+    expect(remaining).toHaveLength(1)
+    expect(remaining[0].id).toBe(artifactToKeep.id)
+
+    const removed = await projectDatabase.artifact.findUniqueOrThrow({
+      where: { id: artifactToRemove.id },
+      include: { instances: true },
+    })
+    expect(removed.instances).toHaveLength(0)
+  })
+
+  test("removes all artifacts when keep list is empty", async ({
+    database,
+    project,
+    createInstanceState,
+    projectDatabase,
+  }) => {
+    const service = new UnitExtraService(database)
+    const instance = await createInstanceState(project.id)
+
+    await projectDatabase.artifact.create({
+      data: {
+        hash: createId(),
+        size: 128,
+        chunkSize: 1024,
+        meta: { title: "artifact-1" },
+        instances: {
+          connect: { id: instance.id },
+        },
+      },
+    })
+
+    await projectDatabase.artifact.create({
+      data: {
+        hash: createId(),
+        size: 256,
+        chunkSize: 1024,
+        meta: { title: "artifact-2" },
+        instances: {
+          connect: { id: instance.id },
+        },
+      },
+    })
+
+    await projectDatabase.$transaction(async tx => {
+      await service.pruneInstanceArtifacts(tx, instance.id, [])
+    })
+
+    const remaining = await projectDatabase.artifact.findMany({
+      where: {
+        instances: {
+          some: {
+            id: instance.id,
+          },
+        },
+      },
+    })
+
+    expect(remaining).toHaveLength(0)
+  })
+})

package/src/business/unit-extra.ts

@@ -146,6 +146,46 @@ export class UnitExtraService {
     return triggerIds
   }
 
+  /**
+   * Disconnects artifacts that are no longer referenced by the instance.
+   *
+   * @param tx The database transaction to use.
+   * @param stateId The ID of the instance state.
+   * @param artifactIds The artifact IDs that should remain connected to the instance.
+   */
+  async pruneInstanceArtifacts(
+    tx: ProjectTransaction,
+    stateId: string,
+    artifactIds: string[],
+  ): Promise<void> {
+    const staleArtifacts = await tx.artifact.findMany({
+      where: {
+        instances: {
+          some: {
+            id: stateId,
+          },
+        },
+        id: artifactIds.length > 0 ? { notIn: artifactIds } : undefined,
+      },
+      select: {
+        id: true,
+      },
+    })
+
+    if (staleArtifacts.length === 0) {
+      return
+    }
+
+    await tx.instanceState.update({
+      where: { id: stateId },
+      data: {
+        artifacts: {
+          disconnect: staleArtifacts.map(artifact => ({ id: artifact.id })),
+        },
+      },
+    })
+  }
+
   /**
    * Gets all triggers for a specific instance.
    *