Something went wrong. Try again.
forked from tangled.org/core
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603package db
import ( "database/sql" "fmt" "strings" "time"
"github.com/bluesky-social/indigo/atproto/syntax" securejoin "github.com/cyphar/filepath-securejoin" "tangled.sh/tangled.sh/core/api/tangled")
type Repo struct { Did string Name string Knot string Rkey string Created time.Time AtUri string Description string Spindle string
// optionally, populate this when querying for reverse mappings RepoStats *RepoStats
// optional Source string}
func (r Repo) RepoAt() syntax.ATURI { return syntax.ATURI(fmt.Sprintf("at://%s/%s/%s", r.Did, tangled.RepoNSID, r.Rkey))}
func (r Repo) DidSlashRepo() string { p, _ := securejoin.SecureJoin(r.Did, r.Name) return p}
func GetAllRepos(e Execer, limit int) ([]Repo, error) { var repos []Repo
rows, err := e.Query( `select did, name, knot, rkey, description, created, source from repos order by created desc limit ? `, limit, ) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var repo Repo err := scanRepo( rows, &repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &repo.Description, &repo.Created, &repo.Source, ) if err != nil { return nil, err } repos = append(repos, repo) }
if err := rows.Err(); err != nil { return nil, err }
return repos, nil}
func GetRepos(e Execer, filters ...filter) ([]Repo, error) { repoMap := make(map[syntax.ATURI]Repo)
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 did, name, knot, rkey, created, description, source, spindle from repos r %s`, whereClause, ) rows, err := e.Query(repoQuery, args...)
if err != nil { return nil, fmt.Errorf("failed to execute repo query: %w ", err) }
for rows.Next() { var repo Repo var createdAt string var description, source, spindle sql.NullString
err := rows.Scan( &repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &createdAt, &description, &source, &spindle, ) if err != nil { return nil, fmt.Errorf("failed to execute repo query: %w ", err) }
if t, err := time.Parse(time.RFC3339, createdAt); err == nil { repo.Created = t } if description.Valid { repo.Description = description.String } if source.Valid { repo.Source = source.String } if spindle.Valid { repo.Spindle = spindle.String }
repoMap[repo.RepoAt()] = repo }
if err = rows.Err(); err != nil { return nil, fmt.Errorf("failed to execute repo query: %w ", err) }
inClause := strings.TrimSuffix(strings.Repeat("?, ", len(repoMap)), ", ") args = make([]any, len(repoMap)) for _, r := range repoMap { args = append(args, r.RepoAt()) }
starCountQuery := fmt.Sprintf( `select repo_at, count(1) from stars where repo_at in (%s) group by repo_at`, inClause, ) rows, err = e.Query(starCountQuery, args...) if err != nil { return nil, fmt.Errorf("failed to execute star-count query: %w ", err) } for rows.Next() { var repoat string var count int if err := rows.Scan(&repoat, &count); err != nil { continue } if r, ok := repoMap[syntax.ATURI(repoat)]; ok { r.RepoStats.StarCount = count } } if err = rows.Err(); err != nil { return nil, fmt.Errorf("failed to execute star-count query: %w ", err) }
issueCountQuery := fmt.Sprintf( `select repo_at, count(case when open = 1 then 1 end) as open_count, count(case when open = 0 then 1 end) as closed_count from issues where repo_at in (%s) group by repo_at`, inClause, ) rows, err = e.Query(issueCountQuery, args...) if err != nil { return nil, fmt.Errorf("failed to execute issue-count query: %w ", err) } for rows.Next() { var repoat string var open, closed int if err := rows.Scan(&repoat, &open, &closed); err != nil { continue } if r, ok := repoMap[syntax.ATURI(repoat)]; ok { r.RepoStats.IssueCount.Open = open r.RepoStats.IssueCount.Closed = closed } } if err = rows.Err(); err != nil { return nil, fmt.Errorf("failed to execute issue-count query: %w ", err) }
pullCountQuery := fmt.Sprintf( `select repo_at, count(case when state = ? then 1 end) as open_count, count(case when state = ? then 1 end) as merged_count, count(case when state = ? then 1 end) as closed_count, count(case when state = ? then 1 end) as deleted_count from pulls where repo_at in (%s) group by repo_at`, inClause, ) args = append([]any{ PullOpen, PullMerged, PullClosed, PullDeleted, }, args...) rows, err = e.Query( pullCountQuery, args..., ) if err != nil { return nil, fmt.Errorf("failed to execute pulls-count query: %w ", err) } for rows.Next() { var repoat string var open, merged, closed, deleted int if err := rows.Scan(&repoat, &open, &merged, &closed, &deleted); err != nil { continue } if r, ok := repoMap[syntax.ATURI(repoat)]; ok { r.RepoStats.PullCount.Open = open r.RepoStats.PullCount.Merged = merged r.RepoStats.PullCount.Closed = closed r.RepoStats.PullCount.Deleted = deleted } } if err = rows.Err(); err != nil { return nil, fmt.Errorf("failed to execute pulls-count query: %w ", err) }
var repos []Repo for _, r := range repoMap { repos = append(repos, r) }
return repos, nil}
func GetAllReposByDid(e Execer, did string) ([]Repo, error) { var repos []Repo
rows, err := e.Query( `select r.did, r.name, r.knot, r.rkey, r.description, r.created, count(s.id) as star_count, r.source from repos r left join stars s on r.at_uri = s.repo_at where r.did = ? group by r.at_uri order by r.created desc`, did) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var repo Repo var repoStats RepoStats var createdAt string var nullableDescription sql.NullString var nullableSource sql.NullString
err := rows.Scan(&repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &nullableDescription, &createdAt, &repoStats.StarCount, &nullableSource) if err != nil { return nil, err }
if nullableDescription.Valid { repo.Description = nullableDescription.String }
if nullableSource.Valid { repo.Source = nullableSource.String }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { repo.Created = time.Now() } else { repo.Created = createdAtTime }
repo.RepoStats = &repoStats
repos = append(repos, repo) }
if err := rows.Err(); err != nil { return nil, err }
return repos, nil}
func GetRepo(e Execer, did, name string) (*Repo, error) { var repo Repo var description, spindle sql.NullString
row := e.QueryRow(` select did, name, knot, created, at_uri, description, spindle from repos where did = ? and name = ? `, did, name, )
var createdAt string if err := row.Scan(&repo.Did, &repo.Name, &repo.Knot, &createdAt, &repo.AtUri, &description, &spindle); err != nil { return nil, err } createdAtTime, _ := time.Parse(time.RFC3339, createdAt) repo.Created = createdAtTime
if description.Valid { repo.Description = description.String }
if spindle.Valid { repo.Spindle = spindle.String }
return &repo, nil}
func GetRepoByAtUri(e Execer, atUri string) (*Repo, error) { var repo Repo var nullableDescription sql.NullString
row := e.QueryRow(`select did, name, knot, created, at_uri, description from repos where at_uri = ?`, atUri)
var createdAt string if err := row.Scan(&repo.Did, &repo.Name, &repo.Knot, &createdAt, &repo.AtUri, &nullableDescription); err != nil { return nil, err } createdAtTime, _ := time.Parse(time.RFC3339, createdAt) repo.Created = createdAtTime
if nullableDescription.Valid { repo.Description = nullableDescription.String } else { repo.Description = "" }
return &repo, nil}
func AddRepo(e Execer, repo *Repo) error { _, err := e.Exec( `insert into repos (did, name, knot, rkey, at_uri, description, source) values (?, ?, ?, ?, ?, ?, ?)`, repo.Did, repo.Name, repo.Knot, repo.Rkey, repo.AtUri, repo.Description, repo.Source, ) return err}
func RemoveRepo(e Execer, did, name string) error { _, err := e.Exec(`delete from repos where did = ? and name = ?`, did, name) return err}
func GetRepoSource(e Execer, repoAt syntax.ATURI) (string, error) { var nullableSource sql.NullString err := e.QueryRow(`select source from repos where at_uri = ?`, repoAt).Scan(&nullableSource) if err != nil { return "", err } return nullableSource.String, nil}
func GetForksByDid(e Execer, did string) ([]Repo, error) { var repos []Repo
rows, err := e.Query( `select did, name, knot, rkey, description, created, at_uri, source from repos where did = ? and source is not null and source != '' order by created desc`, did, ) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var repo Repo var createdAt string var nullableDescription sql.NullString var nullableSource sql.NullString
err := rows.Scan(&repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &nullableDescription, &createdAt, &repo.AtUri, &nullableSource) if err != nil { return nil, err }
if nullableDescription.Valid { repo.Description = nullableDescription.String }
if nullableSource.Valid { repo.Source = nullableSource.String }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { repo.Created = time.Now() } else { repo.Created = createdAtTime }
repos = append(repos, repo) }
if err := rows.Err(); err != nil { return nil, err }
return repos, nil}
func GetForkByDid(e Execer, did string, name string) (*Repo, error) { var repo Repo var createdAt string var nullableDescription sql.NullString var nullableSource sql.NullString
row := e.QueryRow( `select did, name, knot, rkey, description, created, at_uri, source from repos where did = ? and name = ? and source is not null and source != ''`, did, name, )
err := row.Scan(&repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &nullableDescription, &createdAt, &repo.AtUri, &nullableSource) if err != nil { return nil, err }
if nullableDescription.Valid { repo.Description = nullableDescription.String }
if nullableSource.Valid { repo.Source = nullableSource.String }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { repo.Created = time.Now() } else { repo.Created = createdAtTime }
return &repo, nil}
func AddCollaborator(e Execer, collaborator, repoOwnerDid, repoName, repoKnot string) error { _, err := e.Exec( `insert into collaborators (did, repo) values (?, (select id from repos where did = ? and name = ? and knot = ?));`, collaborator, repoOwnerDid, repoName, repoKnot) return err}
func UpdateDescription(e Execer, repoAt, newDescription string) error { _, err := e.Exec( `update repos set description = ? where at_uri = ?`, newDescription, repoAt) return err}
func UpdateSpindle(e Execer, repoAt, spindle string) error { _, err := e.Exec( `update repos set spindle = ? where at_uri = ?`, spindle, repoAt) return err}
func CollaboratingIn(e Execer, collaborator string) ([]Repo, error) { var repos []Repo
rows, err := e.Query( `select r.did, r.name, r.knot, r.rkey, r.description, r.created, count(s.id) as star_count from repos r join collaborators c on r.id = c.repo left join stars s on r.at_uri = s.repo_at where c.did = ? group by r.id;`, collaborator) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var repo Repo var repoStats RepoStats var createdAt string var nullableDescription sql.NullString
err := rows.Scan(&repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &nullableDescription, &createdAt, &repoStats.StarCount) if err != nil { return nil, err }
if nullableDescription.Valid { repo.Description = nullableDescription.String } else { repo.Description = "" }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { repo.Created = time.Now() } else { repo.Created = createdAtTime }
repo.RepoStats = &repoStats
repos = append(repos, repo) }
if err := rows.Err(); err != nil { return nil, err }
return repos, nil}
type RepoStats struct { StarCount int IssueCount IssueCount PullCount PullCount}
func scanRepo(rows *sql.Rows, did, name, knot, rkey, description *string, created *time.Time, source *string) error { var createdAt string var nullableDescription sql.NullString var nullableSource sql.NullString if err := rows.Scan(did, name, knot, rkey, &nullableDescription, &createdAt, &nullableSource); err != nil { return err }
if nullableDescription.Valid { *description = nullableDescription.String } else { *description = "" }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { *created = time.Now() } else { *created = createdAtTime }
if nullableSource.Valid { *source = nullableSource.String } else { *source = "" }
return nil}