fix: cancel usage tracker debounce on dispose

Author: doctarock

src/core/task/Task.ts

@@ -2333,6 +2333,12 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 			console.error("Error disposing file context tracker:", error)
 		}
 
+		try {
+			this.usageTracker.dispose()
+		} catch (error) {
+			console.error("Error disposing usage tracker:", error)
+		}
+
 		try {
 			// If we're not streaming then `abortStream` won't be called.
 			if (this.isStreaming && this.diffViewProvider.isEditing) {

src/core/task/UsageTracker.ts

@@ -74,6 +74,10 @@ export class UsageTracker {
 		this.debouncedEmitTokenUsage.flush()
 	}
 
+	public dispose(): void {
+		this.debouncedEmitTokenUsage.cancel()
+	}
+
 	public recordToolUsage(toolName: ToolName): void {
 		const usage = this.ensureToolUsageEntry(toolName)
 		usage.attempts++

src/core/task/__tests__/Task.dispose.test.ts

@@ -141,6 +141,14 @@ describe("Task dispose method", () => {
 		consoleLogSpy.mockRestore()
 	})
 
+	test("should dispose usage tracker to cancel pending usage emissions", () => {
+		const usageTrackerDisposeSpy = vi.spyOn((task as any).usageTracker, "dispose")
+
+		task.dispose()
+
+		expect(usageTrackerDisposeSpy).toHaveBeenCalledOnce()
+	})
+
 	test("should prevent memory leaks by removing listeners before other cleanup", () => {
 		// Add multiple listeners of different types using type assertion for testing
 		const listeners = {

src/core/task/__tests__/UsageTracker.spec.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it, vi } from "vitest"
+import { afterEach, describe, expect, it, vi } from "vitest"
 import { RooCodeEventName, type ClineMessage } from "@roo-code/types"
 
 import { UsageTracker } from "../UsageTracker.js"
@@ -11,6 +11,10 @@ const createUsageMessage = (text: string, ts = 1000): ClineMessage => ({
 })
 
 describe("UsageTracker", () => {
+	afterEach(() => {
+		vi.useRealTimers()
+	})
+
 	it("recomputes tokenUsage on each read after an emitted snapshot exists", () => {
 		const messages: ClineMessage[] = [
 			{ type: "say", say: "text", text: "task", ts: 1 },
@@ -53,4 +57,31 @@ describe("UsageTracker", () => {
 		)
 		expect((tracker as any).tokenUsageSnapshot.totalTokensIn).toBe(100)
 	})
+
+	it("cancels pending debounced token usage emissions on dispose", () => {
+		vi.useFakeTimers()
+
+		const emit = vi.fn()
+		const messages: ClineMessage[] = [
+			{ type: "say", say: "text", text: "task", ts: 1 },
+			createUsageMessage('{"tokensIn":100,"tokensOut":50,"cost":0.01}', 1000),
+		]
+		const tracker = new UsageTracker({
+			taskId: "task-id",
+			getMessages: () => messages,
+			emit,
+			emitIntervalMs: 1000,
+		})
+
+		tracker.emitTokenUsageUpdate(tracker.getTokenUsage())
+		messages[1] = createUsageMessage('{"tokensIn":250,"tokensOut":50,"cost":0.02}', 1000)
+		tracker.emitTokenUsageUpdate(tracker.getTokenUsage())
+
+		expect(emit).toHaveBeenCalledTimes(1)
+
+		tracker.dispose()
+		vi.advanceTimersByTime(1000)
+
+		expect(emit).toHaveBeenCalledTimes(1)
+	})
 })