Add WebSocket support (#12)

frontend/obsidian-plugin/src/vault-link-plugin.ts

@@ -1,20 +1,25 @@
-import type { WorkspaceLeaf } from "obsidian";
-import { Platform, Plugin } from "obsidian";
-import "./styles.scss";
+import type {
+	Editor,
+	MarkdownFileInfo,
+	MarkdownView,
+	TAbstractFile,
+	WorkspaceLeaf
+} from "obsidian";
+import { Platform, Plugin, TFile } from "obsidian";
 import "../manifest.json";
-import { SyncSettingsTab } from "./views/settings-tab";
-import { HistoryView } from "./views/history-view";
-import { ObsidianFileEventHandler } from "./obisidan-event-handler";
-import { StatusBar } from "./views/status-bar";
-import { LogsView } from "./views/logs-view";
-import { StatusDescription } from "./views/status-description";
+import { HistoryView } from "./views/history/history-view";
+import { StatusBar } from "./views/status-bar/status-bar";
+import { LogsView } from "./views/logs/logs-view";
+import { StatusDescription } from "./views/status-description/status-description";
 import type { LogLine } from "sync-client";
 import { SyncClient, LogLevel } from "sync-client";
 import { ObsidianFileSystemOperations } from "./obsidian-file-system";
+import { SyncSettingsTab } from "./views/settings/settings-tab";
 
 export default class VaultLinkPlugin extends Plugin {
 	private settingsTab: SyncSettingsTab | undefined;
 	private client!: SyncClient;
+
 	private static registerConsoleForLogging(client: SyncClient): void {
 		client.logger.addOnMessageListener((logLine: LogLine) => {
 			const formatted = `${logLine.timestamp.toISOString()} ${logLine.level} ${logLine.message}`;
@@ -38,7 +43,10 @@ export default class VaultLinkPlugin extends Plugin {
 
 	public async onload(): Promise<void> {
 		this.client = await SyncClient.create({
-			fs: new ObsidianFileSystemOperations(this.app.vault),
+			fs: new ObsidianFileSystemOperations(
+				this.app.vault,
+				this.app.workspace
+			),
 			persistence: {
 				load: this.loadData.bind(this),
 				save: this.saveData.bind(this)
@@ -80,35 +88,9 @@ export default class VaultLinkPlugin extends Plugin {
 			async (_: MouseEvent) => this.activateView(LogsView.TYPE)
 		);
 
-		const eventHandler = new ObsidianFileEventHandler(this.client);
-
 		this.app.workspace.onLayoutReady(async () => {
-			this.client.logger.info("Initialising sync handlers");
-
-			[
-				this.app.vault.on(
-					"create",
-					eventHandler.onCreate.bind(eventHandler)
-				),
-				this.app.vault.on(
-					"modify",
-					eventHandler.onModify.bind(eventHandler)
-				),
-				this.app.vault.on(
-					"delete",
-					eventHandler.onDelete.bind(eventHandler)
-				),
-				this.app.vault.on(
-					"rename",
-					eventHandler.onRename.bind(eventHandler)
-				)
-			].forEach((event) => {
-				this.registerEvent(event);
-			});
-
+			this.registerEditorEvents();
 			void this.client.start();
-
-			this.client.logger.info("Sync handlers initialised");
 		});
 	}
 
@@ -145,4 +127,51 @@ export default class VaultLinkPlugin extends Plugin {
 			await workspace.revealLeaf(leaf);
 		}
 	}
+
+	private registerEditorEvents(): void {
+		[
+			this.app.workspace.on(
+				"editor-change",
+				async (
+					_editor: Editor,
+					info: MarkdownView | MarkdownFileInfo
+				) => {
+					const { file } = info;
+					if (file) {
+						await this.client.syncLocallyUpdatedFile({
+							relativePath: file.path
+						});
+					}
+				}
+			),
+			this.app.vault.on("create", async (file: TAbstractFile) => {
+				if (file instanceof TFile) {
+					await this.client.syncLocallyCreatedFile(file.path);
+				}
+			}),
+			this.app.vault.on("modify", async (file: TAbstractFile) => {
+				if (file instanceof TFile) {
+					await this.client.syncLocallyUpdatedFile({
+						relativePath: file.path
+					});
+				}
+			}),
+			this.app.vault.on("delete", async (file: TAbstractFile) => {
+				await this.client.syncLocallyDeletedFile(file.path);
+			}),
+			this.app.vault.on(
+				"rename",
+				async (file: TAbstractFile, oldPath: string) => {
+					if (file instanceof TFile) {
+						await this.client.syncLocallyUpdatedFile({
+							oldPath,
+							relativePath: file.path
+						});
+					}
+				}
+			)
+		].forEach((event) => {
+			this.registerEvent(event);
+		});
+	}
 }