Compare commits

...

6 Commits

Author SHA1 Message Date
a78c8df5a0 Merge branch 'master' into metrics 2025-05-28 23:42:38 +00:00
aa58aa6f2a stacked boxplot 2025-05-29 01:42:01 +02:00
fc29af19b3 gitignore metrics json 2025-05-29 01:05:00 +02:00
71566d9a3a Merge pull request 'metrics' (#1) from metrics into master
Reviewed-on: #1
2025-05-28 23:02:29 +00:00
49a142b076 Sort stats 2025-05-29 01:00:44 +02:00
af1080fb6a Add metrics 2025-05-29 00:52:44 +02:00
6 changed files with 420 additions and 1 deletions

3
.gitignore vendored
View File

@ -1,9 +1,10 @@
old_json/
json/
old_commands/
metrics/metrics.json
# ---> Go
# If you prefer the allow list template instead of the deny list, see community template:
# If you prefer the white list template instead of the nigger list, see community template:
# https://github.com/github/gitignore/blob/main/community/Golang/Go.AllowList.gitignore
#
# Binaries for programs and plugins

View File

@ -1,5 +1,23 @@
package commands
import (
"bytes"
"fmt"
"image/color"
"image/png"
"nignoggobot/metrics"
"sort"
"strings"
"gonum.org/v1/plot"
"gonum.org/v1/plot/plotter"
"gonum.org/v1/plot/vg"
"gonum.org/v1/plot/vg/draw"
"gonum.org/v1/plot/vg/vgimg"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
var AdminIDs = []int64{126131628}
func IsAdmin(userID int64) bool {
@ -10,3 +28,180 @@ func IsAdmin(userID int64) bool {
}
return false
}
type StatsCommand struct{}
func (StatsCommand) Name() string { return "stats" }
func (StatsCommand) Help() string { return "Show bot usage stats (admin only)." }
func (StatsCommand) Execute(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
if !IsAdmin(update.Message.From.ID) {
return
}
stats := metrics.GetStats()
var sb strings.Builder
type cmdStat struct {
Name string
AllTime int
Last30Sum int
Last30 [30]int
}
var statsList []cmdStat
for name, stat := range stats.Commands {
sum := 0
for _, v := range stat.Last30 {
sum += v
}
if sum > 0 {
statsList = append(statsList, cmdStat{Name: name, AllTime: stat.AllTime, Last30Sum: sum, Last30: stat.Last30})
}
}
// Sort by last 30 days usage for the graph
sort.Slice(statsList, func(i, j int) bool {
return statsList[i].Last30Sum > statsList[j].Last30Sum
})
// Prepare data for stacked bar chart
nCmds := len(statsList)
nDays := 30
if nCmds > 0 {
colors := []color.Color{
color.RGBA{0x1f, 0x77, 0xb4, 0xff}, // blue
color.RGBA{0xff, 0x7f, 0x0e, 0xff}, // orange
color.RGBA{0x2c, 0xa0, 0x2c, 0xff}, // green
color.RGBA{0xd6, 0x27, 0x28, 0xff}, // red
color.RGBA{0x94, 0x67, 0xbd, 0xff}, // purple
color.RGBA{0x8c, 0x56, 0x4b, 0xff}, // brown
color.RGBA{0xe3, 0x77, 0xc2, 0xff}, // pink
color.RGBA{0x7f, 0x7f, 0x7f, 0xff}, // gray
color.RGBA{0xbc, 0xbd, 0x22, 0xff}, // yellow-green
color.RGBA{0x17, 0xbe, 0xcf, 0xff}, // cyan
}
for len(colors) < nCmds {
colors = append(colors, color.RGBA{uint8(50 * len(colors)), uint8(100 * len(colors)), uint8(150 * len(colors)), 0xff})
}
p := plot.New()
p.Title.Text = "Command Usage (Last 30 Days)"
p.Y.Label.Text = "Count"
p.X.Label.Text = "Days Ago"
p.Legend.Top = true
p.Legend.Left = false
p.Legend.XOffs = 0
p.Legend.YOffs = 0
// Prepare the stacked values
stacks := make([][]float64, nCmds)
for i := range stacks {
stacks[i] = make([]float64, nDays)
}
for cmdIdx, stat := range statsList {
for day := 0; day < nDays; day++ {
stacks[cmdIdx][day] = float64(stat.Last30[day]) // 0=today, 29=oldest
}
}
barCharts := make([]*plotter.BarChart, nCmds)
barWidth := vg.Points(10)
for cmdIdx := 0; cmdIdx < nCmds; cmdIdx++ {
values := make(plotter.Values, nDays)
for day := 0; day < nDays; day++ {
values[day] = stacks[cmdIdx][day]
}
bar, err := plotter.NewBarChart(values, barWidth)
if err != nil {
continue
}
bar.Color = colors[cmdIdx]
bar.Offset = 0
if cmdIdx > 0 {
bar.StackOn(barCharts[cmdIdx-1])
}
p.Add(bar)
p.Legend.Add(statsList[cmdIdx].Name, bar)
barCharts[cmdIdx] = bar
}
labels := make([]string, nDays)
for i := 0; i < nDays; i++ {
labels[i] = fmt.Sprintf("%d", i) // 0=Today, 29=Oldest
}
p.NominalX(labels...)
img := vgimg.New(800, 400)
dc := draw.New(img)
p.Draw(dc)
buf := new(bytes.Buffer)
png.Encode(buf, img.Image())
photo := tgbotapi.FileBytes{Name: "stats.png", Bytes: buf.Bytes()}
photoMsg := tgbotapi.NewPhoto(update.Message.Chat.ID, photo)
photoMsg.Caption = "Command usage for the last 30 days (0 = today)"
bot.Send(photoMsg)
}
// Text stats as before
sort.Slice(statsList, func(i, j int) bool {
return statsList[i].AllTime > statsList[j].AllTime
})
sb.WriteString("Command usage (all time):\n")
for _, s := range statsList {
sb.WriteString(fmt.Sprintf("/%s: %d\n", s.Name, s.AllTime))
}
sort.Slice(statsList, func(i, j int) bool {
return statsList[i].Last30Sum > statsList[j].Last30Sum
})
sb.WriteString("\nCommand usage (last 30 days):\n")
for _, s := range statsList {
sb.WriteString(fmt.Sprintf("/%s: %d\n", s.Name, s.Last30Sum))
}
groups, privs, blocked := 0, 0, 0
for _, chat := range stats.Chats {
if chat.IsGroup {
groups++
} else {
privs++
}
if chat.Blocked {
blocked++
}
}
sb.WriteString(fmt.Sprintf("\nGroups: %d\nPrivate chats: %d\nBlocked/muted: %d\n", groups, privs, blocked))
msg := tgbotapi.NewMessage(update.Message.Chat.ID, sb.String())
bot.Send(msg)
}
type BroadcastCommand struct{}
func (BroadcastCommand) Name() string { return "broadcast" }
func (BroadcastCommand) Help() string {
return "Broadcast a <message> to all chats (admin only)."
}
func (BroadcastCommand) Execute(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
if !IsAdmin(update.Message.From.ID) {
return
}
args := strings.TrimSpace(update.Message.CommandArguments())
if args == "" {
msg := tgbotapi.NewMessage(update.Message.Chat.ID, "Usage: /broadcast <message>")
bot.Send(msg)
return
}
count := 0
metrics.BroadcastMessage(func(chatID int64) error {
msg := tgbotapi.NewMessage(chatID, args)
_, err := bot.Send(msg)
if err == nil {
count++
}
return err
})
msg := tgbotapi.NewMessage(update.Message.Chat.ID, fmt.Sprintf("Broadcast sent to %d chats.", count))
bot.Send(msg)
}
func init() {
Register(StatsCommand{})
Register(BroadcastCommand{})
}

15
go.mod
View File

@ -3,3 +3,18 @@ module nignoggobot
go 1.24.3
require github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1
require (
codeberg.org/go-fonts/liberation v0.5.0 // indirect
codeberg.org/go-latex/latex v0.1.0 // indirect
codeberg.org/go-pdf/fpdf v0.10.0 // indirect
git.sr.ht/~sbinet/gg v0.6.0 // indirect
github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b // indirect
github.com/campoy/embedmd v1.0.0 // indirect
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
github.com/pmezard/go-difflib v1.0.0 // indirect
golang.org/x/image v0.25.0 // indirect
golang.org/x/text v0.23.0 // indirect
gonum.org/v1/plot v0.16.0 // indirect
rsc.io/pdf v0.1.1 // indirect
)

51
go.sum
View File

@ -1,2 +1,53 @@
codeberg.org/go-fonts/liberation v0.5.0 h1:SsKoMO1v1OZmzkG2DY+7ZkCL9U+rrWI09niOLfQ5Bo0=
codeberg.org/go-fonts/liberation v0.5.0/go.mod h1:zS/2e1354/mJ4pGzIIaEtm/59VFCFnYC7YV6YdGl5GU=
codeberg.org/go-latex/latex v0.1.0 h1:hoGO86rIbWVyjtlDLzCqZPjNykpWQ9YuTZqAzPcfL3c=
codeberg.org/go-latex/latex v0.1.0/go.mod h1:LA0q/AyWIYrqVd+A9Upkgsb+IqPcmSTKc9Dny04MHMw=
codeberg.org/go-pdf/fpdf v0.10.0 h1:u+w669foDDx5Ds43mpiiayp40Ov6sZalgcPMDBcZRd4=
codeberg.org/go-pdf/fpdf v0.10.0/go.mod h1:Y0DGRAdZ0OmnZPvjbMp/1bYxmIPxm0ws4tfoPOc4LjU=
git.sr.ht/~sbinet/gg v0.6.0 h1:RIzgkizAk+9r7uPzf/VfbJHBMKUr0F5hRFxTUGMnt38=
git.sr.ht/~sbinet/gg v0.6.0/go.mod h1:uucygbfC9wVPQIfrmwM2et0imr8L7KQWywX0xpFMm94=
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
github.com/ajstarks/deck v0.0.0-20200831202436-30c9fc6549a9/go.mod h1:JynElWSGnm/4RlzPXRlREEwqTHAN3T56Bv2ITsFT3gY=
github.com/ajstarks/deck/generate v0.0.0-20210309230005-c3f852c02e19/go.mod h1:T13YZdzov6OU0A1+RfKZiZN9ca6VeKdBdyDV+BY97Tk=
github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b h1:slYM766cy2nI3BwyRiyQj/Ud48djTMtMebDqepE95rw=
github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b/go.mod h1:1KcenG0jGWcpt8ov532z81sp/kMMUG485J2InIOyADM=
github.com/campoy/embedmd v1.0.0 h1:V4kI2qTJJLf4J29RzI/MAt2c3Bl4dQSYPuflzwFH2hY=
github.com/campoy/embedmd v1.0.0/go.mod h1:oxyr9RCiSXg0M3VJ3ks0UGfp98BpSSGr0kpiX3MzVl8=
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1 h1:wG8n/XJQ07TmjbITcGiUaOtXxdrINDz1b0J1w0SzqDc=
github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1/go.mod h1:A2S0CWkNylc2phvKXWBBdD3K0iGnDBGbzRpISP2zBl8=
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g=
github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
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/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/image v0.25.0 h1:Y6uW6rH1y5y/LK1J8BPWZtr6yZ7hrsy6hFrXjgsc2fQ=
golang.org/x/image v0.25.0/go.mod h1:tCAmOEGthTtkalusGp1g3xa2gke8J6c2N565dTyl9Rs=
golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY=
golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
golang.org/x/tools v0.1.0/go.mod h1:xkSsbof2nBLbhDlRMhhhyNLN/zl3eTqcnHD5viDpcZ0=
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
gonum.org/v1/plot v0.16.0 h1:dK28Qx/Ky4VmPUN/2zeW0ELyM6ucDnBAj5yun7M9n1g=
gonum.org/v1/plot v0.16.0/go.mod h1:Xz6U1yDMi6Ni6aaXILqmVIb6Vro8E+K7Q/GeeH+Pn0c=
honnef.co/go/tools v0.1.3/go.mod h1:NgwopIslSNH47DimFoV78dnkksY2EFtX0ajyb3K/las=
rsc.io/pdf v0.1.1 h1:k1MczvYDUvJBe93bYd7wrZLLUEcLZAuF824/I4e5Xr4=
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

View File

@ -8,6 +8,7 @@ import (
"syscall"
"nignoggobot/commands"
"nignoggobot/metrics"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
@ -66,10 +67,18 @@ func main() {
continue
}
log.Printf("Received command: %s from chat %d (user %d)", update.Message.Command(), update.Message.Chat.ID, update.Message.From.ID)
cmdName := update.Message.Command()
isGroup := update.Message.Chat.IsGroup() || update.Message.Chat.IsSuperGroup()
metrics.UpdateChat(update.Message.Chat.ID, isGroup)
if cmd, ok := commands.All()[cmdName]; ok {
log.Printf("Executing command: %s", cmdName)
// Bottleneck here locking mutexes and writing to json every time but oh well
metrics.IncrementCommandUsage(cmdName)
cmd.Execute(update, bot)
} else {
log.Printf("Unknown command: %s", cmdName)
msg := tgbotapi.NewMessage(update.Message.Chat.ID, "Unknown command.")
bot.Send(msg)
}

148
metrics/metrics.go Normal file
View File

@ -0,0 +1,148 @@
package metrics
import (
"encoding/json"
"fmt"
"os"
"sync"
"time"
)
type CommandStats struct {
AllTime int `json:"all_time"`
Last30 [30]int `json:"last_30"`
Last30Day int64 `json:"last_30_day"` // Unix day of the first slot
}
type ChatInfo struct {
IsGroup bool `json:"is_group"`
LastActive int64 `json:"last_active"`
Blocked bool `json:"blocked"`
LastError string `json:"last_error,omitempty"`
}
type Metrics struct {
Commands map[string]*CommandStats `json:"commands"`
Chats map[int64]*ChatInfo `json:"chats"`
}
var (
metricsFile = "metrics/metrics.json"
metrics *Metrics
mu sync.Mutex
)
func loadMetrics() {
if metrics != nil {
return
}
metrics = &Metrics{
Commands: make(map[string]*CommandStats),
Chats: make(map[int64]*ChatInfo),
}
f, err := os.Open(metricsFile)
if err != nil {
return
}
defer f.Close()
json.NewDecoder(f).Decode(metrics)
}
func saveMetrics() {
// Do not lock here; caller must hold the lock else it will shitty deadlock
loadMetrics()
f, err := os.Create(metricsFile)
if err != nil {
return
}
defer f.Close()
enc := json.NewEncoder(f)
enc.SetIndent("", " ")
enc.Encode(metrics)
}
func IncrementCommandUsage(cmd string) {
fmt.Printf("[metrics] IncrementCommandUsage called with cmd=%s\n", cmd)
mu.Lock()
defer mu.Unlock()
loadMetrics()
stat, ok := metrics.Commands[cmd]
if !ok {
stat = &CommandStats{Last30Day: time.Now().Unix() / 86400}
metrics.Commands[cmd] = stat
}
today := time.Now().Unix() / 86400
shift := int(today - stat.Last30Day)
if shift > 0 && shift < 30 {
copy(stat.Last30[shift:], stat.Last30[:30-shift])
for i := 0; i < shift; i++ {
stat.Last30[i] = 0
}
stat.Last30Day = today
} else if shift >= 30 {
for i := 0; i < 30; i++ {
stat.Last30[i] = 0
}
stat.Last30Day = today
}
stat.AllTime++
stat.Last30[0]++
saveMetrics()
}
func UpdateChat(chatID int64, isGroup bool) {
fmt.Printf("[metrics] UpdateChat called with chatID=%d isGroup=%v\n", chatID, isGroup)
mu.Lock()
defer mu.Unlock()
loadMetrics()
info, ok := metrics.Chats[chatID]
if !ok {
info = &ChatInfo{IsGroup: isGroup}
metrics.Chats[chatID] = info
}
info.LastActive = time.Now().Unix()
info.Blocked = false
info.LastError = ""
saveMetrics()
}
func MarkChatBlocked(chatID int64, errMsg string) {
fmt.Printf("[metrics] MarkChatBlocked called with chatID=%d errMsg=%s\n", chatID, errMsg)
mu.Lock()
defer mu.Unlock()
loadMetrics()
info, ok := metrics.Chats[chatID]
if !ok {
info = &ChatInfo{}
metrics.Chats[chatID] = info
}
info.Blocked = true
info.LastError = errMsg
saveMetrics()
}
func GetStats() *Metrics {
fmt.Printf("[metrics] GetStats called\n")
mu.Lock()
defer mu.Unlock()
loadMetrics()
copy := *metrics
return &copy
}
func BroadcastMessage(sendFunc func(chatID int64) error) {
fmt.Printf("[metrics] BroadcastMessage called\n")
mu.Lock()
loadMetrics()
ids := make([]int64, 0, len(metrics.Chats))
for id := range metrics.Chats {
ids = append(ids, id)
}
mu.Unlock()
for _, id := range ids {
err := sendFunc(id)
if err != nil {
MarkChatBlocked(id, err.Error())
}
}
}