Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
3.1 kB · 96 lines
Go
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697package model
import ( "context" "errors" "time"
"github.com/bluesky-social/indigo/util" "gorm.io/gorm" "stream.place/streamplace/pkg/placestream")
type PinnedRecord struct { Uri string `gorm:"primaryKey;column:uri"` CID string `gorm:"column:cid"` RepoDID string `json:"repoDID" gorm:"column:repo_did"` Repo *Repo `json:"repo,omitempty" gorm:"foreignKey:DID;references:RepoDID"` PinnedMessage string `gorm:"column:pinned_message" json:"pinnedMessage"` PinnedBy string `gorm:"column:pinned_by" json:"pinnedBy"` IndexedAt *time.Time `gorm:"column:indexed_at" json:"indexedAt"` ExpiresAt *time.Time `gorm:"column:expires_at" json:"expiresAt"` CreatedAt time.Time `gorm:"column:created_at" json:"createdAt"`}
func (p *PinnedRecord) ToStreamplacePinnedRecord() (placestream.ChatPinnedRecord, error) { rec := placestream.ChatPinnedRecord{ LexiconTypeID: "place.stream.chat.pinnedRecord", PinnedMessage: p.PinnedMessage, CreatedAt: p.CreatedAt.UTC().Format(util.ISO8601), } if p.ExpiresAt != nil { s := p.ExpiresAt.UTC().Format(util.ISO8601) rec.ExpiresAt = &s } return rec, nil}
func (p *PinnedRecord) ToStreamplacePinnedRecordView() (placestream.ChatDefs_PinnedRecordView, error) { pr := placestream.ChatPinnedRecord{ LexiconTypeID: "place.stream.chat.pinnedRecord", PinnedMessage: p.PinnedMessage, CreatedAt: p.CreatedAt.UTC().Format(util.ISO8601), } if p.ExpiresAt != nil { s := p.ExpiresAt.UTC().Format(util.ISO8601) pr.ExpiresAt = &s } rec := placestream.ChatDefs_PinnedRecordView{ LexiconTypeID: "place.stream.chat.defs#pinnedRecordView", Record: pr, Cid: p.CID, IndexedAt: p.CreatedAt.UTC().Format(time.RFC3339Nano), // message, pinnedby not included, will fill in later Uri: p.Uri, } return rec, nil}func (m *DBModel) CreatePinnedRecord(ctx context.Context, pin *PinnedRecord) error { return createOrVerify(ctx, m, pin, map[string]any{"uri": pin.Uri})}
func (m *DBModel) GetPinnedRecord(ctx context.Context, uri string) (*PinnedRecord, error) { var pin PinnedRecord err := m.DB.Preload("Repo").Where("uri = ?", uri).First(&pin).Error if errors.Is(err, gorm.ErrRecordNotFound) { return nil, nil } if err != nil { return nil, err } return &pin, nil}
func (m *DBModel) DeletePinnedRecord(ctx context.Context, uri string) error { return m.DB.Where("uri = ?", uri).Delete(&PinnedRecord{}).Error}
func (m *DBModel) DeleteAllPinnedRecords(ctx context.Context, streamerDID string) error { return m.DB.Where("repo_did = ?", streamerDID).Delete(&PinnedRecord{}).Error}
func (m *DBModel) GetActivePinnedRecord(ctx context.Context, streamerDID string) (*PinnedRecord, error) { var pin PinnedRecord now := time.Now() err := m.DB.Preload("Repo"). Where("repo_did = ? AND (expires_at IS NULL OR expires_at > ?)", streamerDID, now). Order("created_at DESC"). First(&pin).Error if errors.Is(err, gorm.ErrRecordNotFound) { return nil, nil } if err != nil { return nil, err } return &pin, nil}