Something went wrong. Try again.
Monorepo for Tangled — https://tangled.org forked from tangled.org/core
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381package db
import ( "database/sql" "errors" "fmt" "log" "slices" "strings" "time"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.org/core/appview/models")
func AddStar(e Execer, star *models.Star) error { query := `insert or ignore into stars (starred_by_did, repo_at, rkey) values (?, ?, ?)` _, err := e.Exec( query, star.StarredByDid, star.RepoAt.String(), star.Rkey, ) return err}
// Get a star recordfunc GetStar(e Execer, starredByDid string, repoAt syntax.ATURI) (*models.Star, error) { query := ` select starred_by_did, repo_at, created, rkey from stars where starred_by_did = ? and repo_at = ?` row := e.QueryRow(query, starredByDid, repoAt)
var star models.Star var created string err := row.Scan(&star.StarredByDid, &star.RepoAt, &created, &star.Rkey) if err != nil { return nil, err }
createdAtTime, err := time.Parse(time.RFC3339, created) if err != nil { log.Println("unable to determine followed at time") star.Created = time.Now() } else { star.Created = createdAtTime }
return &star, nil}
// Remove a starfunc DeleteStar(e Execer, starredByDid string, repoAt syntax.ATURI) error { _, err := e.Exec(`delete from stars where starred_by_did = ? and repo_at = ?`, starredByDid, repoAt) return err}
// Remove a starfunc DeleteStarByRkey(e Execer, starredByDid string, rkey string) error { _, err := e.Exec(`delete from stars where starred_by_did = ? and rkey = ?`, starredByDid, rkey) return err}
func GetStarCount(e Execer, repoAt syntax.ATURI) (int, error) { stars := 0 err := e.QueryRow( `select count(starred_by_did) from stars where repo_at = ?`, repoAt).Scan(&stars) if err != nil { return 0, err } return stars, nil}
// getStarStatuses returns a map of repo URIs to star status for a given user// This is an internal helper function to avoid N+1 queriesfunc getStarStatuses(e Execer, userDid string, repoAts []syntax.ATURI) (map[string]bool, error) { if len(repoAts) == 0 || userDid == "" { return make(map[string]bool), nil }
placeholders := make([]string, len(repoAts)) args := make([]any, len(repoAts)+1) args[0] = userDid
for i, repoAt := range repoAts { placeholders[i] = "?" args[i+1] = repoAt.String() }
query := fmt.Sprintf(` SELECT repo_at FROM stars WHERE starred_by_did = ? AND repo_at IN (%s) `, strings.Join(placeholders, ","))
rows, err := e.Query(query, args...) if err != nil { return nil, err } defer rows.Close()
result := make(map[string]bool) // Initialize all repos as not starred for _, repoAt := range repoAts { result[repoAt.String()] = false }
// Mark starred repos as true for rows.Next() { var repoAt string if err := rows.Scan(&repoAt); err != nil { return nil, err } result[repoAt] = true }
return result, nil}
func GetStarStatus(e Execer, userDid string, repoAt syntax.ATURI) bool { statuses, err := getStarStatuses(e, userDid, []syntax.ATURI{repoAt}) if err != nil { return false } return statuses[repoAt.String()]}
// GetStarStatuses returns a map of repo URIs to star status for a given userfunc GetStarStatuses(e Execer, userDid string, repoAts []syntax.ATURI) (map[string]bool, error) { return getStarStatuses(e, userDid, repoAts)}func GetStars(e Execer, limit int, filters ...filter) ([]models.Star, error) { 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 = fmt.Sprintf(" limit %d", limit) }
repoQuery := fmt.Sprintf( `select starred_by_did, repo_at, created, rkey from stars %s order by created desc %s`, whereClause, limitClause, ) rows, err := e.Query(repoQuery, args...) if err != nil { return nil, err }
starMap := make(map[string][]models.Star) for rows.Next() { var star models.Star var created string err := rows.Scan(&star.StarredByDid, &star.RepoAt, &created, &star.Rkey) if err != nil { return nil, err }
star.Created = time.Now() if t, err := time.Parse(time.RFC3339, created); err == nil { star.Created = t }
repoAt := string(star.RepoAt) starMap[repoAt] = append(starMap[repoAt], star) }
// populate *Repo in each star args = make([]any, len(starMap)) i := 0 for r := range starMap { args[i] = r i++ }
if len(args) == 0 { return nil, nil }
repos, err := GetRepos(e, 0, FilterIn("at_uri", args)) if err != nil { return nil, err }
for _, r := range repos { if stars, ok := starMap[string(r.RepoAt())]; ok { for i := range stars { stars[i].Repo = &r } } }
var stars []models.Star for _, s := range starMap { stars = append(stars, s...) }
slices.SortFunc(stars, func(a, b models.Star) int { if a.Created.After(b.Created) { return -1 } if b.Created.After(a.Created) { return 1 } return 0 })
return stars, nil}
func CountStars(e Execer, filters ...filter) (int64, error) { 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 ") }
repoQuery := fmt.Sprintf(`select count(1) from stars %s`, whereClause) var count int64 err := e.QueryRow(repoQuery, args...).Scan(&count)
if !errors.Is(err, sql.ErrNoRows) && err != nil { return 0, err }
return count, nil}
func GetAllStars(e Execer, limit int) ([]models.Star, error) { var stars []models.Star
rows, err := e.Query(` select s.starred_by_did, s.repo_at, s.rkey, s.created, r.did, r.name, r.knot, r.rkey, r.created from stars s join repos r on s.repo_at = r.at_uri `)
if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var star models.Star var repo models.Repo var starCreatedAt, repoCreatedAt string
if err := rows.Scan( &star.StarredByDid, &star.RepoAt, &star.Rkey, &starCreatedAt, &repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &repoCreatedAt, ); err != nil { return nil, err }
star.Created, err = time.Parse(time.RFC3339, starCreatedAt) if err != nil { star.Created = time.Now() } repo.Created, err = time.Parse(time.RFC3339, repoCreatedAt) if err != nil { repo.Created = time.Now() } star.Repo = &repo
stars = append(stars, star) }
if err := rows.Err(); err != nil { return nil, err }
return stars, nil}
// GetTopStarredReposLastWeek returns the top 8 most starred repositories from the last weekfunc GetTopStarredReposLastWeek(e Execer) ([]models.Repo, error) { // first, get the top repo URIs by star count from the last week query := ` with recent_starred_repos as ( select distinct repo_at from stars where created >= datetime('now', '-7 days') ), repo_star_counts as ( select s.repo_at, count(*) as stars_gained_last_week from stars s join recent_starred_repos rsr on s.repo_at = rsr.repo_at where s.created >= datetime('now', '-7 days') group by s.repo_at ) select rsc.repo_at from repo_star_counts rsc order by rsc.stars_gained_last_week desc limit 8 `
rows, err := e.Query(query) if err != nil { return nil, err } defer rows.Close()
var repoUris []string for rows.Next() { var repoUri string err := rows.Scan(&repoUri) if err != nil { return nil, err } repoUris = append(repoUris, repoUri) }
if err := rows.Err(); err != nil { return nil, err }
if len(repoUris) == 0 { return []models.Repo{}, nil }
// get full repo data repos, err := GetRepos(e, 0, FilterIn("at_uri", repoUris)) if err != nil { return nil, err }
// sort repos by the original trending order repoMap := make(map[string]models.Repo) for _, repo := range repos { repoMap[repo.RepoAt().String()] = repo }
orderedRepos := make([]models.Repo, 0, len(repoUris)) for _, uri := range repoUris { if repo, exists := repoMap[uri]; exists { orderedRepos = append(orderedRepos, repo) } }
return orderedRepos, nil}