top20

Author: zenlor

internal/core/handlers.go

@@ -10,6 +10,7 @@ import (
 	"slices"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/moolite/bot/internal/db"
 	"github.com/moolite/bot/internal/dicer"
@@ -35,7 +36,9 @@ func registerCommands(ctx context.Context, b *tg.Bot) error {
 			// Search
 			{Command: "search", Description: "Search media files"},
 			// Top
-			{Command: "top", Description: "Top 10 media files"},
+			{Command: "top10", Description: "Top 10 media files"},
+			// Top20
+			{Command: "top20", Description: "Top 20 media files"},
 			// grumpyness
 			{Command: "grumpyness", Description: "Grumpyness measuring instrument"},
 		},
@@ -59,9 +62,15 @@ func registerBotHandlers(_ context.Context, b *tg.Bot) {
 		&tg.UpdateHandler{
 			Type:    tg.UPD_STARTSWITH,
 			Param:   "/top",
-			Aliases: []string{"/t"},
+			Aliases: []string{"/t", "/top10"},
 			Fn:      MediaToptenCommand,
 		},
+		&tg.UpdateHandler{
+			Type:    tg.UPD_STARTSWITH,
+			Param:   "/tte",
+			Aliases: []string{"/t2", "/top20"},
+			Fn:      MediaToptwentyCommand,
+		},
 		&tg.UpdateHandler{
 			Type:    tg.UPD_STARTSWITH,
 			Param:   "/pot",
@@ -560,6 +569,25 @@ func MediaBottomtenCommand(ctx context.Context, b *tg.Bot, update *tg.Update) (*
 	return snd, nil
 }
 
+func MediaToptwentyCommand(ctx context.Context, b *tg.Bot, update *tg.Update) (*tg.Sendable, error) {
+	media, err := db.SelectMediaTop(ctx, update.Message.Chat.ID, 20)
+	if err != nil {
+		return nil, err
+	}
+
+	go func() {
+		time.Sleep(time.Millisecond * 200) // wait a reasonable 1/5 of a second
+		snd := mediaCollection(update.Message.Chat.ID, media[10].Description, media[10:])
+
+		if _, err := b.Send(context.Background(), snd); err != nil {
+			slog.Error("error in bot.Send", "err", err)
+		}
+	}()
+
+	snd := mediaCollection(update.Message.Chat.ID, media[0].Description, media[:10])
+	return snd, nil
+}
+
 func MediaRememberCommand(ctx context.Context, b *tg.Bot, update *tg.Update) (*tg.Sendable, error) {
 	kind := ""
 	data := ""