peter/ekster
1
0
Fork 0
Code Issues 3 Pull Requests Projects 3 Releases 1 Wiki Activity

Use microformats module to parse <link> elements

Browse Source
This commit is contained in:
Peter Stuifzand 2018-07-07 20:37:49 +02:00
parent 55b279f363
commit b291019093
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/websub/subscribe.go
View File

@ -9,6 +9,8 @@ import (
"linkheader" "linkheader"
"rss" "rss"
"willnorris.com/go/microformats"
) )
// GetHubURL finds the HubURL for topic // GetHubURL finds the HubURL for topic
@ -51,7 +53,8 @@ func parseBodyLinks(client *http.Client, topic string) (string, error) {
} }
defer resp.Body.Close() defer resp.Body.Close()
if isFeedContentType(resp.Header.Get("Content-Type")) { contentType := resp.Header.Get("Content-Type")
if isFeedContentType(contentType) {
body, err := ioutil.ReadAll(resp.Body) body, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
return "", err return "", err
@ -64,7 +67,16 @@ func parseBodyLinks(client *http.Client, topic string) (string, error) {
if feed.HubURL != "" { if feed.HubURL != "" {
return feed.HubURL, nil return feed.HubURL, nil
} }
return "", fmt.Errorf("No hub url found in RSS feed") return "", fmt.Errorf("No WebSub hub url found in the RSS feed")
} else if strings.HasPrefix(contentType, "text/html") {
topicURL, _ := url.Parse(topic)
md := microformats.Parse(resp.Body, topicURL)
if hubs, e := md.Rels["hub"]; e {
if len(hubs) >= 1 {
return hubs[0], nil
}
}
return "", fmt.Errorf("No WebSub hub url found in HTML <link> elements")
} }
return "", fmt.Errorf("Unknown content type of response: %s", resp.Header.Get("Content-Type")) return "", fmt.Errorf("Unknown content type of response: %s", resp.Header.Get("Content-Type"))