6e16dc6051
- Updated silent.go to replace hardcoded strings with i18n keys for user feedback. - Enhanced telegraph.go to use i18n for error messages and prompts. - Modified update.go to implement i18n for version update notifications and errors. - Refactored dir.go to utilize i18n for help messages related to directory operations. - Updated parse.go to replace static text with i18n keys for parsed text entities. - Enhanced rule.go to use i18n for rule management help messages. - Refactored storage.go to implement i18n for storage selection prompts. - Updated task.go to utilize i18n for task addition notifications. - Added new i18n keys in keys.go for various messages. - Updated Chinese localization in zh-Hans.yaml to reflect new i18n keys and messages.
102 lines
3.3 KiB
Go
102 lines
3.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
|
|
"github.com/celestix/gotgproto/dispatcher"
|
|
"github.com/celestix/gotgproto/ext"
|
|
"github.com/gotd/td/tg"
|
|
"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/parsers"
|
|
)
|
|
|
|
func handleParserCmd(ctx *ext.Context, u *ext.Update) error {
|
|
args := strings.Split(u.EffectiveMessage.Text, " ")
|
|
help := i18n.T(i18nk.BotMsgParserHelpText, nil)
|
|
if len(args) < 2 {
|
|
ctx.Reply(u, ext.ReplyTextString(help), nil)
|
|
return nil
|
|
}
|
|
switch args[1] {
|
|
// case "list":
|
|
// return handleParserListCmd(ctx, u)
|
|
case "install":
|
|
return handleParserInstallCmd(ctx, u)
|
|
// case "uninstall":
|
|
// return handleParserUninstallCmd(ctx, u)
|
|
default:
|
|
}
|
|
return dispatcher.EndGroups
|
|
}
|
|
|
|
func handleParserInstallCmd(ctx *ext.Context, u *ext.Update) error {
|
|
if !config.C().Parser.PluginEnable {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserPluginNotEnabled, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
if u.EffectiveMessage.ReplyToMessage == nil || u.EffectiveMessage.ReplyToMessage.Media == nil {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserPromptReplyWithParserFile, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
media := u.EffectiveMessage.ReplyToMessage.Media
|
|
document, ok := media.(*tg.MessageMediaDocument)
|
|
if !ok {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
value, ok := document.GetDocument()
|
|
if !ok {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
doc, ok := value.AsNotEmpty()
|
|
if !ok {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorNoValidFileInReply, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
if !strings.HasPrefix(doc.MimeType, "text/") {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorWrongFileType, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
if doc.Size > 1024*1024*10 {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorFileTooLarge, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
var fileName string
|
|
for _, attr := range doc.Attributes {
|
|
if fileNameAttr, ok := attr.(*tg.DocumentAttributeFilename); ok {
|
|
fileName = fileNameAttr.FileName
|
|
break
|
|
}
|
|
}
|
|
if fileName == "" {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorGetFilenameFailed, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
if !strings.HasSuffix(fileName, ".js") {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorOnlyJsSupported, nil)), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
data := bytes.NewBuffer(nil)
|
|
_, err := ctx.DownloadMedia(media, ext.DownloadOutputStream{Writer: data}, nil)
|
|
if err != nil {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorDownloadFileFailed, map[string]any{
|
|
"Error": err.Error(),
|
|
})), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
if err := parsers.AddPlugin(ctx, data.String(), fileName); err != nil {
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserErrorInstallPluginFailed, map[string]any{
|
|
"Error": err.Error(),
|
|
})), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
ctx.Reply(u, ext.ReplyTextString(i18n.T(i18nk.BotMsgParserInfoInstallPluginSuccess, map[string]any{
|
|
"Name": fileName,
|
|
})), nil)
|
|
return dispatcher.EndGroups
|
|
}
|