71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"database/sql"
|
|
"net/url"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/joho/godotenv"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
//TODO: add more error handlings
|
|
|
|
var db *sql.DB
|
|
|
|
func main() {
|
|
godotenv.Load()
|
|
|
|
var err error
|
|
connStr := "postgresql://" + os.Getenv("DB_USER") + ":" + url.QueryEscape(os.Getenv("DB_PASSWORD")) + "@" + os.Getenv("DB_SERVER") + ":" + string(os.Getenv("DB_PORT")) + "/" + os.Getenv("DB_NAME") + "?sslmode=disable&application_name=Discord Bot"
|
|
db, err = sql.Open("postgres", connStr)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
initTables()
|
|
discord, err := discordgo.New("Bot " + os.Getenv("BOT_TOKEN"))
|
|
if err != nil {
|
|
fmt.Println("error creating Discord session,", err)
|
|
return
|
|
} else {
|
|
fmt.Println("Discord session created")
|
|
}
|
|
discord.Identify.Intents = discordgo.IntentsGuildMessages | discordgo.IntentsGuilds | discordgo.IntentMessageContent
|
|
discord.AddHandler(ready)
|
|
discord.AddHandler(interactionCreate)
|
|
discord.AddHandler(messageCreate)
|
|
err = discord.Open()
|
|
if err != nil {
|
|
fmt.Println("error opening connection,", err)
|
|
return
|
|
}
|
|
fmt.Printf("\nBot is now running as \"%s\"!", discord.State.User.Username)
|
|
sc := make(chan os.Signal, 1)
|
|
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
|
|
<-sc
|
|
fmt.Println("\nShutting down...")
|
|
discord.Close()
|
|
}
|
|
|
|
func int64Ptr(i int64) *int64 {
|
|
return &i
|
|
}
|
|
|
|
func hexToDecimal(hexColor string) int {
|
|
// Remove the hash symbol if it's present
|
|
hexColor = strings.TrimPrefix(hexColor, "#")
|
|
decimal, err := strconv.ParseInt(hexColor, 16, 64)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return int(decimal)
|
|
}
|