From 1934494f559a992d6f6a48c99b277894f4782a09 Mon Sep 17 00:00:00 2001
From: dusk <y.bera003.06@protonmail.com>
Date: Wed, 13 Aug 2025 06:38:27 +0300
Subject: [PATCH] appview: db: follow: add GetFollowers and GetFollowing
 functions to fetch Follows
Change-Id: vwmxzzsuuxmvxrxskqorsnsplnrqxkrv

Signed-off-by: dusk <y.bera003.06@protonmail.com>
---
 appview/db/follow.go   | 107 ++++++++++++++++++++++++++---------------
 appview/db/timeline.go |   2 +-
 2 files changed, 68 insertions(+), 41 deletions(-)

diff --git a/appview/db/follow.go b/appview/db/follow.go
index 3a5b97f..8eb53c3 100644
--- a/appview/db/follow.go
+++ b/appview/db/follow.go
@@ -1,7 +1,9 @@
 package db
 
 import (
+	"fmt"
 	"log"
+	"strings"
 	"time"
 )
 
@@ -56,7 +58,7 @@ func DeleteFollowByRkey(e Execer, userDid, rkey string) error {
 func GetFollowerFollowingCount(e Execer, did string) (int, int, error) {
 	followers, following := 0, 0
 	err := e.QueryRow(
-		`SELECT 
+		`SELECT
 		COUNT(CASE WHEN subject_did = ? THEN 1 END) AS followers,
 		COUNT(CASE WHEN user_did = ? THEN 1 END) AS following
 		FROM follows;`, did, did).Scan(&followers, &following)
@@ -66,6 +68,70 @@ func GetFollowerFollowingCount(e Execer, did string) (int, int, error) {
 	return followers, following, nil
 }
 
+func GetFollows(e Execer, limit int, filters ...filter) ([]Follow, error) {
+	var follows []Follow
+
+	var conditions []string
+	var args []any
+	for _, filter := range filters {
+		conditions = append(conditions, filter.Condition())
+		args = append(args, filter.Arg()...)
+	}
+
+	whereClause := ""
+	if conditions != nil {
+		whereClause = " where " + strings.Join(conditions, " and ")
+	}
+	limitClause := ""
+	if limit > 0 {
+		limitClause = " limit ?"
+		args = append(args, limit)
+	}
+
+	query := fmt.Sprintf(
+		`select user_did, subject_did, followed_at, rkey
+		from follows
+		%s
+		order by followed_at desc
+		%s
+	`, whereClause, limitClause)
+
+	rows, err := e.Query(query, args...)
+	if err != nil {
+		return nil, err
+	}
+	for rows.Next() {
+		var follow Follow
+		var followedAt string
+		err := rows.Scan(
+			&follow.UserDid,
+			&follow.SubjectDid,
+			&followedAt,
+			&follow.Rkey,
+		)
+		if err != nil {
+			return nil, err
+		}
+		followedAtTime, err := time.Parse(time.RFC3339, followedAt)
+		if err != nil {
+			log.Println("unable to determine followed at time")
+			follow.FollowedAt = time.Now()
+		} else {
+			follow.FollowedAt = followedAtTime
+		}
+		follows = append(follows, follow)
+	}
+	return follows, nil
+}
+
+func GetFollowers(e Execer, did string) ([]Follow, error) {
+	return GetFollows(e, 0, FilterEq("subject_did", did))
+}
+
+func GetFollowing(e Execer, did string) ([]Follow, error) {
+	return GetFollows(e, 0, FilterEq("user_did", did))
+}
+
 type FollowStatus int
 
 const (
@@ -96,42 +162,3 @@ func GetFollowStatus(e Execer, userDid, subjectDid string) FollowStatus {
 		return IsFollowing
 	}
 }
-
-func GetAllFollows(e Execer, limit int) ([]Follow, error) {
-	var follows []Follow
-
-	rows, err := e.Query(`
-		select user_did, subject_did, followed_at, rkey
-		from follows
-		order by followed_at desc
-		limit ?`, limit,
-	)
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-
-	for rows.Next() {
-		var follow Follow
-		var followedAt string
-		if err := rows.Scan(&follow.UserDid, &follow.SubjectDid, &followedAt, &follow.Rkey); err != nil {
-			return nil, err
-		}
-
-		followedAtTime, err := time.Parse(time.RFC3339, followedAt)
-		if err != nil {
-			log.Println("unable to determine followed at time")
-			follow.FollowedAt = time.Now()
-		} else {
-			follow.FollowedAt = followedAtTime
-		}
-
-		follows = append(follows, follow)
-	}
-
-	if err := rows.Err(); err != nil {
-		return nil, err
-	}
-
-	return follows, nil
-}
diff --git a/appview/db/timeline.go b/appview/db/timeline.go
index b4b1638..0354b81 100644
--- a/appview/db/timeline.go
+++ b/appview/db/timeline.go
@@ -137,7 +137,7 @@ func getTimelineStars(e Execer) ([]TimelineEvent, error) {
 }
 
 func getTimelineFollows(e Execer) ([]TimelineEvent, error) {
-	follows, err := GetAllFollows(e, Limit)
+	follows, err := GetFollows(e, Limit)
 	if err != nil {
 		return nil, err
 	}
-- 
2.43.0