@posthog/agent 2.0.0 → 2.0.2

package/src/sagas/apply-snapshot-saga.ts

@@ -0,0 +1,88 @@
+import { mkdir, rm, writeFile } from "node:fs/promises";
+import { join } from "node:path";
+import { Saga } from "@posthog/shared";
+import { ApplyTreeSaga as GitApplyTreeSaga } from "@twig/git/sagas/tree";
+import type { PostHogAPIClient } from "../posthog-api.js";
+import type { TreeSnapshot } from "../types.js";
+
+export interface ApplySnapshotInput {
+  snapshot: TreeSnapshot;
+  repositoryPath: string;
+  apiClient: PostHogAPIClient;
+  taskId: string;
+  runId: string;
+}
+
+export interface ApplySnapshotOutput {
+  treeHash: string;
+}
+
+export class ApplySnapshotSaga extends Saga<
+  ApplySnapshotInput,
+  ApplySnapshotOutput
+> {
+  private archivePath: string | null = null;
+
+  protected async execute(
+    input: ApplySnapshotInput,
+  ): Promise<ApplySnapshotOutput> {
+    const { snapshot, repositoryPath, apiClient, taskId, runId } = input;
+    const tmpDir = join(repositoryPath, ".posthog", "tmp");
+
+    if (!snapshot.archiveUrl) {
+      throw new Error("Cannot apply snapshot: no archive URL");
+    }
+
+    await this.step({
+      name: "create_tmp_dir",
+      execute: () => mkdir(tmpDir, { recursive: true }),
+      rollback: async () => {},
+    });
+
+    this.archivePath = join(tmpDir, `${snapshot.treeHash}.tar.gz`);
+    await this.step({
+      name: "download_archive",
+      execute: async () => {
+        const arrayBuffer = await apiClient.downloadArtifact(
+          taskId,
+          runId,
+          snapshot.archiveUrl!,
+        );
+        if (!arrayBuffer) {
+          throw new Error("Failed to download archive");
+        }
+        const base64Content = Buffer.from(arrayBuffer).toString("utf-8");
+        const binaryContent = Buffer.from(base64Content, "base64");
+        await writeFile(this.archivePath!, binaryContent);
+      },
+      rollback: async () => {
+        if (this.archivePath) {
+          await rm(this.archivePath, { force: true }).catch(() => {});
+        }
+      },
+    });
+
+    const gitApplySaga = new GitApplyTreeSaga(this.log);
+    const applyResult = await gitApplySaga.run({
+      baseDir: repositoryPath,
+      treeHash: snapshot.treeHash,
+      baseCommit: snapshot.baseCommit,
+      changes: snapshot.changes,
+      archivePath: this.archivePath,
+    });
+
+    if (!applyResult.success) {
+      throw new Error(`Failed to apply tree: ${applyResult.error}`);
+    }
+
+    await rm(this.archivePath, { force: true }).catch(() => {});
+
+    this.log.info("Tree snapshot applied", {
+      treeHash: snapshot.treeHash,
+      totalChanges: snapshot.changes.length,
+      deletedFiles: snapshot.changes.filter((c) => c.status === "D").length,
+    });
+
+    return { treeHash: snapshot.treeHash };
+  }
+}