From 3495579ebfa2f0efc98b799f499a3b826f9d2f5f Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Fri, 15 Dec 2023 08:36:08 -0600
Subject: [PATCH] Non-VIP Form getter

---
 api.go   | 73 +++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 types.go |  5 ++++
 2 files changed, 77 insertions(+), 1 deletion(-)

diff --git a/api.go b/api.go
index ae2c8fc..d7f941a 100644
--- a/api.go
+++ b/api.go
@@ -2,11 +2,14 @@ package main
 
 import (
 	"bytes"
+	"fmt"
+	"io"
 	"net/http"
 	"net/http/cookiejar"
 	"regexp"
 	"slices"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/PuerkitoBio/goquery"
@@ -34,7 +37,7 @@ func init() {
 }
 
 func onRequest(req *http.Request) {
-	log.Debugf("GET %s", req.URL.String())
+	log.Debugf("%s %s", req.Method, req.URL.String())
 	requestCounter++
 }
 
@@ -147,3 +150,71 @@ func GetLocations() []Location {
 
 	return cachedLocations
 }
+
+type GetFormResult struct {
+	propertyName     string
+	address          string
+	fields           []Field  // label & inputs in the form
+	hiddenInputs     []string // hidden inputs in the form
+	requireGuestCode bool     // whether a guest code is required
+	err              error    // any error that occurred
+}
+
+func GetForm(id uint) GetFormResult {
+	body := fmt.Sprintf("propertyIdSelected=%d&propertySource=parking-snap", id)
+	req := BuildRequestWithBody("POST", "/register-get-vehicle-form", nil, bytes.NewBufferString(body))
+	SetTypicalHeaders(req, nil, nil, false)
+	onRequest(req)
+
+	res, _ := client.Do(req)
+	onResponse(res)
+
+	html, _ := io.ReadAll(res.Body)
+	htmlString := string(html)
+
+	doc, err := goquery.NewDocumentFromReader(bytes.NewBufferString(htmlString))
+	if err != nil {
+		return GetFormResult{err: err}
+	}
+
+	// Check if this form is a VIP property
+	guestCodeInput := doc.Find("div.form-group > input#guestCode").First()
+	if guestCodeInput.Length() == 1 {
+		log.Debugf("Guest Code input: %v", guestCodeInput)
+		log.Debug("Guest Code form detected")
+		return GetFormResult{
+			requireGuestCode: true,
+		}
+	}
+
+	// Get the hidden inputs & form fields
+	hiddenInputs := make([]string, 0, 5)
+	formFields := make([]Field, 0, 5)
+	doc.Find("#property-name-form > input[type=hidden]").Each(func(i int, s *goquery.Selection) {
+		hiddenInputs = append(hiddenInputs, s.AttrOr("id", ""))
+	})
+	doc.Find("#property-name-form > div.form-group").Each(func(i int, s *goquery.Selection) {
+		input := s.Find("input.form-control").First()
+		label := s.Find("label").First().Text()
+
+		inputId, _ := input.Attr("id")
+
+		formFields = append(formFields, Field{
+			text: label,
+			id:   inputId,
+		})
+	})
+
+	// Acquire the title/address
+	titleElement := doc.Find("div > div > h4").First()
+	title := titleElement.Text()
+	address := strings.TrimSpace(titleElement.Next().Text())
+
+	return GetFormResult{
+		propertyName:     title,
+		address:          address,
+		fields:           formFields,
+		hiddenInputs:     hiddenInputs,
+		requireGuestCode: false,
+	}
+}
diff --git a/types.go b/types.go
index 83c3462..d28d9c2 100644
--- a/types.go
+++ b/types.go
@@ -6,3 +6,8 @@ type Location struct {
 	name    string // Used for autocomplete & location selection
 	address string // Not used in this application so far
 }
+
+type Field struct {
+	text string // The text displayed
+	id   string // The id of the field
+}