add eu cookie support

1 files changed, 99 insertions, 21 deletions

diff --git a/yahoo_crumb.go b/yahoo_crumb.go
index 6a951c4..6a774be 100644
--- a/yahoo_crumb.go
+++ b/yahoo_crumb.go
@@ -5,14 +5,18 @@
 package mop
 
 import (
-	"io/ioutil"
+	"io"
 	"net/http"
+	"net/url"
+	"regexp"
+	"strconv"
 	"strings"
 )
 
 const crumbURL = "https://query1.finance.yahoo.com/v1/test/getcrumb"
-const cookieURL = "https://login.yahoo.com"
+const cookieURL = "https://finance.yahoo.com/"
 const userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/113.0"
+const euConsentURL = "https://consent.yahoo.com/v2/collectConsent?sessionId="
 
 func fetchCrumb(cookies string) string {
 	client := http.Client{}
@@ -42,7 +46,7 @@ func fetchCrumb(cookies string) string {
 	}
 	defer response.Body.Close()
 
-	body, err := ioutil.ReadAll(response.Body)
+	body, err := io.ReadAll(response.Body)
 	if err != nil {
 		panic(err)
 	}
@@ -51,25 +55,27 @@ func fetchCrumb(cookies string) string {
 }
 
 func fetchCookies() string {
+
 	client := http.Client{}
+	var cookies []*http.Cookie
+
+	// Get the session ID from the first request
 	request, err := http.NewRequest("GET", cookieURL, nil)
 	if err != nil {
 		panic(err)
 	}
 
 	request.Header = http.Header{
-		"Accept":                   {"*/*"},
-		"Accept-Encoding":          {"gzip, deflate, br"},
-		"Accept-Language":          {"en-US,en;q=0.5"},
-		"Connection":               {"keep-alive"},
-		"Host":                     {"login.yahoo.com"},
-		"Sec-Fetch-Dest":           {"document"},
-		"Sec-Fetch-Mode":           {"navigate"},
-		"Sec-Fetch-Site":           {"none"},
-		"Sec-Fetch-User":           {"?1"},
-		"TE":                       {"trailers"},
-		"Update-Insecure-Requests": {"1"},
-		"User-Agent":               {userAgent},
+		"Authority":                 {"finance.yahoo.com"},
+		"Accept":                    {"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7"},
+		"Accept-Encoding":           {"gzip, deflate, br"},
+		"Accept-Language":           {"en-US,en;q=0.9"},
+		"Sec-Fetch-Dest":            {"document"},
+		"Sec-Fetch-Mode":            {"navigate"},
+		"Sec-Fetch-Site":            {"none"},
+		"Sec-Fetch-User":            {"?1"},
+		"Upgrade-Insecure-Requests": {"1"},
+		"User-Agent":                {userAgent},
 	}
 
 	response, err := client.Do(request)
@@ -78,12 +84,84 @@ func fetchCookies() string {
 	}
 	defer response.Body.Close()
 
-	var result string
-	for _, cookie := range response.Cookies() {
-		if cookie.Name != "AS" {
-			result += cookie.Name + "=" + cookie.Value + "; "
+	cookieA1 := getA1Cookie(response.Cookies())
+	if cookieA1 != "" {
+		return cookieA1
+	}
+
+	// first pass failed - try EU shenanigans
+
+	sessionRegex := regexp.MustCompile("sessionId=(?:([A-Za-z0-9_-]*))")
+	sessionID := sessionRegex.FindStringSubmatch(response.Request.URL.RawQuery)[1]
+
+	csrfRegex := regexp.MustCompile("gcrumb=(?:([A-Za-z0-9_]*))")
+	csrfToken := csrfRegex.FindStringSubmatch(response.Request.Response.Request.URL.RawQuery)[1]
+
+	gucsCookie := response.Request.Response.Request.Response.Cookies()
+	var gucsCookieString string = ""
+	for _, cookie := range gucsCookie {
+		gucsCookieString += cookie.Name + "=" + cookie.Value + "; "
+	}
+	gucsCookieString = strings.TrimSuffix(gucsCookieString, "; ")
+
+	if len(gucsCookie) == 0 {
+		panic(err)
+	}
+
+	// Create a new request to agree to the EU consent request
+	form := url.Values{}
+	form.Add("csrfToken", csrfToken)
+	form.Add("sessionId", sessionID)
+	form.Add("namespace", "yahoo")
+	form.Add("agree", "agree")
+	request2, err := http.NewRequest("POST", euConsentURL+sessionID, strings.NewReader(form.Encode()))
+	if err != nil {
+		panic(err)
+	}
+
+	contentLength := strconv.FormatInt(int64(len(form.Encode())), 10)
+
+	request2.Header = http.Header{
+		"Accept":                    {"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7"},
+		"Accept-Encoding":           {"gzip, deflate, br"},
+		"Accept-Language":           {"en-US,en;q=0.9"},
+		"Connection":                {"keep-alive"},
+		"Cookie":                    {gucsCookieString},
+		"Content-Length":            {contentLength},
+		"Content-Type":              {"application/x-www-form-urlencoded"},
+		"DNT":                       {"1"},
+		"Host":                      {"consent.yahoo.com"},
+		"Origin":                    {"https://consent.yahoo.com"},
+		"Referer":                   {euConsentURL + sessionID},
+		"Sec-Fetch-Dest":            {"document"},
+		"Sec-Fetch-Mode":            {"navigate"},
+		"Sec-Fetch-Site":            {"same-origin"},
+		"Sec-Fetch-User":            {"?1"},
+		"Upgrade-Insecure-Requests": {"1"},
+		"User-Agent":                {userAgent},
+	}
+
+	response2, err := client.Do(request2)
+	if err != nil {
+		panic(err)
+	}
+	defer response2.Body.Close()
+
+	// redirect festival
+	cookies = response2.Request.Response.Request.Response.Request.Response.Cookies()
+	cookieA1 = getA1Cookie(cookies)
+	if cookieA1 != "" {
+		return cookieA1
+	} else {
+		panic(err)
+	}
+}
+
+func getA1Cookie(cookies []*http.Cookie) string {
+	for _, cookie := range cookies {
+		if cookie.Name == "A1" {
+			return cookie.Name + "=" + cookie.Value + "; "
 		}
 	}
-	result = strings.TrimSuffix(result, "; ")
-	return result
+	return ""
 }