discord-flauschbot/main.go

package main

import (
    "fmt"
    "encoding/json"
    //"strings"
    "os"
    "os/signal"
    "syscall"
    "regexp"
    "log"
    "github.com/bwmarrin/discordgo"
)

type Embed struct {
    Message string
    QuestionsTitle string
    QuestionsText string
    BugsTitle string
    BugsText string
    Image string
}

type Config struct {
    AdminID string
    ServerID string
    LockedRoleID string
    Token string
    WelcomeChannel string
    SendWelcomeDM bool
    RequireAccept bool
    WelcomeEmbed Embed
    RoleCommands map[string]string
}

var config = readConfig()

func readConfig() Config {
    file, _ := os.Open("config2.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
    }

    f, err := os.OpenFile("selphybot.log", os.O_RDWR | os.O_CREATE | os.O_APPEND, 0666)
    if err != nil {
        fmt.Println("error opening log file: %v", err)
    }
    defer f.Close()
    log.SetOutput(f)

    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()
}

func onJoin(s *discordgo.Session, member *discordgo.GuildMemberAdd) {
    if !member.User.Bot && config.RequireAccept {
        s.GuildMemberRoleAdd(config.ServerID, member.User.ID, config.LockedRoleID)
    }
    if !member.User.Bot && config.SendWelcomeDM {
        dm, err := s.UserChannelCreate(member.User.ID)
        if err != nil {
            // todo: @mention or something
        } else {
            embed := &discordgo.MessageEmbed {
                Author:      &discordgo.MessageEmbedAuthor{},
                Color:       0xffb90f,
                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(dm.ID, embed)
        }
    }
    log.Printf("User joined: %s", userToString(member.User))
}

func unlockUser(s *discordgo.Session, id string) {
    s.GuildMemberRoleRemove(config.ServerID, id, config.LockedRoleID)
    log.Printf("Removed lock from user: %s", userToString(getUser(s, id)))
}

func userToString(u *discordgo.User) string {
    return fmt.Sprintf("%s#%s (ID: %s)", u.Username, u.Discriminator, u.ID)
}

func roleName(s *discordgo.State, rid string) string {
    role, _ := s.Role(config.ServerID, rid)
    return role.Name
}

func channelToString(c *discordgo.Channel) string {
    return fmt.Sprintf("%s (ID: %s) on %s", c.Name, c.ID, c.GuildID)
}

func messageToString(m *discordgo.Message) string {
    return fmt.Sprintf("<%s#%s>: %s", m.Author.Username, m.Author.Discriminator, m.Content)
}

func getChannel(s *discordgo.State, cid string) *discordgo.Channel {
    channel, _ := s.Channel(cid)
    return channel
}

func getUser(s *discordgo.Session, uid string) *discordgo.User {
    user, _ := s.User(uid)
    return user
}

/*func undelete(s *discordgo.Session, m *discordgo.MessageDelete) {
    channel, _ := s.State.Channel(m.ChannelID)
    message, _ := s.State.Message(m.ChannelID, m.ID)
    log.Println(fmt.Sprintf("Someone deleted a message in %s: “%s”", channel.Name, messageToString(message)))
}*/

func genericReply(s *discordgo.Session, m *discordgo.MessageCreate) {
    if m.Author.ID == s.State.User.ID {
        return
    }

    if getChannel(s.State, m.ChannelID).Type == discordgo.ChannelTypeDM {
        log.Printf("Received DM from %s with content: “%s”", userToString(m.Author), m.Content)
        fmt.Sprintf("Received DM from %s with content: “%s”", userToString(m.Author), m.Content)
        Member, _ := s.GuildMember(config.ServerID, m.Author.ID)
        for comm, role := range config.RoleCommands {
            if m.Content == comm {
                dm, _ := s.UserChannelCreate(Member.User.ID)
                for _, irole := range config.RoleCommands {
                    for _, mrole := range Member.Roles {
                        if irole == mrole {
                            s.ChannelMessageSend(dm.ID, "Baka, du kannst nur eine der Rollen haben.")
                            log.Printf("Denied Role %s to %s. User already has %s", roleName(s.State, irole), userToString(m.Author), roleName(s.State, irole))
                            return
                        }
                    }
                }
                log.Printf("Giving Role %s to %s", roleName(s.State, role), userToString(m.Author))
                s.ChannelMessageSend(dm.ID, "Haaai, Ryoukai desu~")
                s.GuildMemberRoleAdd(config.ServerID, m.Author.ID, role)
            }
        }
    }

    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)
    if winks {
        s.ChannelMessageSend(m.ChannelID, fmt.Sprintf("<@%s> Oboe!", m.Author.ID))
        s.ChannelMessageDelete(m.ChannelID, m.ID)
        channel := getChannel(s.State, m.ChannelID)
        log.Printf("Deleted message by %s in %s. Content: “%s”", userToString(m.Author), channelToString(channel), m.Content)
        return
    }

    // As per our majesty’s command:
    if m.Content == "\\o" {
        s.ChannelMessageSend(m.ChannelID, "o/")
        log.Printf("o/ at %s", userToString(m.Author))
    } else if m.Content == "o/" {
        s.ChannelMessageSend(m.ChannelID, "\\o")
        log.Printf("\\o at %s", userToString(m.Author))
    } else if m.Content == "ayy" {
        s.ChannelMessageSend(m.ChannelID, "lmao")
        log.Printf("ayy lmao at %s", userToString(m.Author))
    }
}


// Admin stuff down here. This is very server-specific

func replyGodmode(s *discordgo.Session, m *discordgo.MessageCreate) {
/*
    if m.Content == "print_rules()" {
        channel := getChannel(s.State, m.ChannelID)
        log.Printf("print_rules() triggered by %s in %s.", userToString(m.Author), channelToString(channel))
        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)
    }
*/
}