Refetch setup on button press

getsentry · Nov 12, 2024 · c6295fd · c6295fd
1 parent 523391a
commit c6295fd
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 5 deletions.
diff --git a/static/app/components/events/autofix/autofixDrawer.tsx b/static/app/components/events/autofix/autofixDrawer.tsx
@@ -88,7 +88,11 @@ interface AutofixDrawerProps {
 export function AutofixDrawer({group, project, event}: AutofixDrawerProps) {
   const {autofixData, triggerAutofix, reset} = useAiAutofix(group, event);
   const {data: summaryData, isError} = useGroupSummary(group.id, group.issueCategory);
-  const {data: setupData, isPending: isSetupLoading} = useAutofixSetup({
+  const {
+    data: setupData,
+    isPending: isSetupLoading,
+    refetch: refetchSetup,
+  } = useAutofixSetup({
     groupId: group.id,
   });
 
@@ -148,7 +152,8 @@ export function AutofixDrawer({group, project, event}: AutofixDrawerProps) {
             <AutofixSetupContent
               projectId={project.id}
               groupId={group.id}
-              closeModal={() => {}} // Setup is inline, so no modal to close
+              closeModal={() => {}}
+              refetchSetup={refetchSetup}
             />
           </SetupContainer>
         ) : !autofixData ? (

diff --git a/static/app/components/events/autofix/autofixSetupModal.tsx b/static/app/components/events/autofix/autofixSetupModal.tsx
@@ -145,12 +145,19 @@ function AutofixGithubIntegrationStep({
   canStartAutofix,
   closeModal,
   isLastStep,
+  refetchSetup,
 }: {
   autofixSetup: AutofixSetupResponse;
   canStartAutofix: boolean;
   closeModal: () => void;
   isLastStep?: boolean;
+  refetchSetup?: () => void;
 }) {
+  const handleClose = () => {
+    refetchSetup?.();
+    closeModal();
+  };
+
   const sortedRepos = useMemo(
     () =>
       autofixSetup.githubWriteIntegration.repos.toSorted((a, b) => {
@@ -186,7 +193,7 @@ function AutofixGithubIntegrationStep({
               priority="primary"
               size="sm"
               disabled={!canStartAutofix}
-              onClick={closeModal}
+              onClick={handleClose}
             >
               {t("Let's Go!")}
             </Button>
@@ -227,7 +234,7 @@ function AutofixGithubIntegrationStep({
               priority="primary"
               size="sm"
               disabled={!canStartAutofix}
-              onClick={closeModal}
+              onClick={handleClose}
             >
               {t('Skip & Enable Autofix')}
             </Button>
@@ -262,7 +269,7 @@ function AutofixGithubIntegrationStep({
             priority="primary"
             size="sm"
             disabled={!canStartAutofix}
-            onClick={closeModal}
+            onClick={handleClose}
           >
             {t('Skip & Enable Autofix')}
           </Button>
@@ -276,12 +283,14 @@ function AutofixSetupSteps({
   autofixSetup,
   closeModal,
   canStartAutofix,
+  refetchSetup,
 }: {
   autofixSetup: AutofixSetupResponse;
   canStartAutofix: boolean;
   closeModal: () => void;
   groupId: string;
   projectId: string;
+  refetchSetup?: () => void;
 }) {
   return (
     <GuidedSteps>
@@ -304,6 +313,7 @@ function AutofixSetupSteps({
           canStartAutofix={canStartAutofix}
           closeModal={closeModal}
           isLastStep
+          refetchSetup={refetchSetup}
         />
       </GuidedSteps.Step>
     </GuidedSteps>
@@ -314,10 +324,12 @@ export function AutofixSetupContent({
   projectId,
   groupId,
   closeModal,
+  refetchSetup,
 }: {
   closeModal: () => void;
   groupId: string;
   projectId: string;
+  refetchSetup?: () => void;
 }) {
   const organization = useOrganization();
   const {data, canStartAutofix, isPending, isError} = useAutofixSetup(
@@ -363,6 +375,7 @@ export function AutofixSetupContent({
         autofixSetup={data}
         canStartAutofix={canStartAutofix}
         closeModal={closeModal}
+        refetchSetup={refetchSetup}
       />
     </Fragment>
   );