370 lines
8.5 KiB
Go
370 lines
8.5 KiB
Go
package blueskyclient
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Post struct {
|
|
Type string `json:"$type"`
|
|
Text string `json:"text,omitempty"`
|
|
Embed *Embed `json:"embed,omitempty"`
|
|
Langs []string `json:"langs,omitempty"`
|
|
Labels *Labels `json:"labels,omitempty"`
|
|
Reply *Reply `json:"reply,omitempty"`
|
|
Facets *[]Facets `json:"facets,omitempty"`
|
|
CreatedAt time.Time `json:"createdAt"`
|
|
Subject *Record `json:"subject,omitempty"`
|
|
Avatar *Avatar `json:"avatar,omitempty"`
|
|
Banner *Banner `json:"banner,omitempty"`
|
|
Description string `json:"description"`
|
|
DisplayName string `json:"displayName"`
|
|
PinnedPost *Record `json:"pinnedPost,omitempty"`
|
|
Allow []struct {
|
|
Type string `json:"$type"`
|
|
} `json:"allow,omitempty"`
|
|
HiddenReplies []any `json:"hiddenReplies,omitempty"`
|
|
Post string `json:"post,omitempty"`
|
|
}
|
|
|
|
type Banner struct {
|
|
Type string `json:"$type"`
|
|
Ref struct {
|
|
Link string `json:"$link"`
|
|
} `json:"ref"`
|
|
MimeType string `json:"mimeType"`
|
|
Size int `json:"size"`
|
|
}
|
|
|
|
type Avatar struct {
|
|
Type string `json:"$type"`
|
|
Ref struct {
|
|
Link string `json:"$link"`
|
|
} `json:"ref"`
|
|
MimeType string `json:"mimeType"`
|
|
Size int `json:"size"`
|
|
}
|
|
|
|
type Ref struct {
|
|
Link string `json:"$link"`
|
|
}
|
|
|
|
type Thumb struct {
|
|
Type string `json:"$type"`
|
|
Ref *Ref `json:"ref"`
|
|
MimeType string `json:"mimeType"`
|
|
Size int `json:"size"`
|
|
}
|
|
|
|
type External struct {
|
|
URI string `json:"uri"`
|
|
Thumb *Thumb `json:"thumb"`
|
|
Title string `json:"title"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
type Video struct {
|
|
Type string `json:"$type"`
|
|
Ref *Ref `json:"ref"`
|
|
MimeType string `json:"mimeType"`
|
|
Size int `json:"size"`
|
|
}
|
|
|
|
type Image struct {
|
|
Type string `json:"$type"`
|
|
Ref *Ref `json:"ref"`
|
|
MimeType string `json:"mimeType"`
|
|
Size int `json:"size"`
|
|
}
|
|
|
|
type AspectRatio struct {
|
|
Width int `json:"width"`
|
|
Height int `json:"height"`
|
|
}
|
|
|
|
type Images struct {
|
|
Alt string `json:"alt,omitempty"`
|
|
Image *Image `json:"image"`
|
|
AspectRatio *AspectRatio `json:"aspectRatio"`
|
|
}
|
|
|
|
type Media struct {
|
|
Type string `json:"$type"`
|
|
External *External `json:"external,omitempty"`
|
|
Video *Video `json:"video,omitempty"`
|
|
Images *[]Images `json:"images,omitempty"`
|
|
AspectRatio *AspectRatio `json:"aspectRatio,omitempty"`
|
|
}
|
|
|
|
type Record struct {
|
|
Cid string `json:"cid,omitempty"`
|
|
URI string `json:"uri,omitempty"`
|
|
}
|
|
|
|
func (r *Record) GetDID() string {
|
|
return strings.Split(r.URI, "/")[2]
|
|
}
|
|
|
|
func (r *Record) GetCollection() string {
|
|
return strings.Split(r.URI, "/")[3]
|
|
}
|
|
|
|
func (r *Record) GetRKey() string {
|
|
return strings.Split(r.URI, "/")[4]
|
|
}
|
|
|
|
type PostRecord struct {
|
|
Type string `json:"$type"`
|
|
Cid string `json:"cid"`
|
|
URI string `json:"uri"`
|
|
Record *Record `json:"record"`
|
|
}
|
|
|
|
func (r *PostRecord) GetDID() string {
|
|
return strings.Split(r.URI, "/")[2]
|
|
}
|
|
|
|
func (r *PostRecord) GetCollection() string {
|
|
return strings.Split(r.URI, "/")[3]
|
|
}
|
|
|
|
func (r *PostRecord) GetRKey() string {
|
|
return strings.Split(r.URI, "/")[4]
|
|
}
|
|
|
|
type Embed struct {
|
|
Type string `json:"$type"`
|
|
Media *Media `json:"media,omitempty"`
|
|
Images *[]Images `json:"images,omitempty"`
|
|
Video *Video `json:"video,omitempty"`
|
|
Record *PostRecord `json:"record,omitempty"`
|
|
External *External `json:"external,omitempty"`
|
|
}
|
|
|
|
type Values struct {
|
|
Val string `json:"val"`
|
|
}
|
|
|
|
type Labels struct {
|
|
Type string `json:"$type"`
|
|
Values *[]Values `json:"values"`
|
|
}
|
|
|
|
type Reply struct {
|
|
Root *Record `json:"root,omitempty"`
|
|
Parent *Record `json:"parent,omitempty"`
|
|
}
|
|
|
|
type Index struct {
|
|
ByteEnd int `json:"byteEnd"`
|
|
ByteStart int `json:"byteStart"`
|
|
}
|
|
|
|
type Features struct {
|
|
Did string `json:"did"`
|
|
URI string `json:"uri"`
|
|
Tag string `json:"tag"`
|
|
Type string `json:"$type"`
|
|
}
|
|
|
|
type Facets struct {
|
|
Type string `json:"$type"`
|
|
Index *Index `json:"index"`
|
|
Features *[]Features `json:"features"`
|
|
}
|
|
|
|
type ParsedEmbeds struct {
|
|
Type string
|
|
MimeType string
|
|
Ref string
|
|
URI string
|
|
Width int64
|
|
Height int64
|
|
}
|
|
|
|
type Author struct {
|
|
Did string `json:"did"`
|
|
Handle string `json:"handle"`
|
|
DisplayName string `json:"displayName"`
|
|
Avatar string `json:"avatar"`
|
|
Associated struct {
|
|
Chat struct {
|
|
AllowIncoming string `json:"allowIncoming"`
|
|
} `json:"chat"`
|
|
} `json:"associated"`
|
|
Labels []interface{} `json:"labels"`
|
|
CreatedAt time.Time `json:"createdAt"`
|
|
}
|
|
|
|
type FetchedPost struct {
|
|
URI string `json:"uri"`
|
|
Cid string `json:"cid"`
|
|
Author *Author `json:"author"`
|
|
Record *Post `json:"record"`
|
|
Embed *Embed `json:"embed,omitempty"`
|
|
ReplyCount int `json:"replyCount"`
|
|
RepostCount int `json:"repostCount"`
|
|
LikeCount int `json:"likeCount"`
|
|
QuoteCount int `json:"quoteCount"`
|
|
IndexedAt time.Time `json:"indexedAt"`
|
|
Labels []struct {
|
|
Src string `json:"src"`
|
|
URI string `json:"uri"`
|
|
Cid string `json:"cid"`
|
|
Val string `json:"val"`
|
|
Cts time.Time `json:"cts"`
|
|
} `json:"labels"`
|
|
}
|
|
|
|
func (b *BSky) ParsePost(post []byte) (*Post, error) {
|
|
var p = &Post{}
|
|
err := json.Unmarshal(post, &p)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return p, nil
|
|
}
|
|
|
|
func (post *Post) ProcessFacets() string {
|
|
if post == nil {
|
|
return ""
|
|
}
|
|
|
|
if post.Facets == nil {
|
|
return post.Text
|
|
}
|
|
|
|
sort.Slice((*post.Facets), func(i, j int) bool {
|
|
return (*post.Facets)[i].Index.ByteStart < (*post.Facets)[j].Index.ByteStart
|
|
})
|
|
|
|
var result strings.Builder
|
|
lastIndex := 0
|
|
// post.Text = html.EscapeString(post.Text)
|
|
|
|
for _, facet := range *post.Facets {
|
|
start := facet.Index.ByteStart
|
|
end := facet.Index.ByteEnd
|
|
|
|
result.WriteString(post.Text[lastIndex:start])
|
|
|
|
for _, feature := range *facet.Features {
|
|
switch feature.Type {
|
|
case "app.bsky.richtext.facet#mention":
|
|
link := fmt.Sprintf(`<a href="https://bsky.app/profile/%s">%s</a>`, feature.Did, post.Text[start:end])
|
|
result.WriteString(link)
|
|
case "app.bsky.richtext.facet#link":
|
|
link := fmt.Sprintf(`<a href="%s">%s</a>`, feature.URI, post.Text[start:end])
|
|
result.WriteString(link)
|
|
case "app.bsky.richtext.facet#tag":
|
|
link := fmt.Sprintf(`<a href="https://bsky.app/search?q=%%23%s">%s</a>`, feature.Tag, post.Text[start:end])
|
|
result.WriteString(link)
|
|
default:
|
|
result.WriteString(post.Text[start:end])
|
|
}
|
|
}
|
|
lastIndex = end
|
|
}
|
|
result.WriteString(post.Text[lastIndex:])
|
|
return result.String()
|
|
}
|
|
|
|
func (p *Post) GetEmbeds() *[]ParsedEmbeds {
|
|
var parsedEmbeds = &[]ParsedEmbeds{}
|
|
if p.Embed != nil {
|
|
if p.Embed.Video != nil {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: p.Embed.Video.Ref.Link,
|
|
Type: "video",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
if p.Embed.External != nil {
|
|
if strings.Contains(p.Embed.External.URI, "media.tenor.com") {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: p.Embed.External.URI,
|
|
Type: "external",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
}
|
|
if p.Embed.Media != nil {
|
|
if p.Embed.Media.Images != nil {
|
|
for _, image := range *p.Embed.Media.Images {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: image.Image.Ref.Link,
|
|
Type: "image",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
}
|
|
if p.Embed.Media.Video != nil {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: p.Embed.Media.Video.Ref.Link,
|
|
Type: "video",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
if p.Embed.Media.External != nil {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: p.Embed.Media.External.URI,
|
|
Type: "external",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
}
|
|
if p.Embed.Images != nil {
|
|
for _, image := range *p.Embed.Images {
|
|
parsedEmbed := ParsedEmbeds{
|
|
URI: image.Image.Ref.Link,
|
|
Type: "image",
|
|
}
|
|
*parsedEmbeds = append(*parsedEmbeds, parsedEmbed)
|
|
}
|
|
}
|
|
}
|
|
return parsedEmbeds
|
|
}
|
|
|
|
func (p *Post) GetMedia() *Media {
|
|
if p.GetEmbeds() != nil {
|
|
if p.Embed.Media != nil {
|
|
return p.Embed.Media
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Post) GetMediaImages() *[]Images {
|
|
if p.GetMedia() != nil {
|
|
return p.GetMedia().Images
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Post) GetExternal() *External {
|
|
if p.GetMedia() != nil {
|
|
if p.GetMedia().External != nil {
|
|
return p.GetMedia().External
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Post) IsReply() bool {
|
|
return p.Reply != nil
|
|
}
|
|
|
|
func (p *Post) IsQuotePost() bool {
|
|
if p.Embed != nil {
|
|
if p.Embed.Record != nil {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|