chore: enhanced posthog tracking

Author: itsskofficial

src/client/app/chat/page.js

@@ -129,6 +129,7 @@ export default function ChatPage() {
 	const ringtoneAudioRef = useRef(null)
 	const connectedAudioRef = useRef(null)
 	const remoteAudioRef = useRef(null)
+	const voiceModeStartTimeRef = useRef(null)
 
 	const fetchInitialMessages = useCallback(async () => {
 		setIsLoading(true)
@@ -848,6 +849,16 @@ export default function ChatPage() {
 
 		webrtcClientRef.current?.disconnect()
 
+		// --- ADD POSTHOG EVENT TRACKING ---
+		if (voiceModeStartTimeRef.current) {
+			const duration_seconds = Math.round(
+				(Date.now() - voiceModeStartTimeRef.current) / 1000
+			)
+			posthog?.capture("voice_mode_used", { duration_seconds })
+			voiceModeStartTimeRef.current = null // Reset after tracking
+		}
+		// --- END POSTHOG EVENT TRACKING ---
+
 		// 2. Immediately stop any playing audio.
 		if (ringtoneAudioRef.current) {
 			ringtoneAudioRef.current.pause()
@@ -872,6 +883,10 @@ export default function ChatPage() {
 			// Switching TO voice mode, first get permissions
 			const permissionsGranted = await initializeVoiceMode()
 			if (permissionsGranted) {
+				// --- ADD POSTHOG EVENT TRACKING ---
+				posthog?.capture("voice_mode_activated")
+				voiceModeStartTimeRef.current = Date.now() // Set start time
+				// --- END POSTHOG EVENT TRACKING ---
 				setIsVoiceMode(true)
 			}
 		}

src/client/app/settings/page.js

@@ -23,6 +23,7 @@ import {
 } from "@tabler/icons-react"
 import { useState, useEffect, useCallback, Fragment } from "react"
 import { Tooltip } from "react-tooltip"
+import { usePostHog } from "posthog-js/react"
 import { cn } from "@utils/cn"
 import { Switch } from "@headlessui/react"
 
@@ -686,6 +687,7 @@ const TestingTools = () => {
 const ProactivitySettings = () => {
 	const [isEnabled, setIsEnabled] = useState(false)
 	const [isLoading, setIsLoading] = useState(true)
+	const posthog = usePostHog()
 
 	useEffect(() => {
 		const fetchStatus = async () => {
@@ -709,6 +711,12 @@ const ProactivitySettings = () => {
 		const toastId = toast.loading(
 			enabled ? "Enabling proactivity..." : "Disabling proactivity..."
 		)
+
+		// --- ADD POSTHOG EVENT TRACKING ---
+		if (enabled) {
+			posthog?.capture("proactive_assistance_enabled")
+		}
+		// --- END POSTHOG EVENT TRACKING ---
 		try {
 			const response = await fetch("/api/settings/proactivity", {
 				method: "POST",

src/server/main/proactivity/routes.py

@@ -7,6 +7,7 @@
 from main.proactivity.models import SuggestionActionRequest
 from main.proactivity.learning import record_user_feedback
 from workers.tasks import create_task_from_suggestion
+from main.analytics import capture_event
 
 logger = logging.getLogger(__name__)
 router = APIRouter(
@@ -45,6 +46,13 @@ async def handle_suggestion_action(
     if request.user_action == "approved":
         logger.info(f"User '{user_id}' approved suggestion '{request.notification_id}' of type '{suggestion_type}'.")
 
+        # --- ADD POSTHOG EVENT TRACKING ---
+        capture_event(
+            user_id,
+            "proactive_suggestion_approved",
+            {"suggestion_type": suggestion_type}
+        )
+        # --- END POSTHOG EVENT TRACKING ---
         # Re-extract with default for robustness, as suggested by diff
         suggestion_payload = notification.get("suggestion_payload", {})
         gathered_context = suggestion_payload.get("gathered_context")
@@ -64,6 +72,13 @@ async def handle_suggestion_action(
     elif request.user_action == "dismissed":
         logger.info(f"User '{user_id}' dismissed suggestion '{request.notification_id}' of type '{suggestion_type}'.")
 
+        # --- ADD POSTHOG EVENT TRACKING ---
+        capture_event(
+            user_id,
+            "proactive_suggestion_dismissed",
+            {"suggestion_type": suggestion_type}
+        )
+        # --- END POSTHOG EVENT TRACKING ---
         # 3c. Record negative feedback
         await record_user_feedback(user_id, suggestion_type, "negative")
 

src/server/workers/tasks.py

@@ -229,6 +229,15 @@ async def async_refine_and_plan_ai_task(task_id: str, user_id: str):
         parsed_data = JsonExtractor.extract_valid_json(clean_llm_output(response_str))
         if parsed_data:
             # Safeguard: never allow the refiner to nullify or change the user_id
+            # --- ADD POSTHOG EVENT TRACKING ---
+            schedule_type = parsed_data.get("schedule", {}).get("type", "once")
+            capture_event(
+                user_id,
+                "task_created",
+                {"task_id": task_id, "schedule_type": schedule_type, "source": "prompt"}
+            )
+            # --- END POSTHOG EVENT TRACKING ---
+
             if "user_id" in parsed_data:
                 del parsed_data["user_id"]