需求:前台壁纸站支持切换到开源壁纸平台,并能展示后台上传的自家图库, 保留原有「浏览器实时生成」能力(扩展而非替换),PC / 移动双端一致。 设计要点(受服务器 5M 出口带宽约束): - 开源平台图片一律走官方 CDN 直链,后端只代理元数据,零带宽消耗。 - 自建图库落盘到 /data/www/wallpaper,由 nginx 的 /wallpaper/ 直出, Go 服务不参与传图;三档尺寸(480 缩略 / 1920 预览 / 原图仅供下载)。 - 内存缓存平台响应(Bing 1h、搜索类 10min),避免撞第三方配额。 内容: - 平台插件:Bing 每日一图、Picsum、Unsplash、Pexels、Wallhaven。 新增平台 = 写一个 provider_xxx.go 并在 allProviders() 加一行, 刻意不用 init() 自注册,避免隐式副作用。所有适配器支持 config.baseUrl 覆盖,用于绕开 DNS 污染(实测 wallhaven.cc 解析到境外无关 IP)。 - 自建图库:上传(MD5 秒传去重 / 尺寸前置校验 / 失败清理孤儿文件)、 元信息编辑、删除、统计;随机取图用「数总数 → 随机偏移」而非 ORDER BY RAND()。 - 接口:open 组 4 个(sources / list / random / download-track), admin 组 8 个(list / upload / save / delete / stats / source.list|save|test)。 全部 POST 且 URL 无参数,符合工作空间接口规范;上传走 multipart 例外。 - 安全:purity 默认锁死 SFW;apiKey 只回传布尔不回传明文; 保存时留空表示保留旧值;open 组错误信息对外脱敏。 - 配置:clientMaxBodySize 提到 64M(gf 默认 8M 会截断 20MB 原图); wallpaper.root / baseUrl 支持环境变量注入并在占位符未替换时兜底。 依赖:golang.org/x/image@v0.23.0(仅用于缩略图缩放,保持 go 1.23.0 不变)。
107 lines
3.2 KiB
Go
107 lines
3.2 KiB
Go
package wallpaper
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/url"
|
||
"time"
|
||
|
||
"github.com/gogf/gf/v2/encoding/gjson"
|
||
|
||
"service.xpcool.com/internal/model/dto"
|
||
)
|
||
|
||
// PexelsProvider Pexels 官方 API。
|
||
//
|
||
// 特点:免费额度高、图片量大、接口直接给出多档尺寸且支持朝向筛选。
|
||
// 鉴权:Authorization 头直接放 API Key(注意与 Unsplash 不同,**没有** Client-ID 前缀)。
|
||
// 许可:可免费用于个人与商业用途,无需转存,直接用其 CDN 地址即可。
|
||
type PexelsProvider struct{}
|
||
|
||
const pexelsBase = "https://api.pexels.com/v1"
|
||
|
||
func (p *PexelsProvider) Code() string { return "pexels" }
|
||
func (p *PexelsProvider) Name() string { return "Pexels" }
|
||
func (p *PexelsProvider) RequiresKey() bool { return true }
|
||
|
||
func (p *PexelsProvider) List(ctx context.Context, cfg dto.WallpaperSourceConfig, q dto.WallpaperQuery) ([]dto.WallpaperItem, int, error) {
|
||
if cfg.ApiKey == "" {
|
||
return nil, 0, fmt.Errorf("Pexels 未配置 API Key")
|
||
}
|
||
page, size := normalizePage(q.Page, q.Size)
|
||
|
||
key := cacheKey(p.Code(), q)
|
||
if v, err := openCache.Get(ctx, key); err == nil && !v.IsNil() {
|
||
if cached, ok := v.Val().([]dto.WallpaperItem); ok {
|
||
return cached, len(cached), nil
|
||
}
|
||
}
|
||
|
||
// 没有关键词时走「精选」端点,避免请求空 query 的搜索接口
|
||
base := baseOf(cfg.BaseUrl, pexelsBase)
|
||
var u string
|
||
if keyword := firstNonEmpty(q.Query, cfg.DefaultQuery); keyword != "" {
|
||
u = fmt.Sprintf("%s/search?query=%s&page=%d&per_page=%d", base, url.QueryEscape(keyword), page, size)
|
||
} else {
|
||
u = fmt.Sprintf("%s/curated?page=%d&per_page=%d", base, page, size)
|
||
}
|
||
if o := pexelsOrientation(q.Orientation); o != "" {
|
||
u += "&orientation=" + url.QueryEscape(o)
|
||
}
|
||
|
||
headers := map[string]string{"Authorization": cfg.ApiKey}
|
||
j, err := httpGetJSON(ctx, u, headers, 12*time.Second)
|
||
if err != nil {
|
||
return nil, 0, fmt.Errorf("获取 Pexels 列表失败: %w", err)
|
||
}
|
||
|
||
out := make([]dto.WallpaperItem, 0, size)
|
||
for _, raw := range j.Get("photos").Array() {
|
||
item := gjson.New(raw)
|
||
id := item.Get("id").String()
|
||
if id == "" {
|
||
continue
|
||
}
|
||
w := item.Get("width").Int()
|
||
h := item.Get("height").Int()
|
||
if !matchOrientation(w, h, q.Orientation) {
|
||
continue
|
||
}
|
||
out = append(out, dto.WallpaperItem{
|
||
Id: id,
|
||
Title: firstNonEmpty(item.Get("alt").String(), "Pexels "+id),
|
||
Width: w,
|
||
Height: h,
|
||
Orientation: orientationOf(w, h),
|
||
ThumbUrl: item.Get("src.medium").String(),
|
||
PreviewUrl: firstNonEmpty(item.Get("src.large2x").String(), item.Get("src.large").String()),
|
||
FullUrl: item.Get("src.original").String(),
|
||
Source: p.Code(),
|
||
SourceName: p.Name(),
|
||
FromOpen: true,
|
||
Author: item.Get("photographer").String(),
|
||
AuthorUrl: item.Get("photographer_url").String(),
|
||
PageUrl: item.Get("url").String(),
|
||
License: "Pexels License",
|
||
})
|
||
}
|
||
|
||
_ = openCache.Set(ctx, key, out, cacheTTLSearch)
|
||
return out, j.Get("total_results").Int(), nil
|
||
}
|
||
|
||
// pexelsOrientation 把内部朝向编码翻译成 Pexels 的取值。
|
||
// 注意 Pexels 用的是 square(不是 Unsplash 的 squarish)。
|
||
func pexelsOrientation(o int) string {
|
||
switch o {
|
||
case 1:
|
||
return "landscape"
|
||
case 2:
|
||
return "portrait"
|
||
case 3:
|
||
return "square"
|
||
default:
|
||
return ""
|
||
}
|
||
}
|