Compare commits
21 Commits
f1b157038c
...
dev
| Author | SHA1 | Date | |
|---|---|---|---|
| 6cc4c543e2 | |||
| ed4ae89369 | |||
| deec858857 | |||
| d37eedf7cf | |||
| 555bb588a0 | |||
| 6b0dc24721 | |||
| efc178f578 | |||
| 493c093fdb | |||
| 03606d5a3e | |||
| 28f2dc9193 | |||
| ad0e6acf26 | |||
| a3f4786586 | |||
| d9dc739963 | |||
| 9efa490ffe | |||
| e4b86421a6 | |||
| 120b713fb4 | |||
| 6f3e5e3d2b | |||
| 51200014fc | |||
| bfff048d60 | |||
| 75d0078a60 | |||
| 87b3623e60 |
4
.dockerignore
Normal file
4
.dockerignore
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
.git
|
||||||
|
bot.db
|
||||||
|
vendor
|
||||||
|
.env
|
||||||
1
.env.example
Normal file
1
.env.example
Normal file
@@ -0,0 +1 @@
|
|||||||
|
DISCORD_TOKEN="Ein Discord Token"
|
||||||
21
.gitea/workflows/test.yaml
Normal file
21
.gitea/workflows/test.yaml
Normal 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
2
.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
.env
|
||||||
|
*.db
|
||||||
31
Dockerfile
Normal file
31
Dockerfile
Normal 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"]
|
||||||
109
db_management/db_management.go
Normal file
109
db_management/db_management.go
Normal 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)
|
||||||
|
}
|
||||||
|
}
|
||||||
62
db_management/db_management_test.go
Normal file
62
db_management/db_management_test.go
Normal 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
43
docker-compose.yaml
Normal 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
28
go.mod
Normal 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
76
go.sum
Normal 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
305
main.go
Normal 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
29
main_test.go
Normal 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)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user