From 008819e5dca92d2d69f838a601d51d770c28416b Mon Sep 17 00:00:00 2001 From: Kevin Chabowski Date: Sat, 30 Nov 2013 20:15:32 +0100 Subject: Initial commit --- LICENSE | 4 ++ README.md | 15 ++++++++ main.go | 129 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 148 insertions(+) create mode 100644 LICENSE create mode 100644 README.md create mode 100644 main.go diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..c74f13c --- /dev/null +++ b/LICENSE @@ -0,0 +1,4 @@ + DO WHATEVER THE FUCK YOU WANT, PUBLIC LICENSE + TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION + + 0. You just DO WHATEVER THE FUCK YOU WANT. diff --git a/README.md b/README.md new file mode 100644 index 0000000..1f9d32c --- /dev/null +++ b/README.md @@ -0,0 +1,15 @@ +youtube-feed +============ + +Combine multiple YouTube user feeds into a single RSS-Feed. + +Compiling +--------- + +`go get github.com/kch42/youtube-feed` + +Usage +----- + +Put all the usernames of YouTubers you want to receive updates from into the file `~/.youtube-feed`. +`youtube-feed` will now output the combined feed on `stdout`. You can use youtube-feed with [newsbeuter](http://www.newsbeuter.org). Just put `exec:youtube-feed` into your `urls` file. diff --git a/main.go b/main.go new file mode 100644 index 0000000..b6842ce --- /dev/null +++ b/main.go @@ -0,0 +1,129 @@ +package main + +import ( + "bufio" + "encoding/xml" + "fmt" + "io" + "net/http" + "os" + "strings" +) + +type Item struct { + GUID string `xml:"guid"` + PubDate string `xml:"pubDate"` + Title string `xml:"title"` + Description string `xml:"description"` + Link string `xml:"link"` + Author string `xml:"author"` +} + +type Feed struct { + XMLName xml.Name `xml:"rss"` + Version string `xml:"version,attr"` + Items []Item `xml:"channel>item"` + Title string `xml:"channel>title"` +} + +func Parse(r io.Reader) ([]Item, error) { + var f Feed + dec := xml.NewDecoder(r) + err := dec.Decode(&f) + return f.Items, err +} + +func subscribedTo() ([]string, error) { + var chans []string + + p := os.ExpandEnv("$HOME/.youtube-feed") + f, err := os.Open(p) + if err != nil { + return nil, fmt.Errorf("failed reading %s: %s", p, err) + } + defer f.Close() + + scan := bufio.NewScanner(f) + + for scan.Scan() { + chans = append(chans, strings.TrimSpace(scan.Text())) + } + if err := scan.Err(); err != nil { + return nil, fmt.Errorf("failed reading %s: %s", p, err) + } + + return chans, err +} + +const retries = 5 + +func getLatestVideos(ytChan string, itemChan chan<- Item, status chan<- error) { + var resp *http.Response + var err error + + for i := 0; i < retries; i++ { + resp, err = http.Get("http://gdata.youtube.com/feeds/base/users/" + ytChan + "/uploads?alt=rss&v=2&orderby=published&client=ytapi-youtube-profile") + if err == nil { + if resp.StatusCode == http.StatusOK { + break + } else { + resp.Body.Close() + err = fmt.Errorf("Received status %d, expected %d", resp.StatusCode, http.StatusOK) + } + } + } + + if err != nil { + status <- fmt.Errorf("Failed getting channel feed for %s: %s", ytChan, err) + return + } + defer resp.Body.Close() + + items, err := Parse(resp.Body) + if err != nil { + err = fmt.Errorf("Failed parsing feed of %s: %s", ytChan, err) + } + + for _, it := range items { + itemChan <- it + } + + status <- err +} + +func main() { + subs, err := subscribedTo() + if err != nil { + fmt.Fprintf(os.Stderr, "Failed reading channel list: %s\n", err) + os.Exit(1) + } + + itemChan := make(chan Item) + status := make(chan error) + + var items []Item + go func() { + for it := range itemChan { + it.Title = fmt.Sprintf("[%s] %s", it.Author, it.Title) + items = append(items, it) + } + }() + + for _, ytChan := range subs { + go getLatestVideos(ytChan, itemChan, status) + } + + for _ = range subs { + if err := <-status; err != nil { + fmt.Fprintln(os.Stderr, "Warning:", err) + } + } + close(itemChan) + + feed := Feed{Title: "Combined YouTube Feed", Items: items, Version: "2.0"} + enc := xml.NewEncoder(os.Stdout) + if err := enc.Encode(feed); err != nil { + fmt.Fprintf(os.Stderr, "Failed writing feed: %s\n", err) + os.Exit(1) + } +} -- cgit v1.2.3-54-g00ecf