forked from sivchari/gotwtr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mute.go
148 lines (128 loc) · 3.87 KB
/
mute.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
package gotwtr
import (
"bytes"
"context"
"encoding/json"
"errors"
"fmt"
"net/http"
)
func muting(ctx context.Context, c *client, userID string, opt ...*MuteOption) (*MutingResponse, error) {
if userID == "" {
return nil, errors.New("muting: userID parameter is required")
}
ep := fmt.Sprintf(mutingURL, userID)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, ep, nil)
if err != nil {
return nil, fmt.Errorf("muting new request with ctx: %w", err)
}
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.bearerToken))
var fopt MuteOption
switch len(opt) {
case 0:
// do nothing
case 1:
fopt = *opt[0]
default:
return nil, errors.New("muting: only one option is allowed")
}
const (
minimumMaxResults = 1
maximumMaxResults = 1000
defaultMaxResults = 100
)
if fopt.MaxResults == 0 {
fopt.MaxResults = defaultMaxResults
}
if fopt.MaxResults < minimumMaxResults || fopt.MaxResults > maximumMaxResults {
return nil, fmt.Errorf("muting: maxResults must be between %d and %d", minimumMaxResults, maximumMaxResults)
}
fopt.addQuery(req)
resp, err := c.client.Do(req)
if err != nil {
return nil, fmt.Errorf("muting response: %w", err)
}
defer resp.Body.Close()
var m MutingResponse
if err := json.NewDecoder(resp.Body).Decode(&m); err != nil {
return nil, fmt.Errorf("muting: %w", err)
}
if resp.StatusCode != http.StatusOK {
return &m, &HTTPError{
APIName: "muting",
Status: resp.Status,
URL: req.URL.String(),
}
}
return &m, nil
}
func postMuting(ctx context.Context, c *client, userID string, targetUserID string) (*PostMutingResponse, error) {
if userID == "" {
return nil, errors.New("post muting: userID parameter is required")
}
ep := fmt.Sprintf(postMutingURL, userID)
if targetUserID == "" {
return nil, errors.New("post muting: targetUserID parameter is required")
}
body := &MutingBody{
TargetUserID: targetUserID,
}
j, err := json.Marshal(body)
if err != nil {
return nil, fmt.Errorf("post muting json marshal: %w", err)
}
req, err := http.NewRequestWithContext(ctx, http.MethodPost, ep, bytes.NewBuffer(j))
if err != nil {
return nil, fmt.Errorf("post muting new request with ctx: %w", err)
}
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.bearerToken))
req.Header.Set("Content-Type", "application/json")
resp, err := c.client.Do(req)
if err != nil {
return nil, fmt.Errorf("post muting response: %w", err)
}
defer resp.Body.Close()
var postMuting PostMutingResponse
if err := json.NewDecoder(resp.Body).Decode(&postMuting); err != nil {
return nil, fmt.Errorf("post muting decode: %w", err)
}
if resp.StatusCode != http.StatusOK {
return &postMuting, &HTTPError{
APIName: "post muting",
Status: resp.Status,
URL: req.URL.String(),
}
}
return &postMuting, nil
}
func undoMuting(ctx context.Context, c *client, sourceUserID string, targetUserID string) (*UndoMutingResponse, error) {
if sourceUserID == "" {
return nil, errors.New("undo muting: sourceUserID parameter is required")
}
if targetUserID == "" {
return nil, errors.New("undo muting: targetUserID parameter is required")
}
ep := fmt.Sprintf(undoMutingURL, sourceUserID, targetUserID)
req, err := http.NewRequestWithContext(ctx, http.MethodDelete, ep, nil)
if err != nil {
return nil, fmt.Errorf("undo muting new request with ctx: %w", err)
}
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.bearerToken))
resp, err := c.client.Do(req)
if err != nil {
return nil, fmt.Errorf("undo muting response: %w", err)
}
defer resp.Body.Close()
var undoMuting UndoMutingResponse
if err := json.NewDecoder(resp.Body).Decode(&undoMuting); err != nil {
return nil, fmt.Errorf("undo muting decode: %w", err)
}
if resp.StatusCode != http.StatusOK {
return &undoMuting, &HTTPError{
APIName: "undo muting",
Status: resp.Status,
URL: req.URL.String(),
}
}
return &undoMuting, nil
}