2019-12-13 18:08:26 +00:00
|
|
|
package service
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"errors"
|
2019-12-25 04:39:34 +00:00
|
|
|
"fmt"
|
2019-12-14 20:19:02 +00:00
|
|
|
"mime/multipart"
|
2019-12-13 18:08:26 +00:00
|
|
|
"net/url"
|
|
|
|
"strings"
|
|
|
|
|
2020-01-01 15:58:27 +00:00
|
|
|
"bloat/model"
|
|
|
|
"bloat/renderer"
|
|
|
|
"bloat/util"
|
2019-12-13 18:08:26 +00:00
|
|
|
"mastodon"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2020-01-28 17:51:00 +00:00
|
|
|
errInvalidArgument = errors.New("invalid argument")
|
2019-12-13 18:08:26 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type Service interface {
|
2020-01-28 17:51:00 +00:00
|
|
|
ServeErrorPage(ctx context.Context, c *model.Client, err error)
|
|
|
|
ServeSigninPage(ctx context.Context, c *model.Client) (err error)
|
|
|
|
ServeTimelinePage(ctx context.Context, c *model.Client, tType string, maxID string, minID string) (err error)
|
|
|
|
ServeThreadPage(ctx context.Context, c *model.Client, id string, reply bool) (err error)
|
|
|
|
ServeLikedByPage(ctx context.Context, c *model.Client, id string) (err error)
|
|
|
|
ServeRetweetedByPage(ctx context.Context, c *model.Client, id string) (err error)
|
|
|
|
ServeNotificationPage(ctx context.Context, c *model.Client, maxID string, minID string) (err error)
|
2020-01-31 03:38:49 +00:00
|
|
|
ServeUserPage(ctx context.Context, c *model.Client, id string, pageType string,
|
|
|
|
maxID string, minID string) (err error)
|
2020-01-28 17:51:00 +00:00
|
|
|
ServeAboutPage(ctx context.Context, c *model.Client) (err error)
|
|
|
|
ServeEmojiPage(ctx context.Context, c *model.Client) (err error)
|
|
|
|
ServeSearchPage(ctx context.Context, c *model.Client, q string, qType string, offset int) (err error)
|
2020-01-30 15:32:37 +00:00
|
|
|
ServeUserSearchPage(ctx context.Context, c *model.Client, id string, q string, offset int) (err error)
|
2020-01-28 17:51:00 +00:00
|
|
|
ServeSettingsPage(ctx context.Context, c *model.Client) (err error)
|
|
|
|
NewSession(ctx context.Context, instance string) (redirectUrl string, sessionID string, err error)
|
|
|
|
Signin(ctx context.Context, c *model.Client, sessionID string, code string) (token string, err error)
|
|
|
|
Post(ctx context.Context, c *model.Client, content string, replyToID string, format string,
|
|
|
|
visibility string, isNSFW bool, files []*multipart.FileHeader) (id string, err error)
|
|
|
|
Like(ctx context.Context, c *model.Client, id string) (count int64, err error)
|
|
|
|
UnLike(ctx context.Context, c *model.Client, id string) (count int64, err error)
|
|
|
|
Retweet(ctx context.Context, c *model.Client, id string) (count int64, err error)
|
|
|
|
UnRetweet(ctx context.Context, c *model.Client, id string) (count int64, err error)
|
|
|
|
Follow(ctx context.Context, c *model.Client, id string) (err error)
|
|
|
|
UnFollow(ctx context.Context, c *model.Client, id string) (err error)
|
|
|
|
SaveSettings(ctx context.Context, c *model.Client, settings *model.Settings) (err error)
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type service struct {
|
|
|
|
clientName string
|
|
|
|
clientScope string
|
|
|
|
clientWebsite string
|
2019-12-25 16:38:47 +00:00
|
|
|
customCSS string
|
2019-12-26 11:25:29 +00:00
|
|
|
postFormats []model.PostFormat
|
2019-12-13 18:08:26 +00:00
|
|
|
renderer renderer.Renderer
|
2020-01-28 17:51:00 +00:00
|
|
|
sessionRepo model.SessionRepo
|
|
|
|
appRepo model.AppRepo
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func NewService(clientName string,
|
|
|
|
clientScope string,
|
|
|
|
clientWebsite string,
|
|
|
|
customCSS string,
|
|
|
|
postFormats []model.PostFormat,
|
|
|
|
renderer renderer.Renderer,
|
|
|
|
sessionRepo model.SessionRepo,
|
|
|
|
appRepo model.AppRepo,
|
|
|
|
) Service {
|
2019-12-13 18:08:26 +00:00
|
|
|
return &service{
|
|
|
|
clientName: clientName,
|
|
|
|
clientScope: clientScope,
|
|
|
|
clientWebsite: clientWebsite,
|
2019-12-25 16:38:47 +00:00
|
|
|
customCSS: customCSS,
|
2019-12-26 11:25:29 +00:00
|
|
|
postFormats: postFormats,
|
2019-12-13 18:08:26 +00:00
|
|
|
renderer: renderer,
|
|
|
|
sessionRepo: sessionRepo,
|
|
|
|
appRepo: appRepo,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-25 10:07:06 +00:00
|
|
|
func getRendererContext(c *model.Client) *renderer.Context {
|
|
|
|
var settings model.Settings
|
|
|
|
var session model.Session
|
|
|
|
if c != nil {
|
|
|
|
settings = c.Session.Settings
|
|
|
|
session = c.Session
|
|
|
|
} else {
|
|
|
|
settings = *model.NewSettings()
|
|
|
|
}
|
2020-01-14 16:57:16 +00:00
|
|
|
return &renderer.Context{
|
2020-01-25 10:07:06 +00:00
|
|
|
MaskNSFW: settings.MaskNSFW,
|
|
|
|
ThreadInNewTab: settings.ThreadInNewTab,
|
|
|
|
FluorideMode: settings.FluorideMode,
|
|
|
|
DarkMode: settings.DarkMode,
|
|
|
|
CSRFToken: session.CSRFToken,
|
2020-01-14 16:57:16 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func addToReplyMap(m map[string][]mastodon.ReplyInfo, key interface{},
|
|
|
|
val string, number int) {
|
|
|
|
if key == nil {
|
2019-12-13 18:08:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
keyStr, ok := key.(string)
|
|
|
|
if !ok {
|
2019-12-13 18:08:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
_, ok = m[keyStr]
|
|
|
|
if !ok {
|
|
|
|
m[keyStr] = []mastodon.ReplyInfo{}
|
|
|
|
}
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
m[keyStr] = append(m[keyStr], mastodon.ReplyInfo{val, number})
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) getCommonData(ctx context.Context, c *model.Client,
|
|
|
|
title string) (data *renderer.CommonData, err error) {
|
|
|
|
|
|
|
|
data = new(renderer.CommonData)
|
|
|
|
data.HeaderData = &renderer.HeaderData{
|
|
|
|
Title: title + " - " + svc.clientName,
|
|
|
|
NotificationCount: 0,
|
|
|
|
CustomCSS: svc.customCSS,
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
if c == nil || !c.Session.IsLoggedIn() {
|
2019-12-13 18:08:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
notifications, err := c.GetNotifications(ctx, nil)
|
2019-12-13 18:08:26 +00:00
|
|
|
if err != nil {
|
2020-01-28 17:51:00 +00:00
|
|
|
return nil, err
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
var notificationCount int
|
|
|
|
for i := range notifications {
|
|
|
|
if notifications[i].Pleroma != nil &&
|
|
|
|
!notifications[i].Pleroma.IsSeen {
|
|
|
|
notificationCount++
|
|
|
|
}
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
u, err := c.GetAccountCurrentUser(ctx)
|
2019-12-13 18:08:26 +00:00
|
|
|
if err != nil {
|
2020-01-28 17:51:00 +00:00
|
|
|
return nil, err
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data.NavbarData = &renderer.NavbarData{
|
|
|
|
User: u,
|
|
|
|
NotificationCount: notificationCount,
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data.HeaderData.NotificationCount = notificationCount
|
|
|
|
data.HeaderData.CSRFToken = c.Session.CSRFToken
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
return
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeErrorPage(ctx context.Context, c *model.Client, err error) {
|
2019-12-26 09:11:24 +00:00
|
|
|
var errStr string
|
|
|
|
if err != nil {
|
|
|
|
errStr = err.Error()
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, nil, "error")
|
2019-12-26 09:11:24 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
data := &renderer.ErrorData{
|
|
|
|
CommonData: commonData,
|
|
|
|
Error: errStr,
|
|
|
|
}
|
|
|
|
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(c)
|
2020-01-28 17:51:00 +00:00
|
|
|
svc.renderer.RenderErrorPage(rCtx, c.Writer, data)
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeSigninPage(ctx context.Context, c *model.Client) (
|
|
|
|
err error) {
|
|
|
|
|
|
|
|
commonData, err := svc.getCommonData(ctx, nil, "signin")
|
2019-12-13 18:08:26 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-12-26 09:11:24 +00:00
|
|
|
data := &renderer.SigninData{
|
|
|
|
CommonData: commonData,
|
|
|
|
}
|
|
|
|
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(nil)
|
2020-01-28 17:51:00 +00:00
|
|
|
return svc.renderer.RenderSigninPage(rCtx, c.Writer, data)
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeTimelinePage(ctx context.Context, c *model.Client,
|
|
|
|
tType string, maxID string, minID string) (err error) {
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
var nextLink, prevLink, title string
|
|
|
|
var statuses []*mastodon.Status
|
2019-12-13 18:08:26 +00:00
|
|
|
var pg = mastodon.Pagination{
|
2019-12-15 06:55:13 +00:00
|
|
|
MaxID: maxID,
|
|
|
|
MinID: minID,
|
|
|
|
Limit: 20,
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
switch tType {
|
2019-12-25 04:30:21 +00:00
|
|
|
default:
|
2020-01-28 17:51:00 +00:00
|
|
|
return errInvalidArgument
|
2019-12-25 04:30:21 +00:00
|
|
|
case "home":
|
|
|
|
statuses, err = c.GetTimelineHome(ctx, &pg)
|
|
|
|
title = "Timeline"
|
2020-01-28 20:56:15 +00:00
|
|
|
case "direct":
|
|
|
|
statuses, err = c.GetTimelineDirect(ctx, &pg)
|
|
|
|
title = "Local Timeline"
|
2019-12-25 04:30:21 +00:00
|
|
|
case "local":
|
|
|
|
statuses, err = c.GetTimelinePublic(ctx, true, &pg)
|
|
|
|
title = "Local Timeline"
|
|
|
|
case "twkn":
|
|
|
|
statuses, err = c.GetTimelinePublic(ctx, false, &pg)
|
|
|
|
title = "The Whole Known Network"
|
|
|
|
}
|
2019-12-13 18:08:26 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-12-29 11:32:24 +00:00
|
|
|
for i := range statuses {
|
2019-12-31 11:00:21 +00:00
|
|
|
if statuses[i].Reblog != nil {
|
2020-01-05 18:55:37 +00:00
|
|
|
statuses[i].Reblog.RetweetedByID = statuses[i].ID
|
2019-12-31 11:00:21 +00:00
|
|
|
}
|
2019-12-29 11:32:24 +00:00
|
|
|
}
|
|
|
|
|
2019-12-15 06:55:13 +00:00
|
|
|
if len(maxID) > 0 && len(statuses) > 0 {
|
2020-01-28 17:51:00 +00:00
|
|
|
prevLink = fmt.Sprintf("/timeline/%s?min_id=%s", tType,
|
|
|
|
statuses[0].ID)
|
2019-12-15 06:55:13 +00:00
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
|
2019-12-15 06:55:13 +00:00
|
|
|
if len(minID) > 0 && len(pg.MinID) > 0 {
|
2020-01-28 17:51:00 +00:00
|
|
|
newPg := &mastodon.Pagination{MinID: pg.MinID, Limit: 20}
|
|
|
|
newStatuses, err := c.GetTimelineHome(ctx, newPg)
|
2019-12-15 06:55:13 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
newLen := len(newStatuses)
|
|
|
|
if newLen == 20 {
|
|
|
|
prevLink = fmt.Sprintf("/timeline/%s?min_id=%s",
|
|
|
|
tType, pg.MinID)
|
2019-12-15 06:55:13 +00:00
|
|
|
} else {
|
2020-01-28 17:51:00 +00:00
|
|
|
i := 20 - newLen - 1
|
2019-12-15 06:55:13 +00:00
|
|
|
if len(statuses) > i {
|
2020-01-28 17:51:00 +00:00
|
|
|
prevLink = fmt.Sprintf("/timeline/%s?min_id=%s",
|
|
|
|
tType, statuses[i].ID)
|
2019-12-15 06:55:13 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
|
2019-12-13 18:08:26 +00:00
|
|
|
if len(pg.MaxID) > 0 {
|
2020-01-28 17:51:00 +00:00
|
|
|
nextLink = fmt.Sprintf("/timeline/%s?max_id=%s", tType, pg.MaxID)
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2019-12-21 13:26:31 +00:00
|
|
|
postContext := model.PostContext{
|
|
|
|
DefaultVisibility: c.Session.Settings.DefaultVisibility,
|
2019-12-26 11:25:29 +00:00
|
|
|
Formats: svc.postFormats,
|
2019-12-21 13:26:31 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, tType+" timeline ")
|
2019-12-15 17:37:58 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-12-25 14:13:59 +00:00
|
|
|
data := &renderer.TimelineData{
|
|
|
|
Title: title,
|
|
|
|
Statuses: statuses,
|
|
|
|
NextLink: nextLink,
|
|
|
|
PrevLink: prevLink,
|
|
|
|
PostContext: postContext,
|
2019-12-25 16:38:47 +00:00
|
|
|
CommonData: commonData,
|
2019-12-25 14:13:59 +00:00
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(c)
|
2020-01-28 17:51:00 +00:00
|
|
|
return svc.renderer.RenderTimelinePage(rCtx, c.Writer, data)
|
|
|
|
}
|
2019-12-25 14:13:59 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeThreadPage(ctx context.Context, c *model.Client,
|
|
|
|
id string, reply bool) (err error) {
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
var postContext model.PostContext
|
2019-12-13 18:08:26 +00:00
|
|
|
|
|
|
|
status, err := c.GetStatus(ctx, id)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-12-14 17:47:14 +00:00
|
|
|
u, err := c.GetAccountCurrentUser(ctx)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-12-13 20:33:20 +00:00
|
|
|
if reply {
|
2019-12-21 08:36:44 +00:00
|
|
|
var content string
|
2020-01-28 17:51:00 +00:00
|
|
|
var visibility string
|
2019-12-14 17:47:14 +00:00
|
|
|
if u.ID != status.Account.ID {
|
|
|
|
content += "@" + status.Account.Acct + " "
|
|
|
|
}
|
2019-12-18 22:14:02 +00:00
|
|
|
for i := range status.Mentions {
|
2020-01-28 17:51:00 +00:00
|
|
|
if status.Mentions[i].ID != u.ID &&
|
|
|
|
status.Mentions[i].ID != status.Account.ID {
|
2019-12-18 22:14:02 +00:00
|
|
|
content += "@" + status.Mentions[i].Acct + " "
|
2019-12-14 17:47:14 +00:00
|
|
|
}
|
2019-12-13 20:33:20 +00:00
|
|
|
}
|
2019-12-21 13:26:31 +00:00
|
|
|
|
2019-12-27 08:06:43 +00:00
|
|
|
if c.Session.Settings.CopyScope {
|
|
|
|
s, err := c.GetStatus(ctx, id)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
visibility = s.Visibility
|
|
|
|
} else {
|
|
|
|
visibility = c.Session.Settings.DefaultVisibility
|
2019-12-21 13:26:31 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
postContext = model.PostContext{
|
2019-12-27 08:06:43 +00:00
|
|
|
DefaultVisibility: visibility,
|
2019-12-26 11:25:29 +00:00
|
|
|
Formats: svc.postFormats,
|
2019-12-21 13:26:31 +00:00
|
|
|
ReplyContext: &model.ReplyContext{
|
|
|
|
InReplyToID: id,
|
|
|
|
InReplyToName: status.Account.Acct,
|
|
|
|
ReplyContent: content,
|
|
|
|
},
|
2020-01-12 17:16:57 +00:00
|
|
|
DarkMode: c.Session.Settings.DarkMode,
|
2019-12-21 08:36:44 +00:00
|
|
|
}
|
2019-12-13 20:33:20 +00:00
|
|
|
}
|
|
|
|
|
2019-12-18 22:14:02 +00:00
|
|
|
context, err := c.GetStatusContext(ctx, id)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
statuses := append(append(context.Ancestors, status), context.Descendants...)
|
2020-01-28 17:51:00 +00:00
|
|
|
replies := make(map[string][]mastodon.ReplyInfo)
|
2019-12-18 22:14:02 +00:00
|
|
|
|
|
|
|
for i := range statuses {
|
|
|
|
statuses[i].ShowReplies = true
|
2020-01-28 17:51:00 +00:00
|
|
|
statuses[i].ReplyMap = replies
|
|
|
|
addToReplyMap(replies, statuses[i].InReplyToID, statuses[i].ID, i+1)
|
2019-12-18 22:14:02 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, "post by "+status.Account.DisplayName)
|
2019-12-15 17:37:58 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-12-25 14:13:59 +00:00
|
|
|
data := &renderer.ThreadData{
|
|
|
|
Statuses: statuses,
|
|
|
|
PostContext: postContext,
|
2020-01-28 17:51:00 +00:00
|
|
|
ReplyMap: replies,
|
2019-12-25 16:38:47 +00:00
|
|
|
CommonData: commonData,
|
2019-12-25 14:13:59 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderThreadPage(rCtx, c.Writer, data)
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeLikedByPage(ctx context.Context, c *model.Client,
|
|
|
|
id string) (err error) {
|
2019-12-15 17:37:58 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
likers, err := c.GetFavouritedBy(ctx, id, nil)
|
|
|
|
if err != nil {
|
|
|
|
return
|
2019-12-15 17:37:58 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, "likes")
|
2019-12-15 17:37:58 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.LikedByData{
|
|
|
|
CommonData: commonData,
|
|
|
|
Users: likers,
|
2019-12-15 17:37:58 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderLikedByPage(rCtx, c.Writer, data)
|
|
|
|
}
|
2019-12-15 17:37:58 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeRetweetedByPage(ctx context.Context, c *model.Client,
|
|
|
|
id string) (err error) {
|
2019-12-15 17:37:58 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
retweeters, err := c.GetRebloggedBy(ctx, id, nil)
|
2019-12-15 17:37:58 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, "retweets")
|
2019-12-15 17:37:58 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.RetweetedByData{
|
|
|
|
CommonData: commonData,
|
|
|
|
Users: retweeters,
|
|
|
|
}
|
|
|
|
|
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderRetweetedByPage(rCtx, c.Writer, data)
|
2019-12-15 17:37:58 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeNotificationPage(ctx context.Context, c *model.Client,
|
|
|
|
maxID string, minID string) (err error) {
|
|
|
|
|
|
|
|
var nextLink string
|
|
|
|
var unreadCount int
|
|
|
|
var pg = mastodon.Pagination{
|
|
|
|
MaxID: maxID,
|
|
|
|
MinID: minID,
|
|
|
|
Limit: 20,
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
notifications, err := c.GetNotifications(ctx, &pg)
|
2019-12-26 09:11:24 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
for i := range notifications {
|
|
|
|
if notifications[i].Pleroma != nil && !notifications[i].Pleroma.IsSeen {
|
|
|
|
unreadCount++
|
|
|
|
}
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
if unreadCount > 0 {
|
|
|
|
err := c.ReadNotifications(ctx, notifications[0].ID)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
if len(pg.MaxID) > 0 {
|
|
|
|
nextLink = "/notifications?max_id=" + pg.MaxID
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, "notifications")
|
2019-12-26 09:11:24 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.NotificationData{
|
|
|
|
Notifications: notifications,
|
|
|
|
NextLink: nextLink,
|
|
|
|
CommonData: commonData,
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(c)
|
2020-01-28 17:51:00 +00:00
|
|
|
return svc.renderer.RenderNotificationPage(rCtx, c.Writer, data)
|
2019-12-26 09:11:24 +00:00
|
|
|
}
|
2019-12-26 19:18:09 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeUserPage(ctx context.Context, c *model.Client,
|
2020-01-31 03:38:49 +00:00
|
|
|
id string, pageType string, maxID string, minID string) (err error) {
|
2020-01-28 17:51:00 +00:00
|
|
|
|
2019-12-29 03:43:57 +00:00
|
|
|
var nextLink string
|
2020-01-31 03:38:49 +00:00
|
|
|
var statuses []*mastodon.Status
|
|
|
|
var users []*mastodon.Account
|
2019-12-29 03:43:57 +00:00
|
|
|
var pg = mastodon.Pagination{
|
|
|
|
MaxID: maxID,
|
|
|
|
MinID: minID,
|
|
|
|
Limit: 20,
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
user, err := c.GetAccount(ctx, id)
|
2019-12-29 03:43:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-31 03:38:49 +00:00
|
|
|
switch pageType {
|
|
|
|
case "":
|
|
|
|
statuses, err = c.GetAccountStatuses(ctx, id, false, &pg)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if len(statuses) == 20 && len(pg.MaxID) > 0 {
|
|
|
|
nextLink = fmt.Sprintf("/user/%s?max_id=%s", id,
|
|
|
|
pg.MaxID)
|
|
|
|
}
|
|
|
|
case "following":
|
|
|
|
users, err = c.GetAccountFollowing(ctx, id, &pg)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if len(users) == 20 && len(pg.MaxID) > 0 {
|
|
|
|
nextLink = fmt.Sprintf("/user/%s/following?max_id=%s",
|
|
|
|
id, pg.MaxID)
|
|
|
|
}
|
|
|
|
case "followers":
|
|
|
|
users, err = c.GetAccountFollowers(ctx, id, &pg)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if len(users) == 20 && len(pg.MaxID) > 0 {
|
|
|
|
nextLink = fmt.Sprintf("/user/%s/followers?max_id=%s",
|
|
|
|
id, pg.MaxID)
|
|
|
|
}
|
|
|
|
case "media":
|
|
|
|
statuses, err = c.GetAccountStatuses(ctx, id, true, &pg)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if len(statuses) == 20 && len(pg.MaxID) > 0 {
|
|
|
|
nextLink = fmt.Sprintf("/user/%s/media?max_id=%s",
|
|
|
|
id, pg.MaxID)
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return errInvalidArgument
|
2019-12-29 03:43:57 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
commonData, err := svc.getCommonData(ctx, c, user.DisplayName)
|
2019-12-29 03:43:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.UserData{
|
|
|
|
User: user,
|
2020-01-31 03:38:49 +00:00
|
|
|
Type: pageType,
|
|
|
|
Users: users,
|
2020-01-28 17:51:00 +00:00
|
|
|
Statuses: statuses,
|
2019-12-29 03:43:57 +00:00
|
|
|
NextLink: nextLink,
|
2020-01-28 17:51:00 +00:00
|
|
|
CommonData: commonData,
|
2019-12-29 03:43:57 +00:00
|
|
|
}
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(c)
|
2020-01-28 17:51:00 +00:00
|
|
|
return svc.renderer.RenderUserPage(rCtx, c.Writer, data)
|
|
|
|
}
|
2019-12-29 03:43:57 +00:00
|
|
|
|
2020-01-30 15:32:37 +00:00
|
|
|
func (svc *service) ServeUserSearchPage(ctx context.Context, c *model.Client,
|
|
|
|
id string, q string, offset int) (err error) {
|
|
|
|
|
|
|
|
var nextLink string
|
|
|
|
var title = "search"
|
|
|
|
|
|
|
|
user, err := c.GetAccount(ctx, id)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
results, err := c.Search(ctx, q, "statuses", 20, true, offset, id)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(results.Statuses) == 20 {
|
|
|
|
offset += 20
|
|
|
|
nextLink = fmt.Sprintf("/usersearch/%s?q=%s&offset=%d", id, q, offset)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(q) > 0 {
|
|
|
|
title += " \"" + q + "\""
|
|
|
|
}
|
|
|
|
|
|
|
|
commonData, err := svc.getCommonData(ctx, c, title)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
data := &renderer.UserSearchData{
|
|
|
|
CommonData: commonData,
|
|
|
|
User: user,
|
|
|
|
Q: q,
|
|
|
|
Statuses: results.Statuses,
|
|
|
|
NextLink: nextLink,
|
|
|
|
}
|
|
|
|
|
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderUserSearchPage(rCtx, c.Writer, data)
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeAboutPage(ctx context.Context, c *model.Client) (err error) {
|
|
|
|
commonData, err := svc.getCommonData(ctx, c, "about")
|
2019-12-29 03:43:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.AboutData{
|
|
|
|
CommonData: commonData,
|
2019-12-29 03:43:57 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderAboutPage(rCtx, c.Writer, data)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (svc *service) ServeEmojiPage(ctx context.Context, c *model.Client) (err error) {
|
|
|
|
commonData, err := svc.getCommonData(ctx, c, "emojis")
|
2019-12-29 03:43:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
emojis, err := c.GetInstanceEmojis(ctx)
|
2019-12-29 03:43:57 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
data := &renderer.EmojiData{
|
|
|
|
Emojis: emojis,
|
2019-12-29 03:43:57 +00:00
|
|
|
CommonData: commonData,
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderEmojiPage(rCtx, c.Writer, data)
|
2019-12-29 03:43:57 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeSearchPage(ctx context.Context, c *model.Client,
|
|
|
|
q string, qType string, offset int) (err error) {
|
|
|
|
|
2019-12-26 19:18:09 +00:00
|
|
|
var nextLink string
|
2020-01-28 17:51:00 +00:00
|
|
|
var title = "search"
|
2019-12-26 19:18:09 +00:00
|
|
|
|
2020-01-30 15:32:37 +00:00
|
|
|
results, err := c.Search(ctx, q, qType, 20, true, offset, "")
|
2019-12-26 19:18:09 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
if (qType == "accounts" && len(results.Accounts) == 20) ||
|
|
|
|
(qType == "statuses" && len(results.Statuses) == 20) {
|
2019-12-26 19:18:09 +00:00
|
|
|
offset += 20
|
|
|
|
nextLink = fmt.Sprintf("/search?q=%s&type=%s&offset=%d", q, qType, offset)
|
|
|
|
}
|
|
|
|
|
2020-01-10 03:53:02 +00:00
|
|
|
if len(q) > 0 {
|
|
|
|
title += " \"" + q + "\""
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
|
|
|
|
commonData, err := svc.getCommonData(ctx, c, title)
|
2019-12-26 19:18:09 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
data := &renderer.SearchData{
|
|
|
|
CommonData: commonData,
|
|
|
|
Q: q,
|
|
|
|
Type: qType,
|
|
|
|
Users: results.Accounts,
|
|
|
|
Statuses: results.Statuses,
|
|
|
|
NextLink: nextLink,
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
rCtx := getRendererContext(c)
|
|
|
|
return svc.renderer.RenderSearchPage(rCtx, c.Writer, data)
|
2019-12-26 19:18:09 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) ServeSettingsPage(ctx context.Context, c *model.Client) (err error) {
|
|
|
|
commonData, err := svc.getCommonData(ctx, c, "settings")
|
2019-12-27 08:06:43 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
data := &renderer.SettingsData{
|
|
|
|
CommonData: commonData,
|
|
|
|
Settings: &c.Session.Settings,
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
|
2020-01-25 10:07:06 +00:00
|
|
|
rCtx := getRendererContext(c)
|
2020-01-28 17:51:00 +00:00
|
|
|
return svc.renderer.RenderSettingsPage(rCtx, c.Writer, data)
|
|
|
|
}
|
2019-12-27 08:06:43 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) NewSession(ctx context.Context, instance string) (
|
|
|
|
redirectUrl string, sessionID string, err error) {
|
|
|
|
|
|
|
|
var instanceURL string
|
|
|
|
if strings.HasPrefix(instance, "https://") {
|
|
|
|
instanceURL = instance
|
|
|
|
instance = strings.TrimPrefix(instance, "https://")
|
|
|
|
} else {
|
|
|
|
instanceURL = "https://" + instance
|
|
|
|
}
|
|
|
|
|
|
|
|
sessionID, err = util.NewSessionID()
|
2019-12-27 08:06:43 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
csrfToken, err := util.NewCSRFToken()
|
2019-12-27 08:06:43 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
session := model.Session{
|
|
|
|
ID: sessionID,
|
|
|
|
InstanceDomain: instance,
|
|
|
|
CSRFToken: csrfToken,
|
|
|
|
Settings: *model.NewSettings(),
|
|
|
|
}
|
|
|
|
|
2019-12-27 08:06:43 +00:00
|
|
|
err = svc.sessionRepo.Add(session)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
app, err := svc.appRepo.Get(instance)
|
|
|
|
if err != nil {
|
|
|
|
if err != model.ErrAppNotFound {
|
|
|
|
return
|
|
|
|
}
|
2019-12-15 17:37:58 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
mastoApp, err := mastodon.RegisterApp(ctx, &mastodon.AppConfig{
|
|
|
|
Server: instanceURL,
|
|
|
|
ClientName: svc.clientName,
|
|
|
|
Scopes: svc.clientScope,
|
|
|
|
Website: svc.clientWebsite,
|
|
|
|
RedirectURIs: svc.clientWebsite + "/oauth_callback",
|
|
|
|
})
|
2019-12-25 16:38:47 +00:00
|
|
|
if err != nil {
|
2020-01-28 17:51:00 +00:00
|
|
|
return "", "", err
|
2019-12-15 17:37:58 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
app = model.App{
|
|
|
|
InstanceDomain: instance,
|
|
|
|
InstanceURL: instanceURL,
|
|
|
|
ClientID: mastoApp.ClientID,
|
|
|
|
ClientSecret: mastoApp.ClientSecret,
|
2019-12-25 16:38:47 +00:00
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
err = svc.appRepo.Add(app)
|
2019-12-25 16:38:47 +00:00
|
|
|
if err != nil {
|
2020-01-28 17:51:00 +00:00
|
|
|
return "", "", err
|
2019-12-25 16:38:47 +00:00
|
|
|
}
|
2019-12-25 14:13:59 +00:00
|
|
|
}
|
2019-12-15 17:37:58 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
u, err := url.Parse("/oauth/authorize")
|
2020-01-08 18:16:06 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
q := make(url.Values)
|
|
|
|
q.Set("scope", "read write follow")
|
|
|
|
q.Set("client_id", app.ClientID)
|
|
|
|
q.Set("response_type", "code")
|
|
|
|
q.Set("redirect_uri", svc.clientWebsite+"/oauth_callback")
|
|
|
|
u.RawQuery = q.Encode()
|
|
|
|
|
|
|
|
redirectUrl = instanceURL + u.String()
|
|
|
|
|
2019-12-13 18:08:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) Signin(ctx context.Context, c *model.Client,
|
|
|
|
sessionID string, code string) (token string, err error) {
|
|
|
|
|
|
|
|
if len(code) < 1 {
|
|
|
|
err = errInvalidArgument
|
2020-01-08 18:16:06 +00:00
|
|
|
return
|
|
|
|
}
|
2019-12-13 18:08:26 +00:00
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
err = c.AuthenticateToken(ctx, code, svc.clientWebsite+"/oauth_callback")
|
2020-01-08 18:16:06 +00:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
token = c.GetAccessToken(ctx)
|
|
|
|
|
2019-12-13 18:08:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-01-28 17:51:00 +00:00
|
|
|
func (svc *service) Post(ctx context.Context, c *model.Client, content string,
|
|
|
|
replyToID string, format string, visibility string, isNSFW bool,
|
|
|
|
files []*multipart.FileHeader) (id string, err error) {
|
|
|
|
|
|
|
|
var mediaIDs []string
|
2019-12-14 20:19:02 +00:00
|
|
|
for _, f := range files {
|
|
|
|
a, err := c.UploadMediaFromMultipartFileHeader(ctx, f)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
2020-01-28 17:51:00 +00:00
|
|
|
mediaIDs = append(mediaIDs, a.ID)
|
2019-12-14 20:19:02 +00:00
|
|
|
}
|
|
|
|
|
2019-12-13 18:08:26 +00:00
|
|
|
tweet := &mastodon.Toot{
|
|
|
|
Status: content,
|
|
|
|
InReplyToID: replyToID,
|
2020-01-28 17:51:00 +00:00
|
|
|
MediaIDs: mediaIDs,
|
2019-12-26 11:25:29 +00:00
|
|
|
ContentType: format,
|
2019-12-21 13:26:31 +00:00
|
|
|
Visibility: visibility,
|
2019-12-22 16:27:49 +00:00
|
|
|
Sensitive: isNSFW,
|
2019-12-13 18:08:26 +00:00
|
|
|
}
|
2019-12-14 18:12:48 +00:00
|
|
|
|
|
|
|
s, err := c.PostStatus(ctx, tweet)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|