From 9329d796958ca6c6a3beeaf7270eb7bcffe31b3f Mon Sep 17 00:00:00 2001
From: Matt Gabrenya <matt@buildyourweb.app>
Date: Thu, 31 Aug 2023 15:49:12 -0700
Subject: [PATCH] fix: use count_links for followers count

---
 .../follows/src/follower_to_creators.rs       | 28 +++++++++++
 ui/src/components/BaseAgentProfileDetail.vue  |  2 +
 ui/src/pages/AgentProfile.vue                 | 49 +++++++++++++++++--
 3 files changed, 75 insertions(+), 4 deletions(-)

diff --git a/dnas/mewsfeed/zomes/coordinator/follows/src/follower_to_creators.rs b/dnas/mewsfeed/zomes/coordinator/follows/src/follower_to_creators.rs
index 54af1af9..b777ced1 100644
--- a/dnas/mewsfeed/zomes/coordinator/follows/src/follower_to_creators.rs
+++ b/dnas/mewsfeed/zomes/coordinator/follows/src/follower_to_creators.rs
@@ -52,6 +52,34 @@ pub fn get_followers_for_creator(
     Ok(agents)
 }
 
+#[hdk_extern]
+pub fn count_creators_for_follower(
+    follower: AgentPubKey,
+) -> ExternResult<usize> {
+    let query = LinkQuery::new(
+        follower,
+        LinkTypeFilter::single_type(
+            ZomeIndex(2),
+            LinkType(0), // LinkTypes::FollowerToCreators
+        ),
+    );
+    count_links(query)
+}
+
+#[hdk_extern]
+pub fn count_followers_for_creator(
+    creator: AgentPubKey,
+) -> ExternResult<usize> {
+    let query = LinkQuery::new(
+        creator,
+        LinkTypeFilter::single_type(
+            ZomeIndex(2),
+            LinkType(1), // LinkTypes::CreatorToFollowers
+        ),
+    );
+    count_links(query)
+}
+
 #[hdk_extern]
 pub fn get_follower_links_for_creator(
     input: GetFollowersForCreatorInput,
diff --git a/ui/src/components/BaseAgentProfileDetail.vue b/ui/src/components/BaseAgentProfileDetail.vue
index 1545340a..7f9d1d5c 100644
--- a/ui/src/components/BaseAgentProfileDetail.vue
+++ b/ui/src/components/BaseAgentProfileDetail.vue
@@ -95,6 +95,7 @@
             v-if="!isMyProfile"
             :agentPubKey="agentPubKey"
             :big="bigFollowButton"
+            @toggle-follow="(val) => emit('toggle-follow', val)"
           />
           <div v-else-if="!hideEditButton" class="flex flex-col space-y-2">
             <button
@@ -222,6 +223,7 @@ const emit = defineEmits([
   "click-edit-profile",
   "click-creators",
   "click-followers",
+  "toggle-follow",
 ]);
 
 const client = (inject("client") as ComputedRef<AppAgentClient>).value;
diff --git a/ui/src/pages/AgentProfile.vue b/ui/src/pages/AgentProfile.vue
index aa54cbd3..a62d8382 100644
--- a/ui/src/pages/AgentProfile.vue
+++ b/ui/src/pages/AgentProfile.vue
@@ -5,8 +5,8 @@
         :profile="profileWithContext?.profile"
         :joined-timestamp="profileWithContext?.joinedTimestamp"
         :agentPubKey="agentPubKey"
-        :creators-count="creators ? creators.length : 0"
-        :followers-count="followers ? followers.length : 0"
+        :creators-count="creatorsCount ? creatorsCount : 0"
+        :followers-count="followersCount ? followersCount : 0"
         class="bg-base-200/75 rounded-3xl"
         style="-webkit-backdrop-filter: blur(10px)"
         enable-copy-agent-pub-key
@@ -22,6 +22,7 @@
             if (creators && creators.length > 0) showCreatorsListDialog = true;
           }
         "
+        @toggle-follow="refetchFollowersCount"
       />
       <EditAgentProfileDialog
         v-if="profileWithContext"
@@ -240,7 +241,7 @@ const {
 watch(errorProfile, showError);
 
 const fetchFollowers = async () => {
-  const agents: AgentPubKey[] = await await client.callZome({
+  const agents: AgentPubKey[] = await client.callZome({
     role_name: "mewsfeed",
     zome_name: "follows",
     fn_name: "get_followers_for_creator",
@@ -278,7 +279,7 @@ const { data: followers, error: errorFollowers } = useQuery({
 watch(errorFollowers, showError);
 
 const fetchCreators = async () => {
-  const agents: AgentPubKey[] = await await client.callZome({
+  const agents: AgentPubKey[] = await client.callZome({
     role_name: "mewsfeed",
     zome_name: "follows",
     fn_name: "get_creators_for_follower",
@@ -314,4 +315,44 @@ const { data: creators, error: errorCreators } = useQuery({
   queryFn: fetchCreators,
 });
 watch(errorCreators, showError);
+
+const fetchCreatorsCount = async (): Promise<number> =>
+  client.callZome({
+    role_name: "mewsfeed",
+    zome_name: "follows",
+    fn_name: "count_creators_for_follower",
+    payload: decodeHashFromBase64(route.params.agentPubKey as string),
+  });
+
+const { data: creatorsCount, error: errorCreatorsCount } = useQuery({
+  queryKey: [
+    "follows",
+    "count_creators_for_follower",
+    route.params.agentPubKey as string,
+  ],
+  queryFn: fetchCreatorsCount,
+});
+watch(errorCreatorsCount, showError);
+
+const fetchFollowersCount = async (): Promise<number> =>
+  client.callZome({
+    role_name: "mewsfeed",
+    zome_name: "follows",
+    fn_name: "count_followers_for_creator",
+    payload: decodeHashFromBase64(route.params.agentPubKey as string),
+  });
+
+const {
+  data: followersCount,
+  error: errorFollowersCount,
+  refetch: refetchFollowersCount,
+} = useQuery({
+  queryKey: [
+    "follows",
+    "count_followers_for_creator",
+    route.params.agentPubKey as string,
+  ],
+  queryFn: fetchFollowersCount,
+});
+watch(errorFollowersCount, showError);
 </script>