mirror of https://github.com/hak5/scuzzy.git
Create Scuzzy
Features: Hastily add usercolor feature Main: Show prefix in bot status Features: add cat Main/Features/Auth: Add basic Role Name/ID based command authentication and bot configuration. Main: Sanity check for missing flags. Features: Implement command deletion and tag-responses Misc: Add cmd/main to .gitignore Misc: Delete built binary Main: Refactor main.go - Use log.Fatal instead of fmt.Printf and os.Exit - Move Config reading into seperate function - Parse command line flags in main() - Instantiate Features.go struct inside main() - Use simple ticker to set bot status every 15 minutes Auth: Refactor auth - Return early when building Admin role list - Move CheckAdminrole() into auth.go, remove roles.go Features: Refactor features - Remove New() function - Move help.go, ping.go functionality into misc.go Features: Add bot control feature to set status. Features: Implement helper function for embeds, return errors. Main: Rename config file to bot_config.json Features: Use success embed for user color. Features: Add purge messages feature. Main: Add status_text property to config, use it for Bot Status. Features: Send welcome message to new users, implement auto-join roles. Main: Use complex user status Main: Reduce ticker interval to 10 minutes for bot status update. Main: Create helper function for complex custom embeds. Features: Implement info command. Features: Convert requested user color to lowercase. Features: Add markdown helper Features: Fix markdown helper typo Features: Use error embed for handleSetStatus incorrect permissions. Features: Add strikethrough and spoilers information to markdown info. Features: Add info and md to help text. Features: Use backtick instead of single quote for command prefix in help text. Features: Remove markdown info message after 15 seconds. Features: Fix typo in status setting permission error. Features: Add commands to convert between Celsius and Farenheit Features: Update help to include temperature conversion commands. Features/Models/Main: Load color roles from JSON config. Auth/Main/Features/Models: Implement config-based channel white/blacklisting. Main/Models: Remove unused admin_only config field. Features: Simplify error handling. Auth: Remove leftover AdminOnly field. Features: Use command restriction for usercolor commands. Main: Update default config to restrict color and colors commands. Auth/Features: Return true for undefined commands in restrictions list by default. Features: Use better error messages for channel white/blacklist Auth: Improve channel white/blacklist behavior. Main: Replace broken ticker for status updates Features: Get bot avatar via Discord for embed. Features: Delete requesting message for markdown info. Features: Handle potential error in deleting help requester message.master
Marc Egerton
commit
d54d2bb812
|
|
@ -0,0 +1,3 @@
|
||||||
|
.idea/
|
||||||
|
main
|
||||||
|
cmd/main
|
||||||
|
|
@ -0,0 +1,92 @@
|
||||||
|
package auth
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
"github.com/foxtrot/scuzzy/models"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AdminRole struct {
|
||||||
|
Name string
|
||||||
|
ID string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Auth struct {
|
||||||
|
AdminRoles []AdminRole
|
||||||
|
CommandRestrictions []models.CommandRestriction
|
||||||
|
Guild *discordgo.Guild
|
||||||
|
|
||||||
|
Config *models.Configuration
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(config *models.Configuration, guild *discordgo.Guild) *Auth {
|
||||||
|
var ars []AdminRole
|
||||||
|
for _, gRole := range guild.Roles {
|
||||||
|
for _, aRole := range config.AdminRoles {
|
||||||
|
if aRole != gRole.Name {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
ar := AdminRole{
|
||||||
|
Name: gRole.Name,
|
||||||
|
ID: gRole.ID,
|
||||||
|
}
|
||||||
|
ars = append(ars, ar)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var crs []models.CommandRestriction
|
||||||
|
for _, cRes := range config.CommandRestrictions {
|
||||||
|
cr := models.CommandRestriction{
|
||||||
|
Command: cRes.Command,
|
||||||
|
Mode: cRes.Mode,
|
||||||
|
Channels: cRes.Channels,
|
||||||
|
}
|
||||||
|
crs = append(crs, cr)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Auth{
|
||||||
|
AdminRoles: ars,
|
||||||
|
CommandRestrictions: crs,
|
||||||
|
Config: config,
|
||||||
|
Guild: guild,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Auth) CheckAdminRole(m *discordgo.Member) bool {
|
||||||
|
for _, aR := range a.AdminRoles {
|
||||||
|
for _, mID := range m.Roles {
|
||||||
|
if aR.ID == mID {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Auth) CheckCommandRestrictions(m *discordgo.MessageCreate) bool {
|
||||||
|
cName := strings.Split(m.Content, " ")[0]
|
||||||
|
cName = strings.Replace(cName, a.Config.CommandKey, "", 1)
|
||||||
|
cChanID := m.ChannelID
|
||||||
|
|
||||||
|
for _, cR := range a.CommandRestrictions {
|
||||||
|
if cName == cR.Command {
|
||||||
|
for _, cID := range cR.Channels {
|
||||||
|
if cID == cChanID && cR.Mode == "white" {
|
||||||
|
return true
|
||||||
|
} else if cID == cChanID && cR.Mode == "black" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if cR.Mode == "white" {
|
||||||
|
return false
|
||||||
|
} else {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,24 @@
|
||||||
|
{
|
||||||
|
"guild_id": "506629366659153951",
|
||||||
|
|
||||||
|
"status_text": ".help",
|
||||||
|
"welcome_text": "Hi! Welcome to the Hak5 Discord Server. Please be sure to check out the #rules channel and say hello in #general!",
|
||||||
|
"command_key": ".",
|
||||||
|
|
||||||
|
"admin_roles": ["Admin", "Moderator"],
|
||||||
|
"join_role_ids": ["697929841638637778"],
|
||||||
|
|
||||||
|
"command_restrictions": [
|
||||||
|
{ "command": "color", "mode": "white", "channels": ["698567678025138277", "698519835532984470"] },
|
||||||
|
{ "command": "colors", "mode": "white", "channels": ["698567678025138277", "698519835532984470"] }
|
||||||
|
],
|
||||||
|
|
||||||
|
"color_roles": [
|
||||||
|
{ "color": "red", "id": "697930042491142174" },
|
||||||
|
{ "color": "blue", "id": "697930093766639699" },
|
||||||
|
{ "color": "green", "id": "697930441939877970" },
|
||||||
|
{ "color": "purple", "id": "697930578485444610" },
|
||||||
|
{ "color": "yellow", "id": "697930630247350402" },
|
||||||
|
{ "color": "pink", "id": "697965484313935966" }
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,134 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"github.com/foxtrot/scuzzy/auth"
|
||||||
|
"github.com/foxtrot/scuzzy/features"
|
||||||
|
"github.com/foxtrot/scuzzy/models"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Core Bot Properties
|
||||||
|
var (
|
||||||
|
Token string
|
||||||
|
ConfigPath string
|
||||||
|
Config models.Configuration
|
||||||
|
)
|
||||||
|
|
||||||
|
func getConfig() error {
|
||||||
|
cf, err := ioutil.ReadFile(ConfigPath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = json.Unmarshal(cf, &Config)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Parse and Check Flags
|
||||||
|
flag.StringVar(&Token, "t", "", "Bot Token")
|
||||||
|
flag.StringVar(&ConfigPath, "c", "", "Config Path")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
if len(Token) == 0 {
|
||||||
|
log.Fatal("Error: No Auth Token supplied.")
|
||||||
|
}
|
||||||
|
if len(ConfigPath) == 0 {
|
||||||
|
log.Fatal("Error: No Config Path supplied.")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get Config
|
||||||
|
err := getConfig()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Instantiate Bot
|
||||||
|
bot, err := discordgo.New("Bot " + Token)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open Connection
|
||||||
|
err = bot.Open()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setup Auth
|
||||||
|
Config.Guild, err = bot.Guild(Config.GuildID)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
var a *auth.Auth
|
||||||
|
a = auth.New(&Config, Config.Guild)
|
||||||
|
|
||||||
|
// Setup Handlers
|
||||||
|
f := features.Features{
|
||||||
|
Token: Token,
|
||||||
|
Auth: a,
|
||||||
|
Config: Config,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register Handlers
|
||||||
|
bot.AddHandler(f.OnMessageCreate)
|
||||||
|
bot.AddHandler(f.OnUserJoin)
|
||||||
|
|
||||||
|
fmt.Println("Bot Running.")
|
||||||
|
|
||||||
|
// Set Bot Status
|
||||||
|
go func() {
|
||||||
|
usd := discordgo.UpdateStatusData{
|
||||||
|
IdleSince: nil,
|
||||||
|
Game: &discordgo.Game{
|
||||||
|
Name: Config.StatusText,
|
||||||
|
Type: 0,
|
||||||
|
URL: "",
|
||||||
|
Details: "",
|
||||||
|
State: "",
|
||||||
|
TimeStamps: discordgo.TimeStamps{},
|
||||||
|
Assets: discordgo.Assets{},
|
||||||
|
ApplicationID: "",
|
||||||
|
Instance: -1,
|
||||||
|
},
|
||||||
|
AFK: false,
|
||||||
|
Status: "online",
|
||||||
|
}
|
||||||
|
err = bot.UpdateStatusComplex(usd)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// For some reason the bot's status will regularly disappear...
|
||||||
|
for _ = range time.Tick(10 * time.Minute) {
|
||||||
|
err := bot.UpdateStatusComplex(usd)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Catch SIGINT
|
||||||
|
sc := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sc, syscall.SIGINT, syscall.SIGKILL)
|
||||||
|
<-sc
|
||||||
|
|
||||||
|
err = bot.Close()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error: " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,33 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
discordgo "github.com/bwmarrin/discord.go"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) handleSetStatus(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
if !f.Auth.CheckAdminRole(m.Member) {
|
||||||
|
return errors.New("You do not have permissions to use that command.")
|
||||||
|
}
|
||||||
|
|
||||||
|
stSplit := strings.SplitN(m.Content, " ", 2)
|
||||||
|
if len(stSplit) < 2 {
|
||||||
|
return errors.New("You did not specify a status.")
|
||||||
|
}
|
||||||
|
|
||||||
|
st := stSplit[1]
|
||||||
|
|
||||||
|
err := s.UpdateStatus(0, st)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
msg := f.CreateDefinedEmbed("Set Status", "Operation completed successfully.", "success")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,12 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/foxtrot/scuzzy/auth"
|
||||||
|
"github.com/foxtrot/scuzzy/models"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Features struct {
|
||||||
|
Token string
|
||||||
|
Auth *auth.Auth
|
||||||
|
Config models.Configuration
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,171 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) OnMessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Ignore the bot itself
|
||||||
|
if m.Author.ID == s.State.User.ID {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cKey := f.Config.CommandKey
|
||||||
|
cName := strings.Split(m.Content, " ")[0]
|
||||||
|
|
||||||
|
if !strings.HasPrefix(cName, cKey) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
switch cName {
|
||||||
|
/* Misc Commands */
|
||||||
|
case cKey + "help":
|
||||||
|
err = f.handleHelp(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Help)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "info":
|
||||||
|
err = f.handleInfo(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Info)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "ping":
|
||||||
|
err = f.handlePing(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Ping)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "no":
|
||||||
|
err = f.handleCat(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (No)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "md":
|
||||||
|
err = f.handleMarkdownInfo(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Markdown)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "ctof":
|
||||||
|
err = f.handleCtoF(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (CtoF)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "ftoc":
|
||||||
|
err = f.handleFtoC(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (FtoC)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
/* Moderation */
|
||||||
|
case cKey + "purge":
|
||||||
|
err = f.handlePurgeChannel(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Purge)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
/* Bot Control */
|
||||||
|
case cKey + "status":
|
||||||
|
err = f.handleSetStatus(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Status)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
/* Role Colors */
|
||||||
|
case cKey + "colors":
|
||||||
|
err = f.listUserColors(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Colors)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
case cKey + "color":
|
||||||
|
err = f.setUserColor(s, m)
|
||||||
|
if err != nil {
|
||||||
|
eMsg := f.CreateDefinedEmbed("Error (Color)", err.Error(), "error")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) OnUserJoin(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
|
||||||
|
userChannel, err := s.UserChannelCreate(m.User.ID)
|
||||||
|
if err != nil {
|
||||||
|
log.Print("Error (User Join): " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = s.ChannelMessageSend(userChannel.ID, f.Config.WelcomeText)
|
||||||
|
if err != nil {
|
||||||
|
log.Print("Error (User Join): " + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, roleID := range f.Config.JoinRoleIDs {
|
||||||
|
err = s.GuildMemberRoleAdd(f.Auth.Guild.ID, m.User.ID, roleID)
|
||||||
|
if err != nil {
|
||||||
|
log.Print("Error (User Join)" + err.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,94 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
discordgo "github.com/bwmarrin/discord.go"
|
||||||
|
"github.com/foxtrot/scuzzy/models"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) PrintError(component string, error string) {
|
||||||
|
fmt.Printf("Error: %s: %s\n", component, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) CreateDefinedEmbed(title string, desc string, status string) *discordgo.MessageEmbed {
|
||||||
|
msgColor := 0x000000
|
||||||
|
|
||||||
|
switch status {
|
||||||
|
case "error":
|
||||||
|
msgColor = 0xCC0000
|
||||||
|
break
|
||||||
|
case "success":
|
||||||
|
msgColor = 0x00CC00
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
msgColor = 0xFFA500
|
||||||
|
}
|
||||||
|
|
||||||
|
ftr := discordgo.MessageEmbedFooter{
|
||||||
|
Text: "Something broken? Tell foxtrot#1337",
|
||||||
|
IconURL: "https://cdn.discordapp.com/avatars/514163441548656641/a4ede220fea0ad8872b86f3eebc45524.png?size=128",
|
||||||
|
ProxyIconURL: "",
|
||||||
|
}
|
||||||
|
|
||||||
|
msg := discordgo.MessageEmbed{
|
||||||
|
URL: "",
|
||||||
|
Type: "",
|
||||||
|
Title: title,
|
||||||
|
Description: desc,
|
||||||
|
Timestamp: "",
|
||||||
|
Color: msgColor,
|
||||||
|
Footer: &ftr,
|
||||||
|
Image: nil,
|
||||||
|
Thumbnail: nil,
|
||||||
|
Video: nil,
|
||||||
|
Provider: nil,
|
||||||
|
Author: nil,
|
||||||
|
Fields: nil,
|
||||||
|
}
|
||||||
|
|
||||||
|
return &msg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) CreateCustomEmbed(embedData *models.CustomEmbed) *discordgo.MessageEmbed {
|
||||||
|
var ftr discordgo.MessageEmbedFooter
|
||||||
|
var img discordgo.MessageEmbedImage
|
||||||
|
var thm discordgo.MessageEmbedThumbnail
|
||||||
|
var prv discordgo.MessageEmbedProvider
|
||||||
|
var atr discordgo.MessageEmbedAuthor
|
||||||
|
|
||||||
|
ftr.Text = embedData.FooterText
|
||||||
|
ftr.IconURL = embedData.FooterImageURL
|
||||||
|
|
||||||
|
img.URL = embedData.ImageURL
|
||||||
|
img.Height = embedData.ImageH
|
||||||
|
img.Width = embedData.ImageW
|
||||||
|
|
||||||
|
thm.URL = embedData.ThumbnailURL
|
||||||
|
thm.Height = embedData.ThumbnailH
|
||||||
|
thm.Width = embedData.ThumbnailW
|
||||||
|
|
||||||
|
prv.Name = embedData.ProviderText
|
||||||
|
prv.URL = embedData.ProviderURL
|
||||||
|
|
||||||
|
atr.Name = embedData.AuthorText
|
||||||
|
atr.URL = embedData.AuthorURL
|
||||||
|
atr.IconURL = embedData.AuthorImageURL
|
||||||
|
|
||||||
|
msg := discordgo.MessageEmbed{
|
||||||
|
URL: embedData.URL,
|
||||||
|
Type: embedData.Type,
|
||||||
|
Title: embedData.Title,
|
||||||
|
Description: embedData.Desc,
|
||||||
|
Timestamp: "",
|
||||||
|
Color: embedData.Color,
|
||||||
|
Footer: &ftr,
|
||||||
|
Image: &img,
|
||||||
|
Thumbnail: &thm,
|
||||||
|
Video: nil,
|
||||||
|
Provider: &prv,
|
||||||
|
Author: &atr,
|
||||||
|
Fields: nil,
|
||||||
|
}
|
||||||
|
|
||||||
|
return &msg
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,212 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
"github.com/foxtrot/scuzzy/models"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) handleCat(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
if f.Auth.CheckAdminRole(m.Member) {
|
||||||
|
_, _ = s.ChannelMessageSend(m.ChannelID, "https://giphy.com/gifs/cat-cute-no-rCxogJBzaeZuU")
|
||||||
|
_ = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handlePing(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
var r *discordgo.Message
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if !f.Auth.CheckAdminRole(m.Member) {
|
||||||
|
return errors.New("You do not have permissions to use that command.")
|
||||||
|
} else {
|
||||||
|
msg := f.CreateDefinedEmbed("Ping", "Pong", "success")
|
||||||
|
r, err = s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(5 * time.Second)
|
||||||
|
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, r.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handleInfo(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
desc := "**Source**: https://github.com/foxtrot/scuzzy\n"
|
||||||
|
desc += "**Language**: Go\n"
|
||||||
|
desc += "**Commands**: See `" + f.Config.CommandKey + "help`\n\n\n"
|
||||||
|
|
||||||
|
gm, err := s.GuildMember(f.Config.GuildID, s.State.User.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
d := models.CustomEmbed{
|
||||||
|
Title: "Scuzzy Information",
|
||||||
|
Desc: desc,
|
||||||
|
ImageURL: "",
|
||||||
|
ImageH: 100,
|
||||||
|
ImageW: 100,
|
||||||
|
Color: 0xFFA500,
|
||||||
|
URL: "",
|
||||||
|
Type: "",
|
||||||
|
Timestamp: "",
|
||||||
|
FooterText: "Made with ❤ by Foxtrot",
|
||||||
|
FooterImageURL: "https://cdn.discordapp.com/avatars/514163441548656641/a4ede220fea0ad8872b86f3eebc45524.png",
|
||||||
|
ThumbnailURL: gm.User.AvatarURL(""),
|
||||||
|
ThumbnailH: 150,
|
||||||
|
ThumbnailW: 150,
|
||||||
|
ProviderURL: "",
|
||||||
|
ProviderText: "",
|
||||||
|
AuthorText: "",
|
||||||
|
AuthorURL: "",
|
||||||
|
AuthorImageURL: "",
|
||||||
|
}
|
||||||
|
|
||||||
|
msg := f.CreateCustomEmbed(&d)
|
||||||
|
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handleHelp(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
desc := "**Available Commands**\n"
|
||||||
|
desc += "`help` - This help dialog\n"
|
||||||
|
desc += "`info` - Display Scuzzy info\n"
|
||||||
|
desc += "`md` - Display Discord markdown information\n"
|
||||||
|
desc += "`colors` - Available color roles\n"
|
||||||
|
desc += "`color` - Set an available color role\n"
|
||||||
|
desc += "`ctof` - Convert Celsius to Farenheit\n"
|
||||||
|
desc += "`ftoc` - Convert Farenheit to Celsius\n"
|
||||||
|
|
||||||
|
if f.Auth.CheckAdminRole(m.Member) {
|
||||||
|
desc += "\n"
|
||||||
|
desc += "**Admin Commands**\n"
|
||||||
|
desc += "`ping` - Ping the bot\n"
|
||||||
|
desc += "`status` - Set the bot status\n"
|
||||||
|
desc += "`purge` - Purge channel messages\n"
|
||||||
|
}
|
||||||
|
|
||||||
|
desc += "\n\nAll commands are prefixed with `" + f.Config.CommandKey + "`\n"
|
||||||
|
|
||||||
|
msg := f.CreateDefinedEmbed("Help", desc, "")
|
||||||
|
|
||||||
|
_, err := s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handleMarkdownInfo(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
desc := "*Italic* text goes between `*single asterisks*`\n"
|
||||||
|
desc += "**Bold** text goes between `**double asterisks**`\n"
|
||||||
|
desc += "***Bold and Italic*** text goes between `***triple asterisks***`\n"
|
||||||
|
desc += "__Underlined__ text goes between `__double underscore__`\n"
|
||||||
|
desc += "~~Strikethrough~~ text goes between `~~double tilde~~`\n"
|
||||||
|
desc += "||Spoilers|| go between `|| double pipe ||`\n\n"
|
||||||
|
desc += "You can combine the above styles.\n\n"
|
||||||
|
desc += "Inline Code Blocks start and end with a single `````\n"
|
||||||
|
desc += "Multi line Code Blocks start and end with ```````\n"
|
||||||
|
desc += "Multi line Code Blocks can also specify a language with `````language`` at the start\n\n"
|
||||||
|
desc += "Single line quotes start with `>`\n"
|
||||||
|
desc += "Multi line quotes start with `>>>`\n"
|
||||||
|
|
||||||
|
msg := f.CreateDefinedEmbed("Discord Markdown", desc, "")
|
||||||
|
r, err := s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(15 * time.Second)
|
||||||
|
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, r.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handleCtoF(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
inS := strings.Split(m.Content, " ")
|
||||||
|
|
||||||
|
if len(inS) < 2 {
|
||||||
|
return errors.New("You did not specify a temperature")
|
||||||
|
}
|
||||||
|
in := inS[1]
|
||||||
|
|
||||||
|
inF, err := strconv.ParseFloat(in, 2)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("You did not specify a valid number")
|
||||||
|
}
|
||||||
|
|
||||||
|
cels := (inF * 9.0 / 5.0) + 32.0
|
||||||
|
celsF := float64(cels)
|
||||||
|
|
||||||
|
msg := fmt.Sprintf("`%.1f°c` is `%.1f°f`", inF, celsF)
|
||||||
|
|
||||||
|
e := f.CreateDefinedEmbed("Celsius to Farenheit", msg, "")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, e)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) handleFtoC(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
inS := strings.Split(m.Content, " ")
|
||||||
|
|
||||||
|
if len(inS) < 2 {
|
||||||
|
return errors.New("You did not specify a temperature")
|
||||||
|
}
|
||||||
|
in := inS[1]
|
||||||
|
|
||||||
|
inF, err := strconv.ParseFloat(in, 2)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("You did not specify a valid number")
|
||||||
|
}
|
||||||
|
|
||||||
|
faren := (inF - 32) * 5 / 9
|
||||||
|
farenF := float64(faren)
|
||||||
|
|
||||||
|
msg := fmt.Sprintf("`%.1f°f` is `%.1f°c`", inF, farenF)
|
||||||
|
|
||||||
|
e := f.CreateDefinedEmbed("Farenheit to Celsius", msg, "")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, e)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,58 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) handlePurgeChannel(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
if !f.Auth.CheckAdminRole(m.Member) {
|
||||||
|
return errors.New("You do not have permissions to use that command.")
|
||||||
|
}
|
||||||
|
|
||||||
|
purgeSplit := strings.SplitN(m.Content, " ", 2)
|
||||||
|
if len(purgeSplit) < 2 {
|
||||||
|
return errors.New("No message count supplied")
|
||||||
|
}
|
||||||
|
|
||||||
|
msgCount, err := strconv.Atoi(purgeSplit[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if msgCount > 100 {
|
||||||
|
return errors.New("You may only purge upto 100 messages at a time.")
|
||||||
|
}
|
||||||
|
|
||||||
|
chanMsgs, err := s.ChannelMessages(m.ChannelID, msgCount, "", "", "")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
msg := f.CreateDefinedEmbed("Purge Channel", "Purging `"+purgeSplit[1]+"` messages.", "normal")
|
||||||
|
r, err := s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var delMsgs []string
|
||||||
|
for _, v := range chanMsgs {
|
||||||
|
delMsgs = append(delMsgs, v.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.ChannelMessagesBulkDelete(m.ChannelID, delMsgs)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, r.ID)
|
||||||
|
msg = f.CreateDefinedEmbed("Purge Channel", "Purged `"+purgeSplit[1]+"` messages!", "success")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,81 @@
|
||||||
|
package features
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"github.com/bwmarrin/discord.go"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f *Features) listUserColors(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
if !f.Auth.CheckCommandRestrictions(m) {
|
||||||
|
return errors.New("This command is not allowed in this channel.")
|
||||||
|
}
|
||||||
|
|
||||||
|
msgC := "You can choose from the following colors:\n\n"
|
||||||
|
for _, v := range f.Config.ColorRoles {
|
||||||
|
msgC += "<@&" + v.ID + ">\n"
|
||||||
|
}
|
||||||
|
msgC += "\n\nUse `" + f.Config.CommandKey + "color <color_name>` to set.\n"
|
||||||
|
|
||||||
|
msg := f.CreateDefinedEmbed("User Colors", msgC, "")
|
||||||
|
|
||||||
|
_, err := s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Features) setUserColor(s *discordgo.Session, m *discordgo.MessageCreate) error {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if !f.Auth.CheckCommandRestrictions(m) {
|
||||||
|
return errors.New("This command is not allowed in this channel.")
|
||||||
|
}
|
||||||
|
|
||||||
|
rUserID := m.Author.ID
|
||||||
|
|
||||||
|
userInput := strings.Split(m.Content, " ")
|
||||||
|
if len(userInput) < 2 {
|
||||||
|
err = f.listUserColors(s, m)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
roleColorName := userInput[1]
|
||||||
|
roleColorName = strings.ToLower(roleColorName)
|
||||||
|
|
||||||
|
roleColorID := ""
|
||||||
|
for _, role := range f.Config.ColorRoles {
|
||||||
|
if role.Name == roleColorName {
|
||||||
|
roleColorID = role.ID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(roleColorID) == 0 {
|
||||||
|
err = f.listUserColors(s, m)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, role := range f.Config.ColorRoles {
|
||||||
|
// Attempt to remove all color roles regardless of if they have them or not.
|
||||||
|
// Slow because of the REST requests...
|
||||||
|
_ = s.GuildMemberRoleRemove(m.GuildID, rUserID, role.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.GuildMemberRoleAdd(m.GuildID, rUserID, roleColorID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
} else {
|
||||||
|
msg := f.CreateDefinedEmbed("User Color", "<@"+m.Author.ID+">: Your color has been set to <@&"+roleColorID+">!", "success")
|
||||||
|
_, err = s.ChannelMessageSendEmbed(m.ChannelID, msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,31 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
import "github.com/bwmarrin/discord.go"
|
||||||
|
|
||||||
|
type ColorRole struct {
|
||||||
|
Name string `json:"color"`
|
||||||
|
ID string `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CommandRestriction struct {
|
||||||
|
Command string `json:"command"`
|
||||||
|
Mode string `json:"mode"`
|
||||||
|
Channels []string `json:"channels"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Configuration struct {
|
||||||
|
GuildID string `json:"guild_id"`
|
||||||
|
|
||||||
|
StatusText string `json:"status_text"`
|
||||||
|
WelcomeText string `json:"welcome_text"`
|
||||||
|
CommandKey string `json:"command_key"`
|
||||||
|
|
||||||
|
AdminRoles []string `json:"admin_roles"`
|
||||||
|
JoinRoleIDs []string `json:"join_role_ids"`
|
||||||
|
|
||||||
|
CommandRestrictions []CommandRestriction `json:"command_restrictions"`
|
||||||
|
|
||||||
|
ColorRoles []ColorRole `json:"color_roles"`
|
||||||
|
|
||||||
|
Guild *discordgo.Guild `json:"reserved_guild"`
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,28 @@
|
||||||
|
package models
|
||||||
|
|
||||||
|
type CustomEmbed struct {
|
||||||
|
URL string
|
||||||
|
Title string
|
||||||
|
Desc string
|
||||||
|
Type string
|
||||||
|
Timestamp string
|
||||||
|
Color int
|
||||||
|
|
||||||
|
FooterText string
|
||||||
|
FooterImageURL string
|
||||||
|
|
||||||
|
ImageURL string
|
||||||
|
ImageH int
|
||||||
|
ImageW int
|
||||||
|
|
||||||
|
ThumbnailURL string
|
||||||
|
ThumbnailH int
|
||||||
|
ThumbnailW int
|
||||||
|
|
||||||
|
ProviderURL string
|
||||||
|
ProviderText string
|
||||||
|
|
||||||
|
AuthorText string
|
||||||
|
AuthorURL string
|
||||||
|
AuthorImageURL string
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue