forked from indes/flowerss-bot
/
tgraph.go
71 lines (60 loc) · 1.54 KB
/
tgraph.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
package tgraph
import (
"github.com/nerdneilsfield/flowerss-bot/internal/config"
"github.com/indes/telegraph-go"
"go.uber.org/zap"
)
const (
//verbose = false
htmlContent = `<h1>hello</h1>`
)
var (
authToken = config.TelegraphToken
socks5Proxy = config.Socks5
authorUrl = "https://github.com/nerdneilsfield/flowerss-bot/"
authorName = "flowerss"
verbose = false
//client *telegraph.Client
clientPool []*telegraph.Client
)
func init() {
if config.EnableTelegraph {
zap.S().Infow("telegraph enabled",
"token count", len(authToken),
"token list", authToken,
)
telegraph.Verbose = verbose
for _, t := range authToken {
client, err := telegraph.Load(t, socks5Proxy)
if err != nil {
zap.S().Errorw("telegraph load error",
"error", err,
"token", t,
)
} else {
clientPool = append(clientPool, client)
}
}
if len(clientPool) == 0 {
if config.TelegraphAccountName == "" {
config.EnableTelegraph = false
zap.S().Error("telegraph token error, telegraph disabled")
} else if len(authToken) == 0 {
// create account
client, err := telegraph.Create(
config.TelegraphAccountName,
config.TelegraphAuthorName,
config.TelegraphAuthorURL,
config.Socks5,
)
if err != nil {
config.EnableTelegraph = false
zap.S().Errorw("create telegraph account fail, telegraph disabled", "error", err)
}
clientPool = append(clientPool, client)
zap.S().Infow("create telegraph account success",
"telegraph token", client.AccessToken)
}
}
}
}