From 484b09b564da5002392840087248b7e797482292 Mon Sep 17 00:00:00 2001
From: Steven Tey <stevensteel97@gmail.com>
Date: Sat, 19 Oct 2024 18:08:56 -0700
Subject: [PATCH] improve swr revalidate

---
 apps/web/ui/domains/register-domain-form.tsx | 8 ++------
 apps/web/ui/modals/delete-domain-modal.tsx   | 5 +++--
 2 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/apps/web/ui/domains/register-domain-form.tsx b/apps/web/ui/domains/register-domain-form.tsx
index f098f28094..d7649f601a 100644
--- a/apps/web/ui/domains/register-domain-form.tsx
+++ b/apps/web/ui/domains/register-domain-form.tsx
@@ -106,17 +106,13 @@ export function RegisterDomainForm({
     } else {
       toast.success("Domain registered successfully!");
 
-      // Mutate workspace, domains links
+      // Mutate workspace, domains, and links
       await Promise.all([
         mutate(`/api/workspaces/${workspace.slug}`),
         mutate(
           (key) =>
             typeof key === "string" &&
-            (key.startsWith(`/api/domains?workspaceId=${workspace.id}`) ||
-              key.startsWith(`/api/domains/count?workspaceId=${workspace.id}`)),
-        ),
-        mutate(
-          (key) => typeof key === "string" && key.startsWith("/api/links"),
+            (key.startsWith("/api/domains") || key.startsWith("/api/links")),
           undefined,
           { revalidate: true },
         ),
diff --git a/apps/web/ui/modals/delete-domain-modal.tsx b/apps/web/ui/modals/delete-domain-modal.tsx
index 8e0c0cf067..98de227981 100644
--- a/apps/web/ui/modals/delete-domain-modal.tsx
+++ b/apps/web/ui/modals/delete-domain-modal.tsx
@@ -61,8 +61,9 @@ function DeleteDomainModal({
             if (res.status === 200) {
               await mutate(
                 (key) =>
-                  typeof key === "string" &&
-                  key.startsWith(`/api/domains?workspaceId=${id}`),
+                  typeof key === "string" && key.startsWith("/api/domains"),
+                undefined,
+                { revalidate: true },
               );
               setShowDeleteDomainModal(false);
               toast.success("Successfully deleted domain!");