Consolidate goto ref / def posthog events into single event with a source. Added posthog event and audit log for goto/find all refs for chat

brendan-kellam · brendan-kellam · commit 9bf15ce3680a · 2025-07-25T17:50:23.000-07:00
diff --git a/packages/web/src/app/[domain]/browse/[...path]/components/pureCodePreviewPanel.tsx b/packages/web/src/app/[domain]/browse/[...path]/components/pureCodePreviewPanel.tsx
@@ -136,7 +136,9 @@ export const PureCodePreviewPanel = ({
     }, [editorRef, highlightRange]);
 
     const onFindReferences = useCallback((symbolName: string) => {
-        captureEvent('wa_browse_find_references_pressed', {});
+        captureEvent('wa_find_references_pressed', {
+            source: 'browse',
+        });
         createAuditAction({
             action: "user.performed_find_references",
             metadata: {
@@ -160,7 +162,9 @@ export const PureCodePreviewPanel = ({
     // If we resolve multiple matches, instead of navigating to the first match, we should
     // instead popup the bottom sheet with the list of matches.
     const onGotoDefinition = useCallback((symbolName: string, symbolDefinitions: SymbolDefinition[]) => {
-        captureEvent('wa_browse_goto_definition_pressed', {});
+        captureEvent('wa_goto_definition_pressed', {
+            source: 'browse',
+        });
         createAuditAction({
             action: "user.performed_goto_definition",
             metadata: {
diff --git a/packages/web/src/app/[domain]/search/components/codePreviewPanel/codePreview.tsx b/packages/web/src/app/[domain]/search/components/codePreviewPanel/codePreview.tsx
@@ -119,7 +119,9 @@ export const CodePreview = ({
     }, [onSelectedMatchIndexChange]);
 
     const onGotoDefinition = useCallback((symbolName: string, symbolDefinitions: SymbolDefinition[]) => {
-        captureEvent('wa_preview_panel_goto_definition_pressed', {});
+        captureEvent('wa_goto_definition_pressed', {
+            source: 'preview',
+        });
         createAuditAction({
             action: "user.performed_goto_definition",
             metadata: {
@@ -163,7 +165,9 @@ export const CodePreview = ({
     }, [captureEvent, file.filepath, file.language, file.revision, navigateToPath, repoName, domain]);
     
     const onFindReferences = useCallback((symbolName: string) => {
-        captureEvent('wa_preview_panel_find_references_pressed', {});
+        captureEvent('wa_find_references_pressed', {
+            source: 'preview',
+        });
         createAuditAction({
             action: "user.performed_find_references",
             metadata: {
diff --git a/packages/web/src/features/chat/components/chatThread/referencedFileSourceListItem.tsx b/packages/web/src/features/chat/components/chatThread/referencedFileSourceListItem.tsx
@@ -17,7 +17,9 @@ import { ChevronDown, ChevronRight } from "lucide-react";
 import { forwardRef, Ref, useCallback, useImperativeHandle, useMemo, useState } from "react";
 import { FileReference } from "../../types";
 import { createCodeFoldingExtension } from "./codeFoldingExtension";
-
+import useCaptureEvent from "@/hooks/useCaptureEvent";
+import { createAuditAction } from "@/ee/features/audit/actions";
+import { useDomain } from "@/hooks/useDomain";
 
 const lineDecoration = Decoration.line({
     attributes: { class: "cm-range-border-radius chat-lineHighlight" },
@@ -69,6 +71,8 @@ const ReferencedFileSourceListItem = ({
 }: ReferencedFileSourceListItemProps, forwardedRef: Ref<ReactCodeMirrorRef>) => {
     const theme = useCodeMirrorTheme();
     const [editorRef, setEditorRef] = useState<ReactCodeMirrorRef | null>(null);
+    const captureEvent = useCaptureEvent();
+    const domain = useDomain();
 
     useImperativeHandle(
         forwardedRef,
@@ -216,6 +220,16 @@ const ReferencedFileSourceListItem = ({
             return;
         }
 
+        captureEvent('wa_goto_definition_pressed', {
+            source: 'chat',
+        });
+        createAuditAction({
+            action: "user.performed_goto_definition",
+            metadata: {
+                message: symbolName,
+            },
+        }, domain);
+
         if (symbolDefinitions.length === 1) {
             const symbolDefinition = symbolDefinitions[0];
             const { fileName, repoName } = symbolDefinition;
@@ -246,9 +260,19 @@ const ReferencedFileSourceListItem = ({
             });
 
         }
-    }, [navigateToPath, revision, repoName, fileName, language]);
+    }, [captureEvent, domain, navigateToPath, revision, repoName, fileName, language]);
 
     const onFindReferences = useCallback((symbolName: string) => {
+        captureEvent('wa_find_references_pressed', {
+            source: 'chat',
+        });
+        createAuditAction({
+            action: "user.performed_find_references",
+            metadata: {
+                message: symbolName,
+            },
+        }, domain);
+
         navigateToPath({
             repoName,
             revisionName: revision,
@@ -266,7 +290,7 @@ const ReferencedFileSourceListItem = ({
             }
         })
 
-    }, [fileName, language, navigateToPath, repoName, revision]);
+    }, [captureEvent, domain, fileName, language, navigateToPath, repoName, revision]);
 
     const ExpandCollapseIcon = useMemo(() => {
         return isExpanded ? ChevronDown : ChevronRight;
diff --git a/packages/web/src/lib/posthogEvents.ts b/packages/web/src/lib/posthogEvents.ts
@@ -268,11 +268,12 @@ export type PosthogEventMap = {
     wa_api_key_created: {},
     wa_api_key_creation_fail: {},
     //////////////////////////////////////////////////////////////////
-    wa_preview_panel_find_references_pressed: {},
-    wa_preview_panel_goto_definition_pressed: {},
-    //////////////////////////////////////////////////////////////////
-    wa_browse_find_references_pressed: {},
-    wa_browse_goto_definition_pressed: {},
+    wa_goto_definition_pressed: {
+        source: 'chat' | 'browse' | 'preview',
+    },
+    wa_find_references_pressed: {
+        source: 'chat' | 'browse' | 'preview',
+    },
     //////////////////////////////////////////////////////////////////
     wa_explore_menu_reference_clicked: {},
     //////////////////////////////////////////////////////////////////
