-
Notifications
You must be signed in to change notification settings - Fork 119
/
main.go
92 lines (80 loc) · 2.34 KB
/
main.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
package main
import (
"flag"
"fmt"
"io/ioutil"
"net/http"
"net/http/pprof"
rdebug "runtime/debug"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
autodelete "github.com/riking/AutoDelete"
"gopkg.in/yaml.v2"
)
var flagShardID = flag.Int("shard", -1, "shard ID of this bot")
var flagNoHttp = flag.Bool("nohttp", false, "skip http handler")
var flagMetricsPort = flag.Int("metrics", 6130, "port for metrics listener; shard ID is added")
var flagMetricsListen = flag.String("metricslisten", "127.0.0.4", "addr to listen on for metrics handler")
func main() {
var conf autodelete.Config
flag.Parse()
confBytes, err := ioutil.ReadFile("config.yml")
if err != nil {
fmt.Println("Please copy config.yml.example to config.yml and fill out the values")
return
}
err = yaml.Unmarshal(confBytes, &conf)
if err != nil {
fmt.Println("yaml error:", err)
return
}
if conf.BotToken == "" {
fmt.Println("bot token must be specified")
}
if conf.Shards > 0 && *flagShardID == -1 {
fmt.Println("This AutoDelete instance is configured to be sharded; please specify --shard=n")
return
}
if *flagShardID > conf.Shards {
fmt.Println("error: shard nbr is > shard count")
return
}
b := autodelete.New(conf)
err = b.ConnectDiscord(*flagShardID, conf.Shards)
if err != nil {
fmt.Println("connect error:", err)
return
}
var privHttp http.ServeMux
var pubHttp http.ServeMux
go func() {
for {
time.Sleep(time.Hour * 1)
rdebug.FreeOSMemory()
}
}()
go func() {
privHttp.Handle("/debug/pprof/", http.HandlerFunc(pprof.Index))
privHttp.Handle("/metrics", promhttp.HandlerFor(prometheus.DefaultGatherer, promhttp.HandlerOpts{}))
metricSvr := &http.Server{
Handler: &privHttp,
Addr: fmt.Sprintf("%s:%d", *flagMetricsListen, *flagMetricsPort+*flagShardID),
}
err := metricSvr.ListenAndServe()
fmt.Println("exiting metric server", err)
}()
if !*flagNoHttp {
fmt.Printf("url: %s%s\n", conf.HTTP.Public, "/discord_auto_delete/oauth/start")
pubHttp.HandleFunc("/discord_auto_delete/oauth/start", b.HTTPOAuthStart)
pubHttp.HandleFunc("/discord_auto_delete/oauth/callback", b.HTTPOAuthCallback)
pubSrv := &http.Server{
Handler: &pubHttp,
Addr: conf.HTTP.Listen,
}
err = pubSrv.ListenAndServe()
fmt.Println("exiting main()", err)
} else {
select {}
}
}