73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package fqapi
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// HandleMangaImageProxy 处理漫画图片代理请求
|
|
func HandleMangaImageProxy(itemID string, index int) ([]byte, string, error) {
|
|
// 从缓存获取图片信息
|
|
cacheKey := fmt.Sprintf("%s_%d", itemID, index)
|
|
|
|
mangaCacheMutex.RLock()
|
|
imageInfo, exists := mangaImageCache[cacheKey]
|
|
mangaCacheMutex.RUnlock()
|
|
|
|
if !exists {
|
|
return nil, "", fmt.Errorf("图片信息不存在，请先调用 /manga 接口")
|
|
}
|
|
|
|
// 检查缓存是否过期（5分钟）
|
|
if time.Since(imageInfo.CachedAt) > 5*time.Minute {
|
|
// 清理过期缓存
|
|
mangaCacheMutex.Lock()
|
|
delete(mangaImageCache, cacheKey)
|
|
mangaCacheMutex.Unlock()
|
|
return nil, "", fmt.Errorf("图片信息已过期，请重新调用 /manga 接口")
|
|
}
|
|
|
|
// 下载加密图片
|
|
encryptedData, err := downloadImageWithRetry(imageInfo.PicURL, 2)
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("下载图片失败: %w", err)
|
|
}
|
|
|
|
// 解密图片
|
|
keyBytes, err := hex.DecodeString(imageInfo.EncryptKey)
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("密钥解码失败: %w", err)
|
|
}
|
|
|
|
decryptedData, err := decryptImage(encryptedData, keyBytes)
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("解密失败: %w", err)
|
|
}
|
|
|
|
// 识别格式
|
|
format := getImageFormat(decryptedData, imageInfo.PicURL)
|
|
|
|
// 返回图片数据和MIME类型
|
|
mimeType := fmt.Sprintf("image/%s", format)
|
|
if format == "jpg" {
|
|
mimeType = "image/jpeg"
|
|
}
|
|
|
|
return decryptedData, mimeType, nil
|
|
}
|
|
|
|
// CleanExpiredMangaCache 清理过期的漫画图片缓存
|
|
func CleanExpiredMangaCache() {
|
|
mangaCacheMutex.Lock()
|
|
defer mangaCacheMutex.Unlock()
|
|
|
|
now := time.Now()
|
|
for key, info := range mangaImageCache {
|
|
if now.Sub(info.CachedAt) > 5*time.Minute {
|
|
delete(mangaImageCache, key)
|
|
}
|
|
}
|
|
}
|
|
|