From 01e6f7edbbea5ac85e24b7f28317c2748225ff1a Mon Sep 17 00:00:00 2001
From: Laura Hausmann <laura@hausmann.dev>
Date: Sun, 8 Oct 2023 02:40:06 +0200
Subject: [PATCH] [mastodon-client] Respect ffVisibility for follower/following
 counts

---
 .../server/api/mastodon/converters/user.ts    | 30 ++++++++++++-------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/packages/backend/src/server/api/mastodon/converters/user.ts b/packages/backend/src/server/api/mastodon/converters/user.ts
index fb06ec327..665b95c02 100644
--- a/packages/backend/src/server/api/mastodon/converters/user.ts
+++ b/packages/backend/src/server/api/mastodon/converters/user.ts
@@ -1,6 +1,6 @@
-import { User } from "@/models/entities/user.js";
+import { ILocalUser, User } from "@/models/entities/user.js";
 import config from "@/config/index.js";
-import { DriveFiles, UserProfiles, Users } from "@/models/index.js";
+import { DriveFiles, Followings, UserProfiles, Users } from "@/models/index.js";
 import { EmojiConverter } from "@/server/api/mastodon/converters/emoji.js";
 import { populateEmojis } from "@/misc/populate-emojis.js";
 import { escapeMFM } from "@/server/api/mastodon/converters/mfm.js";
@@ -18,6 +18,7 @@ type Field = {
 
 export class UserConverter {
     public static async encode(u: User, ctx: MastoContext): Promise<MastodonEntity.Account> {
+        const localUser = ctx.user as ILocalUser | null;
         const cache = ctx.cache as AccountCache;
         return cache.locks.acquire(u.id, async () => {
             const cacheHit = cache.accounts.find(p => p.id == u.id);
@@ -39,28 +40,37 @@ export class UserConverter {
                 ? (DriveFiles.findOneBy({ id: u.bannerId }))
                     .then(p => p?.url ?? `${config.url}/static-assets/transparent.png`)
                 : `${config.url}/static-assets/transparent.png`;
-            const followersCount = profile.then(profile => {
+
+            const isFollowedOrSelf = !!localUser &&
+                (localUser.id === u.id ||
+                    Followings.exist({
+                        where: {
+                            followeeId: u.id,
+                            followerId: localUser.id,
+                        },
+                    })
+                );
+
+            const followersCount = profile.then(async profile => {
                 if (profile === null) return u.followersCount;
                 switch (profile.ffVisibility) {
                     case "public":
                         return u.followersCount;
                     case "followers":
-                        // FIXME: check following relationship
-                        return 0;
+                        return Promise.resolve(isFollowedOrSelf).then(isFollowedOrSelf => isFollowedOrSelf ? u.followersCount : 0);
                     case "private":
-                        return 0;
+                        return localUser?.id === profile.userId ? u.followersCount : 0;
                 }
             });
-            const followingCount = profile.then(profile => {
+            const followingCount = profile.then(async profile => {
                 if (profile === null) return u.followingCount;
                 switch (profile.ffVisibility) {
                     case "public":
                         return u.followingCount;
                     case "followers":
-                        // FIXME: check following relationship
-                        return 0;
+                        return Promise.resolve(isFollowedOrSelf).then(isFollowedOrSelf => isFollowedOrSelf ? u.followingCount : 0);
                     case "private":
-                        return 0;
+                        return localUser?.id === profile.userId ? u.followingCount : 0;
                 }
             });