114 lines
3.3 KiB
Go
114 lines
3.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/celestix/gotgproto/ext"
|
|
"github.com/charmbracelet/log"
|
|
"github.com/duke-git/lancet/v2/slice"
|
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
|
"github.com/krau/SaveAny-Bot/common/i18n"
|
|
"github.com/krau/SaveAny-Bot/common/i18n/i18nk"
|
|
"github.com/krau/SaveAny-Bot/config"
|
|
"github.com/krau/SaveAny-Bot/pkg/aria2"
|
|
"github.com/krau/SaveAny-Bot/pkg/enums/tasktype"
|
|
"github.com/krau/SaveAny-Bot/pkg/tcbdata"
|
|
"github.com/krau/SaveAny-Bot/storage"
|
|
)
|
|
|
|
func handleDlCmd(ctx *ext.Context, update *ext.Update) error {
|
|
logger := log.FromContext(ctx)
|
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
|
if len(args) < 2 {
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlUsage)), nil)
|
|
return nil
|
|
}
|
|
links := args[1:]
|
|
for i, link := range links {
|
|
links[i] = strings.TrimSpace(link)
|
|
u, err := url.Parse(link)
|
|
if err != nil || u.Scheme == "" || u.Host == "" {
|
|
logger.Warn("invaild link", link)
|
|
links[i] = ""
|
|
}
|
|
}
|
|
links = slice.Compact(links)
|
|
if len(links) == 0 {
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlErrorNoValidLinks)), nil)
|
|
return nil
|
|
}
|
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, update.GetUserChat().GetID()), tcbdata.Add{
|
|
TaskType: tasktype.TaskTypeDirectlinks,
|
|
DirectLinks: links,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlInfoFilesSelectStorage, map[string]any{
|
|
"Count": len(links),
|
|
})), &ext.ReplyOpts{
|
|
Markup: markup,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
var aria2ClientInitOnce sync.Once
|
|
var aria2ClientInitErr error
|
|
var aria2Client *aria2.Client
|
|
|
|
// GetAria2Client returns the shared aria2 client instance
|
|
func GetAria2Client() *aria2.Client {
|
|
return aria2Client
|
|
}
|
|
|
|
func handleAria2DlCmd(ctx *ext.Context, update *ext.Update) error {
|
|
if !config.C().Aria2.Enable {
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2ErrorAria2NotEnabled)), nil)
|
|
return nil
|
|
}
|
|
logger := log.FromContext(ctx)
|
|
args := strings.Split(update.EffectiveMessage.Text, " ")
|
|
if len(args) < 2 {
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlUsage)), nil)
|
|
return nil
|
|
}
|
|
links := args[1:]
|
|
for i, link := range links {
|
|
links[i] = strings.TrimSpace(link)
|
|
}
|
|
links = slice.Compact(links)
|
|
if len(links) == 0 {
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgDlErrorNoValidLinks)), nil)
|
|
return nil
|
|
}
|
|
logger.Debug("Preparing aria2 download", "links", links)
|
|
|
|
// Initialize aria2 client to check connection
|
|
aria2ClientInitOnce.Do(func() {
|
|
aria2Client, aria2ClientInitErr = aria2.NewClient(config.C().Aria2.Url, config.C().Aria2.Secret)
|
|
})
|
|
if aria2ClientInitErr != nil {
|
|
logger.Error("Failed to initialize aria2 client", "error", aria2ClientInitErr)
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2ErrorAria2ClientInitFailed, map[string]any{
|
|
"Error": aria2ClientInitErr.Error(),
|
|
})), nil)
|
|
return nil
|
|
}
|
|
|
|
// Build storage selection keyboard (don't add to aria2 yet)
|
|
markup, err := msgelem.BuildAddSelectStorageKeyboard(storage.GetUserStorages(ctx, update.GetUserChat().GetID()), tcbdata.Add{
|
|
TaskType: tasktype.TaskTypeAria2,
|
|
Aria2URIs: links,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgAria2InfoSelectStorage)), &ext.ReplyOpts{
|
|
Markup: markup,
|
|
})
|
|
return nil
|
|
}
|