🚚 Move seer acknowledgement mutation into it's own file

Author: nsdeschenes

static/app/components/events/autofix/useSeerAcknowledgeMutation.tsx

@@ -0,0 +1,31 @@
+import {promptsUpdate} from 'sentry/actionCreators/prompts';
+import {useMutation, useQueryClient} from 'sentry/utils/queryClient';
+import useApi from 'sentry/utils/useApi';
+import useOrganization from 'sentry/utils/useOrganization';
+
+export const setupCheckQueryKey = (orgSlug: string) =>
+  `/organizations/${orgSlug}/seer/setup-check/`;
+
+export function useSeerAcknowledgeMutation() {
+  const api = useApi();
+  const queryClient = useQueryClient();
+  const organization = useOrganization();
+
+  const {mutate, isPending, isError} = useMutation({
+    mutationKey: [setupCheckQueryKey(organization.slug)],
+    mutationFn: () => {
+      return promptsUpdate(api, {
+        organization,
+        feature: 'seer_autofix_setup_acknowledged',
+        status: 'dismissed',
+      });
+    },
+    onSuccess: () => {
+      queryClient.invalidateQueries({
+        queryKey: [setupCheckQueryKey(organization.slug)],
+      });
+    },
+  });
+
+  return {mutate, isPending, isError};
+}

static/app/components/searchQueryBuilder/askSeer.tsx

@@ -3,55 +3,25 @@ import styled from '@emotion/styled';
 import {useOption} from '@react-aria/listbox';
 import type {ComboBoxState} from '@react-stately/combobox';
 
-import {promptsUpdate} from 'sentry/actionCreators/prompts';
 import InteractionStateLayer from 'sentry/components/core/interactionStateLayer';
 import {makeOrganizationSeerSetupQueryKey} from 'sentry/components/events/autofix/useOrganizationSeerSetup';
+import {
+  setupCheckQueryKey,
+  useSeerAcknowledgeMutation,
+} from 'sentry/components/events/autofix/useSeerAcknowledgeMutation';
 import ExternalLink from 'sentry/components/links/externalLink';
 import LoadingIndicator from 'sentry/components/loadingIndicator';
 import {useSearchQueryBuilder} from 'sentry/components/searchQueryBuilder/context';
 import {IconSeer} from 'sentry/icons';
 import {t, tct} from 'sentry/locale';
 import {space} from 'sentry/styles/space';
 import {trackAnalytics} from 'sentry/utils/analytics';
-import {
-  useIsFetching,
-  useIsMutating,
-  useMutation,
-  useQueryClient,
-} from 'sentry/utils/queryClient';
-import useApi from 'sentry/utils/useApi';
+import {useIsFetching, useIsMutating} from 'sentry/utils/queryClient';
 import useOrganization from 'sentry/utils/useOrganization';
 
 export const ASK_SEER_ITEM_KEY = 'ask_seer';
 export const ASK_SEER_CONSENT_ITEM_KEY = 'ask_seer_consent';
 
-const setupCheckQueryKey = (orgSlug: string) =>
-  `/organizations/${orgSlug}/seer/setup-check/`;
-
-export function useSeerAcknowledgeMutation() {
-  const api = useApi();
-  const queryClient = useQueryClient();
-  const organization = useOrganization();
-
-  const {mutate} = useMutation({
-    mutationKey: [setupCheckQueryKey(organization.slug)],
-    mutationFn: () => {
-      return promptsUpdate(api, {
-        organization,
-        feature: 'seer_autofix_setup_acknowledged',
-        status: 'dismissed',
-      });
-    },
-    onSuccess: () => {
-      queryClient.invalidateQueries({
-        queryKey: [setupCheckQueryKey(organization.slug)],
-      });
-    },
-  });
-
-  return {mutate};
-}
-
 function AskSeerConsentOption<T>({state}: {state: ComboBoxState<T>}) {
   const organization = useOrganization();
   const itemRef = useRef<HTMLDivElement>(null);

static/gsApp/components/ai/AiSetupDataConsent.tsx

@@ -4,20 +4,19 @@ import styled from '@emotion/styled';
 import autofixSetupImg from 'sentry-images/features/autofix-setup.svg';
 
 import {addSuccessMessage} from 'sentry/actionCreators/indicator';
-import {promptsUpdate} from 'sentry/actionCreators/prompts';
 import {Alert} from 'sentry/components/core/alert';
 import {Button} from 'sentry/components/core/button';
 import {Flex} from 'sentry/components/core/layout';
 import {useAutofixSetup} from 'sentry/components/events/autofix/useAutofixSetup';
 import {useOrganizationSeerSetup} from 'sentry/components/events/autofix/useOrganizationSeerSetup';
+import {useSeerAcknowledgeMutation} from 'sentry/components/events/autofix/useSeerAcknowledgeMutation';
 import ExternalLink from 'sentry/components/links/externalLink';
 import LoadingIndicator from 'sentry/components/loadingIndicator';
 import {DATA_CATEGORY_INFO} from 'sentry/constants';
 import {IconRefresh, IconSeer} from 'sentry/icons';
 import {t, tct} from 'sentry/locale';
 import {space} from 'sentry/styles/space';
 import {DataCategory} from 'sentry/types/core';
-import {useMutation, useQueryClient} from 'sentry/utils/queryClient';
 import useApi from 'sentry/utils/useApi';
 import {useNavigate} from 'sentry/utils/useNavigate';
 import useOrganization from 'sentry/utils/useOrganization';
@@ -37,7 +36,6 @@ type AiSetupDataConsentProps = {
 function AiSetupDataConsent({groupId}: AiSetupDataConsentProps) {
   const api = useApi({persistInFlight: true});
   const organization = useOrganization();
-  const queryClient = useQueryClient();
   const navigate = useNavigate();
   const subscription = useSubscription();
 
@@ -82,29 +80,7 @@ function AiSetupDataConsent({groupId}: AiSetupDataConsentProps) {
     !isTouchCustomer &&
     !hasSeerButNeedsPayg;
 
-  const autofixAcknowledgeMutation = useMutation({
-    mutationFn: () => {
-      return promptsUpdate(api, {
-        organization,
-        feature: 'seer_autofix_setup_acknowledged',
-        status: 'dismissed',
-      });
-    },
-    onSuccess: () => {
-      // Invalidate the appropriate query based on mode
-      if (isGroupMode && groupId) {
-        queryClient.invalidateQueries({
-          queryKey: [
-            `/organizations/${organization.slug}/issues/${groupId}/autofix/setup/`,
-          ],
-        });
-      } else {
-        queryClient.invalidateQueries({
-          queryKey: [`/organizations/${organization.slug}/seer/setup-check/`],
-        });
-      }
-    },
-  });
+  const autofixAcknowledgeMutation = useSeerAcknowledgeMutation();
 
   function handlePurchaseSeer() {
     navigate(`/settings/billing/checkout/?referrer=ai_setup_data_consent`);