Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
5.8 kB · 230 lines
Go
at sl/comment
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231package models
import ( "fmt" "sort" "strings" "time"
comatproto "github.com/bluesky-social/indigo/api/atproto" "github.com/bluesky-social/indigo/atproto/syntax" typegen "github.com/whyrusleeping/cbor-gen" "tangled.org/core/api/tangled")
type Comment struct { Id int64
Did syntax.DID Collection syntax.NSID Rkey syntax.RecordKey Cid syntax.CID
// record content Subject comatproto.RepoStrongRef Body tangled.MarkupMarkdown // markup body type. only markdown is supported right now Created time.Time ReplyTo *comatproto.RepoStrongRef // (optional) parent comment PullRoundIdx *int // (optional) pull round number used when subject is sh.tangled.repo.pull
// store on db, but not on PDS Edited *time.Time Deleted *time.Time}
func (c Comment) AtUri() syntax.ATURI { return syntax.ATURI(fmt.Sprintf("at://%s/%s/%s", c.Did, c.Collection, c.Rkey))}
func (c Comment) StrongRef() comatproto.RepoStrongRef { return comatproto.RepoStrongRef{ Uri: c.AtUri().String(), Cid: c.Cid.String(), }}
func (c Comment) AsRecord() typegen.CBORMarshaler { // can't convert to record for legacy types if c.Collection != tangled.FeedCommentNSID { return nil } var pullRoundIdx int64 if c.PullRoundIdx != nil { pullRoundIdx = int64(*c.PullRoundIdx) } return &tangled.FeedComment{ Subject: &c.Subject, Body: &tangled.FeedComment_Body{MarkupMarkdown: &c.Body}, CreatedAt: c.Created.Format(time.RFC3339), ReplyTo: c.ReplyTo, PullRoundIdx: &pullRoundIdx, }}
func (c Comment) EditableBody() string { if c.Body.Original != nil { return *c.Body.Original } return c.Body.Text}
func (c Comment) IsLegacy() bool { return c.Collection != tangled.FeedCommentNSID}
func (c *Comment) IsTopLevel() bool { return c.ReplyTo == nil}
func (c *Comment) IsReply() bool { return c.ReplyTo != nil}
func (c *Comment) Validate() error { // TODO: sanitize the body and then trim space if sb := strings.TrimSpace(c.Body.Text); sb == "" { return fmt.Errorf("body is empty after HTML sanitization") }
// if it's for PR, PullSubmissionId should not be nil subjectAt, err := syntax.ParseATURI(c.Subject.Uri) if err != nil { return fmt.Errorf("subject.uri is not valid at-uri: %w", err) } if subjectAt.Collection().String() == tangled.RepoPullNSID { if c.PullRoundIdx == nil { return fmt.Errorf("pullSubmissionId should not be nil when subject is sh.tangled.repo.pull") } } return nil}
func CommentFromRecord(did syntax.DID, rkey syntax.RecordKey, cid syntax.CID, record tangled.FeedComment) (*Comment, error) { created, err := time.Parse(time.RFC3339, record.CreatedAt) if err != nil { created = time.Now() }
if record.Subject == nil { return nil, fmt.Errorf("subject can't be nil") } subjectAt, err := syntax.ParseATURI(record.Subject.Uri) if err != nil { return nil, fmt.Errorf("invalid subject uri: %w", err) } if _, err = syntax.ParseCID(record.Subject.Cid); err != nil { return nil, fmt.Errorf("invalid subject cid: %w", err) }
if subjectAt.Collection() == tangled.RepoPullNSID { if record.PullRoundIdx == nil { return nil, fmt.Errorf("pullRoundIdx can't be nil when subject is sh.tangled.repo.pull") } }
if record.Body == nil { return nil, fmt.Errorf("body can't be nil") } if record.Body.MarkupMarkdown == nil { return nil, fmt.Errorf("body should be markdown type") }
if record.ReplyTo != nil { if _, err = syntax.ParseATURI(record.ReplyTo.Uri); err != nil { return nil, fmt.Errorf("invalid replyTo uri: %w", err) } if _, err = syntax.ParseCID(record.ReplyTo.Cid); err != nil { return nil, fmt.Errorf("invalid replyTo cid: %w", err) } }
var pullRoundIdx *int if record.PullRoundIdx != nil { pullRoundIdx = new(int) *pullRoundIdx = int(*record.PullRoundIdx) }
return &Comment{ Did: did, Collection: tangled.FeedCommentNSID, Rkey: rkey, Cid: cid,
Subject: *record.Subject, Body: *record.Body.MarkupMarkdown, Created: created, ReplyTo: record.ReplyTo, PullRoundIdx: pullRoundIdx, }, nil}
type CommentListItem struct { Self *Comment Replies []*Comment}
func (it *CommentListItem) Participants() []syntax.DID { participantSet := make(map[syntax.DID]struct{}) participants := []syntax.DID{}
addParticipant := func(did syntax.DID) { if _, exists := participantSet[did]; !exists { participantSet[did] = struct{}{} participants = append(participants, did) } }
addParticipant(syntax.DID(it.Self.Did))
for _, c := range it.Replies { addParticipant(syntax.DID(c.Did)) }
return participants}
func NewCommentList(comments []Comment) []CommentListItem { // Create a map to quickly find comments by their aturi toplevel := make(map[syntax.ATURI]*CommentListItem) var replies []*Comment
// collect top level comments into the map for _, comment := range comments { if comment.IsTopLevel() { toplevel[comment.AtUri()] = &CommentListItem{ Self: &comment, } } else { replies = append(replies, &comment) } }
for _, r := range replies { if r.ReplyTo == nil { continue } if parent, exists := toplevel[syntax.ATURI(r.ReplyTo.Uri)]; exists { parent.Replies = append(parent.Replies, r) } }
var listing []CommentListItem for _, v := range toplevel { listing = append(listing, *v) }
// sort everything sortFunc := func(a, b *Comment) bool { return a.Created.Before(b.Created) } sort.Slice(listing, func(i, j int) bool { return sortFunc(listing[i].Self, listing[j].Self) }) for _, r := range listing { sort.Slice(r.Replies, func(i, j int) bool { return sortFunc(r.Replies[i], r.Replies[j]) }) }
return listing}