close popup on skip fishing

AplinkosMinisterija · Dec 1, 2023 · 7e4638b · 7e4638b
1 parent d7d4075
commit 7e4638b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/components/containers/FishingLocation.tsx b/src/components/containers/FishingLocation.tsx
@@ -32,11 +32,12 @@ const useLocationMutation = (onSuccess: (value: any) => void) => {
   return { getLocation, locationLoading };
 };
 
-const useSkipMutation = () => {
+const useSkipMutation = (onSuccess: () => void) => {
   const { isLoading: skipLoading, mutateAsync: skipFishing } = useMutation(api.skipFishing, {
     onError: ({ response }) => {
       handleAlert(response);
     },
+    onSuccess,
     retry: false,
   });
   return { skipLoading, skipFishing };
@@ -64,7 +65,7 @@ const FishingLocation = ({ setLocation, location, coordinates }: any) => {
   const [locationType, setLocationType] = useState<LocationType | null>(null);
 
   const { startFishing, startLoading } = useStartMutation();
-  const { skipFishing, skipLoading } = useSkipMutation();
+  const { skipFishing, skipLoading } = useSkipMutation(() => setShowSkipFishing(false));
   const { getLocation, locationLoading } = useLocationMutation((value) => setLocation(value));
 
   const handleSelectLocation = (type: LocationType) => () => {