103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/joho/godotenv"
|
|
|
|
// WICHTIG: Pfad entspricht deiner go.mod
|
|
"heydeman/vc-stat-bot-discord/db_management"
|
|
)
|
|
|
|
var startTimes = make(map[string]time.Time)
|
|
|
|
func main() {
|
|
_ = godotenv.Load() // Ignoriert Fehler, falls keine .env existiert (nutzt dann System-Vars)
|
|
|
|
db_management.InitDB()
|
|
|
|
token := os.Getenv("DISCORD_TOKEN")
|
|
if token == "" {
|
|
log.Fatal("DISCORD_TOKEN nicht gefunden!")
|
|
}
|
|
|
|
discord, err := discordgo.New("Bot " + token)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
discord.AddHandler(messageCreate)
|
|
discord.AddHandler(voiceUpdate)
|
|
|
|
discord.Identify.Intents = discordgo.IntentsGuildMessages |
|
|
discordgo.IntentsMessageContent |
|
|
discordgo.IntentsGuildVoiceStates
|
|
|
|
err = discord.Open()
|
|
if err != nil {
|
|
log.Fatal("Error opening connection:", err)
|
|
}
|
|
|
|
fmt.Println("Bot läuft stabil auf Hetzner. Drücke CTRL+C zum Beenden.")
|
|
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
|
|
<-stop
|
|
|
|
// Graceful Shutdown: Alle aktuell aktiven Sessions noch schnell speichern
|
|
fmt.Println("Speichere aktive Sessions vor dem Beenden...")
|
|
for uid, start := range startTimes {
|
|
db_management.SaveSession(uid, int(time.Since(start).Seconds()))
|
|
}
|
|
|
|
discord.Close()
|
|
}
|
|
|
|
func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
if m.Author.Bot {
|
|
return
|
|
}
|
|
|
|
if m.Content == "!stats" {
|
|
totalSeconds := db_management.GetUserStats(m.Author.ID)
|
|
duration := time.Duration(totalSeconds) * time.Second
|
|
|
|
msg := fmt.Sprintf("📊 **30-Tage Statistik für %s**\nVerbrachte Zeit: `%s`",
|
|
m.Author.GlobalName, formatDuration(duration))
|
|
s.ChannelMessageSend(m.ChannelID, msg)
|
|
}
|
|
}
|
|
|
|
func voiceUpdate(s *discordgo.Session, v *discordgo.VoiceStateUpdate) {
|
|
uid := v.UserID
|
|
|
|
// User betritt Sprachkanal
|
|
if v.ChannelID != "" {
|
|
if _, ok := startTimes[uid]; !ok {
|
|
startTimes[uid] = time.Now()
|
|
}
|
|
return
|
|
}
|
|
|
|
// User verlässt Sprachkanal (v.ChannelID == "")
|
|
if v.ChannelID == "" {
|
|
if start, ok := startTimes[uid]; ok {
|
|
db_management.SaveSession(uid, int(time.Since(start).Seconds()))
|
|
delete(startTimes, uid)
|
|
}
|
|
}
|
|
}
|
|
|
|
func formatDuration(d time.Duration) string {
|
|
h := int(d.Hours())
|
|
m := int(d.Minutes()) % 60
|
|
s := int(d.Seconds()) % 60
|
|
return fmt.Sprintf("%dh %dm %ds", h, m, s)
|
|
}
|