go-igdb/search.go

package igdb

import (
	"encoding/json"
	"fmt"
	"github/bestnite/go-igdb/constant"
	"io"
	"net/http"
	"net/url"
	"strings"
	"time"

	pb "github/bestnite/go-igdb/proto"

	"git.nite07.com/nite/flaresolverr"
	"github.com/PuerkitoBio/goquery"
	"google.golang.org/protobuf/proto"
)

func (g *igdb) SearchGame(query string) ([]*pb.Game, error) {
	resp, err := g.Request(constant.IGDBGameURL, query)
	if err != nil {
		return nil, fmt.Errorf("failed to search: %s: %w", query, err)
	}

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

	if len(data.Games) != 0 && data.Games[0].Name == "" {
		return g.WebSearchGame(query)
	}

	return data.Games, nil
}

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

func (g *igdb) WebSearchGame(name string) ([]*pb.Game, error) {
	params := url.Values{}
	params.Add("q", name)
	params.Add("utf8", "✓")
	Url := fmt.Sprintf("%s?%s", constant.IGDBWebSearchURL, params.Encode())

	f, err := g.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 g.GetGames(ids)
}