Init

bot/handlers/inline.go

@@ -0,0 +1,91 @@
+package handlers
+
+import (
+	"context"
+	"govd/bot/core"
+	"govd/models"
+	"govd/util"
+	"strings"
+	"time"
+
+	extractors "govd/ext"
+
+	"github.com/PaulSonOfLars/gotgbot/v2"
+	"github.com/PaulSonOfLars/gotgbot/v2/ext"
+)
+
+func InlineDownloadHandler(
+	bot *gotgbot.Bot,
+	ctx *ext.Context,
+) error {
+	url := strings.TrimSpace(ctx.InlineQuery.Query)
+	if url == "" {
+		ctx.InlineQuery.Answer(bot, []gotgbot.InlineQueryResult{}, &gotgbot.AnswerInlineQueryOpts{
+			CacheTime:  1,
+			IsPersonal: true,
+		})
+		return nil
+	}
+	dlCtx, err := extractors.CtxByURL(url)
+	if err != nil || dlCtx == nil || dlCtx.Extractor == nil {
+		ctx.InlineQuery.Answer(bot, []gotgbot.InlineQueryResult{}, &gotgbot.AnswerInlineQueryOpts{
+			CacheTime:  1,
+			IsPersonal: true,
+		})
+		return nil
+	}
+	return core.HandleInline(bot, ctx, dlCtx)
+}
+
+func InlineDownloadResultHandler(
+	bot *gotgbot.Bot,
+	ctx *ext.Context,
+) error {
+	dlCtx, ok := core.InlineTasks[ctx.ChosenInlineResult.ResultId]
+	if !ok {
+		return nil
+	}
+	defer delete(core.InlineTasks, ctx.ChosenInlineResult.ResultId)
+
+	mediaChan := make(chan *models.Media, 1)
+	errChan := make(chan error, 1)
+	timeout, cancel := context.WithTimeout(
+		context.Background(),
+		5*time.Minute,
+	)
+	defer cancel()
+
+	go core.GetInlineFormat(
+		bot, ctx, dlCtx,
+		mediaChan, errChan,
+	)
+	select {
+	case media := <-mediaChan:
+		err := core.HandleInlineCachedResult(
+			bot, ctx,
+			dlCtx, media,
+		)
+		if err != nil {
+			core.HandleErrorMessage(bot, ctx, err)
+			return nil
+		}
+	case err := <-errChan:
+		core.HandleErrorMessage(bot, ctx, err)
+		return nil
+	case <-timeout.Done():
+		core.HandleErrorMessage(bot, ctx, util.ErrTimeout)
+		return nil
+	}
+	return nil
+}
+
+func InlineLoadingHandler(
+	bot *gotgbot.Bot,
+	ctx *ext.Context,
+) error {
+	ctx.CallbackQuery.Answer(bot, &gotgbot.AnswerCallbackQueryOpts{
+		Text:      "wait !",
+		ShowAlert: true,
+	})
+	return nil
+}