/
twitter.go
65 lines (52 loc) · 1.5 KB
/
twitter.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
package dispatcher
import (
"github.com/dghubble/go-twitter/twitter"
"github.com/dghubble/oauth1"
log "github.com/sirupsen/logrus"
)
type TwitterDispatcher struct {
config *oauth1.Config
token *oauth1.Token
client *twitter.Client
}
func (t *TwitterDispatcher) getCredential(apiKey string, apiSecretKey string, accessToken string, accessTokenSecret string) {
var err error
t.config = oauth1.NewConfig(apiKey, apiSecretKey)
t.token = oauth1.NewToken(accessToken, accessTokenSecret)
if err != nil {
log.Fatal(err)
}
}
func (t *TwitterDispatcher) Start(apiKey string, apiSecretKey string, accessToken string, accessTokenSecret string) {
t.getCredential(apiKey, apiSecretKey, accessToken, accessTokenSecret)
httpClient := t.config.Client(oauth1.NoContext, t.token)
t.client = twitter.NewClient(httpClient)
}
func (t *TwitterDispatcher) SendMessage(msg string) (int64, error) {
tweet, _, err := t.client.Statuses.Update(msg, nil)
return tweet.ID, err
}
func (t *TwitterDispatcher) ReplyMessage(msg string, statusId int64) (int64, error) {
tweet, _, err := t.client.Statuses.Update(msg, &twitter.StatusUpdateParams{
InReplyToStatusID: statusId,
})
return tweet.ID, err
}
func (t *TwitterDispatcher) DeleteEmpty(msg []string) []string {
var r []string
for _, str := range msg {
if str != "" {
r = append(r, str)
}
}
return r
}
func (t *TwitterDispatcher) SplitMessage(msg []string) []string {
var r []string
for _, str := range msg {
if str != "" {
r = append(r, str)
}
}
return r
}