From 1eb2acf3186660157264a306e9f0392b9fb242af Mon Sep 17 00:00:00 2001 From: Vartan Benohanian Date: Tue, 25 Aug 2020 18:45:25 -0400 Subject: [PATCH] Add more api/friend and api/unfriend methods Signed-off-by: Vartan Benohanian --- reddit/moderation.go | 140 ++++++++++++++++++++++++++++++ reddit/moderation_test.go | 177 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 317 insertions(+) diff --git a/reddit/moderation.go b/reddit/moderation.go index 76fd286..c8dd91d 100644 --- a/reddit/moderation.go +++ b/reddit/moderation.go @@ -382,3 +382,143 @@ func (s *ModerationService) Unban(ctx context.Context, subreddit string, usernam return s.client.Do(ctx, req, nil) } + +// BanWiki a user from contributing to the subreddit wiki. +func (s *ModerationService) BanWiki(ctx context.Context, subreddit string, username string, config *BanConfig) (*Response, error) { + path := fmt.Sprintf("r/%s/api/friend", subreddit) + + form, err := query.Values(config) + if err != nil { + return nil, err + } + + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "wikibanned") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// UnbanWiki a user from contributing to the subreddit wiki. +func (s *ModerationService) UnbanWiki(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/unfriend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "wikibanned") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// Mute a user in the subreddit. +func (s *ModerationService) Mute(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/friend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "muted") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// Unmute a user in the subreddit. +func (s *ModerationService) Unmute(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/unfriend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "muted") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// ApproveUser adds a user as an approved user to the subreddit. +func (s *ModerationService) ApproveUser(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/friend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "contributor") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// UnapproveUser removes a user as an approved user to the subreddit. +func (s *ModerationService) UnapproveUser(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/unfriend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "contributor") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// ApproveUserWiki adds a user as an approved wiki contributor in the subreddit. +func (s *ModerationService) ApproveUserWiki(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/friend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "wikicontributor") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} + +// UnapproveUserWiki removes a user as an approved wiki contributor in the subreddit. +func (s *ModerationService) UnapproveUserWiki(ctx context.Context, subreddit string, username string) (*Response, error) { + path := fmt.Sprintf("r/%s/api/unfriend", subreddit) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", username) + form.Set("type", "wikicontributor") + + req, err := s.client.NewRequestWithForm(http.MethodPost, path, form) + if err != nil { + return nil, err + } + + return s.client.Do(ctx, req, nil) +} diff --git a/reddit/moderation_test.go b/reddit/moderation_test.go index 38b1dd5..01e1a58 100644 --- a/reddit/moderation_test.go +++ b/reddit/moderation_test.go @@ -376,3 +376,180 @@ func TestModerationService_Unban(t *testing.T) { _, err := client.Moderation.Unban(ctx, "testsubreddit", "testuser") require.NoError(t, err) } + +func TestModerationService_BanWiki(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/friend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "wikibanned") + form.Set("reason", "test reason") + form.Set("note", "test mod note") + form.Set("duration", "5") + form.Set("ban_message", "test message") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.BanWiki(ctx, "testsubreddit", "testuser", &BanConfig{ + Reason: "test reason", + ModNote: "test mod note", + Days: Int(5), + Message: "test message", + }) + require.NoError(t, err) +} + +func TestModerationService_UnbanWiki(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/unfriend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "wikibanned") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.UnbanWiki(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_Mute(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/friend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "muted") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.Mute(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_Unmuted(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/unfriend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "muted") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.Unmute(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_ApproveUser(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/friend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "contributor") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.ApproveUser(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_UnapproveUser(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/unfriend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "contributor") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.UnapproveUser(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_ApproveUserWiki(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/friend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "wikicontributor") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.ApproveUserWiki(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +} + +func TestModerationService_UnapproveUserWiki(t *testing.T) { + client, mux, teardown := setup() + defer teardown() + + mux.HandleFunc("/r/testsubreddit/api/unfriend", func(w http.ResponseWriter, r *http.Request) { + require.Equal(t, http.MethodPost, r.Method) + + form := url.Values{} + form.Set("api_type", "json") + form.Set("name", "testuser") + form.Set("type", "wikicontributor") + + err := r.ParseForm() + require.NoError(t, err) + require.Equal(t, form, r.PostForm) + }) + + _, err := client.Moderation.UnapproveUserWiki(ctx, "testsubreddit", "testuser") + require.NoError(t, err) +}