Something went wrong. Try again.
forked from tangled.org/core
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142package db
import ( "log" "time"
"github.com/bluesky-social/indigo/atproto/syntax")
type ReactionKind string
const ( Like ReactionKind = "👍" Unlike = "👎" Laugh = "😆" Celebration = "🎉" Confused = "🫤" Heart = "❤️" Rocket = "🚀" Eyes = "👀")
func (rk ReactionKind) String() string { return string(rk)}
var OrderedReactionKinds = []ReactionKind{ Like, Unlike, Laugh, Celebration, Confused, Heart, Rocket, Eyes,}
func ParseReactionKind(raw string) (ReactionKind, bool) { k, ok := (map[string]ReactionKind{ "👍": Like, "👎": Unlike, "😆": Laugh, "🎉": Celebration, "🫤": Confused, "❤️": Heart, "🚀": Rocket, "👀": Eyes, })[raw] return k, ok}
type Reaction struct { ReactedByDid string ThreadAt syntax.ATURI Created time.Time Rkey string Kind ReactionKind}
func AddReaction(e Execer, reactedByDid string, threadAt syntax.ATURI, kind ReactionKind, rkey string) error { query := `insert or ignore into reactions (reacted_by_did, thread_at, kind, rkey) values (?, ?, ?, ?)` _, err := e.Exec(query, reactedByDid, threadAt, kind, rkey) return err}
// Get a reaction recordfunc GetReaction(e Execer, reactedByDid string, threadAt syntax.ATURI, kind ReactionKind) (*Reaction, error) { query := ` select reacted_by_did, thread_at, created, rkey from reactions where reacted_by_did = ? and thread_at = ? and kind = ?` row := e.QueryRow(query, reactedByDid, threadAt, kind)
var reaction Reaction var created string err := row.Scan(&reaction.ReactedByDid, &reaction.ThreadAt, &created, &reaction.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") reaction.Created = time.Now() } else { reaction.Created = createdAtTime }
return &reaction, nil}
// Remove a reactionfunc DeleteReaction(e Execer, reactedByDid string, threadAt syntax.ATURI, kind ReactionKind) error { _, err := e.Exec(`delete from reactions where reacted_by_did = ? and thread_at = ? and kind = ?`, reactedByDid, threadAt, kind) return err}
// Remove a reactionfunc DeleteReactionByRkey(e Execer, reactedByDid string, rkey string) error { _, err := e.Exec(`delete from reactions where reacted_by_did = ? and rkey = ?`, reactedByDid, rkey) return err}
func GetReactionCount(e Execer, threadAt syntax.ATURI, kind ReactionKind) (int, error) { count := 0 err := e.QueryRow( `select count(reacted_by_did) from reactions where thread_at = ? and kind = ?`, threadAt, kind).Scan(&count) if err != nil { return 0, err } return count, nil}
func GetReactionCountMap(e Execer, threadAt syntax.ATURI) (map[ReactionKind]int, error) { countMap := map[ReactionKind]int{} for _, kind := range OrderedReactionKinds { count, err := GetReactionCount(e, threadAt, kind) if err != nil { return map[ReactionKind]int{}, nil } countMap[kind] = count } return countMap, nil}
func GetReactionStatus(e Execer, userDid string, threadAt syntax.ATURI, kind ReactionKind) bool { if _, err := GetReaction(e, userDid, threadAt, kind); err != nil { return false } else { return true }}
func GetReactionStatusMap(e Execer, userDid string, threadAt syntax.ATURI) map[ReactionKind]bool { statusMap := map[ReactionKind]bool{} for _, kind := range OrderedReactionKinds { count := GetReactionStatus(e, userDid, threadAt, kind) statusMap[kind] = count } return statusMap}