-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
101 lines (86 loc) · 2.17 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
93
94
95
96
97
98
99
100
101
package main
import (
"flag"
"fmt"
"github.com/bwmarrin/discordgo"
"io"
"log"
"os"
"os/signal"
"syscall"
"time"
)
// Variables used for command line parameters
var (
ConfPath string
Conf Config
)
func init() {
// Read our command line options
flag.StringVar(&ConfPath, "c", "hades.conf", "Path to Config File")
flag.Parse()
_, err := os.Stat(ConfPath)
if err != nil {
log.Fatal("Config file is missing: ", ConfPath)
}
// Verify we can actually read our config file
err = ReadConfig(ConfPath)
if err != nil {
log.Fatal("error reading config file at: ", ConfPath)
return
}
}
func main() {
fmt.Println("\n\n|| Starting Hades Bot ||")
log.SetOutput(io.Discard)
// Create a new Discord session using the provided bot token.
dg, err := discordgo.New("Bot " + Conf.DiscordConfig.Token)
if err != nil {
fmt.Println("error creating Discord session,", err)
return
}
defer func(dg *discordgo.Session) {
err := dg.Close()
if err != nil {
panic(err)
}
}(dg)
dg.Identify.Intents |= discordgo.IntentsAll
dg.AddHandler(Ready)
dg.AddHandler(func(s *discordgo.Session, r *discordgo.GuildMemberAdd) {
createdAt, err := discordgo.SnowflakeTimestamp(r.Member.User.ID)
if err != nil {
fmt.Println("error getting user creation time,", err)
return
}
// print the number of hours since the account was created
hoursSinceCreation := int64(time.Now().Sub(createdAt).Hours())
if hoursSinceCreation < 48 {
// Ban the user
err := s.GuildBanCreateWithReason(r.GuildID, r.Member.User.ID, "Hades Ban", 0)
if err != nil {
fmt.Println("error banning user,", err)
return
}
}
})
// Open a websocket connection to Discord and begin listening.
err = dg.Open()
if err != nil {
fmt.Println("error opening connection,", err)
return
}
// Wait here until CTRL-C or other term signal is received.
fmt.Println("Bot is now running. Press CTRL-C to exit.")
sc := make(chan os.Signal, 1)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
<-sc
// Cleanly close down the Discord session.
err = dg.Close()
if err != nil {
panic(err)
}
}
func Ready(s *discordgo.Session, event *discordgo.Ready) {
fmt.Println("Bot is up!")
}