-
Notifications
You must be signed in to change notification settings - Fork 0
/
AcademicBot.go
120 lines (102 loc) · 2.93 KB
/
AcademicBot.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
108
109
110
111
112
113
114
115
116
117
118
119
120
package main
import (
"fmt"
"os"
"os/signal"
"strings"
"syscall"
"time"
"github.com/bwmarrin/discordgo"
)
var (
token = "YOUR TOKEN"
voiceStates = make(map[string]map[string]time.Time) // GuildID -> UserID -> JoinTime
welcomeChannel = ""
)
func main() {
dg, err := discordgo.New("Bot " + token)
if err != nil {
fmt.Println("Error creating Discord session:", err)
return
}
dg.AddHandler(messageCreate)
dg.AddHandler(voiceStateUpdate)
dg.AddHandler(guildMemberAdd)
err = dg.Open()
if err != nil {
fmt.Println("Error opening Discord session:", err)
return
}
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
dg.Close()
}
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
if m.Author.ID == s.State.User.ID {
return
}
switch {
case strings.Contains(m.Content, "!ping"):
s.ChannelMessageSend(m.ChannelID, "Pong!")
case strings.HasPrefix(m.Content, "!markattendance"):
guildID := m.GuildID
channelID := m.ChannelID
if voiceStates[guildID] == nil {
s.ChannelMessageSend(channelID, "No users in voice channels.")
return
}
embed := &discordgo.MessageEmbed{
Title: "Attendance List",
Color: 0x00ff00, // Green color
Fields: []*discordgo.MessageEmbedField{},
}
for userID, joinTime := range voiceStates[guildID] {
user, err := s.User(userID)
if err == nil {
duration := time.Since(joinTime).Truncate(time.Second)
embed.Fields = append(embed.Fields, &discordgo.MessageEmbedField{
Name: user.Username,
Value: fmt.Sprintf("Joined %s ago", duration.String()),
Inline: false,
})
}
}
s.ChannelMessageSendEmbed(channelID, embed)
case strings.HasPrefix(m.Content, "!botwelcome"):
channelID := extractChannelID(m.Content)
if channelID != "" {
welcomeChannel = channelID
s.ChannelMessageSend(m.ChannelID, fmt.Sprintf("Welcome messages will be sent to <#%s>.", welcomeChannel))
} else {
s.ChannelMessageSend(m.ChannelID, "Invalid channel. Please mention a valid channel.")
}
}
}
func voiceStateUpdate(s *discordgo.Session, vs *discordgo.VoiceStateUpdate) {
guildID := vs.GuildID
userID := vs.UserID
if vs.ChannelID != "" { // User joined a voice channel
if voiceStates[guildID] == nil {
voiceStates[guildID] = make(map[string]time.Time)
}
voiceStates[guildID][userID] = time.Now()
} else { // User left a voice channel
delete(voiceStates[guildID], userID)
}
}
func guildMemberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
if welcomeChannel != "" {
s.ChannelMessageSend(welcomeChannel, fmt.Sprintf("Welcome, %s, to the server!", m.User.Mention()))
}
}
func extractChannelID(message string) string {
fields := strings.Fields(message)
if len(fields) >= 2 {
channelID := strings.TrimPrefix(fields[1], "<#")
channelID = strings.TrimSuffix(channelID, ">")
return channelID
}
return ""
}