From faef0978a7b802e43fcc0b1ff52c4fa6d974d2a6 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Sun, 15 Dec 2024 12:02:40 +0000
Subject: [PATCH] Add document locks

---
 plugin/src/sync-functions/locks.ts | 39 ++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 plugin/src/sync-functions/locks.ts

diff --git a/plugin/src/sync-functions/locks.ts b/plugin/src/sync-functions/locks.ts
new file mode 100644
index 0000000..279da85
--- /dev/null
+++ b/plugin/src/sync-functions/locks.ts
@@ -0,0 +1,39 @@
+import { DocumentId } from "src/database/document-metadata";
+
+const locked = new Set<DocumentId>();
+const waiters = new Map<DocumentId, Array<() => void>>();
+
+export function tryLockDocument(documentId: DocumentId): boolean {
+	if (locked.has(documentId)) {
+		return false;
+	}
+
+	locked.add(documentId);
+	return true;
+}
+
+export function waitForDocumentLock(documentId: DocumentId): Promise<void> {
+	if (tryLockDocument(documentId)) {
+		return Promise.resolve();
+	}
+
+	return new Promise((resolve) => {
+		if (!waiters.has(documentId)) {
+			waiters.set(documentId, []);
+		}
+
+		waiters.get(documentId)!.push(resolve);
+	});
+}
+
+export function unlockDocument(documentId: DocumentId): void {
+	if (!locked.has(documentId)) {
+		throw new Error(`Document ${documentId} is not locked`);
+	}
+
+	if (waiters.has(documentId)) {
+		waiters.get(documentId)!.shift()?.();
+	} else {
+		locked.delete(documentId);
+	}
+}