Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
5.6 kB · 240 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241package db
import ( "sort"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.org/core/appview/models")
// TODO: this gathers heterogenous events from different sources and aggregates// them in code; if we did this entirely in sql, we could order and limit and paginate easilyfunc MakeTimeline(e Execer, limit int, loggedInUserDid string, limitToUsersIsFollowing bool) ([]models.TimelineEvent, error) { var events []models.TimelineEvent
var userIsFollowing []string if limitToUsersIsFollowing { following, err := GetFollowing(e, loggedInUserDid) if err != nil { return nil, err }
userIsFollowing = make([]string, 0, len(following)) for _, follow := range following { userIsFollowing = append(userIsFollowing, follow.SubjectDid) } }
repos, err := getTimelineRepos(e, limit, loggedInUserDid, userIsFollowing) if err != nil { return nil, err }
stars, err := getTimelineStars(e, limit, loggedInUserDid, userIsFollowing) if err != nil { return nil, err }
follows, err := getTimelineFollows(e, limit, loggedInUserDid, userIsFollowing) if err != nil { return nil, err }
events = append(events, repos...) events = append(events, stars...) events = append(events, follows...)
sort.Slice(events, func(i, j int) bool { return events[i].EventAt.After(events[j].EventAt) })
// Limit the slice to 100 events if len(events) > limit { events = events[:limit] }
return events, nil}
func fetchStarStatuses(e Execer, loggedInUserDid string, repos []models.Repo) (map[string]bool, error) { if loggedInUserDid == "" { return nil, nil }
var repoAts []syntax.ATURI for _, r := range repos { repoAts = append(repoAts, r.RepoAt()) }
return GetStarStatuses(e, loggedInUserDid, repoAts)}
func getRepoStarInfo(repo *models.Repo, starStatuses map[string]bool) (bool, int64) { var isStarred bool if starStatuses != nil { isStarred = starStatuses[repo.RepoAt().String()] }
var starCount int64 if repo.RepoStats != nil { starCount = int64(repo.RepoStats.StarCount) }
return isStarred, starCount}
func getTimelineRepos(e Execer, limit int, loggedInUserDid string, userIsFollowing []string) ([]models.TimelineEvent, error) { filters := make([]filter, 0) if userIsFollowing != nil { filters = append(filters, FilterIn("did", userIsFollowing)) }
repos, err := GetRepos(e, limit, filters...) if err != nil { return nil, err }
// fetch all source repos var args []string for _, r := range repos { if r.Source != "" { args = append(args, r.Source) } }
var origRepos []models.Repo if args != nil { origRepos, err = GetRepos(e, 0, FilterIn("at_uri", args)) } if err != nil { return nil, err }
uriToRepo := make(map[string]models.Repo) for _, r := range origRepos { uriToRepo[r.RepoAt().String()] = r }
starStatuses, err := fetchStarStatuses(e, loggedInUserDid, repos) if err != nil { return nil, err }
var events []models.TimelineEvent for _, r := range repos { var source *models.Repo if r.Source != "" { if origRepo, ok := uriToRepo[r.Source]; ok { source = &origRepo } }
isStarred, starCount := getRepoStarInfo(&r, starStatuses)
events = append(events, models.TimelineEvent{ Repo: &r, EventAt: r.Created, Source: source, IsStarred: isStarred, StarCount: starCount, }) }
return events, nil}
func getTimelineStars(e Execer, limit int, loggedInUserDid string, userIsFollowing []string) ([]models.TimelineEvent, error) { filters := make([]filter, 0) if userIsFollowing != nil { filters = append(filters, FilterIn("did", userIsFollowing)) }
stars, err := GetRepoStars(e, limit, filters...) if err != nil { return nil, err }
var repos []models.Repo for _, s := range stars { repos = append(repos, *s.Repo) }
starStatuses, err := fetchStarStatuses(e, loggedInUserDid, repos) if err != nil { return nil, err }
var events []models.TimelineEvent for _, s := range stars { isStarred, starCount := getRepoStarInfo(s.Repo, starStatuses)
events = append(events, models.TimelineEvent{ RepoStar: &s, EventAt: s.Created, IsStarred: isStarred, StarCount: starCount, }) }
return events, nil}
func getTimelineFollows(e Execer, limit int, loggedInUserDid string, userIsFollowing []string) ([]models.TimelineEvent, error) { filters := make([]filter, 0) if userIsFollowing != nil { filters = append(filters, FilterIn("user_did", userIsFollowing)) }
follows, err := GetFollows(e, limit, filters...) if err != nil { return nil, err }
var subjects []string for _, f := range follows { subjects = append(subjects, f.SubjectDid) }
if subjects == nil { return nil, nil }
profiles, err := GetProfiles(e, FilterIn("did", subjects)) if err != nil { return nil, err }
followStatMap, err := GetFollowerFollowingCounts(e, subjects) if err != nil { return nil, err }
var followStatuses map[string]models.FollowStatus if loggedInUserDid != "" { followStatuses, err = GetFollowStatuses(e, loggedInUserDid, subjects) if err != nil { return nil, err } }
var events []models.TimelineEvent for _, f := range follows { profile, _ := profiles[f.SubjectDid] followStatMap, _ := followStatMap[f.SubjectDid]
followStatus := models.IsNotFollowing if followStatuses != nil { followStatus = followStatuses[f.SubjectDid] }
events = append(events, models.TimelineEvent{ Follow: &f, Profile: profile, FollowStats: &followStatMap, FollowStatus: &followStatus, EventAt: f.FollowedAt, }) }
return events, nil}