Refactor Follow component to use write function

tapexyz · Jan 23, 2024 · c296c4b · c296c4b
1 parent c23aa1e
commit c296c4b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 24 deletions.
diff --git a/apps/web/src/components/Profile/BasicInfo/Follow.tsx b/apps/web/src/components/Profile/BasicInfo/Follow.tsx
@@ -71,6 +71,15 @@ const Follow: FC<Props> = ({ profile, onSubscribe }) => {
     }
   })
 
+  const write = ({ args }: { args: any[] }) => {
+    return writeContract({
+      address: LENSHUB_PROXY_ADDRESS,
+      abi: LENSHUB_PROXY_ABI,
+      functionName: 'follow',
+      args
+    })
+  }
+
   const [broadcast] = useBroadcastOnchainMutation({
     onCompleted: ({ broadcastOnchain }) =>
       onCompleted(broadcastOnchain.__typename),
@@ -101,21 +110,11 @@ const Follow: FC<Props> = ({ profile, onSubscribe }) => {
             variables: { request: { id, signature } }
           })
           if (data?.broadcastOnchain?.__typename === 'RelayError') {
-            return writeContract({
-              address: LENSHUB_PROXY_ADDRESS,
-              abi: LENSHUB_PROXY_ABI,
-              functionName: 'follow',
-              args
-            })
+            return write({ args })
           }
           return
         }
-        return writeContract({
-          address: LENSHUB_PROXY_ADDRESS,
-          abi: LENSHUB_PROXY_ABI,
-          functionName: 'follow',
-          args
-        })
+        return write({ args })
       } catch {
         setLoading(false)
       }

diff --git a/apps/web/src/components/Profile/BasicInfo/SuperFollow.tsx b/apps/web/src/components/Profile/BasicInfo/SuperFollow.tsx
@@ -86,6 +86,15 @@ const SuperFollow: FC<Props> = ({ profile, onJoin }) => {
     onError
   })
 
+  const write = ({ args }: { args: any[] }) => {
+    return writeContract({
+      address: LENSHUB_PROXY_ADDRESS,
+      abi: LENSHUB_PROXY_ABI,
+      functionName: 'follow',
+      args
+    })
+  }
+
   const { data: followModuleData } = useProfileFollowModuleQuery({
     variables: { request: { forProfileId: profile?.id } },
     skip: !profile?.id
@@ -155,21 +164,11 @@ const SuperFollow: FC<Props> = ({ profile, onJoin }) => {
             variables: { request: { id, signature } }
           })
           if (data?.broadcastOnchain.__typename === 'RelayError') {
-            return writeContract({
-              address: LENSHUB_PROXY_ADDRESS,
-              abi: LENSHUB_PROXY_ABI,
-              functionName: 'follow',
-              args
-            })
+            return write({ args })
           }
           return
         }
-        return writeContract({
-          address: LENSHUB_PROXY_ADDRESS,
-          abi: LENSHUB_PROXY_ABI,
-          functionName: 'follow',
-          args
-        })
+        return write({ args })
       } catch {
         setLoading(false)
       }