Compare commits

..

22 Commits

Author SHA1 Message Date
8b09b2e323 Merge pull request 'Closed issue #8 and Created A command to clear the db from entries that are 40 days old or older. The DB now lies in an external volume so the redeploy wont hurt it' (#10) from dev into main
All checks were successful
Bot Unit Tests / test (push) Successful in 10m36s
Reviewed-on: #10
2026-04-02 14:49:46 +00:00
6cc4c543e2 Closed issue #8 and reworked the docker-compose and also made a command to clear the db which should run once a night at 00:00. it deletes all entries that are older then 40 days
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Bot Unit Tests / test (pull_request) Successful in 10m14s
2026-04-02 16:36:29 +02:00
ed4ae89369 Merge pull request 'main' (#9) from main into dev
Some checks failed
Bot Unit Tests / test (push) Failing after 5m17s
Reviewed-on: #9
2026-04-02 14:09:33 +00:00
deec858857 Update db_management/db_management.go
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
2026-04-01 22:01:32 +00:00
d37eedf7cf Update docker-compose.yaml
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
2026-04-01 22:00:21 +00:00
555bb588a0 Merge pull request 'refactored the voice update function to exclude users if their muted or afk and remade the commands' (#6) from dev into main
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Reviewed-on: Jannis/discord-go-bot#6
2026-04-01 21:00:10 +00:00
6b0dc24721 refactored the voice update function to exclude users if their muted or afk and remade the commands
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Bot Unit Tests / test (pull_request) Has been cancelled
2026-04-01 22:50:03 +02:00
efc178f578 Merge pull request 'changed version on the dockerfile' (#5) from dev into main
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Reviewed-on: Jannis/discord-go-bot#5
2026-04-01 19:52:51 +00:00
493c093fdb changed version on the dockerfile
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Bot Unit Tests / test (pull_request) Successful in 10m16s
2026-04-01 21:52:02 +02:00
03606d5a3e Update docker-compose.yaml
All checks were successful
Bot Unit Tests / test (push) Successful in 10m13s
2026-04-01 19:33:25 +00:00
28f2dc9193 Merge pull request 'updated docker-compose' (#4) from dev into main
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Reviewed-on: Jannis/discord-go-bot#4
2026-04-01 19:32:02 +00:00
ad0e6acf26 updated docker-compose
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
Bot Unit Tests / test (pull_request) Has been cancelled
2026-04-01 21:31:12 +02:00
a3f4786586 Merge pull request 'initial merge' (#1) from dev into main
All checks were successful
Bot Unit Tests / test (push) Successful in 10m10s
Reviewed-on: Jannis/discord-go-bot#1
2026-04-01 19:04:09 +00:00
d9dc739963 Dockerized
All checks were successful
Bot Unit Tests / test (push) Successful in 10m14s
Bot Unit Tests / test (pull_request) Successful in 10m14s
2026-04-01 20:45:18 +02:00
9efa490ffe optimized the tests and go.mod file. Switched to older version for better compatibility
All checks were successful
Bot Unit Tests / test (push) Successful in 10m20s
2026-04-01 20:30:40 +02:00
e4b86421a6 updated test.yaml
Some checks failed
Bot Unit Tests / test (push) Has been cancelled
2026-04-01 20:21:57 +02:00
120b713fb4 added tests
Some checks failed
Bot Unit Tests / run-tests (push) Failing after 7s
2026-04-01 20:18:30 +02:00
6f3e5e3d2b Merge branch 'dev' of ssh://gitea.starfour.de:2222/Jannis/discord-go-bot into dev 2026-04-01 20:15:48 +02:00
51200014fc updated so the filename wont matter 2026-04-01 20:14:12 +02:00
bfff048d60 Delete bot.db 2026-04-01 18:13:06 +00:00
75d0078a60 wont save my local test db on the server 2026-04-01 20:12:46 +02:00
87b3623e60 v1 finished. Gonna build a dockerfile later 2026-04-01 20:11:38 +02:00
12 changed files with 711 additions and 0 deletions

4
.dockerignore Normal file
View File

@@ -0,0 +1,4 @@
.git
bot.db
vendor
.env

1
.env.example Normal file
View File

@@ -0,0 +1 @@
DISCORD_TOKEN="Ein Discord Token"

View File

@@ -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

2
.gitignore vendored Normal file
View File

@@ -0,0 +1,2 @@
.env
*.db

31
Dockerfile Normal file
View File

@@ -0,0 +1,31 @@
# STAGE 1: Build
FROM golang:1.25-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"]

View File

@@ -0,0 +1,109 @@
package db_management
import (
"database/sql"
"log"
"os"
"time"
_ "modernc.org/sqlite"
)
var DB *sql.DB
func InitDB() {
var err error
// Öffnet bot.db im aktuellen Verzeichnis
dbPath := os.Getenv("DB_PATH")
if dbPath == "" {
dbPath = "./bot.db"
}
DB, err = sql.Open("sqlite", dbPath)
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 > datetime('now', '-30 days')`
err := DB.QueryRow(query, userID).Scan(&total)
if err != nil || !total.Valid {
return 0
}
return int(total.Int64)
}
type UserStats struct {
UserID string
Seconds int
}
func GetTopUsers(limit int) []UserStats {
query := `
SELECT user_id, SUM(seconds) as total
FROM voice_sessions
WHERE created_at > datetime('now', '-30 days')
GROUP BY user_id
ORDER BY total DESC
LIMIT ?
`
rows, err := DB.Query(query, limit)
if err != nil {
return nil
}
defer rows.Close()
var result []UserStats
for rows.Next() {
var u UserStats
err := rows.Scan(&u.UserID, &u.Seconds)
if err != nil {
continue
}
result = append(result, u)
}
return result
}
func CleanupOldEntries() {
query := `DELETE FROM voice_sessions WHERE created_at < datetime('now', '-40 days)`
result, err := DB.Exec(query)
if err != nil {
log.Printf("Fehler beim DB-Cleanup: %v", err)
return
}
rowsAffected, _ := result.RowsAffected()
if rowsAffected > 0 {
log.Printf("DB Cleanup erfolgreich: %d alte Einträge gelöscht.", rowsAffected)
}
}

View File

@@ -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")
}

43
docker-compose.yaml Normal file
View File

@@ -0,0 +1,43 @@
version: '3.8'
services:
app:
build:
context: .
dockerfile: Dockerfile
image: discord_vc_bot:latest
pull_policy: build
container_name: docker_vc_bot
restart: unless-stopped
volumes:
- discord_db:/app/data
environment:
- DB_PATH=/app/data/bot.db
- DISCORD_TOKEN=${DISCORD_TOKEN}
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
volumes:
discord_db:
external: true

28
go.mod Normal file
View File

@@ -0,0 +1,28 @@
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
github.com/robfig/cron/v3 v3.0.1 // 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
)

76
go.sum Normal file
View File

@@ -0,0 +1,76 @@
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/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
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=

305
main.go Normal file
View File

@@ -0,0 +1,305 @@
package main
import (
"fmt"
"log"
"os"
"os/signal"
"syscall"
"time"
"github.com/bwmarrin/discordgo"
"github.com/joho/godotenv"
"github.com/robfig/cron/v3"
"heydeman/vc-stat-bot-discord/db_management"
)
var sessions = make(map[string]*Session)
type Session struct {
Start time.Time
PausedAt *time.Time
Paused bool
Accum time.Duration
}
func main() {
_ = godotenv.Load()
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)
}
// Handler & Intents
discord.AddHandler(interactionCreate)
discord.AddHandler(voiceUpdate)
discord.Identify.Intents = discordgo.IntentsGuilds | discordgo.IntentsGuildVoiceStates
err = discord.Open()
if err != nil {
log.Fatal("Error opening connection:", err)
}
registerCommands(discord)
// ==========================================
// 🕒 CRON SCHEDULER SETUP
// ==========================================
c := cron.New()
// Alle 5 Minuten: Aktuellen Stand in die DB schieben
_, _ = c.AddFunc("*/5 * * * *", func() {
fmt.Println("🔄 Cron Auto-Save: Speichere Zwischenstände...")
for uid, sess := range sessions {
processAndSave(uid, sess)
}
})
// Jeden Tag um 00:00 Uhr: Alte Daten (40+ Tage) löschen
_, _ = c.AddFunc("0 0 * * *", func() {
fmt.Println("🧹 Cron Cleanup: Lösche Daten älter als 40 Tage...")
db_management.CleanupOldEntries()
})
c.Start()
defer c.Stop()
fmt.Println("Bot läuft stabil. CTRL+C zum Beenden.")
// Shutdown Handling
stop := make(chan os.Signal, 1)
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)
<-stop
fmt.Println("Speichere aktive Sessions vor Shutdown...")
// Beim Beenden alles wegschreiben, was noch im RAM ist
for uid, sess := range sessions {
processAndSave(uid, sess)
}
discord.Close()
}
func registerCommands(s *discordgo.Session) {
// /stats
_, err := s.ApplicationCommandCreate(s.State.User.ID, "", &discordgo.ApplicationCommand{
Name: "stats",
Description: "Zeigt die 30-Tage Statistik eines Users",
Options: []*discordgo.ApplicationCommandOption{
{
Type: discordgo.ApplicationCommandOptionUser,
Name: "user",
Description: "Optional: anderer User",
Required: false,
},
},
})
if err != nil {
log.Println("Fehler bei /stats:", err)
}
// /top
_, err = s.ApplicationCommandCreate(s.State.User.ID, "", &discordgo.ApplicationCommand{
Name: "top",
Description: "Top 10 User nach Zeit (30 Tage)",
})
if err != nil {
log.Println("Fehler bei /top:", err)
}
}
func interactionCreate(s *discordgo.Session, i *discordgo.InteractionCreate) {
if i.Type != discordgo.InteractionApplicationCommand {
return
}
switch i.ApplicationCommandData().Name {
// =====================
// /stats
// =====================
case "stats":
user := i.Member.User
if len(i.ApplicationCommandData().Options) > 0 {
opt := i.ApplicationCommandData().Options[0]
if opt.Name == "user" {
user = opt.UserValue(s)
}
}
totalSeconds := db_management.GetUserStats(user.ID)
duration := time.Duration(totalSeconds) * time.Second
name := user.GlobalName
if name == "" {
name = user.Username
}
msg := fmt.Sprintf(
"📊 **30-Tage Statistik für %s**\nZeit: `%s`",
name,
formatDuration(duration),
)
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: msg,
},
})
// =====================
// /top
// =====================
case "top":
topUsers := db_management.GetTopUsers(10)
if len(topUsers) == 0 {
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: "Keine Daten vorhanden.",
},
})
return
}
msg := "🏆 **Top 10 (30 Tage)**\n\n"
for i, user := range topUsers {
duration := time.Duration(user.Seconds) * time.Second
msg += fmt.Sprintf(
"**%d.** <@%s> — `%s`\n",
i+1,
user.UserID,
formatDuration(duration),
)
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: msg,
},
})
}
}
const excludedChannelID = "1232031563391172649"
func voiceUpdate(s *discordgo.Session, v *discordgo.VoiceStateUpdate) {
uid := v.UserID
// =========================
// ❌ Pause Bedingungen
// =========================
if v.ChannelID == excludedChannelID || v.SelfMute || v.Mute || v.SelfDeaf || v.Deaf {
if sess, ok := sessions[uid]; ok && !sess.Paused {
now := time.Now()
elapsed := now.Sub(sess.Start)
sess.Accum += elapsed
sess.PausedAt = &now
sess.Paused = true
}
return
}
// =========================
// ▶ Resume (User wieder aktiv)
// =========================
if v.ChannelID != "" {
sess, ok := sessions[uid]
if !ok {
// neue Session
sessions[uid] = &Session{
Start: time.Now(),
Paused: false,
}
return
}
// wenn pausiert → weiterlaufen lassen
if sess.Paused {
sess.Start = time.Now()
sess.Paused = false
sess.PausedAt = nil
}
return
}
// =========================
// ❌ User verlässt Voice
// =========================
if sess, ok := sessions[uid]; ok {
total := sess.Accum
if !sess.Paused {
total += time.Since(sess.Start)
}
db_management.SaveSession(uid, int(total.Seconds()))
delete(sessions, 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)
}
func autoSaveSessions() {
ticker := time.NewTicker(5 * time.Minute)
defer ticker.Stop()
for range ticker.C {
fmt.Println("Auto-Save")
for uid, sess := range sessions {
processAndSave(uid, sess)
}
}
}
// Hilfsfunktion, die die Zeit berechnet, speichert und den Timer resettet
func processAndSave(uid string, sess *Session) {
var totalToSave time.Duration
totalToSave += sess.Accum
if !sess.Paused {
totalToSave += time.Since(sess.Start)
}
if totalToSave.Seconds() >= 1 {
db_management.SaveSession(uid, int(totalToSave.Seconds()))
sess.Accum = 0
if !sess.Paused {
sess.Start = time.Now()
}
}
}

29
main_test.go Normal file
View File

@@ -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)
})
}
}