forked from zv0n/lunch-go
77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
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
|
|
}
|