实现直播和搜索建议接口

Author: senshinya

config/subscription.go

@@ -38,26 +38,28 @@ func FetchSubscription(url string) error {
 
 	// 先用 encoding/json 解析以保持键的顺序
 	var rawConfig struct {
-		APISites json.RawMessage  `json:"api_site"`
+		APISites json.RawMessage `json:"api_site"`
+		Lives    json.RawMessage `json:"lives"`
 	}
-	
+
 	if err := json.Unmarshal(decodedData, &rawConfig); err != nil {
 		return fmt.Errorf("failed to unmarshal config JSON: %v", err)
 	}
-	
+
 	// 构建最终的 config
 	var config models.Config
 	config.APISites = make(map[string]models.APISite)
 	config.SiteList = make([]models.APISite, 0)
-	
+	config.Lives = make([]models.LiveSource, 0)
+
 	// 使用 json.Decoder 按顺序解析 api_site 对象
 	dec := json.NewDecoder(bytes.NewReader(rawConfig.APISites))
-	
+
 	// 读取开始的 {
 	if _, err := dec.Token(); err != nil {
 		return fmt.Errorf("failed to parse api_site: %v", err)
 	}
-	
+
 	// 按顺序读取每个键值对
 	for dec.More() {
 		// 读取键
@@ -66,23 +68,54 @@ func FetchSubscription(url string) error {
 			return fmt.Errorf("failed to read key: %v", err)
 		}
 		key := token.(string)
-		
+
 		// 读取值
 		var site models.APISite
 		if err := dec.Decode(&site); err != nil {
 			return fmt.Errorf("failed to decode site: %v", err)
 		}
-		
+
 		site.Key = key
 		config.APISites[key] = site
 		config.SiteList = append(config.SiteList, site)
 	}
-	
+
 	if len(config.APISites) == 0 {
 		return fmt.Errorf("no API sites found in subscription")
 	}
 
+	// 解析 lives 字段（如果存在）
+	if len(rawConfig.Lives) > 0 {
+		liveDec := json.NewDecoder(bytes.NewReader(rawConfig.Lives))
+
+		// 读取开始的 {
+		if _, err := liveDec.Token(); err != nil {
+			return fmt.Errorf("failed to parse lives: %v", err)
+		}
+
+		// 按顺序读取每个键值对
+		for liveDec.More() {
+			// 读取键
+			token, err := liveDec.Token()
+			if err != nil {
+				return fmt.Errorf("failed to read live key: %v", err)
+			}
+			key := token.(string)
+
+			// 读取值
+			var live models.LiveSource
+			if err := liveDec.Decode(&live); err != nil {
+				return fmt.Errorf("failed to decode live source: %v", err)
+			}
+
+			live.Key = key
+			config.Lives = append(config.Lives, live)
+		}
+
+		log.Printf("Loaded %d live sources", len(config.Lives))
+	}
+
 	GlobalConfig = config
-	log.Printf("Subscription config loaded successfully. API sites: %d", len(config.APISites))
+	log.Printf("Subscription config loaded successfully. API sites: %d, Live sources: %d", len(config.APISites), len(config.Lives))
 	return nil
 }

handlers/live.go

@@ -0,0 +1,51 @@
+package handlers
+
+import (
+	"helios/config"
+	"net/http"
+
+	"github.com/bytedance/sonic"
+)
+
+type LiveSourceResponse struct {
+	Key      string `json:"key"`
+	Name     string `json:"name"`
+	URL      string `json:"url"`
+	EPG      string `json:"epg"`
+	UA       string `json:"ua,omitempty"`
+	From     string `json:"from"`
+	Disabled bool   `json:"disabled"`
+}
+
+type LiveSourcesAPIResponse struct {
+	Success bool                 `json:"success"`
+	Data    []LiveSourceResponse `json:"data"`
+}
+
+func LiveSourcesHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	sources := make([]LiveSourceResponse, 0, len(config.GlobalConfig.Lives))
+	for _, live := range config.GlobalConfig.Lives {
+		sources = append(sources, LiveSourceResponse{
+			Key:      live.Key,
+			Name:     live.Name,
+			URL:      live.Url,
+			EPG:      live.Epg,
+			UA:       live.UA,
+			From:     "config", // 来源为配置文件
+			Disabled: false,    // 默认不禁用
+		})
+	}
+
+	response := LiveSourcesAPIResponse{
+		Success: true,
+		Data:    sources,
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	sonic.ConfigDefault.NewEncoder(w).Encode(response)
+}

handlers/suggestions.go

@@ -0,0 +1,97 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+	"helios/config"
+	"helios/lib"
+	"net/http"
+	"time"
+
+	"github.com/bytedance/sonic"
+)
+
+type Suggestion struct {
+	Text string `json:"text"`
+}
+
+type SuggestionsResponse struct {
+	Suggestions []Suggestion `json:"suggestions"`
+}
+
+func SuggestionsHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	query := r.URL.Query().Get("q")
+
+	// 如果没有查询参数，返回空建议
+	if query == "" {
+		w.Header().Set("Content-Type", "application/json")
+		response := SuggestionsResponse{
+			Suggestions: []Suggestion{},
+		}
+		sonic.ConfigDefault.NewEncoder(w).Encode(response)
+		return
+	}
+
+	// 获取第一个 API 站点
+	apiSites := config.GlobalConfig.SiteList
+	if len(apiSites) == 0 {
+		w.Header().Set("Content-Type", "application/json")
+		response := SuggestionsResponse{
+			Suggestions: []Suggestion{},
+		}
+		sonic.ConfigDefault.NewEncoder(w).Encode(response)
+		return
+	}
+
+	firstSite := apiSites[0]
+
+	// 创建带超时的上下文
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	// 使用 channel 来接收结果或超时
+	resultChan := make(chan []Suggestion, 1)
+
+	go func() {
+		// 从第一个源搜索
+		results, err := lib.SearchFromAPI(firstSite, query, 1)
+		if err != nil {
+			resultChan <- []Suggestion{}
+			return
+		}
+
+		// 应用黄色过滤
+		results = lib.FilterYellowContent(results)
+
+		// 转换为建议格式
+		suggestions := make([]Suggestion, 0, len(results))
+		for _, result := range results {
+			suggestions = append(suggestions, Suggestion{
+				Text: result.Title,
+			})
+		}
+
+		resultChan <- suggestions
+	}()
+
+	var suggestions []Suggestion
+	select {
+	case suggestions = <-resultChan:
+		// 成功获取结果
+	case <-ctx.Done():
+		// 超时，返回空结果
+		fmt.Printf("搜索建议超时\n")
+		suggestions = []Suggestion{}
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	response := SuggestionsResponse{
+		Suggestions: suggestions,
+	}
+	sonic.ConfigDefault.NewEncoder(w).Encode(response)
+}

main.go

@@ -50,10 +50,12 @@ func main() {
 	http.HandleFunc("/api/search", handlers.AuthMiddleware(handlers.SearchHandler))
 	http.HandleFunc("/api/search/ws", handlers.AuthMiddleware(handlers.SSESearchHandler))
 	http.HandleFunc("/api/search/resources", handlers.AuthMiddleware(handlers.ResourcesHandler))
+	http.HandleFunc("/api/search/suggestions", handlers.AuthMiddleware(handlers.SuggestionsHandler))
 	http.HandleFunc("/api/detail", handlers.AuthMiddleware(handlers.DetailHandler))
 	http.HandleFunc("/api/favorites", handlers.AuthMiddleware(handlers.FavoritesHandler))
 	http.HandleFunc("/api/searchhistory", handlers.AuthMiddleware(handlers.SearchHistoryHandler))
 	http.HandleFunc("/api/playrecords", handlers.AuthMiddleware(handlers.PlayRecordsHandler))
+	http.HandleFunc("/api/live/sources", handlers.AuthMiddleware(handlers.LiveSourcesHandler))
 
 	port := ":8080"
 	fmt.Printf("Server starting on port %s\n", port)

models/config.go

@@ -3,6 +3,7 @@ package models
 type Config struct {
 	APISites map[string]APISite `json:"api_site"`
 	SiteList []APISite          `json:"site_list"`
+	Lives    []LiveSource       `json:"lives"`
 }
 
 type APISite struct {
@@ -11,3 +12,11 @@ type APISite struct {
 	Name   string `json:"name"`
 	Detail string `json:"detail"`
 }
+
+type LiveSource struct {
+	Key  string `json:"key"`
+	Url  string `json:"url"`
+	Name string `json:"name"`
+	UA   string `json:"ua"`
+	Epg  string `json:"epg"`
+}