275 lines
8.0 KiB
Go
275 lines
8.0 KiB
Go
package plugins
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"trle5.xyz/trbot/utils"
|
|
"trle5.xyz/trbot/utils/configs"
|
|
"trle5.xyz/trbot/utils/flaterr"
|
|
"trle5.xyz/trbot/utils/handler_params"
|
|
"trle5.xyz/trbot/utils/plugin_utils"
|
|
"trle5.xyz/trbot/utils/type/message_utils"
|
|
|
|
"github.com/go-telegram/bot"
|
|
"github.com/go-telegram/bot/models"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
var photoCachedDir string = filepath.Join(configs.CacheDir, "photo/")
|
|
var imageBaseURL string = "https://alist.trle5.xyz/d/cache/photo/"
|
|
|
|
func init() {
|
|
plugin_utils.AddHandlerByMessageTypeHandlers(plugin_utils.ByMessageTypeHandler{
|
|
PluginName: "获取搜图链接",
|
|
ChatType: models.ChatTypePrivate,
|
|
MessageType: message_utils.Photo,
|
|
AllowAutoTrigger: true,
|
|
MessageHandler: searchImageHandler,
|
|
})
|
|
plugin_utils.AddSlashCommandHandlers(plugin_utils.SlashCommand{
|
|
SlashCommand: "searchlinks",
|
|
MessageHandler: sendSearchLinks,
|
|
})
|
|
}
|
|
|
|
type SearchEngines struct {
|
|
Name string `json:"name"`
|
|
URL string `json:"url"`
|
|
}
|
|
|
|
var searchURLs = []SearchEngines{
|
|
{
|
|
Name: "Google",
|
|
URL: "https://www.google.com/searchbyimage?client=app&image_url=%s",
|
|
},
|
|
{
|
|
Name: "Google Lens",
|
|
URL: "https://lens.google.com/uploadbyurl?url=%s",
|
|
},
|
|
{
|
|
Name: "Bing",
|
|
URL: "https://www.bing.com/images/search?q=imgurl:%s&view=detailv2&iss=sbi",
|
|
},
|
|
{
|
|
Name: "Yandex.ru",
|
|
URL: "https://yandex.ru/images/search?rpt=imageview&url=%s",
|
|
},
|
|
{
|
|
Name: "Yandex.com",
|
|
URL: "https://yandex.com/images/search?rpt=imageview&url=%s",
|
|
},
|
|
{
|
|
Name: "SauceNAO",
|
|
URL: "https://saucenao.com/search.php?url=%s",
|
|
},
|
|
{
|
|
Name: "ascii2d",
|
|
URL: "https://ascii2d.net/search/url/%s",
|
|
},
|
|
{
|
|
Name: "Tineye",
|
|
URL: "https://tineye.com/search?url=%s",
|
|
},
|
|
{
|
|
Name: "trace.moe",
|
|
URL: "https://trace.moe/?auto&url=%s",
|
|
},
|
|
// {
|
|
// Name: "IQDB",
|
|
// URL: "http://iqdb.org/?url=%s",
|
|
// },
|
|
// {
|
|
// Name: "3D-IQDB",
|
|
// URL: "http://3d.iqdb.org/?url=%s",
|
|
// },
|
|
}
|
|
|
|
// sendSearchLinks 用于响应 /searchlinks 命令
|
|
func sendSearchLinks(opts *handler_params.Message) error {
|
|
logger := zerolog.Ctx(opts.Ctx).
|
|
With().
|
|
Str("pluginName", "search_images").
|
|
Str(utils.GetCurrentFuncName()).
|
|
Dict(utils.GetUserDict(opts.Message.From)).
|
|
Dict(utils.GetChatDict(&opts.Message.Chat)).
|
|
Logger()
|
|
|
|
wrap := flaterr.NewWrapper(logger.Error())
|
|
|
|
if opts.Message.ReplyToMessage == nil || opts.Message.ReplyToMessage.Photo == nil {
|
|
_, err := opts.Thebot.SendMessage(opts.Ctx, &bot.SendMessageParams{
|
|
ChatID: opts.Message.Chat.ID,
|
|
Text: "使用此命令回复一张图片来获得搜索链接",
|
|
ReplyParameters: &models.ReplyParameters{ MessageID: opts.Message.ID },
|
|
})
|
|
wrap.ErrIf(err).MsgT(flaterr.SendMessage, "need reply to a photo")
|
|
} else {
|
|
photoPath, err := downloadPhoto(opts.Ctx, opts.Thebot, opts.Message.ReplyToMessage)
|
|
if err != nil {
|
|
wrap.Err(err).Msg("error when download photo")
|
|
_, err = opts.Thebot.SendMessage(opts.Ctx, &bot.SendMessageParams{
|
|
ChatID: opts.Message.Chat.ID,
|
|
Text: fmt.Sprintf("缓存图片时发生错误: <blockquote expandable>%s</blockquote>", utils.IgnoreHTMLTags(err.Error())),
|
|
ReplyParameters: &models.ReplyParameters{ MessageID: opts.Message.ReplyToMessage.ID },
|
|
ParseMode: models.ParseModeHTML,
|
|
})
|
|
wrap.ErrIf(err).MsgT(flaterr.SendMessage, "photo cache error")
|
|
} else {
|
|
_, err = opts.Thebot.SendMessage(opts.Ctx, &bot.SendMessageParams{
|
|
ChatID: opts.Message.Chat.ID,
|
|
Text: "选择一个搜索图片的搜索引擎\n此功能灵感来源于 @soutubot",
|
|
ReplyMarkup: buildSearchLinksKeboard(photoPath),
|
|
ReplyParameters: &models.ReplyParameters{ MessageID: opts.Message.ReplyToMessage.ID },
|
|
})
|
|
wrap.ErrIf(err).MsgT(flaterr.SendMessage, "search images link buttons")
|
|
}
|
|
}
|
|
|
|
return wrap.Flat()
|
|
}
|
|
|
|
// searchImageHandler 作为 ByMessageTypeHandler 自动处理图片
|
|
func searchImageHandler(opts *handler_params.Message) error {
|
|
logger := zerolog.Ctx(opts.Ctx).
|
|
With().
|
|
Str("pluginName", "search_images").
|
|
Str(utils.GetCurrentFuncName()).
|
|
Dict(utils.GetUserDict(opts.Message.From)).
|
|
Logger()
|
|
|
|
wrap := flaterr.NewWrapper(logger.Error())
|
|
|
|
photoPath, err := downloadPhoto(opts.Ctx, opts.Thebot, opts.Message)
|
|
if err != nil {
|
|
wrap.Err(err).Str("path", photoPath).Msg("Error when cache photo")
|
|
_, err = opts.Thebot.SendMessage(opts.Ctx, &bot.SendMessageParams{
|
|
ChatID: opts.Message.Chat.ID,
|
|
Text: fmt.Sprintf("缓存图片时发生错误: <blockquote expandable>%s</blockquote>", utils.IgnoreHTMLTags(err.Error())),
|
|
ReplyParameters: &models.ReplyParameters{ MessageID: opts.Message.ID },
|
|
ParseMode: models.ParseModeHTML,
|
|
})
|
|
wrap.ErrIf(err).MsgT(flaterr.SendMessage, "photo cache error")
|
|
return wrap.Flat()
|
|
}
|
|
|
|
_, err = opts.Thebot.SendMessage(opts.Ctx, &bot.SendMessageParams{
|
|
ChatID: opts.Message.Chat.ID,
|
|
Text: "选择一个搜索图片的搜索引擎\n此功能灵感来源于 @soutubot",
|
|
ReplyMarkup: buildSearchLinksKeboard(photoPath),
|
|
ReplyParameters: &models.ReplyParameters{ MessageID: opts.Message.ID },
|
|
})
|
|
wrap.ErrIf(err).MsgT(flaterr.SendMessage, "search images link buttons")
|
|
|
|
return wrap.Flat()
|
|
}
|
|
|
|
func downloadPhoto(ctx context.Context, thebot *bot.Bot, msg *models.Message) (string, error) {
|
|
logger := zerolog.Ctx(ctx).
|
|
With().
|
|
Str("pluginName", "search_images").
|
|
Str(utils.GetCurrentFuncName()).
|
|
Logger()
|
|
|
|
wrap := flaterr.NewWrapper(logger.Error())
|
|
|
|
var photoFileName string = fmt.Sprintf("%d-%s.jpg", msg.From.ID, msg.Photo[len(msg.Photo)-1].FileID)
|
|
var photoFullPath string = filepath.Join(photoCachedDir, photoFileName)
|
|
|
|
_, err := os.Stat(photoFullPath) // 检查图片源文件是否已缓存
|
|
if err != nil {
|
|
// 如果图片源文件未缓存，则下载
|
|
if os.IsNotExist(err) {
|
|
fileInfo, err := thebot.GetFile(ctx, &bot.GetFileParams{
|
|
FileID: msg.Photo[len(msg.Photo)-1].FileID,
|
|
})
|
|
if err != nil {
|
|
wrap.Err(err).
|
|
Str("fileID", msg.Photo[len(msg.Photo)-1].FileID).
|
|
MsgT(flaterr.GetFile, "photo file")
|
|
return "", wrap.Flat()
|
|
} else {
|
|
// 组合链接下载图片源文件
|
|
resp, err := http.Get(thebot.FileDownloadLink(fileInfo))
|
|
if err != nil {
|
|
wrap.Err(err).
|
|
Str("filePath", fileInfo.FilePath).
|
|
Msg("Failed to download photo file")
|
|
return "", wrap.Flat()
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// 创建目录
|
|
err = os.MkdirAll(photoCachedDir, 0755)
|
|
if err != nil {
|
|
wrap.Err(err).
|
|
Str("photoCachedDir", photoCachedDir).
|
|
Msg("Failed to create directory to cache photo")
|
|
return "", wrap.Flat()
|
|
}
|
|
|
|
downloadedPhoto, err := os.Create(photoFullPath)
|
|
if err != nil {
|
|
wrap.Err(err).
|
|
Str("photoFullPath", photoFullPath).
|
|
Msg("Failed to create photo file")
|
|
return "", wrap.Flat()
|
|
}
|
|
defer downloadedPhoto.Close()
|
|
|
|
// 将下载的原图片写入空文件
|
|
_, err = io.Copy(downloadedPhoto, resp.Body)
|
|
if err != nil {
|
|
wrap.Err(err).
|
|
Str("photoFullPath", photoFullPath).
|
|
Msg("Failed to writing photo data to file")
|
|
return "", wrap.Flat()
|
|
}
|
|
}
|
|
} else {
|
|
wrap.Err(err).
|
|
Str("photoFullPath", photoFullPath).
|
|
Msg("Failed to read cached photo file info")
|
|
return "", wrap.Flat()
|
|
}
|
|
} else {
|
|
// 文件已存在，跳过下载
|
|
logger.Debug().
|
|
Str("photoFullPath", photoFullPath).
|
|
Msg("photo file already cached")
|
|
}
|
|
|
|
return photoFileName, nil
|
|
}
|
|
|
|
func buildSearchLinksKeboard(photoPath string) models.ReplyMarkup {
|
|
var button [][]models.InlineKeyboardButton
|
|
var tempButton []models.InlineKeyboardButton
|
|
|
|
for _, url := range searchURLs {
|
|
if len (tempButton) >= 3 {
|
|
button = append(button, tempButton)
|
|
tempButton = []models.InlineKeyboardButton{}
|
|
}
|
|
tempButton = append(tempButton, models.InlineKeyboardButton{
|
|
Text: url.Name,
|
|
URL: fmt.Sprintf(url.URL, imageBaseURL + photoPath),
|
|
})
|
|
}
|
|
|
|
if len(tempButton) > 0 { button = append(button, tempButton) }
|
|
|
|
button = append(button, []models.InlineKeyboardButton{{
|
|
Text: "🚫 关闭菜单",
|
|
CallbackData: "delete_this_message",
|
|
}})
|
|
|
|
return &models.InlineKeyboardMarkup{
|
|
InlineKeyboard: button,
|
|
}
|
|
}
|