diff --git a/mastodon/accounts.go b/mastodon/accounts.go index c02f0bd..995741c 100644 --- a/mastodon/accounts.go +++ b/mastodon/accounts.go @@ -191,6 +191,7 @@ type Relationship struct { Blocking bool `json:"blocking"` Muting bool `json:"muting"` MutingNotifications bool `json:"muting_notifications"` + Subscribing bool `json:"subscribing"` Requested bool `json:"requested"` DomainBlocking bool `json:"domain_blocking"` ShowingReblogs bool `json:"showing_reblogs"` @@ -328,3 +329,23 @@ func (c *Client) GetMutes(ctx context.Context, pg *Pagination) ([]*Account, erro } return accounts, nil } + +// Subscribe to receive notifications for all statuses posted by a user +func (c *Client) Subscribe(ctx context.Context, id string) (*Relationship, error) { + var relationship *Relationship + err := c.doAPI(ctx, http.MethodPost, fmt.Sprintf("/api/v1/pleroma/accounts/%s/subscribe", url.PathEscape(id)), nil, &relationship, nil) + if err != nil { + return nil, err + } + return relationship, nil +} + +// UnSubscribe to stop receiving notifications from user statuses +func (c *Client) UnSubscribe(ctx context.Context, id string) (*Relationship, error) { + var relationship *Relationship + err := c.doAPI(ctx, http.MethodPost, fmt.Sprintf("/api/v1/pleroma/accounts/%s/unsubscribe", url.PathEscape(id)), nil, &relationship, nil) + if err != nil { + return nil, err + } + return relationship, nil +} diff --git a/service/auth.go b/service/auth.go index 7fd238b..dd114b4 100644 --- a/service/auth.go +++ b/service/auth.go @@ -364,6 +364,30 @@ func (s *as) UnBlock(ctx context.Context, c *model.Client, id string) (err error return s.Service.UnBlock(ctx, c, id) } +func (s *as) Subscribe(ctx context.Context, c *model.Client, id string) (err error) { + err = s.authenticateClient(ctx, c) + if err != nil { + return + } + err = checkCSRF(ctx, c) + if err != nil { + return + } + return s.Service.Subscribe(ctx, c, id) +} + +func (s *as) UnSubscribe(ctx context.Context, c *model.Client, id string) (err error) { + err = s.authenticateClient(ctx, c) + if err != nil { + return + } + err = checkCSRF(ctx, c) + if err != nil { + return + } + return s.Service.UnSubscribe(ctx, c, id) +} + func (s *as) SaveSettings(ctx context.Context, c *model.Client, settings *model.Settings) (err error) { err = s.authenticateClient(ctx, c) if err != nil { diff --git a/service/logging.go b/service/logging.go index 2955959..c16cdb3 100644 --- a/service/logging.go +++ b/service/logging.go @@ -269,6 +269,22 @@ func (s *ls) UnBlock(ctx context.Context, c *model.Client, id string) (err error return s.Service.UnBlock(ctx, c, id) } +func (s *ls) Subscribe(ctx context.Context, c *model.Client, id string) (err error) { + defer func(begin time.Time) { + s.logger.Printf("method=%v, id=%v, took=%v, err=%v\n", + "Subscribe", id, time.Since(begin), err) + }(time.Now()) + return s.Service.Subscribe(ctx, c, id) +} + +func (s *ls) UnSubscribe(ctx context.Context, c *model.Client, id string) (err error) { + defer func(begin time.Time) { + s.logger.Printf("method=%v, id=%v, took=%v, err=%v\n", + "UnSubscribe", id, time.Since(begin), err) + }(time.Now()) + return s.Service.UnSubscribe(ctx, c, id) +} + func (s *ls) SaveSettings(ctx context.Context, c *model.Client, settings *model.Settings) (err error) { defer func(begin time.Time) { s.logger.Printf("method=%v, took=%v, err=%v\n", diff --git a/service/service.go b/service/service.go index b039849..3a48e77 100644 --- a/service/service.go +++ b/service/service.go @@ -52,6 +52,8 @@ type Service interface { UnMute(ctx context.Context, c *model.Client, id string) (err error) Block(ctx context.Context, c *model.Client, id string) (err error) UnBlock(ctx context.Context, c *model.Client, id string) (err error) + Subscribe(ctx context.Context, c *model.Client, id string) (err error) + UnSubscribe(ctx context.Context, c *model.Client, id string) (err error) SaveSettings(ctx context.Context, c *model.Client, settings *model.Settings) (err error) MuteConversation(ctx context.Context, c *model.Client, id string) (err error) UnMuteConversation(ctx context.Context, c *model.Client, id string) (err error) @@ -839,6 +841,16 @@ func (svc *service) UnBlock(ctx context.Context, c *model.Client, id string) (er return } +func (svc *service) Subscribe(ctx context.Context, c *model.Client, id string) (err error) { + _, err = c.Subscribe(ctx, id) + return +} + +func (svc *service) UnSubscribe(ctx context.Context, c *model.Client, id string) (err error) { + _, err = c.UnSubscribe(ctx, id) + return +} + func (svc *service) SaveSettings(ctx context.Context, c *model.Client, settings *model.Settings) (err error) { diff --git a/service/transport.go b/service/transport.go index 6540333..108cc18 100644 --- a/service/transport.go +++ b/service/transport.go @@ -549,6 +549,38 @@ func NewHandler(s Service, staticDir string) http.Handler { w.WriteHeader(http.StatusFound) } + subscribe := func(w http.ResponseWriter, req *http.Request) { + c := newClient(w) + ctx := newCtxWithSesionCSRF(req, req.FormValue("csrf_token")) + id, _ := mux.Vars(req)["id"] + + err := s.Subscribe(ctx, c, id) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + s.ServeErrorPage(ctx, c, err) + return + } + + w.Header().Add("Location", req.Header.Get("Referer")) + w.WriteHeader(http.StatusFound) + } + + unSubscribe := func(w http.ResponseWriter, req *http.Request) { + c := newClient(w) + ctx := newCtxWithSesionCSRF(req, req.FormValue("csrf_token")) + id, _ := mux.Vars(req)["id"] + + err := s.UnSubscribe(ctx, c, id) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + s.ServeErrorPage(ctx, c, err) + return + } + + w.Header().Add("Location", req.Header.Get("Referer")) + w.WriteHeader(http.StatusFound) + } + settings := func(w http.ResponseWriter, req *http.Request) { c := newClient(w) ctx := newCtxWithSesionCSRF(req, req.FormValue("csrf_token")) @@ -762,6 +794,8 @@ func NewHandler(s Service, staticDir string) http.Handler { r.HandleFunc("/unmute/{id}", unMute).Methods(http.MethodPost) r.HandleFunc("/block/{id}", block).Methods(http.MethodPost) r.HandleFunc("/unblock/{id}", unBlock).Methods(http.MethodPost) + r.HandleFunc("/subscribe/{id}", subscribe).Methods(http.MethodPost) + r.HandleFunc("/unsubscribe/{id}", unSubscribe).Methods(http.MethodPost) r.HandleFunc("/settings", settings).Methods(http.MethodPost) r.HandleFunc("/muteconv/{id}", muteConversation).Methods(http.MethodPost) r.HandleFunc("/unmuteconv/{id}", unMuteConversation).Methods(http.MethodPost) diff --git a/templates/user.tmpl b/templates/user.tmpl index 6ea79f7..848ca0a 100644 --- a/templates/user.tmpl +++ b/templates/user.tmpl @@ -25,39 +25,51 @@ - {{else}} + {{else}}
{{end}} - {{if .User.Pleroma.Relationship.Requested}} + {{if .User.Pleroma.Relationship.Requested}} - - {{end}} + {{end}} + - + {{if .User.Pleroma.Relationship.Subscribing}} + + {{else}} + + {{end}}