Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
1.9 kB · 89 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990package db
import ( "database/sql" "fmt" "strings"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.org/core/appview/models" "tangled.org/core/orm")
func InsertRepoLanguages(e Execer, langs []models.RepoLanguage) error { stmt, err := e.Prepare( "insert or replace into repo_languages (repo_did, ref, is_default_ref, language, bytes) values (?, ?, ?, ?, ?)", ) if err != nil { return err }
for _, l := range langs { isDefaultRef := 0 if l.IsDefaultRef { isDefaultRef = 1 }
_, err := stmt.Exec(l.RepoDid, l.Ref, isDefaultRef, l.Language, l.Bytes) if err != nil { return err } }
return nil}
func DeleteRepoLanguages(e Execer, filters ...orm.Filter) 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 ") }
query := fmt.Sprintf(`delete from repo_languages %s`, whereClause)
_, err := e.Exec(query, args...) return err}
func GetRepoLanguages(e Execer, repoDid syntax.DID, ref string) (map[string]int64, error) { rows, err := e.Query( `select language, bytes from repo_languages where repo_did = ? and ref = ?`, repoDid, ref, ) if err != nil { return nil, err } defer rows.Close()
out := make(map[string]int64) for rows.Next() { var language string var bytes int64 if err := rows.Scan(&language, &bytes); err != nil { return nil, err } out[language] = bytes }
return out, rows.Err()}
func UpdateRepoLanguages(tx *sql.Tx, repoDid syntax.DID, ref string, langs []models.RepoLanguage) error { err := DeleteRepoLanguages( tx, orm.FilterEq("repo_did", repoDid), orm.FilterEq("ref", ref), ) if err != nil { return fmt.Errorf("failed to delete existing languages: %w", err) }
return InsertRepoLanguages(tx, langs)}