mirror of
https://github.com/tiennm99/tiennm99bot.git
synced 2026-10-13 04:15:17 +00:00
tiennm99
7008a57cf5
BREAKING CHANGE: the Go module path is now github.com/tiennm99/tiennm99bot, the default sticker pack is stickers_by_<bot username>, and the health body, deploy DM, User-Agents and image names say tiennm99bot.
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
// Package stats tracks command usage and exposes /stats subcommands sorted by
|
|
// popularity.
|
|
//
|
|
// Counts are recorded through a CommandHook, so every authorized command
|
|
// invocation of every module is counted without the modules knowing. Storage
|
|
// goes through Mongo aggregations when the collection is MongoDB-backed and
|
|
// falls back to reading the whole collection through the document store
|
|
// otherwise (the in-memory provider).
|
|
package stats
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-telegram/bot/models"
|
|
|
|
"github.com/tiennm99/tiennm99bot/internal/log"
|
|
"github.com/tiennm99/tiennm99bot/internal/modules"
|
|
"github.com/tiennm99/tiennm99bot/internal/storage"
|
|
)
|
|
|
|
// topK caps the rows every /stats view shows.
|
|
const topK = 20
|
|
|
|
// counter owns the stats repository used by the command hook and render views.
|
|
type counter struct {
|
|
store usageStore
|
|
}
|
|
|
|
// Inc records one authorized command invocation. A sender only contributes to
|
|
// user-level stats when Telegram provides a public username; otherwise the
|
|
// invocation still contributes to command totals.
|
|
func (c *counter) Inc(ctx context.Context, name string, update *models.Update) {
|
|
var (
|
|
user usageUser
|
|
hasUser bool
|
|
)
|
|
if update != nil && update.Message != nil && update.Message.From != nil && update.Message.From.Username != "" {
|
|
user = usageUser{
|
|
ID: update.Message.From.ID,
|
|
Username: update.Message.From.Username,
|
|
}
|
|
hasUser = true
|
|
}
|
|
|
|
if err := c.store.Increment(ctx, name, user, hasUser); err != nil {
|
|
if hasUser {
|
|
log.Error("stats: increment failed", "command", name, "user_id", user.ID, "err", err)
|
|
return
|
|
}
|
|
log.Error("stats: increment failed", "command", name, "err", err)
|
|
}
|
|
}
|
|
|
|
func newCounter(coll storage.Collection) *counter {
|
|
return &counter{store: newUsageStore(coll)}
|
|
}
|
|
|
|
// New is the module Factory. Registers a CommandHook that persists counts and
|
|
// a /stats command that displays them.
|
|
func New(deps modules.Deps) modules.Module {
|
|
c := newCounter(deps.Store)
|
|
return modules.Module{
|
|
CommandHook: c.Inc,
|
|
Commands: []modules.Command{
|
|
statsCommand(c),
|
|
},
|
|
}
|
|
}
|