Init

bot/handlers/stats.go

@@ -0,0 +1,89 @@
+package handlers
+
+import (
+	"fmt"
+	"govd/database"
+	"time"
+
+	"github.com/PaulSonOfLars/gotgbot/v2"
+	"github.com/PaulSonOfLars/gotgbot/v2/ext"
+)
+
+type Stats struct {
+	TotalUsers      int64
+	TotalGroups     int64
+	TotalDailyUsers int64
+	TotalMedia      int64
+	UpdatedAt       time.Time
+}
+
+var lastSavedStats *Stats
+
+var statsMessage = "users: %d\nusers today: %d\ngroups: %d\ndownloads: %d\n\nupdates every 10 minutes"
+
+func StatsHandler(bot *gotgbot.Bot, ctx *ext.Context) error {
+	if ctx.EffectiveMessage.Chat.Type != "private" {
+		return nil
+	}
+	ctx.CallbackQuery.Answer(bot, nil)
+	stats := GetStats()
+	ctx.EffectiveMessage.EditText(
+		bot,
+		fmt.Sprintf(
+			statsMessage,
+			stats.TotalUsers,
+			stats.TotalDailyUsers,
+			stats.TotalGroups,
+			stats.TotalMedia,
+		),
+		&gotgbot.EditMessageTextOpts{
+			ReplyMarkup: gotgbot.InlineKeyboardMarkup{
+				InlineKeyboard: [][]gotgbot.InlineKeyboardButton{
+					{
+						{
+							Text:         "back",
+							CallbackData: "start",
+						},
+					},
+				},
+			},
+		},
+	)
+	return nil
+}
+
+func UpdateStats() {
+	totalUsers, err := database.GetUsersCount()
+	if err != nil {
+		return
+	}
+	totalGroups, err := database.GetGroupsCount()
+	if err != nil {
+		return
+	}
+	totalDailyUsers, err := database.GetDailyUserCount()
+	if err != nil {
+		return
+	}
+	totalMedia, err := database.GetMediaCount()
+	if err != nil {
+		return
+	}
+	lastSavedStats = &Stats{
+		TotalUsers:      totalUsers,
+		TotalGroups:     totalGroups,
+		TotalDailyUsers: totalDailyUsers,
+		TotalMedia:      totalMedia,
+		UpdatedAt:       time.Now(),
+	}
+}
+
+func GetStats() *Stats {
+	if lastSavedStats == nil {
+		UpdateStats()
+	}
+	if lastSavedStats.UpdatedAt.Add(10 * time.Minute).Before(time.Now()) {
+		UpdateStats()
+	}
+	return lastSavedStats
+}