From 501c95f0ab50eb3dc4a8ff39bce33d49f8da7f5c Mon Sep 17 00:00:00 2001
From: Martti Malmi <sirius@iki.fi>
Date: Tue, 13 Jul 2021 15:12:46 +0300
Subject: [PATCH] show own content in follows

---
 src/js/Session.js                | 11 +++++++++++
 src/js/components/MessageFeed.js |  4 +---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/js/Session.js b/src/js/Session.js
index 5040ace9..b2704c67 100644
--- a/src/js/Session.js
+++ b/src/js/Session.js
@@ -50,6 +50,15 @@ function getExtendedFollows(callback, k, maxDepth = 3, currentDepth = 1) {
     callback(k, follows[k]);
   }
 
+  function removeFollow(k, followDistance, follower) {
+    if (follows[k]) {
+      follows[k].followers.delete(follower);
+      if (followDistance === 1) {
+        State.local.get('groups').get('follows').get(k).put(false);
+      }
+    }
+  }
+
   addFollow(k, currentDepth - 1);
 
   State.public.user(k).get('follow').map().on((isFollowing, followedKey) => { // TODO: unfollow
@@ -59,6 +68,8 @@ function getExtendedFollows(callback, k, maxDepth = 3, currentDepth = 1) {
       if (currentDepth < maxDepth) {
         getExtendedFollows(callback, followedKey, maxDepth, currentDepth + 1);
       }
+    } else {
+      removeFollow(followedKey, currentDepth, k);
     }
   });
 
diff --git a/src/js/components/MessageFeed.js b/src/js/components/MessageFeed.js
index b4b15d18..312fff12 100644
--- a/src/js/components/MessageFeed.js
+++ b/src/js/components/MessageFeed.js
@@ -36,9 +36,7 @@ class MessageFeed extends Component {
     if (this.props.node) {
       this.props.node.map().on((...args) => this.handleMessage(...args));
     } else if (this.props.group && this.props.path) { // TODO: make group use the same basic gun api
-      const group = this.props.group === 'follows' ?
-        State.public.user().get('follow') :
-        State.local.get('groups').get(this.props.group);
+      const group = State.local.get('groups').get(this.props.group);
       State.group(group).map(this.props.path, (...args) => this.handleMessage(...args));
     }
   }