Split logic to multiple files

1 files changed, 84 insertions, 0 deletions

diff --git a/parse.go b/parse.go
new file mode 100644
index 0000000..6be416e
--- /dev/null
+++ b/parse.go
@@ -0,0 +1,84 @@
+package main
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"strconv"
+	"strings"
+
+	"golang.org/x/net/html"
+)
+
+func parse(body []byte) ([]flat, error) {
+	doc, err := html.Parse(bytes.NewReader(body))
+	if err != nil {
+		return make([]flat, 0), err
+	}
+	flats := make([]flat, 0)
+	for _, n := range findNodes(doc) {
+		flat, err := parseNode(n)
+		if err != nil {
+			continue
+		}
+		flats = append(flats, flat)
+	}
+	return flats, nil
+}
+
+func findNodes(root *html.Node) []*html.Node {
+	flats := make([]*html.Node, 0)
+	for n := range root.Descendants() {
+		if n.Type != html.ElementNode {
+			continue
+		}
+		if n.Data != "a" {
+			continue
+		}
+		attr := matchAttr(n, "data-testid")
+		if attr == nil || attr.Val != "property-price" {
+			continue
+		}
+		flats = append(flats, n)
+	}
+	return flats
+}
+
+func matchAttr(n *html.Node, key string) *html.Attribute {
+	for _, attr := range n.Attr {
+		if attr.Key == key {
+			return &attr
+		}
+	}
+	return nil
+}
+
+func parseNode(root *html.Node) (flat, error) {
+	url := matchAttr(root, "href")
+	if url == nil {
+		return flat{}, errors.New("Couldn't find URL")
+	}
+	ID, err := parseID(url.Val)
+	if err != nil {
+		return flat{}, err
+	}
+	f := flat{ID, ""}
+	for n := range root.Descendants() {
+		if price, found := strings.CutSuffix(n.Data, " pcm"); found {
+			f.Price = price
+			return f, nil
+		}
+	}
+	return flat{}, errors.New("Couldn't find price")
+}
+
+func parseID(path string) (int, error) {
+	s, _ := strings.CutPrefix(path, "/properties/")
+	maybeID, _ := strings.CutSuffix(s, "#/?channel=RES_LET")
+	ID, err := strconv.Atoi(maybeID)
+	if err != nil {
+		err := fmt.Errorf("Couldn't extract ID from %q", path)
+		return -1, err
+	}
+	return ID, err
+}