types.go

package brave

import (
	"fmt"
	"regexp"
	"strconv"
	"strings"
	"time"

	anytime "github.com/ijt/go-anytime"
)

type ResultContainer[T any] struct {
	Type             string `json:"type"`
	Results          []T    `json:"results"`
	MutatedByGoggles bool   `json:"mutated_by_goggles"`
}

type Query struct {
	Original             string    `json:"original"`
	ShowStrictWarning    bool      `json:"show_strict_warning"`
	Altered              string    `json:"altered"`
	Safesearch           bool      `json:"safesearch"`
	IsNavigational       bool      `json:"is_navigational"`
	IsGeolocal           bool      `json:"is_geolocal"`
	LocalDecision        string    `json:"local_decision"`
	LocalLocationsIdx    int       `json:"local_locations_idx"`
	IsTrending           bool      `json:"is_trending"`
	IsNewsBreaking       bool      `json:"is_news_breaking"`
	AskForLocation       bool      `json:"ask_for_location"`
	Language             *Language `json:"language"`
	SpellcheckOff        bool      `json:"spellcheck_off"`
	Country              string    `json:"country"`
	BadResults           bool      `json:"bad_results"`
	ShouldFallback       bool      `json:"should_fallback"`
	Lat                  string    `json:"lat"`
	Long                 string    `json:"long"`
	PostalCode           string    `json:"postal_code"`
	City                 string    `json:"city"`
	State                string    `json:"state"`
	HeaderCountry        string    `json:"header_country"`
	MoreResultsAvailable bool      `json:"more_results_available"`
	CustomLocationLabel  string    `json:"custom_location_label"`
	RedditCluster        string    `json:"reddit_cluster"`
	SummaryKey           string    `json:"summary_key"`
}

type Language struct {
	Main string `json:"main"`
}

type Mixed struct {
	Type string `json:"type"`
	Main []ResultReference
	Top  []ResultReference
	Side []ResultReference
}

type ResultReference struct {
	Type  string `json:"type"`
	Index int    `json:"index"`
	All   bool   `json:"all"`
}

type Result struct {
	Title          string     `json:"title"`
	URL            string     `json:"url"`
	IsSourceLocal  bool       `json:"is_source_local"`
	IsSourceBoth   bool       `json:"is_source_both"`
	Description    string     `json:"description"`
	PageAge        *Timestamp `json:"page_age"`
	PageFetched    string     `json:"page_fetched"`
	Profile        *Profile   `json:"profile"`
	Language       string     `json:"language"`
	FamilyFriendly bool       `json:"family_friendly"`
}

type Profile struct {
	Name     string `json:"name"`
	LongName string `json:"long_name"`
	URL      string `json:"url"`
	Image    string `json:"img"`
}

type NewsResult struct {
	Result
	MetaURL   MetaURL    `json:"meta_url"`
	Source    string     `json:"source"`
	Breaking  bool       `json:"breaking"`
	Thumbnail *Thumbnail `json:"thumbnail"`
	Age       *Timestamp `json:"age"`
}

type VideoResult struct {
	Result
	Type      string     `json:"type"`
	Data      *VideoData `json:"video"`
	MetaURL   MetaURL    `json:"meta_url"`
	Thumbnail *Thumbnail `json:"thumbnail"`
	Age       *Timestamp `json:"age"`
}

type VideoData struct {
	Duration  *Duration  `json:"duration"`
	Views     VideoViews `json:"views"`
	Creator   string     `json:"creator"`
	Publisher string     `json:"publisher"`
	Thumbnail *Thumbnail `json:"thumbnail"`
}

type VideoViews int

func (v *VideoViews) UnmarshalJSON(in []byte) error {
	str := string(in)
	if strings.Contains(str, `"`) {
		str = strings.ToLower(strings.Trim(str, `"`))
	}

	// this is so stupid.
	if strings.HasSuffix(str, "k") {
		str = strings.TrimSuffix(str, "k") + "000"
	} else if strings.HasSuffix(str, "m") {
		str = strings.TrimSuffix(str, "m") + "000000"
	}

	vv, err := strconv.Atoi(str)
	if err != nil {
		return err
	}

	*v = VideoViews(vv)
	return nil
}

type MetaURL struct {
	Scheme   string `json:"scheme"`
	NetLoc   string `json:"netloc"`
	Hostname string `json:"hostname"`
	Favicon  string `json:"favicon"`
	Path     string `json:"path"`
}

type Thumbnail struct {
	Src             string `json:"src"`
	Height          int    `json:"height"`
	Width           int    `json:"width"`
	BackgroundColor string `json:"bg_color"`
	Original        string `json:"original"`
	Logo            bool   `json:"logo"`
	Duplicated      bool   `json:"duplicated"`
	Theme           string `json:"theme"`
}

type SearchResult struct {
	Result
	Type        string          `json:"type"`
	Subtype     string          `json:"subtype"`
	DeepResults *DeepResult     `json:"deep_results"`
	Schemas     any             `json:"schemas"`
	MetaURL     MetaURL         `json:"meta_url"`
	Thumbnail   *Thumbnail      `json:"thumbnail"`
	Age         *Timestamp      `json:"age"`
	Language    string          `json:"language"`
	Restaurant  *LocationResult `json:"restaurant"`
	Locations   *Locations      `json:"locations"`
	Video       *VideoData      `json:"video"`
	Movie       *MovieData      `json:"movie"`
	FAQ         *FAQ            `json:"faq"`
	QA          *QAPage         `json:"qa"`
	Book        *Book           `json:"book"`
	Rating      *Rating         `json:"rating"`
	Article     *Article        `json:"article"`
	// Product     any             `json:"product"`
	ProductCluster []Product       `json:"product_cluster"`
	ClusterType    string          `json:"cluster_type"`
	Cluster        []Result        `json:"cluster"`
	CreativeWork   *CreativeWork   `json:"creative_work"`
	MusicRecording *MusicRecording `json:"music_recording"`
	Review         *Review         `json:"review"`
	Software       *Software       `json:"software"`
	ContentType    string          `json:"content_type"`
}

type ImageResult struct {
	Type        string           `json:"type"`
	Title       string           `json:"title"`
	URL         string           `json:"url"`
	Source      string           `json:"source"`
	PageFetched *Timestamp       `json:"page_fetched"`
	Thumbnail   *Thumbnail       `json:"thumbnail"`
	Properties  *ImageProperties `json:"properties"`
	MetaURL     *MetaURL         `json:"meta_url"`
}

type DeepResult struct {
	News    []NewsResult            `json:"news"`
	Buttons []ButtonResult          `json:"buttons"`
	Social  []KnowledgeGraphProfile `json:"social"`
	Videos  []VideoResult           `json:"videos"`
	Images  []Image                 `json:"images"`
}

type ButtonResult struct {
	Type  string `json:"type"`
	Title string `json:"title"`
	URL   string `json:"url"`
}

type KnowledgeGraphProfile struct {
	KnowledgeGraphEntity

	URL         string `json:"url"`
	Description string `json:"description"`
}

type KnowledgeGraphEntity struct {
	Title       string `json:"title"`
	Description string `json:"description"`
	URL         *URL   `json:"url"`
	Thumbnail   *URL   `json:"thumbnail"`
}

type URL struct {
	Original     string        `json:"original"`
	Display      string        `json:"display"`
	Alternatives []string      `json:"alternatives"`
	Canonical    string        `json:"canonical"`
	Mobile       MobileURLItem `json:"mobile"`
}

type MobileURLItem struct {
	Original string `json:"original"`
	AMP      string `json:"amp"`
	Android  string `json:"android"`
	IOS      string `json:"ios"`
}

type Image struct {
	Thumbnail  *Thumbnail       `json:"thumbnail"`
	URL        string           `json:"url"`
	Properties *ImageProperties `json:"properties"`
	Text       string           `json:"text"`
}

type ImageProperties struct {
	URL         string `json:"url"`
	Resized     string `json:"resized"`
	Height      int    `json:"height"`
	Width       int    `json:"width"`
	Format      string `json:"format"`
	ContentSize string `json:"content_size"`
	Placeholder string `json:"placeholder"`
}

type LocationResult struct {
	Result

	Type           string          `json:"type"`
	ProviderURL    string          `json:"provider_url"`
	Coordinates    []float32       `json:"coordinates"`
	ZoomLevel      int             `json:"zoom_level"`
	Thumbnail      *Thumbnail      `json:"thumbnail"`
	PostalAddress  *PostalAddress  `json:"postal_address"`
	OpeningHours   *OpeningHours   `json:"opening_hours"`
	Contact        *Contact        `json:"contact"`
	PriceRange     string          `json:"price_range"`
	Rating         *Rating         `json:"rating"`
	Distance       *Unit           `json:"distance"`
	Profiles       []DataProvider  `json:"profiles"`
	Reviews        *Reviews        `json:"reviews"`
	Pictures       *PictureResults `json:"pictures"`
	ServesCuisine  []string        `json:"serves_cuisine"`
	Timezone       string          `json:"timezone"`
	TimezoneOffset float32         `json:"timezone_offset"`
}

type PostalAddress struct {
	Type            string `json:"type"`
	Country         string `json:"country"`
	PostalCode      string `json:"postalCode"`
	StreetAddress   string `json:"streetAddress"`
	AddressRegion   string `json:"addressRegion"`
	AddressLocality string `json:"addressLocality"`
	DisplayAddress  string `json:"displayAddress"`
}

type OpeningHours struct {
	CurrentDay []DayOpeningHours   `json:"current_day"`
	Days       [][]DayOpeningHours `json:"days"`
}

type DayOpeningHours struct {
	AbbrName string `json:"abbr_name"`
	FullName string `json:"full_name"`
	Opens    string `json:"opens"`
	Closes   string `json:"closes"`
}

type Contact struct {
	Email     string `json:"email"`
	Telephone string `json:"telephone"`
}

type Rating struct {
	RatingValue   float32  `json:"ratingValue"`
	BestRating    float32  `json:"bestRating"`
	ReviewCount   int      `json:"reviewCount"`
	Profile       *Profile `json:"profile"`
	IsTripadvisor bool     `json:"is_tripadvisor"`
}

type DataProvider struct {
	Type     string `json:"type"`
	Name     string `json:"name"`
	URL      string `json:"url"`
	LongName string `json:"long_name"`
	Image    string `json:"img"`
}

type Unit struct {
	Value float32 `json:"value"`
	Units string  `json:"units"`
}

type Reviews struct {
	Results                  []TripAdvisorReview `json:"results"`
	ViewMoreURL              string              `json:"viewMoreUrl"`
	ReviewsInForeignLanguage bool                `json:"reviews_in_foreign_language"`
}

type TripAdvisorReview struct {
	Title       string  `json:"title"`
	Description string  `json:"description"`
	Date        string  `json:"date"`
	Rating      *Rating `json:"rating"`
	Author      *Person `json:"author"`
	ReviewURL   string  `json:"review_url"`
	Language    string  `json:"language"`
}

type Person struct {
	Type      string     `json:"type"`
	Name      string     `json:"name"`
	URL       string     `json:"url"`
	Thumbnail *Thumbnail `json:"thumbnail"`
}

type PictureResults struct {
	Results     []Thumbnail `json:"results"`
	ViewMoreURL string      `json:"viewMoreUrl"`
}

type Locations struct {
	Type    string           `json:"type"`
	Results []LocationResult `json:"results"`
}

type MovieData struct {
	Name        string     `json:"name"`
	Description string     `json:"description"`
	URL         string     `json:"url"`
	Thumbnail   *Thumbnail `json:"thumbnail"`
	Release     string     `json:"release"`
	Directors   []Person   `json:"directors"`
	Actors      []Person   `json:"actors"`
	Rating      *Rating    `json:"rating"`
}

type FAQ struct {
	Type    string `json:"type"`
	Results []QA   `json:"results"`
}

type QA struct {
	Question string  `json:"question"`
	Answer   string  `json:"answer"`
	Title    string  `json:"title"`
	URL      string  `json:"url"`
	MetaURL  MetaURL `json:"meta_url"`
}

type QAPage struct {
	Question string  `json:"question"`
	Answer   *Answer `json:"answer"`
}

type Answer struct {
	Text          string `json:"text"`
	Author        string `json:"author"`
	UpvoteCount   int    `json:"upvoteCount"`
	DownvoteCount int    `json:"downvoteCount"`
}

type Book struct {
	Title     string   `json:"title"`
	Author    []Person `json:"author"`
	Date      string   `json:"date"`
	Price     *Price   `json:"price"`
	Pages     Number   `json:"pages"`
	Publisher *Person  `json:"publisher"`
	Rating    *Rating  `json:"rating"`
}

type Price struct {
	Price         string `json:"price"`
	PriceCurrency string `json:"price_currency"`
}

type Article struct {
	Author              []Person      `json:"author"`
	Date                string        `json:"date"`
	Publisher           *Organization `json:"publisher"`
	Thumbnail           *Thumbnail    `json:"thumbnail"`
	IsAccessibleForFree bool          `json:"isAccessibleForFree"`
}

type Organization struct {
	Type      string     `json:"type"`
	Name      string     `json:"name"`
	Thumbnail *Thumbnail `json:"thumbnail"`
}

type CreativeWork struct {
	Name      string     `json:"name"`
	Thumbnail *Thumbnail `json:"thumbnail"`
	Rating    *Rating    `json:"rating"`
}

type MusicRecording struct {
	Name      string     `json:"name"`
	Thumbnail *Thumbnail `json:"thumbnail"`
	Rating    *Rating    `json:"rating"`
}

type Review struct {
	Type        string     `json:"type"`
	Name        string     `json:"name"`
	Thumbnail   *Thumbnail `json:"thumbnail"`
	Description string     `json:"description"`
	Rating      *Rating    `json:"rating"`
}

type Software struct {
	Name           string `json:"name"`
	Author         string `json:"author"`
	Version        string `json:"version"`
	CodeRepository string `json:"codeRepository"`
	Homepage       string `json:"homepage"`
	DatePublished  string `json:"datePublisher"`
	IsNPM          bool   `json:"is_npm"`
	IsPyPi         bool   `json:"is_pypi"`
}

type Summarizer struct {
	Type string `json:"type"`
	Key  string `json:"key"`
}

type SummaryMessage struct {
	Type string `json:"type"`
	Data string `json:"data"`
}

type SummaryEnrichments struct {
	Raw      string           `json:"raw"`
	Images   []Image          `json:"images"`
	QA       []SummaryAnswer  `json:"qa"`
	Entities []SummaryEntity  `json:"entities"`
	Context  []SummaryContext `json:"context"`
}

type SummaryAnswer struct {
	Answer    string        `json:"answer"`
	Score     float32       `json:"score"`
	Highlight *TextLocation `json:"highlight"`
}

type SummaryEntity struct {
	UUID      string         `json:"uuid"`
	Name      string         `json:"name"`
	URL       string         `json:"url"`
	Text      string         `json:"text"`
	Images    []Image        `json:"images"`
	Highlight []TextLocation `json:"highlight"`
}

type SummaryContext struct {
	Title   string   `json:"title"`
	URL     string   `json:"url"`
	MetaURL *MetaURL `json:"meta_url"`
}

type TextLocation struct {
	Start Number `json:"start"`
	End   Number `json:"end"`
}

type SuggestResult struct {
	Query       string `json:"string"`
	IsEntity    bool   `json:"is_entity"`
	Title       string `json:"title"`
	Description string `json:"description"`
	Image       string `json:"img"`
}

type SpellcheckResultItem struct {
	Query string `json:"query"`
}

type DiscussionResult struct {
	SearchResult

	Type string    `json:"type"`
	Data ForumData `json:"data"`
}

type ForumData struct {
	ForumName  string `json:"forum_name"`
	NumAnswers int    `json:"num_answers"`
	Score      string `json:"score"`
	Question   string `json:"question"`
	TopComment string `json:"top_comment"`
}

type GraphInfoBox struct {
	Result

	Type            string         `json:"type"`
	Position        int            `json:"position"`
	Label           string         `json:"label"`
	Category        string         `json:"category"`
	LongDesc        string         `json:"long_desc"`
	Thumbnail       *Thumbnail     `json:"thumbnail"`
	Attributes      []any          `json:"attributes"`
	Profiles        []Profile      `json:"profiles"`
	WebsiteURL      string         `json:"website_url"`
	AttributesShown int            `json:"attributes_shown"`
	Ratings         []Rating       `json:"ratings"`
	Providers       []DataProvider `json:"providers"`
	Distance        *Unit          `json:"distance"`
	Images          []Thumbnail    `json:"images"`
	Movie           *MovieData     `json:"movie"`
}

type Product struct {
	Type        string     `json:"type"`
	Name        string     `json:"name"`
	Price       string     `json:"price"`
	Thumbnail   *Thumbnail `json:"thumbnail"`
	Description string     `json:"description"`
	Offers      []Offer    `json:"offers"`
	Rating      *Rating    `json:"rating"`
}

type Offer struct {
	URL           string `json:"url"`
	Price         string `json:"price"`
	PriceCurrency string `json:"priceCurrency"`
}

type ErrorResponse struct {
	ID     string     `json:"id"`
	Status int        `json:"status"`
	Code   string     `json:"code"`
	Detail string     `json:"detail"`
	Meta   ErrorMeta  `json:"meta"`
	Time   *Timestamp `json:"-"`
}

func (er ErrorResponse) Error() string {
	return er.Detail
}

type ErrorMeta struct {
	Component string           `json:"component"`
	Errors    []ErrorMetaError `json:"errors"`
}

type ErrorMetaError struct {
	Loc     []string     `json:"loc"`
	Message string       `json:"msg"`
	Type    string       `json:"type"`
	Context ErrorContext `json:"ctx"`
}

type ErrorContext struct {
	EnumValues []string `json:"enum_values"`
}

type Duration time.Duration

func (d *Duration) Duration() *time.Duration {
	if d == nil {
		return nil
	}

	tt := time.Duration(*d)
	return &tt
}

func (d *Duration) UnmarshalJSON(in []byte) error {
	str := string(in)
	if !strings.Contains(str, `"`) {
		return nil
	}

	str = strings.Trim(string(in), `"`)
	matches := durationRegex.FindAllString(str, -1)
	if l := len(matches); l < 3 {
		for l < 3 {
			matches = append([]string{"00"}, matches...)
			l++
		}
	}

	dur, err := time.ParseDuration(fmt.Sprintf("%sh%sm%ss", matches[0], matches[1], matches[2]))
	if err != nil {
		return err
	}

	*d = Duration(dur)
	return nil
}

type Timestamp time.Time

func (t *Timestamp) Time() *time.Time {
	if t == nil {
		return nil
	}

	tt := time.Time(*t)
	return &tt
}

func (t *Timestamp) UnmarshalJSON(in []byte) (err error) {
	var res time.Time
	str := string(in)
	if strings.Contains(str, `"`) {
		str = strings.Trim(string(in), `"`)
		var err error
		for _, fmt := range timeFormats {
			res, err = time.Parse(fmt, str)
			if err == nil {
				err = nil
				break
			}
		}

		if err != nil {
			var err2 error
			res, err2 = anytime.Parse(str, time.Now())
			if err2 != nil && strings.Contains(str, "second") {
				matches := durationRegex.FindAllString(str, 1)
				if len(matches) == 0 {
					return nil
				}

				seconds, _ := strconv.Atoi(matches[0])
				if seconds == 0 {
					res = time.Now()
				} else {
					res = time.Now().Add(-time.Duration(seconds) * time.Second)
				}
			} else if err2 != nil {
				return nil
			}
		}
	} else {
		i, err := strconv.Atoi(str)
		if err != nil {
			return err
		}

		res = time.Unix(int64(i), 0)
	}

	*t = Timestamp(res)
	return nil
}

type Number int

func (n *Number) UnmarshalJSON(b []byte) error {
	str := string(b)
	str = strings.Trim(str, `"`)
	num, err := strconv.Atoi(str)
	if err != nil {
		return nil
	}

	*n = Number(num)
	return nil
}

var (
	timeFormats = []string{
		time.RFC3339,
		"2006-01-02T15:04:05",
		"January 2, 2006",
	}

	durationRegex = regexp.MustCompile(`(\d{1,2})`)
)