feat(func): media library support change api url

Author: PIKACHUIM

internal/bootstrap/data/setting.go

@@ -245,7 +245,9 @@ func InitialSettings() []model.SettingItem {
 
 		// media settings
 		{Key: conf.MediaTMDBKey, Value: "", Type: conf.TypeString, Group: model.MEDIA, Flag: model.PRIVATE},
+		{Key: conf.MediaTMDBAPIURL, Value: "https://api.themoviedb.org", Type: conf.TypeString, Group: model.MEDIA, Flag: model.PRIVATE},
 		{Key: conf.MediaDiscogsToken, Value: "", Type: conf.TypeString, Group: model.MEDIA, Flag: model.PRIVATE},
+		{Key: conf.MediaDiscogsAPIURL, Value: "https://api.discogs.com", Type: conf.TypeString, Group: model.MEDIA, Flag: model.PRIVATE},
 		{Key: conf.MediaStoreThumbnail, Value: "false", Type: conf.TypeBool, Group: model.MEDIA, Flag: model.PRIVATE},
 		{Key: conf.MediaThumbnailMode, Value: "base64", Type: conf.TypeSelect, Options: "base64,local", Group: model.MEDIA, Flag: model.PRIVATE},
 		{Key: conf.MediaThumbnailPath, Value: "/.thumbnail", Type: conf.TypeString, Group: model.MEDIA, Flag: model.PRIVATE},

internal/conf/const.go

@@ -164,7 +164,9 @@ const (
 
 	// media
 	MediaTMDBKey          = "media_tmdb_key"
+	MediaTMDBAPIURL       = "media_tmdb_api_url"
 	MediaDiscogsToken     = "media_discogs_token"
+	MediaDiscogsAPIURL    = "media_discogs_api_url"
 	MediaThumbnailMode    = "media_thumbnail_mode"
 	MediaThumbnailPath    = "media_thumbnail_path"
 	MediaStoreThumbnail   = "media_store_thumbnail"

internal/media/scraper/discogs.go

@@ -12,19 +12,26 @@ import (
 	"github.com/OpenListTeam/OpenList/v4/internal/model"
 )
 
-const discogsBaseURL = "https://api.discogs.com"
+const discogsDefaultBaseURL = "https://api.discogs.com"
 
 // DiscogsScraper Discogs音乐刮削器
 type DiscogsScraper struct {
-	Token  string
-	client *http.Client
+	Token   string
+	BaseURL string // 可自定义的 Discogs API 地址
+	client  *http.Client
 }
 
 // NewDiscogsScraper 创建Discogs刮削器
-func NewDiscogsScraper(token string) *DiscogsScraper {
+// baseURL 为可自定义的 Discogs API 服务地址，为空时使用默认 https://api.discogs.com
+func NewDiscogsScraper(token string, baseURL string) *DiscogsScraper {
+	base := strings.TrimRight(strings.TrimSpace(baseURL), "/")
+	if base == "" {
+		base = discogsDefaultBaseURL
+	}
 	return &DiscogsScraper{
-		Token:  token,
-		client: &http.Client{Timeout: 15 * time.Second},
+		Token:   token,
+		BaseURL: base,
+		client:  &http.Client{Timeout: 15 * time.Second},
 	}
 }
 
@@ -121,7 +128,7 @@ func parseMusicFileName(fileName string) []string {
 // doDiscogsSearch 执行单次Discogs搜索
 func (s *DiscogsScraper) doDiscogsSearch(query string) (*discogsSearchResult, error) {
 	searchURL := fmt.Sprintf("%s/database/search?q=%s&type=release&token=%s",
-		discogsBaseURL, url.QueryEscape(query), s.Token)
+		s.BaseURL, url.QueryEscape(query), s.Token)
 
 	req, _ := http.NewRequest("GET", searchURL, nil)
 	req.Header.Set("User-Agent", "OpenList/4.0 +https://github.com/OpenListTeam/OpenList")
@@ -214,7 +221,7 @@ func (s *DiscogsScraper) ScrapeMusic(item *model.MediaItem) error {
 	first := searchResult.Results[0]
 
 	// 获取详情
-	detailURL := fmt.Sprintf("%s/releases/%d?token=%s", discogsBaseURL, first.ID, s.Token)
+	detailURL := fmt.Sprintf("%s/releases/%d?token=%s", s.BaseURL, first.ID, s.Token)
 	detailReq, _ := http.NewRequest("GET", detailURL, nil)
 	detailReq.Header.Set("User-Agent", "OpenList/4.0 +https://github.com/OpenListTeam/OpenList")
 

internal/media/scraper/tmdb.go

@@ -88,20 +88,31 @@ func parseVideoFileName(fileName string) parsedVideoTitle {
 	return result
 }
 
-const tmdbBaseURL = "https://api.themoviedb.org/3"
+const tmdbDefaultBaseURL = "https://api.themoviedb.org"
 const tmdbImageBase = "https://image.tmdb.org/t/p/w500"
 
 // TMDBScraper TMDB视频刮削器
 type TMDBScraper struct {
-	APIKey string
-	client *http.Client
+	APIKey  string
+	BaseURL string // 含 /3 版本路径的最终请求地址，例如 https://api.themoviedb.org/3
+	client  *http.Client
 }
 
 // NewTMDBScraper 创建TMDB刮削器
-func NewTMDBScraper(apiKey string) *TMDBScraper {
+// baseURL 为可自定义的 TMDB API 服务地址（不含版本路径），为空时使用默认 https://api.themoviedb.org
+func NewTMDBScraper(apiKey string, baseURL string) *TMDBScraper {
+	base := strings.TrimRight(strings.TrimSpace(baseURL), "/")
+	if base == "" {
+		base = tmdbDefaultBaseURL
+	}
+	// 若用户填入的地址未带 /3 版本路径，则自动追加
+	if !strings.HasSuffix(base, "/3") {
+		base = base + "/3"
+	}
 	return &TMDBScraper{
-		APIKey: apiKey,
-		client: &http.Client{Timeout: 15 * time.Second},
+		APIKey:  apiKey,
+		BaseURL: base,
+		client:  &http.Client{Timeout: 15 * time.Second},
 	}
 }
 
@@ -143,7 +154,7 @@ type tmdbMovieDetail struct {
 // doTMDBSearch 执行一次TMDB搜索请求
 func (s *TMDBScraper) doTMDBSearch(query, year string) (*tmdbSearchResult, error) {
 	searchURL := fmt.Sprintf("%s/search/multi?api_key=%s&query=%s&language=zh-CN&search_type=ngram",
-		tmdbBaseURL, s.APIKey, url.QueryEscape(query))
+		s.BaseURL, s.APIKey, url.QueryEscape(query))
 	if year != "" {
 		searchURL += "&year=" + year
 	}
@@ -237,7 +248,7 @@ func (s *TMDBScraper) ScrapeVideo(item *model.MediaItem) error {
 
 	// 获取详情
 	detailURL := fmt.Sprintf("%s/%s/%d?api_key=%s&language=zh-CN&append_to_response=credits",
-		tmdbBaseURL, mediaType, first.ID, s.APIKey)
+		s.BaseURL, mediaType, first.ID, s.APIKey)
 
 	detailResp, err := s.client.Get(detailURL)
 	if err != nil {

server/handles/media.go

@@ -372,7 +372,9 @@ func StartMediaScrape(c *gin.Context) {
 
 	// 从系统设置中读取刮削配置
 	tmdbKey := setting.GetStr(conf.MediaTMDBKey)
+	tmdbAPIURL := setting.GetStr(conf.MediaTMDBAPIURL, "https://api.themoviedb.org")
 	discogsToken := setting.GetStr(conf.MediaDiscogsToken)
+	discogsAPIURL := setting.GetStr(conf.MediaDiscogsAPIURL, "https://api.discogs.com")
 	thumbnailMode := setting.GetStr(conf.MediaThumbnailMode, "base64")
 	thumbnailPath := setting.GetStr(conf.MediaThumbnailPath, "/.thumbnail")
 	storeThumbnail := setting.GetBool(conf.MediaStoreThumbnail)
@@ -400,10 +402,10 @@ func StartMediaScrape(c *gin.Context) {
 
 			switch req.MediaType {
 			case model.MediaTypeVideo:
-				s := scraper.NewTMDBScraper(tmdbKey)
+				s := scraper.NewTMDBScraper(tmdbKey, tmdbAPIURL)
 				scrapeErr = s.ScrapeVideo(item)
 			case model.MediaTypeMusic:
-				s := scraper.NewDiscogsScraper(discogsToken)
+				s := scraper.NewDiscogsScraper(discogsToken, discogsAPIURL)
 				scrapeErr = s.ScrapeMusic(item)
 			case model.MediaTypeBook:
 				doubanScraper := scraper.NewDoubanScraperWithConfig(