diff --git a/.dockerignore b/.dockerignore new file mode 100644 index 0000000..1ab4ce8 --- /dev/null +++ b/.dockerignore @@ -0,0 +1,4 @@ +.git +bot.db +vendor +.env \ No newline at end of file diff --git a/.env.example b/.env.example new file mode 100644 index 0000000..4e63954 --- /dev/null +++ b/.env.example @@ -0,0 +1 @@ +DISCORD_TOKEN="Ein Discord Token" \ No newline at end of file diff --git a/.gitea/workflows/test.yaml b/.gitea/workflows/test.yaml new file mode 100644 index 0000000..0495083 --- /dev/null +++ b/.gitea/workflows/test.yaml @@ -0,0 +1,21 @@ +name: Bot Unit Tests + +on: + push: + branches: [ "main", "master", "dev" ] + pull_request: + branches: [ "main", "master" ] + +jobs: + test: + runs-on: ubuntu-latest + steps: + - uses: actions/checkout@v4 + - name: Set up Go + uses: actions/setup-go@v4 + with: + go-version: '1.25' + check-latest: false + cache: true + - name: Run Tests + run: go test ./... -v \ No newline at end of file diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..cb7b2f0 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.env +*.db \ No newline at end of file diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..5c6fe30 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,31 @@ +# STAGE 1: Build +FROM golang:1.24-bookworm AS builder + +# Arbeitsverzeichnis im Container +WORKDIR /app + +# Erst Abhängigkeiten kopieren (für besseres Caching) +COPY go.mod go.sum ./ +RUN go mod download + +# Restlichen Code kopieren +COPY . . + +# Statisch kompilieren (wichtig für alpine) +RUN CGO_ENABLED=0 GOOS=linux go build -o bot main.go + +# STAGE 2: Run +FROM alpine:latest + +# Zertifikate für Discord (HTTPS) installieren +RUN apk --no-cache add ca-certificates + +WORKDIR /root/ + +# Nur die Binary vom Builder kopieren +COPY --from=builder /app/bot . +# Falls du eine .env lokal nutzt (auf dem Server meist via Environment im Stack) +# COPY .env . + +# Startbefehl +CMD ["./bot"] \ No newline at end of file diff --git a/db_management/db_management.go b/db_management/db_management.go new file mode 100644 index 0000000..6268a60 --- /dev/null +++ b/db_management/db_management.go @@ -0,0 +1,48 @@ +package db_management + +import ( + "database/sql" + "time" + + _ "modernc.org/sqlite" +) + +var DB *sql.DB + +func InitDB() { + var err error + // Öffnet bot.db im aktuellen Verzeichnis + DB, err = sql.Open("sqlite", "./bot.db") + if err != nil { + panic(err) + } + + // Tabelle für einzelne Sessions + statement := ` + CREATE TABLE IF NOT EXISTS voice_sessions ( + user_id TEXT, + seconds INTEGER, + created_at TIMESTAMP + );` + _, err = DB.Exec(statement) + if err != nil { + panic(err) + } +} + +func SaveSession(userID string, seconds int) { + query := `INSERT INTO voice_sessions (user_id, seconds, created_at) VALUES (?, ?, ?)` + DB.Exec(query, userID, seconds, time.Now()) +} + +func GetUserStats(userID string) int { + var total sql.NullInt64 // NullInt64 fängt NULL-Ergebnisse ab + query := `SELECT SUM(seconds) FROM voice_sessions + WHERE user_id = ? AND created_at > date('now', '-30 days')` + + err := DB.QueryRow(query, userID).Scan(&total) + if err != nil || !total.Valid { + return 0 + } + return int(total.Int64) +} diff --git a/db_management/db_management_test.go b/db_management/db_management_test.go new file mode 100644 index 0000000..32c6a0b --- /dev/null +++ b/db_management/db_management_test.go @@ -0,0 +1,62 @@ +package db_management + +import ( + "database/sql" + "testing" + "time" + + "github.com/stretchr/testify/assert" + _ "modernc.org/sqlite" +) + +func setupTestDB(t *testing.T) { + var err error + // ":memory:" sorgt dafür, dass die DB nach dem Test gelöscht wird + DB, err = sql.Open("sqlite", ":memory:") + assert.NoError(t, err) + + statement := ` + CREATE TABLE voice_sessions ( + user_id TEXT, + seconds INTEGER, + created_at TIMESTAMP + );` + _, err = DB.Exec(statement) + assert.NoError(t, err) +} + +func TestSaveAndGetStats(t *testing.T) { + setupTestDB(t) + defer DB.Close() + + userID := "123456789" + + // Test 1: Neue Session speichern + SaveSession(userID, 3600) // 1 Stunde + stats := GetUserStats(userID) + assert.Equal(t, 3600, stats, "User sollte 3600 Sekunden haben") + + // Test 2: Zweite Session addieren + SaveSession(userID, 600) // +10 Minuten + stats = GetUserStats(userID) + assert.Equal(t, 4200, stats, "Summe sollte 4200 sein") +} + +func TestTimeFiltering(t *testing.T) { + setupTestDB(t) + defer DB.Close() + + userID := "999" + + // Eine Session von heute + SaveSession(userID, 1000) + + // Eine "alte" Session manuell einfügen (31 Tage alt) + oldDate := time.Now().AddDate(0, 0, -31) + _, _ = DB.Exec("INSERT INTO voice_sessions (user_id, seconds, created_at) VALUES (?, ?, ?)", + userID, 5000, oldDate) + + // GetUserStats sollte NUR die 1000 Sekunden von heute zählen + stats := GetUserStats(userID) + assert.Equal(t, 1000, stats, "Alte Sessions dürfen nicht gezählt werden") +} diff --git a/docker-compose.yaml b/docker-compose.yaml new file mode 100644 index 0000000..36dfe8b --- /dev/null +++ b/docker-compose.yaml @@ -0,0 +1,26 @@ +version: '3.8' + +services: + app: + image: alpine:latest # Oder ein spezielles "slim" / "alpine" Image deiner Wahl + container_name: mein_sparsames_projekt + restart: unless-stopped + # Ressourcen-Limits verhindern, dass ein Memory Leak den Host killt + deploy: + resources: + limits: + cpus: '0.50' + memory: 256M + reservations: + memory: 64M + logging: + driver: "json-file" + options: + max-size: "10m" + max-file: "3" + networks: + - discord_backend_net + +networks: + discord_backend_net: + driver: bridge \ No newline at end of file diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..8646adc --- /dev/null +++ b/go.mod @@ -0,0 +1,27 @@ +module heydeman/vc-stat-bot-discord + +go 1.25.0 + +require ( + github.com/bwmarrin/discordgo v0.29.0 + github.com/joho/godotenv v1.5.1 + github.com/stretchr/testify v1.11.1 + modernc.org/sqlite v1.48.0 +) + +require ( + github.com/davecgh/go-spew v1.1.1 // indirect + github.com/dustin/go-humanize v1.0.1 // indirect + github.com/google/uuid v1.6.0 // indirect + github.com/gorilla/websocket v1.4.2 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/ncruces/go-strftime v1.0.0 // indirect + github.com/pmezard/go-difflib v1.0.0 // indirect + github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect + golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect + golang.org/x/sys v0.42.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect + modernc.org/libc v1.70.0 // indirect + modernc.org/mathutil v1.7.1 // indirect + modernc.org/memory v1.11.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..8517d5a --- /dev/null +++ b/go.sum @@ -0,0 +1,74 @@ +github.com/bwmarrin/discordgo v0.29.0 h1:FmWeXFaKUwrcL3Cx65c20bTRW+vOb6k8AnaP+EgjDno= +github.com/bwmarrin/discordgo v0.29.0/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY= +github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto= +github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e h1:ijClszYn+mADRFY17kjQEVQ1XRhq2/JR1M3sGqeJoxs= +github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc= +github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/hashicorp/golang-lru/v2 v2.0.7 h1:a+bsQ5rvGLjzHuww6tVxozPZFVghXaHOwFs4luLUK2k= +github.com/hashicorp/golang-lru/v2 v2.0.7/go.mod h1:QeFd9opnmA6QUJc5vARoKUSoFhyfM2/ZepoAG6RGpeM= +github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0= +github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/ncruces/go-strftime v1.0.0 h1:HMFp8mLCTPp341M/ZnA4qaf7ZlsbTc+miZjCLOFAw7w= +github.com/ncruces/go-strftime v1.0.0/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE= +github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo= +github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U= +github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg= +golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4= +golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8= +golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4= +golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo= +golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k= +golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +modernc.org/cc/v4 v4.27.1 h1:9W30zRlYrefrDV2JE2O8VDtJ1yPGownxciz5rrbQZis= +modernc.org/cc/v4 v4.27.1/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0= +modernc.org/ccgo/v4 v4.32.0 h1:hjG66bI/kqIPX1b2yT6fr/jt+QedtP2fqojG2VrFuVw= +modernc.org/ccgo/v4 v4.32.0/go.mod h1:6F08EBCx5uQc38kMGl+0Nm0oWczoo1c7cgpzEry7Uc0= +modernc.org/fileutil v1.4.0 h1:j6ZzNTftVS054gi281TyLjHPp6CPHr2KCxEXjEbD6SM= +modernc.org/fileutil v1.4.0/go.mod h1:EqdKFDxiByqxLk8ozOxObDSfcVOv/54xDs/DUHdvCUU= +modernc.org/gc/v2 v2.6.5 h1:nyqdV8q46KvTpZlsw66kWqwXRHdjIlJOhG6kxiV/9xI= +modernc.org/gc/v2 v2.6.5/go.mod h1:YgIahr1ypgfe7chRuJi2gD7DBQiKSLMPgBQe9oIiito= +modernc.org/gc/v3 v3.1.2 h1:ZtDCnhonXSZexk/AYsegNRV1lJGgaNZJuKjJSWKyEqo= +modernc.org/gc/v3 v3.1.2/go.mod h1:HFK/6AGESC7Ex+EZJhJ2Gni6cTaYpSMmU/cT9RmlfYY= +modernc.org/goabi0 v0.2.0 h1:HvEowk7LxcPd0eq6mVOAEMai46V+i7Jrj13t4AzuNks= +modernc.org/goabi0 v0.2.0/go.mod h1:CEFRnnJhKvWT1c1JTI3Avm+tgOWbkOu5oPA8eH8LnMI= +modernc.org/libc v1.70.0 h1:U58NawXqXbgpZ/dcdS9kMshu08aiA6b7gusEusqzNkw= +modernc.org/libc v1.70.0/go.mod h1:OVmxFGP1CI/Z4L3E0Q3Mf1PDE0BucwMkcXjjLntvHJo= +modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU= +modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg= +modernc.org/memory v1.11.0 h1:o4QC8aMQzmcwCK3t3Ux/ZHmwFPzE6hf2Y5LbkRs+hbI= +modernc.org/memory v1.11.0/go.mod h1:/JP4VbVC+K5sU2wZi9bHoq2MAkCnrt2r98UGeSK7Mjw= +modernc.org/opt v0.1.4 h1:2kNGMRiUjrp4LcaPuLY2PzUfqM/w9N23quVwhKt5Qm8= +modernc.org/opt v0.1.4/go.mod h1:03fq9lsNfvkYSfxrfUhZCWPk1lm4cq4N+Bh//bEtgns= +modernc.org/sortutil v1.2.1 h1:+xyoGf15mM3NMlPDnFqrteY07klSFxLElE2PVuWIJ7w= +modernc.org/sortutil v1.2.1/go.mod h1:7ZI3a3REbai7gzCLcotuw9AC4VZVpYMjDzETGsSMqJE= +modernc.org/sqlite v1.48.0 h1:ElZyLop3Q2mHYk5IFPPXADejZrlHu7APbpB0sF78bq4= +modernc.org/sqlite v1.48.0/go.mod h1:hWjRO6Tj/5Ik8ieqxQybiEOUXy0NJFNp2tpvVpKlvig= +modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0= +modernc.org/strutil v1.2.1/go.mod h1:EHkiggD70koQxjVdSBM3JKM7k6L0FbGE5eymy9i3B9A= +modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y= +modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..5ecfed3 --- /dev/null +++ b/main.go @@ -0,0 +1,102 @@ +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) +} diff --git a/main_test.go b/main_test.go new file mode 100644 index 0000000..0328c31 --- /dev/null +++ b/main_test.go @@ -0,0 +1,29 @@ +package main + +import ( + "testing" + "time" + + "github.com/stretchr/testify/assert" +) + +func TestFormatDuration(t *testing.T) { + tests := []struct { + name string + seconds int + expected string + }{ + {"Nur Sekunden", 45, "0h 0m 45s"}, + {"Genau eine Minute", 60, "0h 1m 0s"}, + {"Stunden und Minuten", 3661, "1h 1m 1s"}, + {"Null Zeit", 0, "0h 0m 0s"}, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + d := time.Duration(tt.seconds) * time.Second + result := formatDuration(d) + assert.Equal(t, tt.expected, result) + }) + } +}