Add flag to specify number of articles per source

openring.go

@@ -28,11 +28,12 @@ type Article struct {
 func main() {
 	var (
 		narticles  int = 3
+		perSource  int = 1
 		summaryLen int = 256
 		sources    []*url.URL
 	)
 
-	opts, optind, err := getopt.Getopts(os.Args[1:], "l:n:s:")
+	opts, optind, err := getopt.Getopts(os.Args[1:], "l:n:p:s:")
 	if err != nil {
 		panic(err)
 	}
@@ -48,6 +49,11 @@ func main() {
 			if err != nil {
 				panic(err)
 			}
+		case 'p':
+			perSource, err = strconv.Atoi(opt.Value)
+			if err != nil {
+				panic(err)
+			}
 		case 's':
 			u, err := url.Parse(opt.Value)
 			if err != nil {
@@ -103,7 +109,11 @@ func main() {
 			log.Printf("Warning: feed %s has no items", feed.Title)
 			continue
 		}
-		item := feed.Items[0]
+		items := feed.Items
+		if len(items) > perSource {
+			items = items[:perSource]
+		}
+		for _, item := range items {
 			summary := runewidth.Truncate(
 				policy.Sanitize(item.Summary), summaryLen, "…")
 			articles = append(articles, &Article{
@@ -115,6 +125,7 @@ func main() {
 				Link:        item.Link,
 			})
 		}
+	}
 	sort.Slice(articles, func(i, j int) bool {
 		return articles[i].Date.After(articles[j].Date)
 	})