52 lines
1.0 KiB
Go
52 lines
1.0 KiB
Go
package bot
|
|
|
|
import (
|
|
"bitbot/pb"
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/charmbracelet/log"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
CryptoToken string
|
|
BotToken string
|
|
OpenAIToken string
|
|
)
|
|
|
|
func Run() {
|
|
discord, err := discordgo.New("Bot " + BotToken)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
discord.AddHandler(newMessage)
|
|
|
|
discord.Open()
|
|
defer discord.Close()
|
|
log.Info("BitBot is running...")
|
|
pb.Run()
|
|
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt)
|
|
<-c
|
|
}
|
|
|
|
func newMessage(discord *discordgo.Session, message *discordgo.MessageCreate) {
|
|
if message.Author.ID == discord.State.User.ID {
|
|
return
|
|
}
|
|
|
|
switch {
|
|
case strings.Contains(message.Content, "!bit"):
|
|
gptResponse := chatGPT(message.Content)
|
|
discord.ChannelTyping(message.ChannelID)
|
|
discord.ChannelMessageSendComplex(message.ChannelID, gptResponse)
|
|
case strings.Contains(message.Content, "!cry"):
|
|
currentCryptoPrice := getCurrentCryptoPrice(message.Content)
|
|
discord.ChannelMessageSendComplex(message.ChannelID, currentCryptoPrice)
|
|
}
|
|
|
|
}
|