Add account {,un}subscribe

6 files changed, 127 insertions, 8 deletions

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 @@
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="unfollow" class="btn-link">
 			</form>
-			{{else}} 
+			{{else}}
 			<form class="d-inline" action="/follow/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="{{if .User.Pleroma.Relationship.Requested}}resend request{{else}}follow{{end}}" class="btn-link">
 			</form>
 			{{end}}
-			{{if .User.Pleroma.Relationship.Requested}} 
+			{{if .User.Pleroma.Relationship.Requested}}
 			-
 			<form class="d-inline" action="/unfollow/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="cancel request" class="btn-link">
 			</form>
-			{{end}} 
+			{{end}}
+			-
+			{{if .User.Pleroma.Relationship.Subscribing}}
+			<form class="d-inline" action="/unsubscribe/{{.User.ID}}" method="post">
+				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
+				<input type="submit" value="unsubscribe" class="btn-link">
+			</form>
+			{{else}}
+			<form class="d-inline" action="/subscribe/{{.User.ID}}" method="post">
+				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
+				<input type="submit" value="subscribe" class="btn-link">
+			</form>
+			{{end}}
 		</div>
 		<div>
-			{{if .User.Pleroma.Relationship.Blocking}} 
+			{{if .User.Pleroma.Relationship.Blocking}}
 			<form class="d-inline" action="/unblock/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="unblock" class="btn-link">
 			</form>
-			{{else}} 
+			{{else}}
 			<form class="d-inline" action="/block/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="block" class="btn-link">
 			</form>
-			{{end}} 
+			{{end}}
 			-
-			{{if .User.Pleroma.Relationship.Muting}} 
+			{{if .User.Pleroma.Relationship.Muting}}
 			<form class="d-inline" action="/unmute/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="unmute" class="btn-link">
 			</form>
-			{{else}} 
+			{{else}}
 			<form class="d-inline" action="/mute/{{.User.ID}}" method="post">
 				<input type="hidden" name="csrf_token" value="{{$.Ctx.CSRFToken}}">
 				<input type="submit" value="mute" class="btn-link">