feat: added log requests to backend for observability

Author: James Mtendamema

packages/types/src/vscode-extension-host.ts

@@ -372,6 +372,10 @@ export type ExtensionState = Pick<
 	mdmCompliant?: boolean
 	taskSyncEnabled: boolean
 	openAiCodexIsAuthenticated?: boolean
+	zooCodeIsAuthenticated?: boolean
+	zooCodeUserName?: string
+	zooCodeUserEmail?: string
+	zooCodeUserImage?: string
 	debug?: boolean
 
 	/**
@@ -505,6 +509,7 @@ export interface WebviewMessage {
 		| "rooCloudManualUrl"
 		| "openAiCodexSignIn"
 		| "openAiCodexSignOut"
+		| "zooCodeSignOut"
 		| "switchOrganization"
 		| "condenseTaskContextRequest"
 		| "requestIndexingStatus"

src/activate/handleUri.ts

@@ -2,25 +2,24 @@ import * as vscode from "vscode"
 
 import { getRouterUnavailableSignInMessage } from "../core/config/routerRemoval"
 import { ClineProvider } from "../core/webview/ClineProvider"
+import { handleAuthCallback as handleZooCodeAuthCallback, setZooCodeUserInfo } from "../services/zoo-code-auth"
 
 export const handleUri = async (uri: vscode.Uri) => {
 	const path = uri.path
 	const query = new URLSearchParams(uri.query.replace(/\+/g, "%2B"))
 	const visibleProvider = ClineProvider.getVisibleInstance()
 
-	if (!visibleProvider) {
-		return
-	}
-
 	switch (path) {
 		case "/openrouter": {
+			if (!visibleProvider) return
 			const code = query.get("code")
 			if (code) {
 				await visibleProvider.handleOpenRouterCallback(code)
 			}
 			break
 		}
 		case "/requesty": {
+			if (!visibleProvider) return
 			const code = query.get("code")
 			const baseUrl = query.get("baseUrl")
 			if (code) {
@@ -32,6 +31,33 @@ export const handleUri = async (uri: vscode.Uri) => {
 			vscode.window.showInformationMessage(getRouterUnavailableSignInMessage())
 			break
 		}
+		case "/auth-callback": {
+			const token = query.get("token")
+			if (token) {
+				// Extract user info from callback URL params with proper URL decoding
+				const name = query.get("name") ? decodeURIComponent(query.get("name")!.replace(/\+/g, " ")) : undefined
+				const email = query.get("email")
+					? decodeURIComponent(query.get("email")!.replace(/\+/g, " "))
+					: undefined
+				const image = query.get("image") ? decodeURIComponent(query.get("image")!) : undefined
+
+				// Store user info if provided
+				if (name || email || image) {
+					await setZooCodeUserInfo({
+						name,
+						email,
+						image,
+					})
+				}
+
+				const success = await handleZooCodeAuthCallback(token)
+				if (success && visibleProvider) {
+					// Update the active API configuration to use zoo-code with the new token
+					await visibleProvider.handleZooCodeCallback(token)
+				}
+			}
+			break
+		}
 		default:
 			break
 	}

src/core/task/Task.ts

@@ -3155,6 +3155,25 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 									cacheReadTokens: tokens.cacheRead,
 									cost: tokens.total ?? costResult.totalCost,
 								})
+
+								// Zoo Code observability telemetry
+								import("../../services/zoo-telemetry")
+									.then(({ sendLlmTelemetry }) =>
+										sendLlmTelemetry({
+											taskId: this.taskId,
+											provider: this.apiConfiguration?.apiProvider ?? "unknown",
+											model: this.apiConfiguration
+												? (getModelId(this.apiConfiguration) ?? "unknown")
+												: "unknown",
+											mode: this.taskMode,
+											inputTokens: costResult.totalInputTokens,
+											outputTokens: costResult.totalOutputTokens,
+											cacheReadTokens: tokens.cacheRead ?? 0,
+											cacheWriteTokens: tokens.cacheWrite ?? 0,
+											totalCost: tokens.total ?? costResult.totalCost,
+										}).catch(() => {}),
+									)
+									.catch(() => {})
 							}
 						}
 

src/core/webview/ClineProvider.ts

@@ -240,8 +240,9 @@ export class ClineProvider
 
 			// Create named listener functions so we can remove them later.
 			const onTaskStarted = () => this.emit(RooCodeEventName.TaskStarted, instance.taskId)
-			const onTaskCompleted = (taskId: string, tokenUsage: TokenUsage, toolUsage: ToolUsage) =>
+			const onTaskCompleted = (taskId: string, tokenUsage: TokenUsage, toolUsage: ToolUsage) => {
 				this.emit(RooCodeEventName.TaskCompleted, taskId, tokenUsage, toolUsage)
+			}
 			const onTaskAborted = async () => {
 				this.emit(RooCodeEventName.TaskAborted, instance.taskId)
 
@@ -1681,6 +1682,15 @@ export class ClineProvider
 		await this.upsertProviderProfile(currentApiConfigName, newConfiguration)
 	}
 
+	// Zoo Code Auth (for observability telemetry)
+
+	async handleZooCodeCallback(token: string) {
+		const { handleAuthCallback } = await import("../../services/zoo-code-auth")
+		await handleAuthCallback(token)
+		// Refresh webview state to show updated auth status
+		await this.postStateToWebview()
+	}
+
 	// Requesty
 
 	async handleRequestyCallback(code: string, baseUrl: string | null) {
@@ -2279,6 +2289,31 @@ export class ClineProvider
 					return false
 				}
 			})(),
+			zooCodeIsAuthenticated: await (async () => {
+				try {
+					const { isZooCodeAuthenticated } = await import("../../services/zoo-code-auth")
+					return await isZooCodeAuthenticated()
+				} catch {
+					return false
+				}
+			})(),
+			...(() => {
+				try {
+					const { getCachedZooCodeUserInfo } = require("../../services/zoo-code-auth")
+					const userInfo = getCachedZooCodeUserInfo()
+					return {
+						zooCodeUserName: userInfo.name,
+						zooCodeUserEmail: userInfo.email,
+						zooCodeUserImage: userInfo.image,
+					}
+				} catch {
+					return {
+						zooCodeUserName: undefined,
+						zooCodeUserEmail: undefined,
+						zooCodeUserImage: undefined,
+					}
+				}
+			})(),
 			debug: vscode.workspace.getConfiguration(Package.name).get<boolean>("debug", false),
 		}
 	}

src/core/webview/webviewMessageHandler.ts

@@ -2405,6 +2405,18 @@ export const webviewMessageHandler = async (
 			await provider.postStateToWebview()
 			break
 		}
+		case "zooCodeSignOut": {
+			try {
+				const { clearZooCodeToken } = await import("../../services/zoo-code-auth")
+				await clearZooCodeToken()
+				await provider.postStateToWebview()
+			} catch (error) {
+				provider.log(
+					`Failed to sign out of Zoo Code: ${error instanceof Error ? error.message : String(error)}`,
+				)
+			}
+			break
+		}
 		case "switchOrganization": {
 			try {
 				const organizationId = message.organizationId ?? null

src/extension.ts

@@ -49,6 +49,7 @@ import {
 } from "./activate"
 import { initializeI18n } from "./i18n"
 import { initializeModelCacheRefresh } from "./api/providers/fetchers/modelCache"
+import { initZooCodeAuth } from "./services/zoo-code-auth"
 
 /**
  * Built using https://github.com/microsoft/vscode-webview-ui-toolkit
@@ -158,6 +159,9 @@ export async function activate(context: vscode.ExtensionContext) {
 	// Initialize OpenAI Codex OAuth manager for ChatGPT subscription-based access.
 	openAiCodexOAuthManager.initialize(context, (message) => outputChannel.appendLine(message))
 
+	// Initialize Zoo Code auth service for extension session token management.
+	initZooCodeAuth(context)
+
 	// Get default commands from configuration.
 	const defaultCommands = vscode.workspace.getConfiguration(Package.name).get<string[]>("allowedCommands") || []