forked from chanbakjsd/TerraOceanPlugin
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package modules
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
|
|
"TerraOceanBot/discord/backend"
|
|
"TerraOceanBot/discord/config"
|
|
"TerraOceanBot/discord/message"
|
|
)
|
|
|
|
//adminID is my Discord user ID (chanbakjsd#7968).
|
|
const adminID = "218983355746746369"
|
|
|
|
func adminOnly(passOn CommandHandler) CommandHandler {
|
|
return func(s *discordgo.Session, m *discordgo.MessageCreate, command []string) {
|
|
if m.Author.ID != adminID {
|
|
return
|
|
}
|
|
passOn(s, m, command)
|
|
}
|
|
}
|
|
|
|
var sendAs = adminOnly(enforceArgumentCount(config.SendAsUsage, 3,
|
|
//!sendas <channel ID> <message>
|
|
func(s *discordgo.Session, m *discordgo.MessageCreate, command []string) {
|
|
msg, err := s.ChannelMessageSend(command[1], command[2])
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
message.InitNewEmbed(config.SuccessTitle, config.SendAsSuccessfulDescription, config.SuccessColour).
|
|
AddField(config.SendAsSuccessfulMessageID, msg.ID).
|
|
Send(s, m.ChannelID)
|
|
},
|
|
))
|
|
|
|
var editAs = adminOnly(enforceArgumentCount(config.EditAsUsage, 4,
|
|
//!editas <channel ID> <message ID> <message>
|
|
func(s *discordgo.Session, m *discordgo.MessageCreate, command []string) {
|
|
_, err := s.ChannelMessageEdit(command[1], command[2], command[3])
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
message.InitNewEmbed(config.SuccessTitle, config.EditAsSuccessfulDescription, config.SuccessColour).
|
|
Send(s, m.ChannelID)
|
|
},
|
|
))
|
|
|
|
var viewTrustLevel = adminOnly(func(s *discordgo.Session, m *discordgo.MessageCreate, command []string) {
|
|
if len(command) > 1 {
|
|
value, err := backend.GetTrust(s, command[1])
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
total, err := backend.GetTotalTrust(s)
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
message.NewEmbed().SetColour(config.VTLColour).
|
|
SetTitle(fmt.Sprintf(config.VTLSingleUserTitle, command[1], command[1])).
|
|
AddField(config.VTLSingleUserValue, formatFloat(value, 1)).
|
|
AddField(config.VTLSingleUserServerTotal, formatFloat(total, 1)).
|
|
AddField(config.VTLSingleUserPercentage, formatFloat(value/total*100, 2)).
|
|
Send(s, m.ChannelID)
|
|
return
|
|
}
|
|
|
|
total, err := backend.GetTotalTrust(s)
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
members, err := backend.GetAllMembers(s)
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
|
|
//Generate message
|
|
msg := message.NewEmbed().SetColour(config.VTLColour).SetTitle(config.VTLGlobalTitle)
|
|
for _, v := range members {
|
|
value, err := backend.GetTrust(s, v.User.ID)
|
|
if err == backend.ErrNotMember {
|
|
continue
|
|
}
|
|
if err != nil {
|
|
message.AuditError(s, m.ChannelID, err)
|
|
return
|
|
}
|
|
msg.AddInlineField(
|
|
v.User.Username+"#"+v.User.Discriminator,
|
|
fmt.Sprintf(config.VTLGlobalFieldFormat, value, total, value/total*100),
|
|
)
|
|
}
|
|
msg.Send(s, m.ChannelID)
|
|
})
|