feat: find running trackers use timekeep registry

The timekeep locators model will now use the registry if enabled for
faster locating of timekeeps and when disabled will still use the
registry logic just without the registry caching

src/commands/findRunningTrackers.ts

@@ -1,11 +1,18 @@
 import { App, Command } from "obsidian";
 
 import { TimekeepLocatorModal } from "@/modals/TimekeepLocatorModal";
+import { TimekeepRegistry } from "@/service/registry";
+import { TimekeepSettings } from "@/settings";
+import { Store } from "@/store";
 
-export default function (app: App): Command {
+export default function (
+	app: App,
+	registry: TimekeepRegistry,
+	settings: Store<TimekeepSettings>
+): Command {
 	return {
 		id: `find`,
 		name: `Find running trackers`,
-		callback: () => new TimekeepLocatorModal(app).open(),
+		callback: () => new TimekeepLocatorModal(app, registry, settings).open(),
 	};
 }

src/main.ts

@@ -121,7 +121,7 @@ export default class TimekeepPlugin extends Plugin {
 
 		// Register commands
 		this.addCommand(insertTracker());
-		this.addCommand(findRunningTrackers(this.app));
+		this.addCommand(findRunningTrackers(this.app, this.registry, this.settingsStore));
 		this.addCommand(createMerged(this.app, this.settingsStore));
 		this.addCommand(exportMergedPdf(this.app, this.settingsStore));
 		this.addCommand(stopAllTimekeepsCommand(this.app));

src/modals/TimekeepLocatorModal.ts

@@ -1,8 +1,13 @@
-import { App, TFile, SuggestModal } from "obsidian";
+import type { App, TFile } from "obsidian";
 
-import { extractTimekeepCodeblocks } from "@/timekeep/parser";
+import { SuggestModal } from "obsidian";
+
+import type { TimekeepSettings } from "@/settings";
+import type { Store } from "@/store";
+import type { Timekeep, TimeEntry } from "@/timekeep/schema";
+
+import { TimekeepEntryItemType, TimekeepRegistry, TimekeepRegistryEntry } from "@/service/registry";
 import { getRunningEntry } from "@/timekeep/queries";
-import { Timekeep, TimeEntry } from "@/timekeep/schema";
 
 interface TimekeepResult {
 	timekeep: Timekeep;
@@ -12,14 +17,31 @@ interface TimekeepResult {
 
 export class TimekeepLocatorModal extends SuggestModal<TimekeepResult> {
 	results: TimekeepResult[] | undefined = undefined;
+	registry: TimekeepRegistry;
+	settings: Store<TimekeepSettings>;
 
-	constructor(app: App) {
+	constructor(app: App, registry: TimekeepRegistry, settings: Store<TimekeepSettings>) {
 		super(app);
+		this.registry = registry;
+		this.settings = settings;
 	}
 
 	async getSuggestions(query: string): Promise<TimekeepResult[]> {
+		const settings = this.settings.getState();
+
+		// When the registry is enabled source the entries from the registry
+		// as this is much faster than triggering a search
+		if (settings.registryEnabled) {
+			const entries = this.registry.entries.getState();
+			const results: TimekeepResult[] = TimekeepLocatorModal.getResultsFromEntries(entries);
+			this.results = results;
+		}
+
+		// Fallback to searching using the registry logic without caching it if the
+		// registry is disabled
 		if (this.results === undefined) {
-			this.results = await this.getResults();
+			const entries = await TimekeepRegistry.getTimekeepsWithinVault(this.app.vault);
+			this.results = TimekeepLocatorModal.getResultsFromEntries(entries);
 		}
 
 		const queryLower = query.toLowerCase();
@@ -32,37 +54,6 @@ export class TimekeepLocatorModal extends SuggestModal<TimekeepResult> {
 		});
 	}
 
-	async getResults(): Promise<TimekeepResult[]> {
-		const markdownFiles = this.app.vault.getMarkdownFiles();
-		const batchSize = 10;
-
-		const results: TimekeepResult[] = [];
-
-		for (let i = 0; i < markdownFiles.length; i += batchSize) {
-			const batch = markdownFiles.slice(i, i + batchSize);
-
-			await Promise.allSettled(
-				batch.map(async (file) => {
-					const content = await this.app.vault.cachedRead(file);
-					const timekeeps = extractTimekeepCodeblocks(content);
-
-					for (const timekeep of timekeeps) {
-						const running = getRunningEntry(timekeep.entries);
-						if (running === null) continue;
-
-						results.push({
-							timekeep,
-							running,
-							file,
-						});
-					}
-				})
-			);
-		}
-
-		return results;
-	}
-
 	renderSuggestion(value: TimekeepResult, el: HTMLElement) {
 		el.createEl("div", { text: value.running.name });
 		el.createEl("small", { text: value.file.path });
@@ -71,4 +62,33 @@ export class TimekeepLocatorModal extends SuggestModal<TimekeepResult> {
 	onChooseSuggestion(item: TimekeepResult, _evt: MouseEvent | KeyboardEvent) {
 		void this.app.workspace.getLeaf().openFile(item.file);
 	}
+
+	static getResultsFromEntries(entries: TimekeepRegistryEntry[]): TimekeepResult[] {
+		const results: TimekeepResult[] = [];
+		for (const entry of entries) {
+			switch (entry.type) {
+				case TimekeepEntryItemType.FILE: {
+					const timekeep = entry.timekeep;
+					const running = getRunningEntry(timekeep.entries);
+					if (running !== null) {
+						results.push({ file: entry.file, running: running, timekeep });
+					}
+					break;
+				}
+				case TimekeepEntryItemType.MARKDOWN: {
+					for (const timekeepWithPosition of entry.timekeeps) {
+						const timekeep = timekeepWithPosition.timekeep;
+						const running = getRunningEntry(timekeep.entries);
+						if (running !== null) {
+							results.push({ file: entry.file, running: running, timekeep });
+						}
+					}
+
+					break;
+				}
+			}
+		}
+
+		return results;
+	}
 }