41 lines
1.2 KiB
Go
41 lines
1.2 KiB
Go
package main
|
|
|
|
import "github.com/bwmarrin/discordgo"
|
|
|
|
const guildID = "626424729234046987"
|
|
const memberRoleID = "626434632614805526"
|
|
|
|
func enforceDM(s *discordgo.Session, m *discordgo.MessageCreate) bool {
|
|
if m.GuildID != "" {
|
|
//This command can only be used in DM to protect the invite creator.
|
|
s.ChannelMessageDelete(m.ChannelID, m.ID)
|
|
s.ChannelMessageSend(
|
|
m.ChannelID,
|
|
m.Author.Mention()+"，这个指令只能在私信中使用。\n"+m.Author.Mention()+", this command can only be used in DM.",
|
|
)
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func membersOnly(s *discordgo.Session, m *discordgo.MessageCreate) bool {
|
|
member, err := s.GuildMember(guildID, m.Author.ID)
|
|
if err != nil {
|
|
s.ChannelMessageSend(
|
|
m.ChannelID,
|
|
m.Author.Mention()+"，请在成为会员后才使用这个指令。\n"+m.Author.Mention()+", please use this command after you become a member.",
|
|
)
|
|
return false
|
|
}
|
|
for _, v := range member.Roles {
|
|
if v == memberRoleID {
|
|
return true
|
|
}
|
|
}
|
|
s.ChannelMessageSend(
|
|
m.ChannelID,
|
|
m.Author.Mention()+"，请在成为会员后才使用这个指令。\n"+m.Author.Mention()+", please use this command after you become a member.",
|
|
)
|
|
return false
|
|
}
|