Working on login script

main.go

@@ -1,37 +1,27 @@
 package main
 
 import (
-	"fmt"
-	"github.com/PuerkitoBio/goquery"
 	"log"
-	"net/http"
+	"os"
 )
 
 func main() {
 	db := openDatabase()
 	defer db.Close()
 
-	resp, err := http.Get("https://vivaplus.tv/supporters/sign_in")
+	username := os.Getenv("VIVAPLUS_USER")
-	if err != nil {
+	password := os.Getenv("VIVAPLUS_PASS")
-		log.Fatalf("error getting sign in page: %v", err)
-	}
-	defer resp.Body.Close()
-
-	doc, err := goquery.NewDocumentFromReader(resp.Body)
-	if err != nil {
-		log.Fatalf("error parsing sign in page: %v", err)
-	}
-
-	doc.Find("meta[name='csrf-token']").Each(func(i int, s *goquery.Selection) {
-		fmt.Println(s.Attr("content"))
-	})
-
-	//username := os.Getenv("VIVAPLUS_USER")
 	//password, err := base64.StdEncoding.DecodeString(os.Getenv("VIVAPLUS_PASS"))
 	//if err != nil {
 	//	log.Fatalf("error decoding password: %v", err)
 	//}
-	//
+
+	w := NewWebClient()
+	err := w.VivaLogin(username, string(password))
+	if err != nil {
+		log.Fatalf("error login in: %v", err)
+	}
+
 	//form := url.Values{}
 	//form.Set("email", username)
 	//form.Set("password", string(password))

vivaweb.go

@@ -0,0 +1,105 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"github.com/PuerkitoBio/goquery"
+	"log"
+	"net/http"
+	"net/http/cookiejar"
+	"net/url"
+	"strings"
+)
+
+const BASE_URL = "https://vivaplus.tv/"
+const SIGN_IN_URL = BASE_URL + "supporters/sign_in"
+
+type WebClient struct {
+	csrfToken string
+	cookies   *cookiejar.Jar
+}
+
+func NewWebClient() *WebClient {
+	jar, err := cookiejar.New(nil)
+	if err != nil {
+		log.Fatalf("error creating cookiejar: %w", err)
+	}
+	return &WebClient{
+		cookies: jar,
+	}
+}
+
+func (w *WebClient) RegisterCookies(rawUrl string, resp *http.Response) {
+	u, err := url.Parse(rawUrl)
+	if err != nil {
+		log.Fatalf("error parsing url: %v", err)
+	}
+	w.cookies.SetCookies(u, resp.Cookies())
+}
+
+func (w *WebClient) UseCookies(rawUrl string, req *http.Request) {
+	u, err := url.Parse(rawUrl)
+	if err != nil {
+		log.Fatalf("error parsing url: %v", err)
+	}
+	for _, c := range w.cookies.Cookies(u) {
+		req.AddCookie(c)
+	}
+}
+
+func (w *WebClient) FetchCsrfToken() error {
+	resp, err := http.Get(SIGN_IN_URL)
+	if err != nil {
+		return fmt.Errorf("error getting sign in page: %w", err)
+	}
+	defer resp.Body.Close()
+
+	doc, err := goquery.NewDocumentFromReader(resp.Body)
+	if err != nil {
+		return fmt.Errorf("error parsing sign in page: %w", err)
+	}
+	w.RegisterCookies(SIGN_IN_URL, resp)
+
+	csrfTokenElement := doc.Find("meta[name='csrf-token']").First()
+	if csrfTokenElement == nil {
+		return errors.New("error getting csrf token element")
+	}
+	var exists bool
+	w.csrfToken, exists = csrfTokenElement.Attr("content")
+	if !exists {
+		return errors.New("error content attribute does not exist")
+	}
+	return nil
+}
+
+func (w *WebClient) VivaLogin(username, password string) error {
+	err := w.FetchCsrfToken()
+	if err != nil {
+		return err
+	}
+
+	form := url.Values{}
+	form.Set("email", username)
+	form.Set("password", password)
+
+	println("Encoded form:", form.Encode())
+	println("CSRF token:", w.csrfToken)
+	req, err := http.NewRequest("POST", SIGN_IN_URL, strings.NewReader(form.Encode()))
+	w.UseCookies(SIGN_IN_URL, req)
+	for _, c := range req.Cookies() {
+		println("Cookie:", c.Name, "=", c.Value)
+	}
+	if err != nil {
+		return fmt.Errorf("error creating login request: %w", err)
+	}
+	req.Header.Add("x-csrf-token", w.csrfToken)
+	req.Header.Add("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8")
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("error logging in: %w", err)
+	}
+
+	log.Printf("Status code: %d", resp.StatusCode)
+
+	return nil
+}