@hyperframes/studio 0.8.80 → 0.8.82

package/src/utils/studioFileHistory.ts

@@ -27,7 +27,7 @@ interface SaveProjectFilesWithHistoryInput {
   label: string;
   coalesceKey?: string;
   coalesceMs?: number;
-  files: Record<string, string>;
+  files: Record<string, (contentInsideFileQueue: string) => string>;
   readFile: (path: string) => Promise<string>;
   writeFile: ProjectFileWriter;
   recordEdit: (entry: RecordEditInput) => Promise<void>;
@@ -68,6 +68,34 @@ export async function saveProjectFilesWithHistory(
   );
 }
 
+/**
+ * A server-side rewrite and its undo entry, holding the file's queue from the read to the history write.
+ * `rewrite` returns what the server left on disk (and the edit's final content if it goes further), or null.
+ */
+export async function saveServerRewriteWithHistory(input: {
+  projectId: string;
+  path: string;
+  label: string;
+  coalesceKey?: string;
+  writeFile: ProjectFileWriter;
+  recordEdit: (entry: RecordEditInput) => Promise<void>;
+  rewrite: (original: string) => Promise<{ disk: string; after?: string } | null>;
+}): Promise<boolean> {
+  const { projectId, path, writeFile } = input;
+  return serializeStudioFileMutations(writeFile, [path], async () => {
+    const original = await readProjectFileContent(projectId, path);
+    const result = await input.rewrite(original);
+    if (!result) return false;
+    await writeProjectFilesWithHistoryInQueue({
+      ...input,
+      files: { [path]: () => result.after ?? result.disk },
+      readFile: async () => original,
+      diskContent: { [path]: result.disk },
+    });
+    return true;
+  });
+}
+
 export async function writeProjectFilesWithHistoryInQueue({
   label,
   coalesceKey,
@@ -79,8 +107,9 @@ export async function writeProjectFilesWithHistoryInQueue({
   diskContent,
 }: SaveProjectFilesWithHistoryInput): Promise<string[]> {
   const snapshots: Record<string, { before: string; after: string }> = {};
-  for (const [path, after] of Object.entries(files)) {
+  for (const [path, build] of Object.entries(files)) {
     const before = await readFile(path);
+    const after = build(before);
     if (before !== after) {
       snapshots[path] = { before, after };
     }