Mirrors/telephant
2
0
Fork 0
mirror of https://github.com/muesli/telephant synced 2024-11-14 15:57:07 +00:00
Code Issues Projects Releases Packages Wiki Activity

Seed feeds with Mastodon posts on startup

Browse source
This commit is contained in:
Christian Muehlhaeuser 2019-05-05 14:43:20 +02:00
parent 2abf333c19
commit 0c927a4d0e
No known key found for this signature in database
GPG key ID: 3CF9FA45CA1EBB7E
1 changed files with 125 additions and 94 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

135
accounts/mastodon/mastodon.go
View file

@ -87,10 +87,33 @@ func (mod *Account) Run(eventChan chan interface{}) {
Name: mod.self.DisplayName,
Avatar: mod.self.Avatar,
ProfileURL: mod.self.URL,
Posts: mod.self.StatusesCount,
Follows: mod.self.FollowingCount,
Followers: mod.self.FollowersCount,
}
mod.evchan <- ev
// FIXME: retrieve initial feed
// seed feeds initially
nn, err := mod.client.GetNotifications(context.Background(), &mastodon.Pagination{
Limit: initialNotificationsCount,
})
if err != nil {
panic(err)
}
for _, n := range nn {
mod.handleNotification(n)
}
tt, err := mod.client.GetTimelineHome(context.Background(), &mastodon.Pagination{
Limit: initialNotificationsCount,
})
if err != nil {
panic(err)
}
for _, t := range tt {
mod.handleStatus(t)
}
mod.handleStream()
}
@ -168,106 +191,114 @@ func parseBody(body string) string {
*/
}
func (mod *Account) handleStreamEvent(item interface{}) {
spw := &spew.ConfigState{Indent: " ", DisableCapacities: true, DisablePointerAddresses: true}
log.Println("Message received:", spw.Sdump(item))
switch status := item.(type) {
case *mastodon.NotificationEvent:
func (mod *Account) handleNotification(n *mastodon.Notification) {
var ev accounts.MessageEvent
if status.Notification.Status != nil {
if n.Status != nil {
ev = accounts.MessageEvent{
Account: "mastodon",
Name: "post",
Notification: true,
Post: accounts.Post{
MessageID: string(status.Notification.Status.ID),
Body: status.Notification.Status.Content,
Author: status.Notification.Account.Username,
AuthorName: status.Notification.Account.DisplayName,
AuthorURL: status.Notification.Account.URL,
Avatar: status.Notification.Account.Avatar,
CreatedAt: time.Now(),
URL: status.Notification.Status.URL,
MessageID: string(n.Status.ID),
Body: parseBody(n.Status.Content),
Author: n.Account.Username,
AuthorName: n.Account.DisplayName,
AuthorURL: n.Account.URL,
Avatar: n.Account.Avatar,
CreatedAt: n.CreatedAt,
URL: n.Status.URL,
},
}
for _, v := range status.Notification.Status.MediaAttachments {
for _, v := range n.Status.MediaAttachments {
ev.Media = append(ev.Media, v.PreviewURL)
}
}
switch status.Notification.Type {
switch n.Type {
case "mention":
if status.Notification.Status.InReplyToID != nil {
if n.Status.InReplyToID != nil {
ev.Mention = true
ev.Post.ReplyToAuthor = status.Notification.Status.InReplyToAccountID.(string)
ev.Post.ReplyToID = status.Notification.Status.InReplyToID.(string)
ev.Post.ReplyToAuthor = n.Status.InReplyToAccountID.(string)
ev.Post.ReplyToID = n.Status.InReplyToID.(string)
}
case "reblog":
ev.Forward = true
ev.Post.Author = status.Notification.Status.Account.Username
ev.Post.AuthorName = status.Notification.Status.Account.DisplayName
ev.Post.AuthorURL = status.Notification.Status.Account.URL
// ev.Post.Avatar = status.Notification.Status.Account.Avatar
ev.Post.Actor = status.Notification.Account.Username
ev.Post.ActorName = status.Notification.Account.DisplayName
ev.Post.Author = n.Status.Account.Username
ev.Post.AuthorName = n.Status.Account.DisplayName
ev.Post.AuthorURL = n.Status.Account.URL
// ev.Post.Avatar = n.Status.Account.Avatar
ev.Post.Actor = n.Account.Username
ev.Post.ActorName = n.Account.DisplayName
case "favourite":
ev.Like = true
ev.Post.Author = status.Notification.Status.Account.Username
ev.Post.AuthorName = status.Notification.Status.Account.DisplayName
ev.Post.AuthorURL = status.Notification.Status.Account.URL
// ev.Post.Avatar = status.Notification.Status.Account.Avatar
ev.Post.Actor = status.Notification.Account.Username
ev.Post.ActorName = status.Notification.Account.DisplayName
ev.Post.Author = n.Status.Account.Username
ev.Post.AuthorName = n.Status.Account.DisplayName
ev.Post.AuthorURL = n.Status.Account.URL
// ev.Post.Avatar = n.Status.Account.Avatar
ev.Post.Actor = n.Account.Username
ev.Post.ActorName = n.Account.DisplayName
default:
fmt.Println("Unknown type:", status.Notification.Type)
fmt.Println("Unknown type:", n.Type)
return
}
mod.evchan <- ev
}
case *mastodon.UpdateEvent:
func (mod *Account) handleStatus(s *mastodon.Status) {
ev := accounts.MessageEvent{
Account: "mastodon",
Name: "post",
Post: accounts.Post{
MessageID: string(status.Status.ID),
Body: status.Status.Content,
Author: status.Status.Account.Acct,
AuthorName: status.Status.Account.DisplayName,
AuthorURL: status.Status.Account.URL,
Avatar: status.Status.Account.Avatar,
CreatedAt: time.Now(),
URL: status.Status.URL,
MessageID: string(s.ID),
Body: parseBody(s.Content),
Author: s.Account.Acct,
AuthorName: s.Account.DisplayName,
AuthorURL: s.Account.URL,
Avatar: s.Account.Avatar,
CreatedAt: s.CreatedAt,
URL: s.URL,
},
}
for _, v := range status.Status.MediaAttachments {
for _, v := range s.MediaAttachments {
ev.Media = append(ev.Media, v.PreviewURL)
}
if status.Status.Reblog != nil {
if s.Reblog != nil {
ev.Forward = true
for _, v := range status.Status.Reblog.MediaAttachments {
for _, v := range s.Reblog.MediaAttachments {
ev.Media = append(ev.Media, v.PreviewURL)
}
ev.Post.URL = status.Status.Reblog.URL
ev.Post.Author = status.Status.Reblog.Account.Username
ev.Post.AuthorName = status.Status.Reblog.Account.DisplayName
ev.Post.AuthorURL = status.Status.Reblog.Account.URL
ev.Post.Actor = status.Status.Account.DisplayName
ev.Post.ActorName = status.Status.Account.Username
ev.Post.URL = s.Reblog.URL
ev.Post.Author = s.Reblog.Account.Username
ev.Post.AuthorName = s.Reblog.Account.DisplayName
ev.Post.AuthorURL = s.Reblog.Account.URL
ev.Post.Actor = s.Account.DisplayName
ev.Post.ActorName = s.Account.Username
}
mod.evchan <- ev
}
func (mod *Account) handleStreamEvent(item interface{}) {
spw := &spew.ConfigState{Indent: " ", DisableCapacities: true, DisablePointerAddresses: true}
log.Println("Message received:", spw.Sdump(item))
switch e := item.(type) {
case *mastodon.NotificationEvent:
mod.handleNotification(e.Notification)
case *mastodon.UpdateEvent:
mod.handleStatus(e.Status)
}
}