From 101ac7c606932ce11cabc524c4c5666c6b2cd284 Mon Sep 17 00:00:00 2001
From: DIYgod <i@diygod.me>
Date: Mon, 30 Jan 2023 15:44:54 +0000
Subject: [PATCH] feat: useIsOwner hook

---
 src/components/dashboard/DashboardLayout.tsx | 41 ++++----------------
 src/hooks/useIsOwner.ts                      | 33 ++++++++++++++++
 2 files changed, 40 insertions(+), 34 deletions(-)
 create mode 100644 src/hooks/useIsOwner.ts

diff --git a/src/components/dashboard/DashboardLayout.tsx b/src/components/dashboard/DashboardLayout.tsx
index 5735d075..93dcfd26 100644
--- a/src/components/dashboard/DashboardLayout.tsx
+++ b/src/components/dashboard/DashboardLayout.tsx
@@ -1,8 +1,6 @@
 import clsx from "clsx"
-import Link from "next/link"
 import { useRouter } from "next/router"
-import React, { useState, useEffect } from "react"
-import { useAccountState } from "@crossbell/connect-kit"
+import React, { useEffect } from "react"
 
 import { APP_NAME } from "~/lib/env"
 import { getSiteLink } from "~/lib/helpers"
@@ -11,13 +9,14 @@ import { UniLink } from "../ui/UniLink"
 import { DashboardSidebar } from "./DashboardSidebar"
 import { useAccountSites } from "~/queries/site"
 import { ConnectButton } from "~/components/common/ConnectButton"
-import { useGetSite, useIsOperators } from "~/queries/site"
+import { useGetSite } from "~/queries/site"
 import { toGateway } from "~/lib/ipfs-parser"
 import { Avatar } from "~/components/ui/Avatar"
 import {
   useShowNotificationModal,
   useNotifications,
 } from "@crossbell/notification"
+import { useIsOwner } from "~/hooks/useIsOwner"
 
 export function DashboardLayout({
   children,
@@ -29,41 +28,15 @@ export function DashboardLayout({
   const router = useRouter()
   const subdomain = router.query.subdomain as string
   const site = useGetSite(subdomain)
-  const [ssrReady, isConnected, address] = useAccountState((s) => [
-    s.ssrReady,
-    !!s.computed.account,
-    s.computed.account?.address,
-  ])
-
   const userSite = useAccountSites()
 
-  const isOperator = useIsOperators({
-    characterId: +(site.data?.metadata?.proof || 0),
-    operator: address,
-  })
+  const isOwner = useIsOwner()
 
   useEffect(() => {
-    if (ssrReady && !isConnected) {
-      router.push("/")
+    if (isOwner.isSuccess && !isOwner.data) {
+      router.push("/dashboard")
     }
-
-    if (userSite.isSuccess && subdomain) {
-      if (
-        !userSite.data?.find((site) => site.username === subdomain) &&
-        !isOperator.data
-      ) {
-        router.push("/dashboard")
-      }
-    }
-  }, [
-    ssrReady,
-    isConnected,
-    router,
-    userSite.isSuccess,
-    userSite.data,
-    subdomain,
-    isOperator.data,
-  ])
+  }, [router, isOwner])
 
   const showNotificationModal = useShowNotificationModal()
   const { isAllRead } = useNotifications()
diff --git a/src/hooks/useIsOwner.ts b/src/hooks/useIsOwner.ts
new file mode 100644
index 00000000..9c8ec36e
--- /dev/null
+++ b/src/hooks/useIsOwner.ts
@@ -0,0 +1,33 @@
+import { useAccountBalance as _useAccountBalance } from "@crossbell/connect-kit"
+import { useRouter } from "next/router"
+import { useAccountSites } from "~/queries/site"
+import { useGetSite, useIsOperators } from "~/queries/site"
+import { useAccountState } from "@crossbell/connect-kit"
+
+export function useIsOwner() {
+  const router = useRouter()
+  const subdomain = router.query.subdomain as string
+  const site = useGetSite(subdomain)
+
+  const userSite = useAccountSites()
+  const [address] = useAccountState((s) => [s.computed.account?.address])
+
+  const isOperator = useIsOperators({
+    characterId: +(site.data?.metadata?.proof || 0),
+    operator: address,
+  })
+
+  return {
+    isSuccess:
+      subdomain &&
+      address &&
+      site.isSuccess &&
+      userSite.isSuccess &&
+      isOperator.isSuccess,
+    data:
+      subdomain &&
+      address &&
+      (userSite.data?.find((site) => site.username === subdomain) ||
+        isOperator.data),
+  }
+}