/
malachite.go
107 lines (85 loc) · 1.99 KB
/
malachite.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
package main
import (
"fmt"
"io/ioutil"
"log"
"os"
"os/signal"
"strings"
"syscall"
"github.com/bwmarrin/discordgo"
"github.com/cull-methi/malachite/cmd/messagedelete"
"gopkg.in/yaml.v2"
)
var prefix = "./"
// Creds struct to read in the creds yaml file
type Creds struct {
AppToken string
AppID string
BotToken string
BotName string
BotDiscriminator string
BotPermissionsInt int
}
func main() {
configFilePath := os.Getenv("CONFIG_FILE")
creds := Creds{}
err := fillCredsFromConfigFile(configFilePath, &creds)
if err != nil {
log.Fatal(err)
}
dg, err := discordgo.New("Bot " + creds.BotToken)
if err != nil {
log.Fatal(err)
}
dg.AddHandler(messageCreateHandler)
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, os.Kill)
<-sc
// Cleanly close down the Discord session.
dg.Close()
}
func fillCredsFromConfigFile(configFilePath string, creds *Creds) (err error) {
file, err := os.Open(configFilePath)
if err != nil {
return err
}
data, err := ioutil.ReadAll(file)
if err != nil {
return err
}
err = yaml.Unmarshal(data, &creds)
if err != nil {
return err
}
return nil
}
func messageCreateHandler(s *discordgo.Session, m *discordgo.MessageCreate) {
if m.Author.ID == s.State.User.ID {
return
}
content := m.Content
if !strings.HasPrefix(content, prefix) {
return
}
content = content[2:len(content)]
fmt.Println(content)
// If the message is "ping" reply with "Pong!"
if content == "ping" {
s.ChannelMessageSend(m.ChannelID, "Yeah, yeah, I'm here bruh")
}
if content == "test" {
s.ChannelMessageSend(m.ChannelID, "Here's your test, fuckwad")
}
err := error{}
if strings.HasPrefix(content, "delete") {
err = messagedelete.Entrypoint(s, m)
}
}