package verify

import (
	"strconv"

	"github.com/sjlleo/netflix-verify/util"
)

type IPv6Verifier struct {
	Config
	IP              string
	unblockStatus   int
	unblockTestChan chan UnblockTestResult
}

func (v *IPv6Verifier) Execute() *VerifyResponse {
	var err error
	var response VerifyResponse
	v.unblockStatus = AreaUnavailable
	response.Type = 2

	if v.IP, err = util.DnsResolver(6); err != nil {
		response.StatusCode = NetworkUnrachable
		return &response
	}

	v.unblockTestChan = make(chan UnblockTestResult)
	defer close(v.unblockTestChan)

	if v.Custom == "" {
		go v.UnblockTest(AreaAvailableID)
		go v.UnblockTest(SelfMadeAvailableID)
		go v.UnblockTest(NonSelfMadeAvailableID)

		for i := 0; i < 3; i++ {
			switch res := <-v.unblockTestChan; {

			case res.err != nil:
				v.unblockStatus = NetworkUnrachable

			case res.CountryCode != "":
				switch res.movieID {
				case AreaAvailableID:
					v.upgradeStatus(AreaAvailable)
				case SelfMadeAvailableID:
					v.upgradeStatus(UnblockSelfMadeMovie)
				case NonSelfMadeAvailableID:
					v.upgradeStatus(UnblockNonSelfMadeMovie)
				}
				response.CountryCode = res.CountryCode
				response.CountryName = util.CountryCodeToCountryName(res.CountryCode)
			default:
			}
		}
	} else {
		if customMovieID, err := strconv.Atoi(v.Custom); err == nil {
			go v.UnblockTest(customMovieID)
		}
		res := <-v.unblockTestChan
		if res.CountryCode != "" {
			v.unblockStatus = CustomMovieUnblock
			response.CountryCode = res.CountryCode
			response.CountryName = util.CountryCodeToCountryName(res.CountryCode)
		} else {
			v.unblockStatus = CustomMovieBlock
		}
	}
	response.StatusCode = v.unblockStatus
	return &response
}

func (v *IPv6Verifier) upgradeStatus(status int) {
	if v.unblockStatus < status {
		v.unblockStatus = status
	}
}

func (v *IPv6Verifier) UnblockTest(MoiveID int) {

	testURL := NetflixURL_PREFIX + strconv.Itoa(MoiveID)
	if reCode, err := util.RequestIP(testURL, v.IP, v.LocalAddr, v.Proxy); err != nil {
		if err.Error() == "Banned" {
			v.unblockTestChan <- UnblockTestResult{MoiveID, "", nil}
		} else {
			v.unblockTestChan <- UnblockTestResult{MoiveID, "", err}
		}
	} else {
		v.unblockTestChan <- UnblockTestResult{MoiveID, reCode, nil}
	}
}