forked from tangled.org/core
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497package db
import ( "database/sql" "time"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.sh/tangled.sh/core/appview/pagination")
type Issue struct { RepoAt syntax.ATURI OwnerDid string IssueId int IssueAt string Created time.Time Title string Body string Open bool
// optionally, populate this when querying for reverse mappings // like comment counts, parent repo etc. Metadata *IssueMetadata}
type IssueMetadata struct { CommentCount int Repo *Repo // labels, assignee etc.}
type Comment struct { OwnerDid string RepoAt syntax.ATURI Rkey string Issue int CommentId int Body string Created *time.Time Deleted *time.Time Edited *time.Time}
func NewIssue(tx *sql.Tx, issue *Issue) error { defer tx.Rollback()
_, err := tx.Exec(` insert or ignore into repo_issue_seqs (repo_at, next_issue_id) values (?, 1) `, issue.RepoAt) if err != nil { return err }
var nextId int err = tx.QueryRow(` update repo_issue_seqs set next_issue_id = next_issue_id + 1 where repo_at = ? returning next_issue_id - 1 `, issue.RepoAt).Scan(&nextId) if err != nil { return err }
issue.IssueId = nextId
_, err = tx.Exec(` insert into issues (repo_at, owner_did, issue_id, title, body) values (?, ?, ?, ?, ?) `, issue.RepoAt, issue.OwnerDid, issue.IssueId, issue.Title, issue.Body) if err != nil { return err }
if err := tx.Commit(); err != nil { return err }
return nil}
func SetIssueAt(e Execer, repoAt syntax.ATURI, issueId int, issueAt string) error { _, err := e.Exec(`update issues set issue_at = ? where repo_at = ? and issue_id = ?`, issueAt, repoAt, issueId) return err}
func GetIssueAt(e Execer, repoAt syntax.ATURI, issueId int) (string, error) { var issueAt string err := e.QueryRow(`select issue_at from issues where repo_at = ? and issue_id = ?`, repoAt, issueId).Scan(&issueAt) return issueAt, err}
func GetIssueId(e Execer, repoAt syntax.ATURI) (int, error) { var issueId int err := e.QueryRow(`select next_issue_id from repo_issue_seqs where repo_at = ?`, repoAt).Scan(&issueId) return issueId - 1, err}
func GetIssueOwnerDid(e Execer, repoAt syntax.ATURI, issueId int) (string, error) { var ownerDid string err := e.QueryRow(`select owner_did from issues where repo_at = ? and issue_id = ?`, repoAt, issueId).Scan(&ownerDid) return ownerDid, err}
func GetIssues(e Execer, repoAt syntax.ATURI, isOpen bool, page pagination.Page) ([]Issue, error) { var issues []Issue openValue := 0 if isOpen { openValue = 1 }
rows, err := e.Query( ` with numbered_issue as ( select i.owner_did, i.issue_id, i.created, i.title, i.body, i.open, count(c.id) as comment_count, row_number() over (order by i.created desc) as row_num from issues i left join comments c on i.repo_at = c.repo_at and i.issue_id = c.issue_id where i.repo_at = ? and i.open = ? group by i.id, i.owner_did, i.issue_id, i.created, i.title, i.body, i.open ) select owner_did, issue_id, created, title, body, open, comment_count from numbered_issue where row_num between ? and ?`, repoAt, openValue, page.Offset+1, page.Offset+page.Limit) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var issue Issue var createdAt string var metadata IssueMetadata err := rows.Scan(&issue.OwnerDid, &issue.IssueId, &createdAt, &issue.Title, &issue.Body, &issue.Open, &metadata.CommentCount) if err != nil { return nil, err }
createdTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { return nil, err } issue.Created = createdTime issue.Metadata = &metadata
issues = append(issues, issue) }
if err := rows.Err(); err != nil { return nil, err }
return issues, nil}
// timeframe here is directly passed into the sql query filter, and any// timeframe in the past should be negative; e.g.: "-3 months"func GetIssuesByOwnerDid(e Execer, ownerDid string, timeframe string) ([]Issue, error) { var issues []Issue
rows, err := e.Query( `select i.owner_did, i.repo_at, i.issue_id, i.created, i.title, i.body, i.open, r.did, r.name, r.knot, r.rkey, r.created from issues i join repos r on i.repo_at = r.at_uri where i.owner_did = ? and i.created >= date ('now', ?) order by i.created desc`, ownerDid, timeframe) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var issue Issue var issueCreatedAt, repoCreatedAt string var repo Repo err := rows.Scan( &issue.OwnerDid, &issue.RepoAt, &issue.IssueId, &issueCreatedAt, &issue.Title, &issue.Body, &issue.Open, &repo.Did, &repo.Name, &repo.Knot, &repo.Rkey, &repoCreatedAt, ) if err != nil { return nil, err }
issueCreatedTime, err := time.Parse(time.RFC3339, issueCreatedAt) if err != nil { return nil, err } issue.Created = issueCreatedTime
repoCreatedTime, err := time.Parse(time.RFC3339, repoCreatedAt) if err != nil { return nil, err } repo.Created = repoCreatedTime
issue.Metadata = &IssueMetadata{ Repo: &repo, }
issues = append(issues, issue) }
if err := rows.Err(); err != nil { return nil, err }
return issues, nil}
func GetIssue(e Execer, repoAt syntax.ATURI, issueId int) (*Issue, error) { query := `select owner_did, created, title, body, open from issues where repo_at = ? and issue_id = ?` row := e.QueryRow(query, repoAt, issueId)
var issue Issue var createdAt string err := row.Scan(&issue.OwnerDid, &createdAt, &issue.Title, &issue.Body, &issue.Open) if err != nil { return nil, err }
createdTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { return nil, err } issue.Created = createdTime
return &issue, nil}
func GetIssueWithComments(e Execer, repoAt syntax.ATURI, issueId int) (*Issue, []Comment, error) { query := `select owner_did, issue_id, created, title, body, open from issues where repo_at = ? and issue_id = ?` row := e.QueryRow(query, repoAt, issueId)
var issue Issue var createdAt string err := row.Scan(&issue.OwnerDid, &issue.IssueId, &createdAt, &issue.Title, &issue.Body, &issue.Open) if err != nil { return nil, nil, err }
createdTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { return nil, nil, err } issue.Created = createdTime
comments, err := GetComments(e, repoAt, issueId) if err != nil { return nil, nil, err }
return &issue, comments, nil}
func NewIssueComment(e Execer, comment *Comment) error { query := `insert into comments (owner_did, repo_at, rkey, issue_id, comment_id, body) values (?, ?, ?, ?, ?, ?)` _, err := e.Exec( query, comment.OwnerDid, comment.RepoAt, comment.Rkey, comment.Issue, comment.CommentId, comment.Body, ) return err}
func GetComments(e Execer, repoAt syntax.ATURI, issueId int) ([]Comment, error) { var comments []Comment
rows, err := e.Query(` select owner_did, issue_id, comment_id, rkey, body, created, edited, deleted from comments where repo_at = ? and issue_id = ? order by created asc`, repoAt, issueId, ) if err == sql.ErrNoRows { return []Comment{}, nil } if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var comment Comment var createdAt string var deletedAt, editedAt, rkey sql.NullString err := rows.Scan(&comment.OwnerDid, &comment.Issue, &comment.CommentId, &rkey, &comment.Body, &createdAt, &editedAt, &deletedAt) if err != nil { return nil, err }
createdAtTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { return nil, err } comment.Created = &createdAtTime
if deletedAt.Valid { deletedTime, err := time.Parse(time.RFC3339, deletedAt.String) if err != nil { return nil, err } comment.Deleted = &deletedTime }
if editedAt.Valid { editedTime, err := time.Parse(time.RFC3339, editedAt.String) if err != nil { return nil, err } comment.Edited = &editedTime }
if rkey.Valid { comment.Rkey = rkey.String }
comments = append(comments, comment) }
if err := rows.Err(); err != nil { return nil, err }
return comments, nil}
func GetComment(e Execer, repoAt syntax.ATURI, issueId, commentId int) (*Comment, error) { query := ` select owner_did, body, rkey, created, deleted, edited from comments where repo_at = ? and issue_id = ? and comment_id = ? ` row := e.QueryRow(query, repoAt, issueId, commentId)
var comment Comment var createdAt string var deletedAt, editedAt, rkey sql.NullString err := row.Scan(&comment.OwnerDid, &comment.Body, &rkey, &createdAt, &deletedAt, &editedAt) if err != nil { return nil, err }
createdTime, err := time.Parse(time.RFC3339, createdAt) if err != nil { return nil, err } comment.Created = &createdTime
if deletedAt.Valid { deletedTime, err := time.Parse(time.RFC3339, deletedAt.String) if err != nil { return nil, err } comment.Deleted = &deletedTime }
if editedAt.Valid { editedTime, err := time.Parse(time.RFC3339, editedAt.String) if err != nil { return nil, err } comment.Edited = &editedTime }
if rkey.Valid { comment.Rkey = rkey.String }
comment.RepoAt = repoAt comment.Issue = issueId comment.CommentId = commentId
return &comment, nil}
func EditComment(e Execer, repoAt syntax.ATURI, issueId, commentId int, newBody string) error { _, err := e.Exec( ` update comments set body = ?, edited = strftime('%Y-%m-%dT%H:%M:%SZ', 'now') where repo_at = ? and issue_id = ? and comment_id = ? `, newBody, repoAt, issueId, commentId) return err}
func DeleteComment(e Execer, repoAt syntax.ATURI, issueId, commentId int) error { _, err := e.Exec( ` update comments set body = "", deleted = strftime('%Y-%m-%dT%H:%M:%SZ', 'now') where repo_at = ? and issue_id = ? and comment_id = ? `, repoAt, issueId, commentId) return err}
func CloseIssue(e Execer, repoAt syntax.ATURI, issueId int) error { _, err := e.Exec(`update issues set open = 0 where repo_at = ? and issue_id = ?`, repoAt, issueId) return err}
func ReopenIssue(e Execer, repoAt syntax.ATURI, issueId int) error { _, err := e.Exec(`update issues set open = 1 where repo_at = ? and issue_id = ?`, repoAt, issueId) return err}
type IssueCount struct { Open int Closed int}
func GetIssueCount(e Execer, repoAt syntax.ATURI) (IssueCount, error) { row := e.QueryRow(` select 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 = ?`, repoAt, )
var count IssueCount if err := row.Scan(&count.Open, &count.Closed); err != nil { return IssueCount{0, 0}, err }
return count, nil}