Fixes last tab being opened instead of last active tab

Fixes https://discord.com/channels/731730685944922173/881073965047636018/937518022548131891
GraphiteEditor · Jan 13, 2025 · addf7e8 · addf7e8
1 parent 3582126
commit addf7e8
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/frontend/src/io-managers/persistence.ts b/frontend/src/io-managers/persistence.ts
@@ -16,6 +16,17 @@ export function createPersistenceManager(editor: Editor, portfolio: PortfolioSta
 		await set("documents_tab_order", documentOrder, graphiteStore);
 	}
 
+	async function storeCurrentDocumentIndex() {
+		const documentIndex = getFromStore(portfolio).activeDocumentIndex;
+		const documentId = getFromStore(portfolio).documents[documentIndex].id;
+
+		await storeDocumentDocumentByID(String(documentId));
+	}
+
+	async function storeDocumentDocumentByID(documentId: string) {
+		await set("current_document_id", String(documentId), graphiteStore);
+	}
+
 	async function storeDocument(autoSaveDocument: TriggerIndexedDbWriteDocument) {
 		await update<Record<string, TriggerIndexedDbWriteDocument>>(
 			"documents",
@@ -28,6 +39,7 @@ export function createPersistenceManager(editor: Editor, portfolio: PortfolioSta
 		);
 
 		await storeDocumentOrder();
+		await storeDocumentDocumentByID(autoSaveDocument.details.id);
 	}
 
 	async function removeDocument(id: string) {
@@ -42,18 +54,23 @@ export function createPersistenceManager(editor: Editor, portfolio: PortfolioSta
 		);
 
 		await storeDocumentOrder();
+		await storeCurrentDocumentIndex();
 	}
 
 	async function loadDocuments() {
 		const previouslySavedDocuments = await get<Record<string, TriggerIndexedDbWriteDocument>>("documents", graphiteStore);
 		const documentOrder = await get<string[]>("documents_tab_order", graphiteStore);
+		const currentDocumentId = await get<string>("current_document_id", graphiteStore);
 		if (!previouslySavedDocuments || !documentOrder) return;
 
 		const orderedSavedDocuments = documentOrder.flatMap((id) => (previouslySavedDocuments[id] ? [previouslySavedDocuments[id]] : []));
 
 		orderedSavedDocuments?.forEach(async (doc: TriggerIndexedDbWriteDocument) => {
 			editor.handle.openAutoSavedDocument(BigInt(doc.details.id), doc.details.name, doc.details.isSaved, doc.document);
 		});
+		if (currentDocumentId) {
+			editor.handle.selectDocument(BigInt(currentDocumentId));
+		}
 	}
 
 	// PREFERENCES
@@ -91,5 +108,6 @@ export function createPersistenceManager(editor: Editor, portfolio: PortfolioSta
 
 export async function wipeDocuments() {
 	await del("documents_tab_order", graphiteStore);
+	await del("current_document_id", graphiteStore);
 	await del("documents", graphiteStore);
 }