package main import ( "encoding/json" "io/ioutil" "log" "os" "os/signal" "path" "path/filepath" "strings" "syscall" "github.com/animenotifier/arn" "github.com/bwmarrin/discordgo" ) // Session provides access to the Discord session. var discord *discordgo.Session func main() { var err error exe, err := os.Executable() if err != nil { panic(err) } dir := path.Dir(exe) var apiKeysPath string apiKeysPath, err = filepath.Abs(dir + "/../../security/api-keys.json") if err != nil { panic(err) } var apiKeys arn.APIKeys data, _ := ioutil.ReadFile(apiKeysPath) err = json.Unmarshal(data, &apiKeys) if err != nil { panic(err) } discord, _ = discordgo.New() discord.Token = "Bot " + apiKeys.Discord.Token // Verify a Token was provided if discord.Token == "" { log.Println("You must provide a Discord authentication token.") return } // Verify the Token is valid and grab user information discord.State.User, err = discord.User("@me") if err != nil { log.Printf("Error fetching user information: %s\n", err) } // Open a websocket connection to Discord err = discord.Open() if err != nil { log.Printf("Error opening connection to Discord, %s\n", err) } defer discord.Close() // Receive messages discord.AddHandler(onMessage) // Wait for a CTRL-C log.Printf("Tsundere is ready. Press CTRL-C to exit.") sc := make(chan os.Signal, 1) signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill) <-sc } // This function will be called every time a new message is created on any channel. func onMessage(s *discordgo.Session, m *discordgo.MessageCreate) { // Ignore all messages created by the bot itself if m.Author.ID == s.State.User.ID { return } if m.Content == "!commands" { s.ChannelMessageSend(m.ChannelID, ` **!user** [username] **!anime** [id] **!animelist** [username] **!tag** [forum tag]`) } if strings.HasPrefix(m.Content, "!user ") { s.ChannelMessageSend(m.ChannelID, "https://notify.moe/+"+strings.Split(m.Content, " ")[1]) return } if strings.HasPrefix(m.Content, "!anime ") { s.ChannelMessageSend(m.ChannelID, "https://notify.moe/anime/"+strings.Split(m.Content, " ")[1]) return } if strings.HasPrefix(m.Content, "!animelist ") { s.ChannelMessageSend(m.ChannelID, "https://notify.moe/+"+strings.Split(m.Content, " ")[1]+"/animelist") return } if strings.HasPrefix(m.Content, "!tag ") { s.ChannelMessageSend(m.ChannelID, "https://notify.moe/forum/"+strings.ToLower(strings.Split(m.Content, " ")[1])) return } }