package client

import (
	"fmt"
	"net/http"

	"github.com/PuerkitoBio/goquery"
)

type Review struct {
	Source string
	Review string
}

type IMDB struct {
}

func NewIMDB() *IMDB {
	return &IMDB{}
}

func (i *IMDB) GetReviews(imdbID string) (map[string]string, error) {
	url := fmt.Sprintf("https://www.imdb.com/title/%s/reviews", imdbID)
	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		return nil, err
	}

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		return nil, err
	}
	if res.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("unexpected status code: %d", res.StatusCode)
	}

	doc, err := goquery.NewDocumentFromReader(res.Body)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()

	reviews := make(map[string]string)
	doc.Find(".lister-item-content").Each(func(i int, reviewNode *goquery.Selection) {

		var permaLink string
		reviewNode.Find("a").Each(func(i int, s *goquery.Selection) {
			if s.Text() == "Permalink" {
				link, exists := s.Attr("href")
				if exists {
					permaLink = link
				}
			}
		})

		if permaLink == "" {
			return
		}

		reviews[permaLink] = reviewNode.Text()
	})

	return reviews, nil
}