hak5
/
scuzzy
mirror of https://github.com/hak5/scuzzy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

yo

master
Marc 2021-05-29 00:09:48 +01:00
parent 2bcf6b7136
commit 7fb7473e73
No known key found for this signature in database
GPG Key ID: 0657563F705ACAAE
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
overwatch/overwatch.go
View File

@ -7,12 +7,12 @@ import (
) )
type UserMessageStat struct { type UserMessageStat struct {
UserID string UserID string
Username string Username string
MessagesLastDay uint64 MessagesLastDay uint64
MessagesLastHour uint64 MessagesLastHour uint64
MessagesLastFiveMins uint64 MessagesLastFiveMins uint64
MessagesLastTenSecs uint64 MessagesLastThirtySecs uint64
} }
type Overwatch struct { type Overwatch struct {
@ -38,12 +38,12 @@ func (o *Overwatch) handleUserStat(s *discordgo.Session, m *discordgo.MessageCre
user, ok := o.UserMessages[userID] user, ok := o.UserMessages[userID]
if !ok { if !ok {
o.UserMessages[userID] = &UserMessageStat{ o.UserMessages[userID] = &UserMessageStat{
UserID: userID, UserID: userID,
Username: m.Author.Username, Username: m.Author.Username,
MessagesLastDay: 0, MessagesLastDay: 0,
MessagesLastHour: 0, MessagesLastHour: 0,
MessagesLastFiveMins: 0, MessagesLastFiveMins: 0,
MessagesLastTenSecs: 0, MessagesLastThirtySecs: 0,
} }
user = o.UserMessages[userID] user = o.UserMessages[userID]
} }
@ -51,7 +51,7 @@ func (o *Overwatch) handleUserStat(s *discordgo.Session, m *discordgo.MessageCre
user.MessagesLastDay++ user.MessagesLastDay++
user.MessagesLastHour++ user.MessagesLastHour++
user.MessagesLastFiveMins++ user.MessagesLastFiveMins++
user.MessagesLastTenSecs++ user.MessagesLastThirtySecs++
return nil return nil
} }
@ -60,11 +60,11 @@ func (o *Overwatch) handleUserStat(s *discordgo.Session, m *discordgo.MessageCre
func (o *Overwatch) Run() { func (o *Overwatch) Run() {
// Five second loop // Five second loop
go func() { go func() {
for range time.Tick(10 * time.Second) { for range time.Tick(5 * time.Second) {
for _, user := range o.UserMessages { for _, user := range o.UserMessages {
log.Printf("User: %+v", user) log.Printf("User: %+v", user)
// load the threshold from the config file // load the threshold from the config file
if user.MessagesLastTenSecs > 10 { if user.MessagesLastThirtySecs > 10 {
// Set slow mode, kick user? add kick count? // Set slow mode, kick user? add kick count?
log.Printf("[*] User %s (%s) has triggered the message threshold.", user.Username, user.UserID) log.Printf("[*] User %s (%s) has triggered the message threshold.", user.Username, user.UserID)
} }
@ -74,10 +74,10 @@ func (o *Overwatch) Run() {
// Clear Counters // Clear Counters
go func() { go func() {
for range time.Tick(10 * time.Second) { for range time.Tick(30 * time.Second) {
log.Printf("[*] Resetting all users 10 second message counters") log.Printf("[*] Resetting all users 10 second message counters")
for _, user := range o.UserMessages { for _, user := range o.UserMessages {
user.MessagesLastTenSecs = 0 user.MessagesLastThirtySecs = 0
} }
} }