From 7a88b342a62d32aa7a86368cbcc6a6a9cb013ebb Mon Sep 17 00:00:00 2001
From: Arnaud <arnaud@status.im>
Date: Fri, 27 Sep 2024 11:41:09 +0200
Subject: [PATCH] Remove mutationKey to avoid confusion with invalidate query

---
 src/components/Availability/AvailabilityForm.tsx            | 2 +-
 src/components/Availability/useAvailabilityMutation.ts      | 2 +-
 src/components/LogLevel/LogLevel.tsx                        | 1 -
 .../StorageRequestSetup/useStorageRequestMutation.ts        | 6 ++++--
 4 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/components/Availability/AvailabilityForm.tsx b/src/components/Availability/AvailabilityForm.tsx
index 2d6a92c..f9a9c43 100644
--- a/src/components/Availability/AvailabilityForm.tsx
+++ b/src/components/Availability/AvailabilityForm.tsx
@@ -60,7 +60,7 @@ export function AvailabilityForm({
 
   const max = availabilityMax(space);
   const isValid = isAvailabilityValid(availability, max);
-  console.info(isValid, max);
+
   const helper = isValid
     ? "Total size of availability's storage in bytes"
     : "The total size cannot exceed the space available.";
diff --git a/src/components/Availability/useAvailabilityMutation.ts b/src/components/Availability/useAvailabilityMutation.ts
index 8a36fcb..99c12ae 100644
--- a/src/components/Availability/useAvailabilityMutation.ts
+++ b/src/components/Availability/useAvailabilityMutation.ts
@@ -21,7 +21,6 @@ export function useAvailabilityMutation(
   const [error, setError] = useState<Error | null>(null);
 
   const { mutateAsync } = useMutation({
-    mutationKey: ["availabilities"],
     mutationFn: ({
       totalSize,
       totalSizeUnit,
@@ -47,6 +46,7 @@ export function useAvailabilityMutation(
       }).then((s) => Promises.rejectOnError(s));
     },
     onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["availabilities"] });
       queryClient.invalidateQueries({ queryKey: ["space"] });
 
       WebStorage.delete("availability");
diff --git a/src/components/LogLevel/LogLevel.tsx b/src/components/LogLevel/LogLevel.tsx
index 66b9f1a..0219b78 100644
--- a/src/components/LogLevel/LogLevel.tsx
+++ b/src/components/LogLevel/LogLevel.tsx
@@ -15,7 +15,6 @@ export function LogLevel() {
   const queryClient = useQueryClient();
   const [level, setLevel] = useState<CodexLogLevel>("DEBUG");
   const { mutateAsync, isPending } = useMutation({
-    mutationKey: ["debug"],
     mutationFn: (level: CodexLogLevel) =>
       CodexSdk.debug.setLogLevel(level).then((s) => Promises.rejectOnError(s)),
     onSuccess: () => {
diff --git a/src/components/StorageRequestSetup/useStorageRequestMutation.ts b/src/components/StorageRequestSetup/useStorageRequestMutation.ts
index fcfb03a..82c2ef5 100644
--- a/src/components/StorageRequestSetup/useStorageRequestMutation.ts
+++ b/src/components/StorageRequestSetup/useStorageRequestMutation.ts
@@ -1,6 +1,6 @@
 import { CodexCreateStorageRequestInput } from "@codex-storage/sdk-js";
 import { CodexSdk } from "../../sdk/codex";
-import { useMutation } from "@tanstack/react-query";
+import { useMutation, useQueryClient } from "@tanstack/react-query";
 import { Promises } from "../../utils/promises";
 import { WebStorage } from "../../utils/web-storage";
 import {
@@ -15,14 +15,16 @@ export function useStorageRequestMutation(
   state: StepperState
 ) {
   const [error, setError] = useState<Error | null>(null);
+  const queryClient = useQueryClient();
 
   const { mutateAsync } = useMutation({
-    mutationKey: ["purchases"],
     mutationFn: (input: CodexCreateStorageRequestInput) =>
       CodexSdk.marketplace
         .createStorageRequest(input)
         .then((s) => Promises.rejectOnError(s)),
     onSuccess: async () => {
+      queryClient.invalidateQueries({ queryKey: ["mutationKey"] });
+
       //   if (!requestId.startsWith("0x")) {
       //     requestId = "0x" + requestId;
       //   }