Add command to start and stop the tracker, and once it stops, automatically update YAML metadata of note and its ascendants

Author: lucaslopes

src/main.ts

@@ -1,11 +1,4 @@
-import {
-	Plugin,
-	TFile,
-	Notice,
-	Editor,
-	MarkdownSectionInformation,
-	ButtonComponent,
-} from "obsidian";
+import { Plugin, TFile, Notice, Editor, MarkdownView } from "obsidian";
 import { defaultSettings, TimeTreeSettings } from "./settings";
 import { TimeTreeSettingsTab } from "./settings-tab";
 import { FrontMatterManager } from "./front-matter-manager";
@@ -17,6 +10,7 @@ export default class TimeTreePlugin extends Plugin {
 	private frontMatterManager: FrontMatterManager;
 	private calculator: TimeTreeCalculator;
 	private computeIntervalHandle: any;
+	private buttonObserver: MutationObserver | null = null;
 
 	async onload(): Promise<void> {
 		await this.loadSettings();
@@ -62,10 +56,38 @@ export default class TimeTreePlugin extends Plugin {
 			},
 		});
 
+		this.buttonObserver = new MutationObserver((mutations) => {
+			mutations.forEach((mutation) => {
+				mutation.addedNodes.forEach((node) => {
+					if (node instanceof HTMLElement) {
+						const btn = node.querySelector(
+							".simple-time-tracker-btn"
+						) as HTMLButtonElement | null;
+						if (btn) {
+							btn.addEventListener("click", () => {
+								const btnStatus =
+									btn.getAttribute("aria-label");
+								if (btnStatus === "End") {
+									this.elapsedTime();
+								}
+							});
+						}
+					}
+				});
+			});
+		});
+		this.buttonObserver.observe(document.body, {
+			childList: true,
+			subtree: true,
+		});
+
 		this.scheduleComputeTimeTree();
 	}
 
 	onunload(): void {
+		if (this.buttonObserver) {
+			this.buttonObserver.disconnect();
+		}
 		if (this.computeIntervalHandle) {
 			clearInterval(this.computeIntervalHandle);
 		}
@@ -84,7 +106,76 @@ export default class TimeTreePlugin extends Plugin {
 		this.scheduleComputeTimeTree();
 	}
 
+	async adjustCursorOutsideTracker(editor: Editor): Promise<void> {
+		// Get the full content and split it into lines.
+		const content = editor.getValue();
+		const lines = content.split("\n");
+
+		// Determine the end of YAML front matter if present. Line L is the last line of YAML metadata.
+		let yamlEnd = 0;
+		if (lines[0].trim() === "---") {
+			for (let i = 1; i < lines.length; i++) {
+				if (lines[i].trim() === "---") {
+					yamlEnd = i + 1; // YAML metadata ends at line L
+					break;
+				}
+			}
+		}
+
+		// Collect tracker block boundaries, but only after the YAML metadata.
+		const trackerBlocks: { start: number; end: number }[] = [];
+		for (let i = yamlEnd; i < lines.length; i++) {
+			if (lines[i].trimEnd() === "```simple-time-tracker") {
+				const blockStart = i;
+				// Look for the closing marker.
+				for (let j = i + 1; j < lines.length; j++) {
+					if (lines[j].trimEnd() === "```") {
+						trackerBlocks.push({ start: blockStart, end: j });
+						i = j; // Skip the rest of this block
+						break;
+					}
+				}
+			}
+		}
+
+		// Define a helper to check if a given line index is inside any tracker block.
+		const isLineInTracker = (line: number): boolean => {
+			return trackerBlocks.some(
+				(block) => line >= block.start && line <= block.end
+			);
+		};
+
+		// Find the first line after YAML metadata that is not inside any tracker block.
+		let targetLine = yamlEnd;
+		while (targetLine < lines.length && isLineInTracker(targetLine)) {
+			targetLine++;
+		}
+		if (targetLine >= lines.length) {
+			targetLine = lines.length - 1;
+		}
+
+		// Get the current cursor position.
+		const cursor = editor.getCursor();
+
+		// If the cursor is not inside any tracker block, do nothing.
+		const cursorInTracker = trackerBlocks.some(
+			(block) => cursor.line >= block.start && cursor.line <= block.end
+		);
+		if (!cursorInTracker) {
+			return;
+		}
+
+		// Move the cursor to the first line after YAML metadata that is not part of any tracker block.
+		editor.setCursor({ line: targetLine, ch: 0 });
+	}
+
 	async startStopTracker(): Promise<void> {
+		const activeView = this.app.workspace.getActiveViewOfType(MarkdownView);
+		if (activeView) {
+			await this.adjustCursorOutsideTracker(activeView.editor);
+		} else {
+			new Notice("No active Markdown editor found.");
+		}
 		const btn = document.querySelector(
 			".simple-time-tracker-btn"
 		) as HTMLButtonElement | null;
@@ -103,6 +194,14 @@ export default class TimeTreePlugin extends Plugin {
 		}
 		let elapsed = 0;
 		elapsed = await this.calculator.calculateElapsedTime(activeFile);
+		await new Promise((resolve) => setTimeout(resolve, 10));
+		await this.frontMatterManager.updateProperty(
+			activeFile,
+			(frontmatter) => {
+				frontmatter.elapsed = elapsed;
+				return frontmatter;
+			}
+		);
 		await this.calculator.communicateAscendants(activeFile);
 		const rootPath = this.settings.rootNotePath;
 		if (rootPath) {

src/settings.ts

@@ -1,15 +1,15 @@
 export const defaultSettings: TimeTreeSettings = {
-	onlyFirstTracker: false,
+	onlyFirstTracker: true,
 	rootNotePath: "",
 	RootFolderPath: "/",
-	considerSubdirs: false,
-	computeIntervalMinutes: 0, // New property: 0 means disabled by default
+	considerSubdirs: true,
+	computeIntervalMinutes: 0, // 0 means disabled by default
 };
 
 export interface TimeTreeSettings {
 	onlyFirstTracker: boolean;
 	rootNotePath: string;
 	RootFolderPath: string;
 	considerSubdirs: boolean;
-	computeIntervalMinutes: number; // New property
+	computeIntervalMinutes: number;
 }

src/time-tree-calculator.ts

@@ -35,15 +35,15 @@ export class TimeTreeCalculator {
 				}
 			}
 		}
-		await this.frontMatterManager.updateProperty(file, (frontmatter) => {
-			frontmatter.elapsed = localElapsed;
-			return frontmatter;
-		});
 		return localElapsed;
 	}
 
 	async calculateRecursiveElapsedTime(file: TFile): Promise<number> {
 		let localElapsed = await this.calculateElapsedTime(file);
+		await this.frontMatterManager.updateProperty(file, (frontmatter) => {
+			frontmatter.elapsed = localElapsed;
+			return frontmatter;
+		});
 		const fileCache = this.app.metadataCache.getFileCache(file);
 		if (fileCache && fileCache.links && fileCache.links.length > 0) {
 			for (const link of fileCache.links) {

test-vault/.obsidian/community-plugins.json

@@ -1,5 +1,5 @@
 [
+  "custom-node-size",
   "simple-time-tracker",
-  "time-tree",
-  "custom-node-size"
+  "time-tree"
 ]

test-vault/.obsidian/graph.json

@@ -32,6 +32,6 @@
   "repelStrength": 20,
   "linkStrength": 1,
   "linkDistance": 184,
-  "scale": 0.4016008904932634,
+  "scale": 0.38564307862722413,
   "close": true
 }

test-vault/.obsidian/plugins/simple-time-tracker/manifest.json

@@ -1,11 +1,10 @@
-{
-    "id": "simple-time-tracker",
-    "name": "Super Simple Time Tracker",
-    "version": "1.0.3",
-    "minAppVersion": "1.2.8",
-    "description": "Multi-purpose time trackers for your notes!",
-    "author": "Ellpeck",
-    "authorUrl": "https://ellpeck.de",
-    "fundingUrl": "https://ellpeck.de/support",
-    "isDesktopOnly": false
-}
+{
+    "id": "simple-time-tracker",
+    "name": "Super Simple Time Tracker",
+    "version": "1.0.3",
+    "minAppVersion": "1.2.8",
+    "description": "Multi-purpose time trackers for your notes!",
+    "author": "Ellpeck",
+    "authorUrl": "https://ellpeck.de",
+    "isDesktopOnly": false
+}