Graphite/frontend/src/lifetime/auto-save.ts

import { AutoSaveDocument, RemoveAutoSaveDocument } from "@/dispatcher/js-messages";
import { DocumentsState } from "@/state/documents";
import { EditorState, getWasmInstance } from "@/state/wasm-loader";

const GRAPHITE_INDEXED_DB_VERSION = 2;
const GRAPHITE_INDEXED_DB_NAME = "graphite-indexed-db";
const GRAPHITE_AUTO_SAVE_STORE = "auto-save-documents";
const GRAPHITE_AUTO_SAVE_ORDER_KEY = "auto-save-documents-order";

const databaseConnection: Promise<IDBDatabase> = new Promise((resolve) => {
	const dbOpenRequest = indexedDB.open(GRAPHITE_INDEXED_DB_NAME, GRAPHITE_INDEXED_DB_VERSION);

	dbOpenRequest.onupgradeneeded = (): void => {
		const db = dbOpenRequest.result;
		// Wipes out all auto-save data on upgrade
		if (db.objectStoreNames.contains(GRAPHITE_AUTO_SAVE_STORE)) {
			db.deleteObjectStore(GRAPHITE_AUTO_SAVE_STORE);
		}

		db.createObjectStore(GRAPHITE_AUTO_SAVE_STORE, { keyPath: "details.id" });
	};

	dbOpenRequest.onerror = (): void => {
		// eslint-disable-next-line no-console
		console.error("Graphite IndexedDb error:", dbOpenRequest.error);
	};

	dbOpenRequest.onsuccess = (): void => {
		resolve(dbOpenRequest.result);
	};
});

// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
export function createAutoSaveManager(editor: EditorState, documents: DocumentsState) {
	const openAutoSavedDocuments = async (): Promise<void> => {
		const db = await databaseConnection;
		const transaction = db.transaction(GRAPHITE_AUTO_SAVE_STORE, "readonly");
		const request = transaction.objectStore(GRAPHITE_AUTO_SAVE_STORE).getAll();

		return new Promise((resolve) => {
			request.onsuccess = (): void => {
				const previouslySavedDocuments: AutoSaveDocument[] = request.result;

				const documentOrder: string[] = JSON.parse(window.localStorage.getItem(GRAPHITE_AUTO_SAVE_ORDER_KEY) || "[]");
				const orderedSavedDocuments = documentOrder.map((id) => previouslySavedDocuments.find((autoSave) => autoSave.details.id === id)).filter((x) => x !== undefined) as AutoSaveDocument[];

				const currentDocumentVersion = getWasmInstance().graphite_version();
				orderedSavedDocuments.forEach((doc: AutoSaveDocument) => {
					if (doc.version === currentDocumentVersion) {
						editor.instance.open_auto_saved_document(BigInt(doc.details.id), doc.details.name, doc.details.is_saved, doc.document);
					} else {
						removeDocument(doc.details.id);
					}
				});
				resolve(undefined);
			};
		});
	};

	const storeDocumentOrder = (): void => {
		// Make sure to store as string since JSON does not play nice with BigInt
		const documentOrder = documents.state.documents.map((doc) => doc.id.toString());
		window.localStorage.setItem(GRAPHITE_AUTO_SAVE_ORDER_KEY, JSON.stringify(documentOrder));
	};

	const removeDocument = async (id: string): Promise<void> => {
		const db = await databaseConnection;
		const transaction = db.transaction(GRAPHITE_AUTO_SAVE_STORE, "readwrite");
		transaction.objectStore(GRAPHITE_AUTO_SAVE_STORE).delete(id);
		storeDocumentOrder();
	};

	editor.dispatcher.subscribeJsMessage(AutoSaveDocument, async (autoSaveDocument) => {
		const db = await databaseConnection;
		const transaction = db.transaction(GRAPHITE_AUTO_SAVE_STORE, "readwrite");
		transaction.objectStore(GRAPHITE_AUTO_SAVE_STORE).put(autoSaveDocument);
		storeDocumentOrder();
	});

	editor.dispatcher.subscribeJsMessage(RemoveAutoSaveDocument, async (removeAutoSaveDocument) => {
		removeDocument(removeAutoSaveDocument.document_id);
	});

	// On creation
	openAutoSavedDocuments();

	return {
		openAutoSavedDocuments,
	};
}