track in-app ai consent approval and use context query client

Author: charlesvien

apps/code/src/renderer/features/ai-approval/components/AiApprovalScreen.test.tsx

@@ -36,13 +36,6 @@ vi.mock("@renderer/trpc/client", () => ({
   trpcClient: {},
 }));
 
-vi.mock("@utils/queryClient", async () => {
-  const { QueryClient: TestQueryClient } = await import(
-    "@tanstack/react-query"
-  );
-  return { queryClient: new TestQueryClient() };
-});
-
 import { AiApprovalScreen } from "./AiApprovalScreen";
 
 function renderInTheme(isAdmin: boolean) {

apps/code/src/renderer/features/ai-approval/components/AiApprovalScreen.tsx

@@ -8,9 +8,8 @@ import { GearSix, Robot, SignOut, WarningCircle } from "@phosphor-icons/react";
 import { Button, Callout, Flex, Spinner, Text } from "@radix-ui/themes";
 import { SHORTCUTS } from "@renderer/constants/keyboard-shortcuts";
 import { ANALYTICS_EVENTS } from "@shared/types/analytics";
-import { useMutation } from "@tanstack/react-query";
+import { useMutation, useQueryClient } from "@tanstack/react-query";
 import { track } from "@utils/analytics";
-import { queryClient } from "@utils/queryClient";
 import { motion } from "framer-motion";
 import { useEffect } from "react";
 import { useHotkeys } from "react-hotkeys-hook";
@@ -24,12 +23,14 @@ export function AiApprovalScreen({ orgName, isAdmin }: AiApprovalScreenProps) {
   const logoutMutation = useLogoutMutation();
   const openSettings = useSettingsDialogStore((s) => s.open);
   const client = useAuthenticatedClient();
+  const queryClient = useQueryClient();
 
   const approveMutation = useMutation({
     mutationFn: async () => {
       await client.approveAiDataProcessing();
     },
     onSuccess: async () => {
+      track(ANALYTICS_EVENTS.AI_CONSENT_GRANTED_INAPP);
       await queryClient.invalidateQueries({
         queryKey: authKeys.currentUsers(),
       });

apps/code/src/shared/types/analytics.ts

@@ -656,6 +656,7 @@ export const ANALYTICS_EVENTS = {
   ONBOARDING_ABANDONED: "Onboarding abandoned",
   AI_CONSENT_GATE_SHOWN: "Ai consent gate shown",
   AI_CONSENT_APPROVED: "Ai consent approved",
+  AI_CONSENT_GRANTED_INAPP: "Ai consent granted in-app",
 
   // Setup / onboarding events
   SETUP_DISCOVERY_STARTED: "Setup discovery started",
@@ -768,6 +769,7 @@ export type EventPropertyMap = {
   [ANALYTICS_EVENTS.ONBOARDING_ABANDONED]: OnboardingAbandonedProperties;
   [ANALYTICS_EVENTS.AI_CONSENT_GATE_SHOWN]: AiConsentGateShownProperties;
   [ANALYTICS_EVENTS.AI_CONSENT_APPROVED]: never;
+  [ANALYTICS_EVENTS.AI_CONSENT_GRANTED_INAPP]: never;
 
   // Setup / onboarding events
   [ANALYTICS_EVENTS.SETUP_DISCOVERY_STARTED]: SetupDiscoveryStartedProperties;