package main

import (
	log "github.com/Sirupsen/logrus"
	"regexp"
	"strings"

	"github.com/PuerkitoBio/goquery"
	"github.com/gocolly/colly"
)

func ScrapeWhic(shop Shop) []Angebot {
	Shop_url := "https://whic.de/angebote"
	Whiskys := []Angebot{}

	c := colly.NewCollector(
		colly.AllowedDomains("whic.de"),
	)

	c.OnHTML("li.item", func(e *colly.HTMLElement) {

		if e.Request.URL.String() != Shop_url {
			return
		}

		W := Angebot{}

		whisky_name := e.ChildAttr("a", "title")
		whisky_url := e.ChildAttr("a", "href")

		W.Name = whisky_name
		W.Url = whisky_url

		var err error

		e.ForEach(".price-box", func(i int, e *colly.HTMLElement) {
			e.ForEach(".old-price", func(i int, e *colly.HTMLElement) {
				W.Original_price, err = convert_price(e.ChildText(".price"))
				if err != nil {
					log.Fatal(err)
				}
			})
			e.ForEach(".special-price", func(i int, e *colly.HTMLElement) {
				W.Discounted_price, err = convert_price(e.ChildText(".price"))
				if err != nil {
					log.Fatal(err)
				}
			})
		})

		base_price_noisy := e.ChildText(".base-price")
		W.Base_price, err = sanitize_base_price(base_price_noisy)
		if err != nil {
			log.Fatal(err)
		}

		/*
		 * colly does not parse a <noscript>, thus we are reading the content and parse it as html.
		 */
		img_link_noisy := e.ChildText(".product-image")

		doc, err := goquery.NewDocumentFromReader(strings.NewReader(img_link_noisy))
		if err != nil {
			log.Fatal(err)
		}
		W.Image_url, _ = doc.Find("img").Attr("src")

		e.Request.Visit(W.Url)
		W.Volume = get_volume(e)
		W.Abv = get_abv(e)

		W.Shop = shop.Id
		W.Spirit_type = "Whisky"
		W.Website = e.Request.Ctx.Get("website")

		Whiskys = append(Whiskys, W)
	})

	c.OnHTML("#product-view-head-txt-extra-info", func(e *colly.HTMLElement) {
		text_noisy := e.Text

		r_volume, err := regexp.Compile("Volumen: ([0-9]+([.,][0-9]+)) Liter")
		if err != nil {
			log.Fatal(err)
		}
		volume := r_volume.FindStringSubmatch(text_noisy)
		if volume == nil || len(volume) < 2 {
			return
		}

		e.Request.Ctx.Put("volume", volume[1]+"l")

		r_abv, err := regexp.Compile("Alkoholgehalt: ([0-9]+([.,][0-9]+))%")
		if err != nil {
			log.Fatal(err)
		}
		abv := r_abv.FindStringSubmatch(text_noisy)
		if abv == nil || len(abv) < 2 {
			return
		}

		e.Request.Ctx.Put("abv", abv[1]+"%")
		e.Request.Ctx.Put("website", string(e.Response.Body))
	})

	c.Visit(Shop_url)

	return Whiskys
}