Initial working version

pkg/restaurants/html.go

@@ -0,0 +1,76 @@
+package restaurants
+
+import (
+	"errors"
+
+	"golang.org/x/net/html"
+	"golang.org/x/text/encoding/charmap"
+)
+
+func getAttribute(node *html.Node, key string) (string, error) {
+	for _, attr := range node.Attr {
+		if attr.Key == key {
+			return attr.Val, nil
+		}
+	}
+
+	return "", errors.New("couldn't find the provided key")
+}
+
+func hasClass(node *html.Node, class string) bool {
+	if node.Type == html.ElementNode {
+		c, err := getAttribute(node, "class")
+		if err != nil {
+			return false
+		}
+		return c == class
+	}
+	return false
+}
+
+func findNodeByClass(node *html.Node, class string) (*html.Node, error) {
+	if hasClass(node, class) {
+		return node, nil
+	}
+	for n := node.FirstChild; n != nil; n = n.NextSibling {
+		c, err := findNodeByClass(n, class)
+		if err == nil {
+			return c, nil
+		}
+	}
+	return nil, errors.New("couldn't find a node with provided class")
+}
+
+func getTextInternal(node *html.Node) (string, error) {
+	if node.Type == html.TextNode {
+		return node.Data, nil
+	}
+	return "", errors.New("not a text node")
+}
+
+func getText(node *html.Node) (string, error) {
+	if node.Type == html.TextNode {
+		return node.Data, nil
+	}
+	for n := node.FirstChild; n != nil; n = n.NextSibling {
+		text, err := getTextInternal(n)
+		if err == nil {
+			return text, nil
+		}
+	}
+	return "", errors.New("couldn't find a text node")
+}
+
+func getTextDecodeWindows1250(node *html.Node) (string, error) {
+	text, err := getText(node)
+	if err != nil {
+		return text, err
+	}
+	return decodeWindows1250(text)
+}
+
+func decodeWindows1250(text string) (string, error) {
+	dec := charmap.Windows1250.NewDecoder()
+	out, err := dec.String(text)
+	return out, err
+}