mirror of
https://git.fightbot.fun/hxuanyu/BingPaper.git
synced 2026-02-15 07:29:33 +08:00
569 lines
17 KiB
Go
569 lines
17 KiB
Go
package handlers
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
|
||
"BingPaper/internal/config"
|
||
"BingPaper/internal/model"
|
||
"BingPaper/internal/service/image"
|
||
"BingPaper/internal/storage"
|
||
"BingPaper/internal/util"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
type ImageVariantResp struct {
|
||
Variant string `json:"variant"`
|
||
Format string `json:"format"`
|
||
Size int64 `json:"size"`
|
||
URL string `json:"url"`
|
||
StorageKey string `json:"storage_key"`
|
||
}
|
||
|
||
type ImageMetaResp struct {
|
||
Date string `json:"date"`
|
||
Mkt string `json:"mkt"`
|
||
Title string `json:"title"`
|
||
Copyright string `json:"copyright"`
|
||
CopyrightLink string `json:"copyrightlink"`
|
||
Quiz string `json:"quiz"`
|
||
StartDate string `json:"startdate"`
|
||
FullStartDate string `json:"fullstartdate"`
|
||
HSH string `json:"hsh"`
|
||
Variants []ImageVariantResp `json:"variants"`
|
||
}
|
||
|
||
// GetToday 获取今日图片
|
||
// @Summary 获取今日图片
|
||
// @Description 根据参数返回今日必应图片流或重定向
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Param variant query string false "分辨率 (UHD, 1920x1080, 1366x768, 1280x720, 1024x768, 800x600, 800x480, 640x480, 640x360, 480x360, 400x240, 320x240)" default(UHD)
|
||
// @Param format query string false "格式 (jpg)" default(jpg)
|
||
// @Produce image/jpeg
|
||
// @Success 200 {file} binary
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/today [get]
|
||
func GetToday(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetTodayImage(mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
handleImageResponse(c, imgRegion, 7200) // 2小时
|
||
}
|
||
|
||
// GetTodayMeta 获取今日图片元数据
|
||
// @Summary 获取今日图片元数据
|
||
// @Description 获取今日必应图片的标题、版权等元数据
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {object} ImageMetaResp
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/today/meta [get]
|
||
func GetTodayMeta(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetTodayImage(mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "public, max-age=7200") // 2小时
|
||
c.JSON(http.StatusOK, formatMeta(imgRegion))
|
||
}
|
||
|
||
// GetRandom 获取随机图片
|
||
// @Summary 获取随机图片
|
||
// @Description 随机返回一张已抓取的图片流或重定向
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Param variant query string false "分辨率" default(UHD)
|
||
// @Param format query string false "格式" default(jpg)
|
||
// @Produce image/jpeg
|
||
// @Success 200 {file} binary
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/random [get]
|
||
// GetRandom 获取随机图片
|
||
func GetRandom(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetRandomImage(mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
handleImageResponse(c, imgRegion, 0) // 禁用缓存
|
||
}
|
||
|
||
// GetRandomMeta 获取随机图片元数据
|
||
// @Summary 获取随机图片元数据
|
||
// @Description 随机获取一张已抓取图片的元数据
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {object} ImageMetaResp
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/random/meta [get]
|
||
func GetRandomMeta(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetRandomImage(mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
c.JSON(http.StatusOK, formatMeta(imgRegion))
|
||
}
|
||
|
||
// GetByDate 获取指定日期图片
|
||
// @Summary 获取指定日期图片
|
||
// @Description 根据日期返回图片流或重定向 (yyyy-mm-dd)
|
||
// @Tags image
|
||
// @Param date path string true "日期 (yyyy-mm-dd)"
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Param variant query string false "分辨率" default(UHD)
|
||
// @Param format query string false "格式" default(jpg)
|
||
// @Produce image/jpeg
|
||
// @Success 200 {file} binary
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/date/{date} [get]
|
||
func GetByDate(c *gin.Context) {
|
||
date := c.Param("date")
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetImageByDate(date, mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
handleImageResponse(c, imgRegion, 604800) // 7天
|
||
}
|
||
|
||
// GetByDateMeta 获取指定日期图片元数据
|
||
// @Summary 获取指定日期图片元数据
|
||
// @Description 根据日期获取图片元数据 (yyyy-mm-dd)
|
||
// @Tags image
|
||
// @Param date path string true "日期 (yyyy-mm-dd)"
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {object} ImageMetaResp
|
||
// @Success 202 {object} map[string]string "按需抓取任务已启动"
|
||
// @Failure 404 {object} map[string]string "图片未找到，响应体包含具体原因"
|
||
// @Router /image/date/{date}/meta [get]
|
||
func GetByDateMeta(c *gin.Context) {
|
||
date := c.Param("date")
|
||
mkt := c.Query("mkt")
|
||
imgRegion, err := image.GetImageByDate(date, mkt)
|
||
if err == image.ErrFetchStarted {
|
||
c.JSON(http.StatusAccepted, gin.H{"message": fmt.Sprintf("On-demand fetch started for region [%s]. Please try again later.", mkt)})
|
||
return
|
||
}
|
||
if err != nil {
|
||
sendImageNotFound(c, mkt)
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "public, max-age=604800") // 7天
|
||
c.JSON(http.StatusOK, formatMeta(imgRegion))
|
||
}
|
||
|
||
// ListImages 获取图片列表
|
||
// @Summary 获取图片列表
|
||
// @Description 分页获取已抓取的图片元数据列表。支持分页(page, page_size)、限制数量(limit)和按月份过滤(month, 格式: YYYY-MM)。
|
||
// @Tags image
|
||
// @Param limit query int false "限制数量 (如果不使用分页)" default(30)
|
||
// @Param page query int false "页码 (从1开始)"
|
||
// @Param page_size query int false "每页数量"
|
||
// @Param month query string false "按月份过滤 (格式: YYYY-MM)"
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {array} ImageMetaResp
|
||
// @Router /images [get]
|
||
func ListImages(c *gin.Context) {
|
||
limitStr := c.Query("limit")
|
||
pageStr := c.Query("page")
|
||
pageSizeStr := c.Query("page_size")
|
||
month := c.Query("month")
|
||
mkt := c.Query("mkt")
|
||
|
||
// 记录请求参数，便于排查过滤失效问题
|
||
util.Logger.Debug("ListImages parameters",
|
||
zap.String("month", month),
|
||
zap.String("mkt", mkt),
|
||
zap.String("page", pageStr),
|
||
zap.String("page_size", pageSizeStr),
|
||
zap.String("limit", limitStr))
|
||
|
||
var limit, offset int
|
||
|
||
if pageStr != "" && pageSizeStr != "" {
|
||
page, _ := strconv.Atoi(pageStr)
|
||
pageSize, _ := strconv.Atoi(pageSizeStr)
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if pageSize < 1 {
|
||
pageSize = 30
|
||
}
|
||
limit = pageSize
|
||
offset = (page - 1) * pageSize
|
||
} else {
|
||
if limitStr == "" {
|
||
limit = 30
|
||
} else {
|
||
limit, _ = strconv.Atoi(limitStr)
|
||
}
|
||
offset = 0
|
||
}
|
||
|
||
images, err := image.GetImageList(limit, offset, month, mkt)
|
||
if err != nil {
|
||
util.Logger.Error("ListImages service call failed", zap.Error(err))
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
result := []gin.H{}
|
||
for _, img := range images {
|
||
result = append(result, formatMetaSummary(&img))
|
||
}
|
||
c.JSON(http.StatusOK, result)
|
||
}
|
||
|
||
// ListGlobalTodayImages 获取所有地区的今日图片列表
|
||
// @Summary 获取所有地区的今日图片列表
|
||
// @Description 获取配置文件中所有已开启地区的今日必应图片元数据（缩略图）
|
||
// @Tags image
|
||
// @Produce json
|
||
// @Success 200 {array} ImageMetaResp
|
||
// @Router /images/global/today [get]
|
||
func ListGlobalTodayImages(c *gin.Context) {
|
||
images, err := image.GetAllRegionsTodayImages()
|
||
if err != nil {
|
||
util.Logger.Error("ListGlobalTodayImages service call failed", zap.Error(err))
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||
return
|
||
}
|
||
|
||
result := []gin.H{}
|
||
for _, img := range images {
|
||
result = append(result, formatMetaSummary(&img))
|
||
}
|
||
c.JSON(http.StatusOK, result)
|
||
}
|
||
|
||
func sendImageNotFound(c *gin.Context, mkt string) {
|
||
cfg := config.GetConfig().API
|
||
message := "image not found"
|
||
|
||
if mkt != "" {
|
||
reasons := []string{}
|
||
if !util.IsValidRegion(mkt) {
|
||
reasons = append(reasons, fmt.Sprintf("[%s] is not a standard region code", mkt))
|
||
} else {
|
||
if !cfg.EnableOnDemandFetch {
|
||
reasons = append(reasons, "on-demand fetch is disabled")
|
||
}
|
||
if !cfg.EnableMktFallback {
|
||
reasons = append(reasons, "region fallback is disabled")
|
||
}
|
||
}
|
||
|
||
if len(reasons) > 0 {
|
||
message = fmt.Sprintf("Image not found for region [%s]. Reasons: %s.", mkt, strings.Join(reasons, ", "))
|
||
} else {
|
||
message = fmt.Sprintf("Image not found for region [%s] even after on-demand fetch and fallback attempts.", mkt)
|
||
}
|
||
}
|
||
|
||
c.JSON(http.StatusNotFound, gin.H{"error": message})
|
||
}
|
||
|
||
func handleImageResponse(c *gin.Context, m *model.ImageRegion, maxAge int) {
|
||
variant := c.DefaultQuery("variant", "UHD")
|
||
format := c.DefaultQuery("format", "jpg")
|
||
|
||
var selected *model.ImageVariant
|
||
for _, v := range m.Variants {
|
||
if v.Variant == variant && v.Format == format {
|
||
selected = &v
|
||
break
|
||
}
|
||
}
|
||
|
||
if selected == nil && len(m.Variants) > 0 {
|
||
// 回退逻辑
|
||
selected = &m.Variants[0]
|
||
}
|
||
|
||
if selected == nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "variant not found"})
|
||
return
|
||
}
|
||
|
||
mode := config.GetConfig().API.Mode
|
||
if mode == "redirect" {
|
||
if selected.PublicURL != "" {
|
||
if maxAge > 0 {
|
||
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%d", maxAge))
|
||
} else {
|
||
c.Header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
}
|
||
c.Redirect(http.StatusFound, selected.PublicURL)
|
||
} else if m.URLBase != "" {
|
||
// 兜底重定向到原始 Bing
|
||
bingURL := fmt.Sprintf("https://www.bing.com%s_%s.jpg", m.URLBase, selected.Variant)
|
||
if maxAge > 0 {
|
||
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%d", maxAge))
|
||
} else {
|
||
c.Header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
}
|
||
c.Redirect(http.StatusFound, bingURL)
|
||
} else {
|
||
serveLocal(c, selected.StorageKey, m.Date, maxAge)
|
||
}
|
||
} else {
|
||
serveLocal(c, selected.StorageKey, m.Date, maxAge)
|
||
}
|
||
}
|
||
|
||
func serveLocal(c *gin.Context, key string, etag string, maxAge int) {
|
||
if etag != "" {
|
||
c.Header("ETag", fmt.Sprintf("\"%s\"", etag))
|
||
if c.GetHeader("If-None-Match") == fmt.Sprintf("\"%s\"", etag) {
|
||
c.AbortWithStatus(http.StatusNotModified)
|
||
return
|
||
}
|
||
}
|
||
|
||
reader, contentType, err := storage.GlobalStorage.Get(context.Background(), key)
|
||
if err != nil {
|
||
util.Logger.Error("Failed to get image from storage", zap.String("key", key), zap.Error(err))
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to get image"})
|
||
return
|
||
}
|
||
defer reader.Close()
|
||
|
||
if contentType != "" {
|
||
c.Header("Content-Type", contentType)
|
||
}
|
||
|
||
if maxAge > 0 {
|
||
c.Header("Cache-Control", fmt.Sprintf("public, max-age=%d", maxAge))
|
||
} else {
|
||
c.Header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
}
|
||
io.Copy(c.Writer, reader)
|
||
}
|
||
|
||
func formatMetaSummary(m *model.ImageRegion) gin.H {
|
||
cfg := config.GetConfig()
|
||
|
||
// 找到最小的变体
|
||
var smallest *model.ImageVariant
|
||
for i := range m.Variants {
|
||
v := &m.Variants[i]
|
||
if smallest == nil {
|
||
smallest = v
|
||
continue
|
||
}
|
||
|
||
// 如果当前变体 Size 更小且不为 0，或者 smallest 的 Size 为 0
|
||
if v.Size > 0 && (smallest.Size == 0 || v.Size < smallest.Size) {
|
||
smallest = v
|
||
} else if v.Size == smallest.Size {
|
||
// 如果 Size 相同（包括都为 0），根据分辨率名称判断
|
||
if compareResolution(v.Variant, smallest.Variant) < 0 {
|
||
smallest = v
|
||
}
|
||
}
|
||
}
|
||
|
||
variants := []gin.H{}
|
||
if smallest != nil {
|
||
url := smallest.PublicURL
|
||
if url == "" && cfg.API.Mode == "redirect" && m.URLBase != "" {
|
||
url = fmt.Sprintf("https://www.bing.com%s_%s.jpg", m.URLBase, smallest.Variant)
|
||
} else if cfg.API.Mode == "local" || url == "" {
|
||
url = fmt.Sprintf("%s/api/v1/image/date/%s?variant=%s&format=%s&mkt=%s", cfg.Server.BaseURL, m.Date, smallest.Variant, smallest.Format, m.Mkt)
|
||
}
|
||
variants = append(variants, gin.H{
|
||
"variant": smallest.Variant,
|
||
"format": smallest.Format,
|
||
"size": smallest.Size,
|
||
"url": url,
|
||
"storage_key": smallest.StorageKey,
|
||
})
|
||
}
|
||
|
||
return gin.H{
|
||
"date": m.Date,
|
||
"mkt": m.Mkt,
|
||
"title": m.Title,
|
||
"copyright": m.Copyright,
|
||
"copyrightlink": m.CopyrightLink,
|
||
"quiz": m.Quiz,
|
||
"startdate": m.StartDate,
|
||
"fullstartdate": m.FullStartDate,
|
||
"hsh": m.HSH,
|
||
"variants": variants,
|
||
}
|
||
}
|
||
|
||
func formatMeta(m *model.ImageRegion) gin.H {
|
||
cfg := config.GetConfig()
|
||
variants := []gin.H{}
|
||
for _, v := range m.Variants {
|
||
url := v.PublicURL
|
||
if url == "" && cfg.API.Mode == "redirect" && m.URLBase != "" {
|
||
url = fmt.Sprintf("https://www.bing.com%s_%s.jpg", m.URLBase, v.Variant)
|
||
} else if cfg.API.Mode == "local" || url == "" {
|
||
url = fmt.Sprintf("%s/api/v1/image/date/%s?variant=%s&format=%s&mkt=%s", cfg.Server.BaseURL, m.Date, v.Variant, v.Format, m.Mkt)
|
||
}
|
||
variants = append(variants, gin.H{
|
||
"variant": v.Variant,
|
||
"format": v.Format,
|
||
"size": v.Size,
|
||
"url": url,
|
||
"storage_key": v.StorageKey,
|
||
})
|
||
}
|
||
|
||
return gin.H{
|
||
"date": m.Date,
|
||
"mkt": m.Mkt,
|
||
"title": m.Title,
|
||
"copyright": m.Copyright,
|
||
"copyrightlink": m.CopyrightLink,
|
||
"quiz": m.Quiz,
|
||
"startdate": m.StartDate,
|
||
"fullstartdate": m.FullStartDate,
|
||
"hsh": m.HSH,
|
||
"variants": variants,
|
||
}
|
||
}
|
||
|
||
// GetRegions 获取支持的地区列表
|
||
// @Summary 获取支持的地区列表
|
||
// @Description 返回系统支持的所有必应地区编码及标签。如果配置中指定了抓取地区，这些地区将排在列表最前面（置顶）。
|
||
// @Tags image
|
||
// @Produce json
|
||
// @Success 200 {array} util.Region
|
||
// @Router /regions [get]
|
||
func GetRegions(c *gin.Context) {
|
||
cfg := config.GetConfig()
|
||
pinned := cfg.Fetcher.Regions
|
||
|
||
if len(pinned) == 0 {
|
||
// 如果没有配置抓取地区，返回所有支持的地区
|
||
c.JSON(http.StatusOK, util.AllRegions)
|
||
return
|
||
}
|
||
|
||
// 创建一个 Map 用于快速查找配置的地区
|
||
pinnedMap := make(map[string]bool)
|
||
for _, v := range pinned {
|
||
pinnedMap[v] = true
|
||
}
|
||
|
||
// 只返回配置中的地区，并保持配置中的顺序
|
||
var result []util.Region
|
||
// 为了保持配置顺序，我们遍历 pinned 而不是 AllRegions
|
||
for _, pVal := range pinned {
|
||
for _, r := range util.AllRegions {
|
||
if r.Value == pVal {
|
||
result = append(result, r)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果配置了一些不在 AllRegions 里的 mkt，上述循环可能漏掉
|
||
// 但根据之前的逻辑，AllRegions 是已知的 17 个地区。
|
||
// 如果用户配置了 fr-CA (不在 17 个内)，我们也应该返回它吗？
|
||
// 需求说 "前端页面对地区进行约束"，如果配置了，前端就该显示。
|
||
// 如果不在 AllRegions 里的，我们直接返回原始编码作为 label 或者查找一下。
|
||
|
||
if len(result) < len(pinned) {
|
||
// 补全不在 AllRegions 里的地区
|
||
for _, pVal := range pinned {
|
||
found := false
|
||
for _, r := range result {
|
||
if r.Value == pVal {
|
||
found = true
|
||
break
|
||
}
|
||
}
|
||
if !found {
|
||
result = append(result, util.Region{Value: pVal, Label: pVal})
|
||
}
|
||
}
|
||
}
|
||
|
||
c.JSON(http.StatusOK, result)
|
||
}
|
||
|
||
// compareResolution 比较两个分辨率变体的大小。
|
||
// 返回 < 0 表示 v1 < v2，返回 > 0 表示 v1 > v2，返回 0 表示相等。
|
||
func compareResolution(v1, v2 string) int {
|
||
resOrder := map[string]int{
|
||
"320x240": 1,
|
||
"400x240": 2,
|
||
"480x360": 3,
|
||
"640x360": 4,
|
||
"640x480": 5,
|
||
"800x480": 6,
|
||
"800x600": 7,
|
||
"1024x768": 8,
|
||
"1280x720": 9,
|
||
"1366x768": 10,
|
||
"1920x1080": 11,
|
||
"UHD": 12,
|
||
}
|
||
|
||
o1, ok1 := resOrder[v1]
|
||
o2, ok2 := resOrder[v2]
|
||
|
||
if !ok1 && !ok2 {
|
||
return strings.Compare(v1, v2)
|
||
}
|
||
if !ok1 {
|
||
return 1 // 未知的分辨率认为比已知的大
|
||
}
|
||
if !ok2 {
|
||
return -1
|
||
}
|
||
|
||
return o1 - o2
|
||
}
|