mirror of
https://git.fightbot.fun/hxuanyu/BingPaper.git
synced 2026-02-15 07:29:33 +08:00
421 lines
12 KiB
Go
421 lines
12 KiB
Go
package handlers
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"sort"
|
||
"strconv"
|
||
|
||
"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
|
||
// @Router /image/today [get]
|
||
func GetToday(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetTodayImage(mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
handleImageResponse(c, img, 7200) // 2小时
|
||
}
|
||
|
||
// GetTodayMeta 获取今日图片元数据
|
||
// @Summary 获取今日图片元数据
|
||
// @Description 获取今日必应图片的标题、版权等元数据
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {object} ImageMetaResp
|
||
// @Router /image/today/meta [get]
|
||
func GetTodayMeta(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetTodayImage(mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "public, max-age=7200") // 2小时
|
||
c.JSON(http.StatusOK, formatMeta(img))
|
||
}
|
||
|
||
// 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
|
||
// @Router /image/random [get]
|
||
func GetRandom(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetRandomImage(mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
handleImageResponse(c, img, 0) // 禁用缓存
|
||
}
|
||
|
||
// GetRandomMeta 获取随机图片元数据
|
||
// @Summary 获取随机图片元数据
|
||
// @Description 随机获取一张已抓取图片的元数据
|
||
// @Tags image
|
||
// @Param mkt query string false "地区编码 (如 zh-CN, en-US)"
|
||
// @Produce json
|
||
// @Success 200 {object} ImageMetaResp
|
||
// @Router /image/random/meta [get]
|
||
func GetRandomMeta(c *gin.Context) {
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetRandomImage(mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "no-cache, no-store, must-revalidate")
|
||
c.JSON(http.StatusOK, formatMeta(img))
|
||
}
|
||
|
||
// 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
|
||
// @Router /image/date/{date} [get]
|
||
func GetByDate(c *gin.Context) {
|
||
date := c.Param("date")
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetImageByDate(date, mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
handleImageResponse(c, img, 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
|
||
// @Router /image/date/{date}/meta [get]
|
||
func GetByDateMeta(c *gin.Context) {
|
||
date := c.Param("date")
|
||
mkt := c.Query("mkt")
|
||
img, err := image.GetImageByDate(date, mkt)
|
||
if err != nil {
|
||
c.JSON(http.StatusNotFound, gin.H{"error": "not found"})
|
||
return
|
||
}
|
||
c.Header("Cache-Control", "public, max-age=604800") // 7天
|
||
c.JSON(http.StatusOK, formatMeta(img))
|
||
}
|
||
|
||
// 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)
|
||
}
|
||
|
||
func handleImageResponse(c *gin.Context, img *model.Image, maxAge int) {
|
||
variant := c.DefaultQuery("variant", "UHD")
|
||
format := c.DefaultQuery("format", "jpg")
|
||
|
||
var selected *model.ImageVariant
|
||
for _, v := range img.Variants {
|
||
if v.Variant == variant && v.Format == format {
|
||
selected = &v
|
||
break
|
||
}
|
||
}
|
||
|
||
if selected == nil && len(img.Variants) > 0 {
|
||
// 回退逻辑
|
||
selected = &img.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 img.URLBase != "" {
|
||
// 兜底重定向到原始 Bing
|
||
bingURL := fmt.Sprintf("https://www.bing.com%s_%s.jpg", img.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, img.Date, maxAge)
|
||
}
|
||
} else {
|
||
serveLocal(c, selected.StorageKey, img.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(img *model.Image) gin.H {
|
||
cfg := config.GetConfig()
|
||
|
||
// 找到最小的变体（Size 最小）
|
||
var smallest *model.ImageVariant
|
||
for i := range img.Variants {
|
||
v := &img.Variants[i]
|
||
if smallest == nil || v.Size < smallest.Size {
|
||
smallest = v
|
||
}
|
||
}
|
||
|
||
variants := []gin.H{}
|
||
if smallest != nil {
|
||
url := smallest.PublicURL
|
||
if url == "" && cfg.API.Mode == "redirect" && img.URLBase != "" {
|
||
url = fmt.Sprintf("https://www.bing.com%s_%s.jpg", img.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, img.Date, smallest.Variant, smallest.Format, img.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": img.Date,
|
||
"mkt": img.Mkt,
|
||
"title": img.Title,
|
||
"copyright": img.Copyright,
|
||
"copyrightlink": img.CopyrightLink,
|
||
"quiz": img.Quiz,
|
||
"startdate": img.StartDate,
|
||
"fullstartdate": img.FullStartDate,
|
||
"hsh": img.HSH,
|
||
"variants": variants,
|
||
}
|
||
}
|
||
|
||
func formatMeta(img *model.Image) gin.H {
|
||
cfg := config.GetConfig()
|
||
variants := []gin.H{}
|
||
for _, v := range img.Variants {
|
||
url := v.PublicURL
|
||
if url == "" && cfg.API.Mode == "redirect" && img.URLBase != "" {
|
||
url = fmt.Sprintf("https://www.bing.com%s_%s.jpg", img.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, img.Date, v.Variant, v.Format, img.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": img.Date,
|
||
"mkt": img.Mkt,
|
||
"title": img.Title,
|
||
"copyright": img.Copyright,
|
||
"copyrightlink": img.CopyrightLink,
|
||
"quiz": img.Quiz,
|
||
"startdate": img.StartDate,
|
||
"fullstartdate": img.FullStartDate,
|
||
"hsh": img.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
|
||
|
||
// 创建副本以避免修改原始全局变量
|
||
all := make([]util.Region, len(util.AllRegions))
|
||
copy(all, util.AllRegions)
|
||
|
||
if len(pinned) > 0 {
|
||
// 创建一个 Map 用于快速查找置顶地区及其顺序
|
||
pinnedMap := make(map[string]int)
|
||
for i, v := range pinned {
|
||
pinnedMap[v] = i
|
||
}
|
||
|
||
// 对列表进行稳定排序，使置顶地区排在前面
|
||
sort.SliceStable(all, func(i, j int) bool {
|
||
idxI, okI := pinnedMap[all[i].Value]
|
||
idxJ, okJ := pinnedMap[all[j].Value]
|
||
|
||
if okI && okJ {
|
||
return idxI < idxJ
|
||
}
|
||
if okI {
|
||
return true
|
||
}
|
||
if okJ {
|
||
return false
|
||
}
|
||
return false // 保持非置顶地区的原有相对顺序
|
||
})
|
||
}
|
||
|
||
c.JSON(http.StatusOK, all)
|
||
}
|