go-igdb/endpoint/search.go

package endpoint

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strings"
	"time"

	pb "github.com/bestnite/go-igdb/proto"
	"github.com/go-resty/resty/v2"

	"github.com/PuerkitoBio/goquery"
	"github.com/bestnite/go-flaresolverr"
	"google.golang.org/protobuf/proto"
)

var webSearchCFCookies struct {
	cookies []*http.Cookie
	expires time.Time
}

type Search struct {
	endpointName Name
	request      func(URL string, dataBody any) (*resty.Response, error)
	flaresolverr *flaresolverr.Flaresolverr
}

func NewSearch(request func(URL string, dataBody any) (*resty.Response, error)) *Search {
	return &Search{
		endpointName: EPSearch,
		request:      request,
	}
}

func (a *Search) Search(query string) ([]*pb.Search, error) {
	resp, err := a.request(fmt.Sprintf("https://api.igdb.com/v4/%s.pb", a.endpointName), query)
	if err != nil {
		return nil, fmt.Errorf("failed to request: %w", err)
	}

	data := pb.SearchResult{}
	if err = proto.Unmarshal(resp.Body(), &data); err != nil {
		return nil, fmt.Errorf("failed to unmarshal: %w", err)
	}

	if len(data.Searches) == 0 {
		return nil, fmt.Errorf("no results: %s", query)
	}

	return data.Searches, nil
}

func (a *Search) getFlaresolverr() (*flaresolverr.Flaresolverr, error) {
	if a.flaresolverr == nil {
		return nil, fmt.Errorf("flaresolverr is not initialized")
	}
	return a.flaresolverr, nil
}

func (a *Search) WebSearchGameIDs(name string) ([]uint64, error) {
	params := url.Values{}
	params.Add("q", name)
	params.Add("utf8", "✓")
	Url := fmt.Sprintf("%s?%s", "https://www.igdb.com/search", params.Encode())

	f, err := a.getFlaresolverr()
	if err != nil {
		return nil, fmt.Errorf("failed to get flaresolverr: %w", err)
	}

	var respBody io.Reader
	if webSearchCFCookies.cookies == nil || time.Now().After(webSearchCFCookies.expires) {
		resp, err := f.GetV1(Url, nil)
		if err != nil {
			return nil, fmt.Errorf("failed to web search: %s: %w", name, err)
		}
		webSearchCFCookies.cookies = resp.Solution.Cookies
		webSearchCFCookies.expires = time.Now().Add(3 * time.Hour)
		respBody = strings.NewReader(resp.Solution.Response)
	} else if time.Now().Before(webSearchCFCookies.expires) {
		resp, err := f.SimulateGet(Url, &flaresolverr.SimulateOptions{
			HttpCookies: webSearchCFCookies.cookies,
		})
		if err != nil {
			return nil, fmt.Errorf("failed to search IGDB ID: %s: %w", name, err)
		}
		respBody = strings.NewReader(resp.Body)
	}

	doc, err := goquery.NewDocumentFromReader(respBody)
	if err != nil {
		return nil, fmt.Errorf("failed to parse IGDB web search response: %w", err)
	}
	selection := doc.Find("script[data-component-name='GameEntries']")
	if selection.Length() == 0 {
		return nil, fmt.Errorf("no search results found for: %s", name)
	}
	innerJson := selection.First().Text()
	data := struct {
		Games []struct {
			Id   uint64 `json:"id"`
			Name string `json:"name"`
		} `json:"games"`
	}{}
	if err := json.Unmarshal([]byte(innerJson), &data); err != nil {
		return nil, fmt.Errorf("failed to unmarshal IGDB web search response: %w", err)
	}

	ids := make([]uint64, len(data.Games))
	for i, game := range data.Games {
		ids[i] = game.Id
	}

	return ids, nil
}