From 162802914fc309377150a0d622dd598ca643604e Mon Sep 17 00:00:00 2001
From: Alessio <me@playfulpachyderm.com>
Date: Sat, 9 Sep 2023 18:24:01 -0300
Subject: [PATCH] Add a parameter to a Cursor to fetch a user's likes

---
 pkg/persistence/compound_queries_test.go | 18 ++++++++++++++
 pkg/persistence/compound_ssf_queries.go  | 31 +++++++++++++++++++++++-
 2 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/pkg/persistence/compound_queries_test.go b/pkg/persistence/compound_queries_test.go
index 9e68fb4..a83c8b3 100644
--- a/pkg/persistence/compound_queries_test.go
+++ b/pkg/persistence/compound_queries_test.go
@@ -146,6 +146,24 @@ func TestUserFeedWithTombstone(t *testing.T) {
 	assert.Equal(tombstone_tweet.TombstoneText, "This Tweet was deleted by the Tweet author")
 }
 
+func TestUserLikesFeed(t *testing.T) {
+	require := require.New(t)
+	assert := assert.New(t)
+
+	profile, err := persistence.LoadProfile("../../sample_data/profile")
+	require.NoError(err)
+
+	// Fetch @Peter_Nimitz user feed while logged in as @MysteryGrove
+	c := persistence.NewUserFeedLikesCursor(UserHandle("MysteryGrove"))
+	feed, err := profile.NextPage(c, UserID(0))
+	require.NoError(err)
+
+	// Should have "liked" 1 tweet
+	require.Len(feed.Tweets, 1)
+	_, is_ok := feed.Tweets[1413646595493568516]
+	assert.True(is_ok)
+}
+
 func TestTweetDetailWithReplies(t *testing.T) {
 	require := require.New(t)
 	assert := assert.New(t)
diff --git a/pkg/persistence/compound_ssf_queries.go b/pkg/persistence/compound_ssf_queries.go
index 8ce5576..6550a7f 100644
--- a/pkg/persistence/compound_ssf_queries.go
+++ b/pkg/persistence/compound_ssf_queries.go
@@ -110,6 +110,7 @@ type Cursor struct {
 	RetweetedByUserHandle scraper.UserHandle
 	ByUserHandle          scraper.UserHandle
 	ToUserHandles         []scraper.UserHandle
+	LikedByUserHandle     scraper.UserHandle
 	SinceTimestamp        scraper.Timestamp
 	UntilTimestamp        scraper.Timestamp
 	FilterLinks           Filter
@@ -170,6 +171,22 @@ func NewUserFeedCursor(h scraper.UserHandle) Cursor {
 	}
 }
 
+// Generate a cursor for a User's Likes
+func NewUserFeedLikesCursor(h scraper.UserHandle) Cursor {
+	return Cursor{
+		Keywords:       []string{},
+		ToUserHandles:  []scraper.UserHandle{},
+		SinceTimestamp: scraper.TimestampFromUnix(0),
+		UntilTimestamp: scraper.TimestampFromUnix(0),
+		CursorPosition: CURSOR_START,
+		CursorValue:    0,
+		SortOrder:      SORT_ORDER_NEWEST,
+		PageSize:       50,
+
+		LikedByUserHandle: h,
+	}
+}
+
 func NewCursorFromSearchQuery(q string) (Cursor, error) {
 	ret := NewCursor()
 	is_in_quotes := false
@@ -351,13 +368,23 @@ func (p Profile) NextPage(c Cursor, current_user_id scraper.UserID) (Feed, error
 		where_clauses = append(where_clauses, "retweet_id = 0")
 	}
 
+	liked_by_filter_join_clause := ""
+	if c.LikedByUserHandle != "" {
+		liked_by_filter_join_clause = " join likes filter_likes on tweets.id = filter_likes.tweet_id "
+		where_clauses = append(where_clauses, "filter_likes.user_id = (select id from users where handle like ?) ")
+		bind_values = append(bind_values, c.LikedByUserHandle)
+	}
+
 	// Pagination
 	if c.CursorPosition != CURSOR_START {
 		where_clauses = append(where_clauses, c.SortOrder.PaginationWhereClause())
 		bind_values = append(bind_values, c.CursorValue)
 	}
 
-	where_clause := "where " + strings.Join(where_clauses, " and ")
+	where_clause := ""
+	if len(where_clauses) > 0 {
+		where_clause = "where " + strings.Join(where_clauses, " and ")
+	}
 
 	q := `select * from (
 	select ` + TWEETS_ALL_SQL_FIELDS + `,
@@ -366,6 +393,7 @@ func (p Profile) NextPage(c Cursor, current_user_id scraper.UserID) (Feed, error
       from tweets
  left join tombstone_types on tweets.tombstone_type = tombstone_types.rowid
  left join likes on tweets.id = likes.tweet_id and likes.user_id = ?
+     ` + liked_by_filter_join_clause + `
      ` + where_clause + ` ` + c.SortOrder.OrderByClause() + ` limit ?
     )
 
@@ -379,6 +407,7 @@ func (p Profile) NextPage(c Cursor, current_user_id scraper.UserID) (Feed, error
  left join tweets on retweets.tweet_id = tweets.id
  left join tombstone_types on tweets.tombstone_type = tombstone_types.rowid
  left join likes on tweets.id = likes.tweet_id and likes.user_id = ?
+     ` + liked_by_filter_join_clause + `
      ` + where_clause + `
      ` + c.SortOrder.OrderByClause() + `
      limit ?