2018-05-17 16:30:29 +02:00
package main
import (
"fmt"
"encoding/json"
2018-05-25 17:21:17 +02:00
//"strings"
2018-05-17 16:30:29 +02:00
"os"
"os/signal"
"syscall"
2018-05-25 17:21:17 +02:00
"regexp"
2018-05-17 16:30:29 +02:00
"github.com/bwmarrin/discordgo"
)
2018-05-26 04:23:10 +02:00
type Embed struct {
Message string
RulesTitle string
RulesText string
RulesText2 string
QuestionsTitle string
QuestionsText string
BugsTitle string
BugsText string
Image string
}
2018-05-17 16:30:29 +02:00
type Config struct {
2018-05-26 04:23:10 +02:00
AdminID string
ServerID string
LockedRoleID string
2018-05-17 16:30:29 +02:00
Token string
WelcomeChannel string
2018-05-26 04:23:10 +02:00
WelcomeEmbed Embed
2018-05-17 16:30:29 +02:00
}
var config = readConfig ( )
func readConfig ( ) Config {
file , _ := os . Open ( "config.json" )
conf := Config { }
_ = json . NewDecoder ( file ) . Decode ( & conf )
file . Close ( )
return conf
}
func main ( ) {
dg , err := discordgo . New ( "Bot " + config . Token )
if err != nil {
fmt . Println ( "error: " , err )
return
}
dg . AddHandler ( genericReply )
dg . AddHandler ( onJoin )
err = dg . Open ( )
if err != nil {
fmt . Println ( "no connection, " , err )
return
}
fmt . Println ( "bot running. selphyWoo" )
sc := make ( chan os . Signal , 1 )
signal . Notify ( sc , syscall . SIGINT , syscall . SIGTERM , os . Interrupt , os . Kill )
<- sc
dg . Close ( )
}
2018-05-26 04:23:10 +02:00
func onJoin ( s * discordgo . Session , member * discordgo . GuildMemberAdd ) {
if ! member . User . Bot {
s . GuildMemberRoleAdd ( config . ServerID , member . User . ID , config . LockedRoleID )
}
}
func unlockUser ( s * discordgo . Session , id string ) {
s . GuildMemberRoleRemove ( config . ServerID , id , config . LockedRoleID )
}
2018-05-17 16:30:29 +02:00
func genericReply ( s * discordgo . Session , m * discordgo . MessageCreate ) {
if m . Author . ID == s . State . User . ID {
return
}
2018-05-26 04:23:10 +02:00
if m . Author . ID == config . AdminID {
replyGodmode ( s , m )
} else if m . ChannelID == config . WelcomeChannel {
s . ChannelMessageDelete ( m . ChannelID , m . ID )
if m . Content == "!accept" {
unlockUser ( s , m . Author . ID )
}
return
}
// In case this doesn’t work with your font: the last character is a winking emoji.
winks , _ := regexp . MatchString ( "([()|DoO];|;[()|DoOpP]|:wink:|😉)" , m . Content )
2018-05-25 17:21:17 +02:00
if winks {
2018-05-17 16:30:29 +02:00
s . ChannelMessageSend ( m . ChannelID , fmt . Sprintf ( "<@%s> faggot" , m . Author . ID ) )
2018-05-25 17:21:17 +02:00
s . ChannelMessageDelete ( m . ChannelID , m . ID )
2018-05-17 16:30:29 +02:00
return
}
// As per our majesty’s command:
if m . Content == "\\o" {
s . ChannelMessageSend ( m . ChannelID , "o/" )
} else if m . Content == "o/" {
s . ChannelMessageSend ( m . ChannelID , "\\o" )
}
}
2018-05-26 04:23:10 +02:00
// Admin stuff down here. This is very server-specific
func replyGodmode ( s * discordgo . Session , m * discordgo . MessageCreate ) {
if m . Content == "print_rules()" {
embedColor := 0xffb90f // kageru gold
embed := & discordgo . MessageEmbed {
Author : & discordgo . MessageEmbedAuthor { } ,
Color : embedColor ,
Description : config . WelcomeEmbed . Message ,
Fields : [ ] * discordgo . MessageEmbedField {
& discordgo . MessageEmbedField {
Name : config . WelcomeEmbed . QuestionsTitle ,
Value : config . WelcomeEmbed . QuestionsText ,
Inline : true ,
} ,
& discordgo . MessageEmbedField {
Name : config . WelcomeEmbed . BugsTitle ,
Value : fmt . Sprintf ( config . WelcomeEmbed . BugsText , config . AdminID ) ,
Inline : true ,
} ,
} ,
Thumbnail : & discordgo . MessageEmbedThumbnail {
URL : config . WelcomeEmbed . Image ,
} ,
}
s . ChannelMessageSendEmbed ( m . ChannelID , embed )
embed = & discordgo . MessageEmbed {
Author : & discordgo . MessageEmbedAuthor { } ,
Color : embedColor ,
Fields : [ ] * discordgo . MessageEmbedField {
& discordgo . MessageEmbedField {
Name : config . WelcomeEmbed . RulesTitle ,
Value : config . WelcomeEmbed . RulesText ,
Inline : true ,
} ,
} ,
}
s . ChannelMessageSendEmbed ( m . ChannelID , embed )
embed = & discordgo . MessageEmbed {
Author : & discordgo . MessageEmbedAuthor { } ,
Color : embedColor ,
Fields : [ ] * discordgo . MessageEmbedField {
& discordgo . MessageEmbedField {
Name : config . WelcomeEmbed . RulesTitle ,
Value : config . WelcomeEmbed . RulesText2 ,
Inline : true ,
} ,
} ,
}
s . ChannelMessageSendEmbed ( m . ChannelID , embed )
2018-05-17 16:30:29 +02:00
}
}