mirror of
https://gitea.phreedom.club/localhost_frssoft/bloat.git
synced 2024-11-27 01:13:00 +00:00
1007 lines
23 KiB
Go
1007 lines
23 KiB
Go
package service
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"mime/multipart"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"bloat/mastodon"
|
|
"bloat/model"
|
|
"bloat/renderer"
|
|
"bloat/util"
|
|
)
|
|
|
|
var (
|
|
errInvalidArgument = errors.New("invalid argument")
|
|
errInvalidSession = errors.New("invalid session")
|
|
errInvalidCSRFToken = errors.New("invalid csrf token")
|
|
)
|
|
|
|
type service struct {
|
|
cname string
|
|
cscope string
|
|
cwebsite string
|
|
css string
|
|
instance string
|
|
postFormats []model.PostFormat
|
|
renderer renderer.Renderer
|
|
sessionRepo model.SessionRepo
|
|
appRepo model.AppRepo
|
|
}
|
|
|
|
func NewService(cname string, cscope string, cwebsite string,
|
|
css string, instance string, postFormats []model.PostFormat,
|
|
renderer renderer.Renderer, sessionRepo model.SessionRepo,
|
|
appRepo model.AppRepo) *service {
|
|
return &service{
|
|
cname: cname,
|
|
cscope: cscope,
|
|
cwebsite: cwebsite,
|
|
css: css,
|
|
instance: instance,
|
|
postFormats: postFormats,
|
|
renderer: renderer,
|
|
sessionRepo: sessionRepo,
|
|
appRepo: appRepo,
|
|
}
|
|
}
|
|
|
|
func (s *service) authenticate(c *client, sid string, csrf string, ref string, t int) (err error) {
|
|
var sett *model.Settings
|
|
defer func() {
|
|
if sett == nil {
|
|
sett = model.NewSettings()
|
|
}
|
|
c.rctx = &renderer.Context{
|
|
HideAttachments: sett.HideAttachments,
|
|
MaskNSFW: sett.MaskNSFW,
|
|
ThreadInNewTab: sett.ThreadInNewTab,
|
|
FluorideMode: sett.FluorideMode,
|
|
DarkMode: sett.DarkMode,
|
|
CSRFToken: c.s.CSRFToken,
|
|
UserID: c.s.UserID,
|
|
AntiDopamineMode: sett.AntiDopamineMode,
|
|
UserCSS: sett.CSS,
|
|
Referrer: ref,
|
|
}
|
|
}()
|
|
if t < SESSION {
|
|
return
|
|
}
|
|
if len(sid) < 1 {
|
|
return errInvalidSession
|
|
}
|
|
c.s, err = s.sessionRepo.Get(sid)
|
|
if err != nil {
|
|
return errInvalidSession
|
|
}
|
|
sett = &c.s.Settings
|
|
app, err := s.appRepo.Get(c.s.InstanceDomain)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.Client = mastodon.NewClient(&mastodon.Config{
|
|
Server: app.InstanceURL,
|
|
ClientID: app.ClientID,
|
|
ClientSecret: app.ClientSecret,
|
|
AccessToken: c.s.AccessToken,
|
|
})
|
|
if t >= CSRF && (len(csrf) < 1 || csrf != c.s.CSRFToken) {
|
|
return errInvalidCSRFToken
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *service) cdata(c *client, title string, count int, rinterval int,
|
|
target string) (data *renderer.CommonData) {
|
|
data = &renderer.CommonData{
|
|
Title: title + " - " + s.cname,
|
|
CustomCSS: s.css,
|
|
Count: count,
|
|
RefreshInterval: rinterval,
|
|
Target: target,
|
|
}
|
|
if c != nil && c.s.IsLoggedIn() {
|
|
data.CSRFToken = c.s.CSRFToken
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *service) ErrorPage(c *client, err error, retry bool) error {
|
|
var errStr string
|
|
var sessionErr bool
|
|
if err != nil {
|
|
errStr = err.Error()
|
|
if me, ok := err.(mastodon.Error); ok && me.IsAuthError() ||
|
|
err == errInvalidSession || err == errInvalidCSRFToken {
|
|
sessionErr = true
|
|
}
|
|
}
|
|
cdata := s.cdata(nil, "error", 0, 0, "")
|
|
data := &renderer.ErrorData{
|
|
CommonData: cdata,
|
|
Err: errStr,
|
|
Retry: retry,
|
|
SessionErr: sessionErr,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.ErrorPage, data)
|
|
}
|
|
|
|
func (s *service) SigninPage(c *client) (err error) {
|
|
cdata := s.cdata(nil, "signin", 0, 0, "")
|
|
data := &renderer.SigninData{
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.SigninPage, data)
|
|
}
|
|
|
|
func (s *service) RootPage(c *client) (err error) {
|
|
data := &renderer.RootData{
|
|
Title: s.cname,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.RootPage, data)
|
|
}
|
|
|
|
func (s *service) NavPage(c *client) (err error) {
|
|
u, err := c.GetAccountCurrentUser(c.ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
pctx := model.PostContext{
|
|
DefaultVisibility: c.s.Settings.DefaultVisibility,
|
|
DefaultFormat: c.s.Settings.DefaultFormat,
|
|
Formats: s.postFormats,
|
|
}
|
|
cdata := s.cdata(c, "nav", 0, 0, "main")
|
|
data := &renderer.NavData{
|
|
User: u,
|
|
CommonData: cdata,
|
|
PostContext: pctx,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.NavPage, data)
|
|
}
|
|
|
|
func (s *service) TimelinePage(c *client, tType string, instance string,
|
|
maxID string, minID string) (err error) {
|
|
|
|
var nextLink, prevLink, title string
|
|
var statuses []*mastodon.Status
|
|
var pg = mastodon.Pagination{
|
|
MaxID: maxID,
|
|
MinID: minID,
|
|
Limit: 20,
|
|
}
|
|
|
|
switch tType {
|
|
default:
|
|
return errInvalidArgument
|
|
case "home":
|
|
statuses, err = c.GetTimelineHome(c.ctx, &pg)
|
|
title = "Timeline"
|
|
case "direct":
|
|
statuses, err = c.GetTimelineDirect(c.ctx, &pg)
|
|
title = "Direct Timeline"
|
|
case "local":
|
|
statuses, err = c.GetTimelinePublic(c.ctx, true, "", &pg)
|
|
title = "Local Timeline"
|
|
case "remote":
|
|
if len(instance) > 0 {
|
|
statuses, err = c.GetTimelinePublic(c.ctx, false, instance, &pg)
|
|
}
|
|
title = "Remote Timeline"
|
|
case "twkn":
|
|
statuses, err = c.GetTimelinePublic(c.ctx, false, "", &pg)
|
|
title = "The Whole Known Network"
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := range statuses {
|
|
if statuses[i].Reblog != nil {
|
|
statuses[i].Reblog.RetweetedByID = statuses[i].ID
|
|
}
|
|
}
|
|
|
|
if (len(maxID) > 0 || len(minID) > 0) && len(statuses) > 0 {
|
|
v := make(url.Values)
|
|
v.Set("min_id", statuses[0].ID)
|
|
if len(instance) > 0 {
|
|
v.Set("instance", instance)
|
|
}
|
|
prevLink = "/timeline/" + tType + "?" + v.Encode()
|
|
}
|
|
|
|
if len(minID) > 0 || (len(pg.MaxID) > 0 && len(statuses) == 20) {
|
|
v := make(url.Values)
|
|
v.Set("max_id", pg.MaxID)
|
|
if len(instance) > 0 {
|
|
v.Set("instance", instance)
|
|
}
|
|
nextLink = "/timeline/" + tType + "?" + v.Encode()
|
|
}
|
|
|
|
cdata := s.cdata(c, tType+" timeline ", 0, 0, "")
|
|
data := &renderer.TimelineData{
|
|
Title: title,
|
|
Type: tType,
|
|
Instance: instance,
|
|
Statuses: statuses,
|
|
NextLink: nextLink,
|
|
PrevLink: prevLink,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.TimelinePage, data)
|
|
}
|
|
|
|
func addToReplyMap(m map[string][]mastodon.ReplyInfo, key interface{},
|
|
val string, number int) {
|
|
if key == nil {
|
|
return
|
|
}
|
|
keyStr, ok := key.(string)
|
|
if !ok {
|
|
return
|
|
}
|
|
_, ok = m[keyStr]
|
|
if !ok {
|
|
m[keyStr] = []mastodon.ReplyInfo{}
|
|
}
|
|
m[keyStr] = append(m[keyStr], mastodon.ReplyInfo{val, number})
|
|
}
|
|
|
|
func (s *service) ThreadPage(c *client, id string, reply bool) (err error) {
|
|
var pctx model.PostContext
|
|
|
|
status, err := c.GetStatus(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if reply {
|
|
var content string
|
|
var visibility string
|
|
if c.s.UserID != status.Account.ID {
|
|
content += "@" + status.Account.Acct + " "
|
|
}
|
|
for i := range status.Mentions {
|
|
if status.Mentions[i].ID != c.s.UserID &&
|
|
status.Mentions[i].ID != status.Account.ID {
|
|
content += "@" + status.Mentions[i].Acct + " "
|
|
}
|
|
}
|
|
|
|
isDirect := status.Visibility == "direct"
|
|
if isDirect || c.s.Settings.CopyScope {
|
|
visibility = status.Visibility
|
|
} else {
|
|
visibility = c.s.Settings.DefaultVisibility
|
|
}
|
|
|
|
pctx = model.PostContext{
|
|
DefaultVisibility: visibility,
|
|
DefaultFormat: c.s.Settings.DefaultFormat,
|
|
Formats: s.postFormats,
|
|
ReplyContext: &model.ReplyContext{
|
|
InReplyToID: id,
|
|
InReplyToName: status.Account.Acct,
|
|
ReplyContent: content,
|
|
ForceVisibility: isDirect,
|
|
},
|
|
}
|
|
}
|
|
|
|
context, err := c.GetStatusContext(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
statuses := append(append(context.Ancestors, status), context.Descendants...)
|
|
replies := make(map[string][]mastodon.ReplyInfo)
|
|
idNumbers := make(map[string]int)
|
|
|
|
for i := range statuses {
|
|
statuses[i].ShowReplies = true
|
|
|
|
statuses[i].IDNumbers = idNumbers
|
|
idNumbers[statuses[i].ID] = i + 1
|
|
|
|
statuses[i].IDReplies = replies
|
|
addToReplyMap(replies, statuses[i].InReplyToID, statuses[i].ID, i+1)
|
|
}
|
|
|
|
cdata := s.cdata(c, "post by "+status.Account.DisplayName, 0, 0, "")
|
|
data := &renderer.ThreadData{
|
|
Statuses: statuses,
|
|
PostContext: pctx,
|
|
ReplyMap: replies,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.ThreadPage, data)
|
|
}
|
|
|
|
func (s *service) QuickReplyPage(c *client, id string) (err error) {
|
|
status, err := c.GetStatus(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
var ancestor *mastodon.Status
|
|
if status.InReplyToID != nil {
|
|
ancestor, err = c.GetStatus(c.ctx, status.InReplyToID.(string))
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
var content string
|
|
if c.s.UserID != status.Account.ID {
|
|
content += "@" + status.Account.Acct + " "
|
|
}
|
|
for i := range status.Mentions {
|
|
if status.Mentions[i].ID != c.s.UserID &&
|
|
status.Mentions[i].ID != status.Account.ID {
|
|
content += "@" + status.Mentions[i].Acct + " "
|
|
}
|
|
}
|
|
|
|
var visibility string
|
|
isDirect := status.Visibility == "direct"
|
|
if isDirect || c.s.Settings.CopyScope {
|
|
visibility = status.Visibility
|
|
} else {
|
|
visibility = c.s.Settings.DefaultVisibility
|
|
}
|
|
|
|
pctx := model.PostContext{
|
|
DefaultVisibility: visibility,
|
|
DefaultFormat: c.s.Settings.DefaultFormat,
|
|
Formats: s.postFormats,
|
|
ReplyContext: &model.ReplyContext{
|
|
InReplyToID: id,
|
|
InReplyToName: status.Account.Acct,
|
|
QuickReply: true,
|
|
ReplyContent: content,
|
|
ForceVisibility: isDirect,
|
|
},
|
|
}
|
|
|
|
cdata := s.cdata(c, "post by "+status.Account.DisplayName, 0, 0, "")
|
|
data := &renderer.QuickReplyData{
|
|
Ancestor: ancestor,
|
|
Status: status,
|
|
PostContext: pctx,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.QuickReplyPage, data)
|
|
}
|
|
|
|
func (svc *service) StatusPopup(c *client, id string) (err error) {
|
|
status, err := c.GetStatus(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return svc.renderer.Render(c.rctx, c.w, renderer.StatusPopup, status)
|
|
}
|
|
|
|
func (s *service) LikedByPage(c *client, id string) (err error) {
|
|
likers, err := c.GetFavouritedBy(c.ctx, id, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cdata := s.cdata(c, "likes", 0, 0, "")
|
|
data := &renderer.LikedByData{
|
|
CommonData: cdata,
|
|
Users: likers,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.LikedByPage, data)
|
|
}
|
|
|
|
func (s *service) RetweetedByPage(c *client, id string) (err error) {
|
|
retweeters, err := c.GetRebloggedBy(c.ctx, id, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cdata := s.cdata(c, "retweets", 0, 0, "")
|
|
data := &renderer.RetweetedByData{
|
|
CommonData: cdata,
|
|
Users: retweeters,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.RetweetedByPage, data)
|
|
}
|
|
|
|
func (s *service) NotificationPage(c *client, maxID string,
|
|
minID string) (err error) {
|
|
|
|
var nextLink string
|
|
var unreadCount int
|
|
var readID string
|
|
var includes, excludes []string
|
|
var pg = mastodon.Pagination{
|
|
MaxID: maxID,
|
|
MinID: minID,
|
|
Limit: 20,
|
|
}
|
|
|
|
if c.s.Settings.HideUnsupportedNotifs {
|
|
// Explicitly include the supported types.
|
|
// For now, only Pleroma supports this option, Mastadon
|
|
// will simply ignore the unknown params.
|
|
includes = []string{"follow", "follow_request", "mention", "reblog", "favourite"}
|
|
}
|
|
if c.s.Settings.AntiDopamineMode {
|
|
excludes = append(excludes, "follow", "favourite", "reblog")
|
|
}
|
|
|
|
notifications, err := c.GetNotifications(c.ctx, &pg, includes, excludes)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
for i := range notifications {
|
|
if notifications[i].Pleroma != nil && !notifications[i].Pleroma.IsSeen {
|
|
unreadCount++
|
|
}
|
|
}
|
|
|
|
if unreadCount > 0 {
|
|
readID = notifications[0].ID
|
|
}
|
|
if len(notifications) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = "/notifications?max_id=" + pg.MaxID
|
|
}
|
|
|
|
cdata := s.cdata(c, "notifications", unreadCount,
|
|
c.s.Settings.NotificationInterval, "main")
|
|
data := &renderer.NotificationData{
|
|
Notifications: notifications,
|
|
UnreadCount: unreadCount,
|
|
ReadID: readID,
|
|
NextLink: nextLink,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.NotificationPage, data)
|
|
}
|
|
|
|
func (s *service) UserPage(c *client, id string, pageType string,
|
|
maxID string, minID string) (err error) {
|
|
|
|
var nextLink string
|
|
var statuses []*mastodon.Status
|
|
var users []*mastodon.Account
|
|
var pg = mastodon.Pagination{
|
|
MaxID: maxID,
|
|
MinID: minID,
|
|
Limit: 20,
|
|
}
|
|
|
|
user, err := c.GetAccount(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
isCurrent := c.s.UserID == user.ID
|
|
|
|
switch pageType {
|
|
case "":
|
|
statuses, err = c.GetAccountStatuses(c.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(c.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(c.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(c.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)
|
|
}
|
|
case "bookmarks":
|
|
if !isCurrent {
|
|
return errInvalidArgument
|
|
}
|
|
statuses, err = c.GetBookmarks(c.ctx, &pg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(statuses) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = fmt.Sprintf("/user/%s/bookmarks?max_id=%s",
|
|
id, pg.MaxID)
|
|
}
|
|
case "mutes":
|
|
if !isCurrent {
|
|
return errInvalidArgument
|
|
}
|
|
users, err = c.GetMutes(c.ctx, &pg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(users) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = fmt.Sprintf("/user/%s/mutes?max_id=%s",
|
|
id, pg.MaxID)
|
|
}
|
|
case "blocks":
|
|
if !isCurrent {
|
|
return errInvalidArgument
|
|
}
|
|
users, err = c.GetBlocks(c.ctx, &pg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(users) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = fmt.Sprintf("/user/%s/blocks?max_id=%s",
|
|
id, pg.MaxID)
|
|
}
|
|
case "likes":
|
|
if !isCurrent {
|
|
return errInvalidArgument
|
|
}
|
|
statuses, err = c.GetFavourites(c.ctx, &pg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(statuses) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = fmt.Sprintf("/user/%s/likes?max_id=%s",
|
|
id, pg.MaxID)
|
|
}
|
|
case "requests":
|
|
if !isCurrent {
|
|
return errInvalidArgument
|
|
}
|
|
users, err = c.GetFollowRequests(c.ctx, &pg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(users) == 20 && len(pg.MaxID) > 0 {
|
|
nextLink = fmt.Sprintf("/user/%s/requests?max_id=%s",
|
|
id, pg.MaxID)
|
|
}
|
|
default:
|
|
return errInvalidArgument
|
|
}
|
|
|
|
for i := range statuses {
|
|
if statuses[i].Reblog != nil {
|
|
statuses[i].Reblog.RetweetedByID = statuses[i].ID
|
|
}
|
|
}
|
|
|
|
cdata := s.cdata(c, user.DisplayName+" @"+user.Acct, 0, 0, "")
|
|
data := &renderer.UserData{
|
|
User: user,
|
|
IsCurrent: isCurrent,
|
|
Type: pageType,
|
|
Users: users,
|
|
Statuses: statuses,
|
|
NextLink: nextLink,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.UserPage, data)
|
|
}
|
|
|
|
func (s *service) UserSearchPage(c *client,
|
|
id string, q string, offset int) (err error) {
|
|
|
|
var nextLink string
|
|
var title = "search"
|
|
|
|
user, err := c.GetAccount(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
var results *mastodon.Results
|
|
if len(q) > 0 {
|
|
results, err = c.Search(c.ctx, q, "statuses", 20, true, offset, id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
results = &mastodon.Results{}
|
|
}
|
|
|
|
if len(results.Statuses) == 20 {
|
|
offset += 20
|
|
nextLink = fmt.Sprintf("/usersearch/%s?q=%s&offset=%d", id,
|
|
url.QueryEscape(q), offset)
|
|
}
|
|
|
|
if len(q) > 0 {
|
|
title += " \"" + q + "\""
|
|
}
|
|
|
|
cdata := s.cdata(c, title, 0, 0, "")
|
|
data := &renderer.UserSearchData{
|
|
CommonData: cdata,
|
|
User: user,
|
|
Q: q,
|
|
Statuses: results.Statuses,
|
|
NextLink: nextLink,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.UserSearchPage, data)
|
|
}
|
|
|
|
func (s *service) AboutPage(c *client) (err error) {
|
|
cdata := s.cdata(c, "about", 0, 0, "")
|
|
data := &renderer.AboutData{
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.AboutPage, data)
|
|
}
|
|
|
|
func (s *service) EmojiPage(c *client) (err error) {
|
|
emojis, err := c.GetInstanceEmojis(c.ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cdata := s.cdata(c, "emojis", 0, 0, "")
|
|
data := &renderer.EmojiData{
|
|
Emojis: emojis,
|
|
CommonData: cdata,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.EmojiPage, data)
|
|
}
|
|
|
|
func (s *service) SearchPage(c *client,
|
|
q string, qType string, offset int) (err error) {
|
|
|
|
var nextLink string
|
|
var title = "search"
|
|
|
|
var results *mastodon.Results
|
|
if len(q) > 0 {
|
|
results, err = c.Search(c.ctx, q, qType, 20, true, offset, "")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
results = &mastodon.Results{}
|
|
}
|
|
|
|
if (qType == "accounts" && len(results.Accounts) == 20) ||
|
|
(qType == "statuses" && len(results.Statuses) == 20) {
|
|
offset += 20
|
|
nextLink = fmt.Sprintf("/search?q=%s&type=%s&offset=%d",
|
|
url.QueryEscape(q), qType, offset)
|
|
}
|
|
|
|
if len(q) > 0 {
|
|
title += " \"" + q + "\""
|
|
}
|
|
|
|
cdata := s.cdata(c, title, 0, 0, "")
|
|
data := &renderer.SearchData{
|
|
CommonData: cdata,
|
|
Q: q,
|
|
Type: qType,
|
|
Users: results.Accounts,
|
|
Statuses: results.Statuses,
|
|
NextLink: nextLink,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.SearchPage, data)
|
|
}
|
|
|
|
func (s *service) SettingsPage(c *client) (err error) {
|
|
cdata := s.cdata(c, "settings", 0, 0, "")
|
|
data := &renderer.SettingsData{
|
|
CommonData: cdata,
|
|
Settings: &c.s.Settings,
|
|
PostFormats: s.postFormats,
|
|
}
|
|
return s.renderer.Render(c.rctx, c.w, renderer.SettingsPage, data)
|
|
}
|
|
|
|
func (svc *service) FiltersPage(c *client) (err error) {
|
|
filters, err := c.GetFilters(c.ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cdata := svc.cdata(c, "filters", 0, 0, "")
|
|
data := &renderer.FiltersData{
|
|
CommonData: cdata,
|
|
Filters: filters,
|
|
}
|
|
return svc.renderer.Render(c.rctx, c.w, renderer.FiltersPage, data)
|
|
}
|
|
|
|
func (s *service) SingleInstance() (instance string, ok bool) {
|
|
if len(s.instance) > 0 {
|
|
instance = s.instance
|
|
ok = true
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *service) NewSession(c *client, instance string) (rurl string, sid string, err error) {
|
|
var instanceURL string
|
|
if strings.HasPrefix(instance, "https://") {
|
|
instanceURL = instance
|
|
instance = strings.TrimPrefix(instance, "https://")
|
|
} else {
|
|
instanceURL = "https://" + instance
|
|
}
|
|
|
|
sid, err = util.NewSessionID()
|
|
if err != nil {
|
|
return
|
|
}
|
|
csrf, err := util.NewCSRFToken()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
sess := model.Session{
|
|
ID: sid,
|
|
InstanceDomain: instance,
|
|
CSRFToken: csrf,
|
|
Settings: *model.NewSettings(),
|
|
}
|
|
err = s.sessionRepo.Add(sess)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
app, err := s.appRepo.Get(instance)
|
|
if err != nil {
|
|
if err != model.ErrAppNotFound {
|
|
return
|
|
}
|
|
mastoApp, err := mastodon.RegisterApp(c.ctx, &mastodon.AppConfig{
|
|
Server: instanceURL,
|
|
ClientName: s.cname,
|
|
Scopes: s.cscope,
|
|
Website: s.cwebsite,
|
|
RedirectURIs: s.cwebsite + "/oauth_callback",
|
|
})
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
app = model.App{
|
|
InstanceDomain: instance,
|
|
InstanceURL: instanceURL,
|
|
ClientID: mastoApp.ClientID,
|
|
ClientSecret: mastoApp.ClientSecret,
|
|
}
|
|
err = s.appRepo.Add(app)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
}
|
|
|
|
u, err := url.Parse("/oauth/authorize")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
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", s.cwebsite+"/oauth_callback")
|
|
u.RawQuery = q.Encode()
|
|
|
|
rurl = instanceURL + u.String()
|
|
return
|
|
}
|
|
|
|
func (s *service) Signin(c *client, code string) (err error) {
|
|
if len(code) < 1 {
|
|
err = errInvalidArgument
|
|
return
|
|
}
|
|
err = c.AuthenticateToken(c.ctx, code, s.cwebsite+"/oauth_callback")
|
|
if err != nil {
|
|
return
|
|
}
|
|
u, err := c.GetAccountCurrentUser(c.ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
c.s.AccessToken = c.GetAccessToken(c.ctx)
|
|
c.s.UserID = u.ID
|
|
return s.sessionRepo.Add(c.s)
|
|
}
|
|
|
|
func (s *service) Signout(c *client) (err error) {
|
|
s.sessionRepo.Remove(c.s.ID)
|
|
return
|
|
}
|
|
|
|
func (s *service) Post(c *client, content string, replyToID string,
|
|
format string, visibility string, isNSFW bool,
|
|
files []*multipart.FileHeader) (id string, err error) {
|
|
|
|
var mediaIDs []string
|
|
for _, f := range files {
|
|
a, err := c.UploadMediaFromMultipartFileHeader(c.ctx, f)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
mediaIDs = append(mediaIDs, a.ID)
|
|
}
|
|
|
|
tweet := &mastodon.Toot{
|
|
Status: content,
|
|
InReplyToID: replyToID,
|
|
MediaIDs: mediaIDs,
|
|
ContentType: format,
|
|
Visibility: visibility,
|
|
Sensitive: isNSFW,
|
|
}
|
|
st, err := c.PostStatus(c.ctx, tweet)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return st.ID, nil
|
|
}
|
|
|
|
func (s *service) Like(c *client, id string) (count int64, err error) {
|
|
st, err := c.Favourite(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
count = st.FavouritesCount
|
|
return
|
|
}
|
|
|
|
func (s *service) UnLike(c *client, id string) (count int64, err error) {
|
|
st, err := c.Unfavourite(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
count = st.FavouritesCount
|
|
return
|
|
}
|
|
|
|
func (s *service) Retweet(c *client, id string) (count int64, err error) {
|
|
st, err := c.Reblog(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if st.Reblog != nil {
|
|
count = st.Reblog.ReblogsCount
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *service) UnRetweet(c *client, id string) (
|
|
count int64, err error) {
|
|
st, err := c.Unreblog(c.ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
count = st.ReblogsCount
|
|
return
|
|
}
|
|
|
|
func (s *service) Vote(c *client, id string, choices []string) (err error) {
|
|
_, err = c.Vote(c.ctx, id, choices)
|
|
return
|
|
}
|
|
|
|
func (s *service) Follow(c *client, id string, reblogs *bool) (err error) {
|
|
_, err = c.AccountFollow(c.ctx, id, reblogs)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnFollow(c *client, id string) (err error) {
|
|
_, err = c.AccountUnfollow(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) Accept(c *client, id string) (err error) {
|
|
return c.FollowRequestAuthorize(c.ctx, id)
|
|
}
|
|
|
|
func (s *service) Reject(c *client, id string) (err error) {
|
|
return c.FollowRequestReject(c.ctx, id)
|
|
}
|
|
|
|
func (s *service) Mute(c *client, id string, notifications *bool) (err error) {
|
|
_, err = c.AccountMute(c.ctx, id, notifications)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnMute(c *client, id string) (err error) {
|
|
_, err = c.AccountUnmute(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) Block(c *client, id string) (err error) {
|
|
_, err = c.AccountBlock(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnBlock(c *client, id string) (err error) {
|
|
_, err = c.AccountUnblock(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) Subscribe(c *client, id string) (err error) {
|
|
_, err = c.Subscribe(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnSubscribe(c *client, id string) (err error) {
|
|
_, err = c.UnSubscribe(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) SaveSettings(c *client, settings *model.Settings) (err error) {
|
|
switch settings.NotificationInterval {
|
|
case 0, 30, 60, 120, 300, 600:
|
|
default:
|
|
return errInvalidArgument
|
|
}
|
|
if len(settings.CSS) > 1<<20 {
|
|
return errInvalidArgument
|
|
}
|
|
sess, err := s.sessionRepo.Get(c.s.ID)
|
|
if err != nil {
|
|
return
|
|
}
|
|
sess.Settings = *settings
|
|
return s.sessionRepo.Add(sess)
|
|
}
|
|
|
|
func (s *service) MuteConversation(c *client, id string) (err error) {
|
|
_, err = c.MuteConversation(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnMuteConversation(c *client, id string) (err error) {
|
|
_, err = c.UnmuteConversation(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) Delete(c *client, id string) (err error) {
|
|
return c.DeleteStatus(c.ctx, id)
|
|
}
|
|
|
|
func (s *service) ReadNotifications(c *client, maxID string) (err error) {
|
|
return c.ReadNotifications(c.ctx, maxID)
|
|
}
|
|
|
|
func (s *service) Bookmark(c *client, id string) (err error) {
|
|
_, err = c.Bookmark(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (s *service) UnBookmark(c *client, id string) (err error) {
|
|
_, err = c.Unbookmark(c.ctx, id)
|
|
return
|
|
}
|
|
|
|
func (svc *service) Filter(c *client, phrase string, wholeWord bool) (err error) {
|
|
fctx := []string{"home", "notifications", "public", "thread"}
|
|
return c.AddFilter(c.ctx, phrase, fctx, true, wholeWord, nil)
|
|
}
|
|
|
|
func (svc *service) UnFilter(c *client, id string) (err error) {
|
|
return c.RemoveFilter(c.ctx, id)
|
|
}
|