diff options
| -rw-r--r-- | mastodon/accounts.go | 21 | ||||
| -rw-r--r-- | service/auth.go | 24 | ||||
| -rw-r--r-- | service/logging.go | 16 | ||||
| -rw-r--r-- | service/service.go | 12 | ||||
| -rw-r--r-- | service/transport.go | 34 | ||||
| -rw-r--r-- | templates/user.tmpl | 28 | 
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"> | 
