Initial commit for AI Media Hub

backend/services/search.go

@@ -0,0 +1,64 @@
+package services
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"os"
+)
+
+type SearchResultItem struct {
+	Title string `json:"title"`
+	Link  string `json:"link"`
+	Pagemap struct {
+		CseImage []struct {
+			Src string `json:"src"`
+		} `json:"cse_image"`
+		CseThumbnail []struct {
+			Src string `json:"src"`
+		} `json:"cse_thumbnail"`
+	} `json:"pagemap"`
+}
+
+type SearchResponse struct {
+	Items []SearchResultItem `json:"items"`
+}
+
+// PerformSearch calls Google Custom Search API targeting YouTube, TikTok, Envato, Artgrid
+func PerformSearch(query string) ([]string, error) {
+	apiKey := os.Getenv("GOOGLE_CSE_API_KEY")
+	cx := os.Getenv("GOOGLE_CSE_ID")
+
+	if apiKey == "" || cx == "" {
+		return nil, fmt.Errorf("Google CSE credentials not configured")
+	}
+
+	// We append "site:youtube.com OR site:tiktok.com OR site:envato.com OR site:artgrid.io" to restrict
+	// depending on CSE settings, but the easiest is doing it in query string
+	fullQuery := fmt.Sprintf("%s site:youtube.com OR site:tiktok.com OR site:envato.com OR site:artgrid.io", query)
+	url := fmt.Sprintf("https://www.googleapis.com/customsearch/v1?key=%s&cx=%s&q=%s&searchType=image&num=10", apiKey, cx, fullQuery)
+
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		return nil, fmt.Errorf("Google CSE API returned status: %d", resp.StatusCode)
+	}
+
+	body, _ := ioutil.ReadAll(resp.Body)
+	var res SearchResponse
+	if err := json.Unmarshal(body, &res); err != nil {
+		return nil, err
+	}
+
+	var thumbnails []string
+	for _, item := range res.Items {
+		thumbnails = append(thumbnails, item.Link)
+	}
+
+	return thumbnails, nil
+}