added node search for video

video.go

@@ -30,6 +30,7 @@ var (
 	}
 	disabledInstances = make(map[string]bool)
 	mu                sync.Mutex
+	videoResultsChan  = make(chan []VideoResult) // Channel to receive video results from other nodes
 )
 
 // VideoAPIResponse matches the structure of the JSON response from the Piped API
@@ -151,32 +152,10 @@ func makeHTMLRequest(query, safe, lang string, page int) (*VideoAPIResponse, err
 func handleVideoSearch(w http.ResponseWriter, query, safe, lang string, page int) {
 	start := time.Now()
 
-	apiResp, err := makeHTMLRequest(query, safe, lang, page)
-	if err != nil {
-		log.Printf("Error fetching video results: %v", err)
-		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
-		return
-	}
-
-	var results []VideoResult
-	for _, item := range apiResp.Items {
-		if item.Type == "channel" || item.Type == "playlist" {
-			continue
-		}
-		if item.UploadedDate == "" {
-			item.UploadedDate = "Now"
-		}
-
-		results = append(results, VideoResult{
-			Href:      fmt.Sprintf("https://youtube.com%s", item.URL),
-			Title:     item.Title,
-			Date:      item.UploadedDate,
-			Views:     formatViews(item.Views),
-			Creator:   item.UploaderName,
-			Publisher: "Piped",
-			Image:     fmt.Sprintf("/img_proxy?url=%s", url.QueryEscape(item.Thumbnail)),
-			Duration:  formatDuration(item.Duration),
-		})
+	results := fetchVideoResults(query, safe, lang, page)
+	if len(results) == 0 {
+		log.Printf("No results from primary search, trying other nodes")
+		results = tryOtherNodesForVideoSearch(query, safe, lang, page)
 	}
 
 	elapsed := time.Since(start)
@@ -230,3 +209,72 @@ func fetchVideoResults(query, safe, lang string, page int) []VideoResult {
 	}
 	return results
 }
+
+func tryOtherNodesForVideoSearch(query, safe, lang string, page int) []VideoResult {
+	for _, nodeAddr := range peers {
+		results, err := sendVideoSearchRequestToNode(nodeAddr, query, safe, lang, page)
+		if err != nil {
+			log.Printf("Error contacting node %s: %v", nodeAddr, err)
+			continue
+		}
+		if len(results) > 0 {
+			return results
+		}
+	}
+	return nil
+}
+
+func sendVideoSearchRequestToNode(nodeAddr, query, safe, lang string, page int) ([]VideoResult, error) {
+	searchParams := struct {
+		Query        string `json:"query"`
+		Safe         string `json:"safe"`
+		Lang         string `json:"lang"`
+		Page         int    `json:"page"`
+		ResponseAddr string `json:"responseAddr"`
+	}{
+		Query:        query,
+		Safe:         safe,
+		Lang:         lang,
+		Page:         page,
+		ResponseAddr: fmt.Sprintf("http://localhost:%d/node", config.Port),
+	}
+
+	msgBytes, err := json.Marshal(searchParams)
+	if err != nil {
+		return nil, fmt.Errorf("failed to marshal search parameters: %v", err)
+	}
+
+	msg := Message{
+		ID:      hostID,
+		Type:    "search-video",
+		Content: string(msgBytes),
+	}
+
+	err = sendMessage(nodeAddr, msg)
+	if err != nil {
+		return nil, fmt.Errorf("failed to send search request to node %s: %v", nodeAddr, err)
+	}
+
+	// Wait for results
+	select {
+	case res := <-videoResultsChan:
+		return res, nil
+	case <-time.After(20 * time.Second):
+		return nil, fmt.Errorf("timeout waiting for results from node %s", nodeAddr)
+	}
+}
+
+func handleVideoResultsMessage(msg Message) {
+	var results []VideoResult
+	err := json.Unmarshal([]byte(msg.Content), &results)
+	if err != nil {
+		log.Printf("Error unmarshalling video results: %v", err)
+		return
+	}
+
+	log.Printf("Received video results: %+v", results)
+	// Send results to videoResultsChan
+	go func() {
+		videoResultsChan <- results
+	}()
+}