From 7b32f59281953ec7ca3a457109591e3187dc6c76 Mon Sep 17 00:00:00 2001
From: Aled Griffiths <aled.griffiths@gmail.com>
Date: Wed, 26 Feb 2020 21:32:18 +0000
Subject: [PATCH] Add refreshAllFeeds to the API

Adds refreshAllFeeds to the API, following the suggestion given in the
comments of #359.
---
 api/api.go             |  6 ++++--
 api/feed.go            | 15 +++++++++++++++
 api/handler.go         |  2 ++
 service/httpd/httpd.go |  2 +-
 4 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/api/api.go b/api/api.go
index 167693f1..f86d6e6a 100644
--- a/api/api.go
+++ b/api/api.go
@@ -7,13 +7,14 @@ package api // import "miniflux.app/api"
 import (
 	"miniflux.app/reader/feed"
 	"miniflux.app/storage"
+	"miniflux.app/worker"
 
 	"github.com/gorilla/mux"
 )
 
 // Serve declares API routes for the application.
-func Serve(router *mux.Router, store *storage.Storage, feedHandler *feed.Handler) {
-	handler := &handler{store, feedHandler}
+func Serve(router *mux.Router, store *storage.Storage, pool *worker.Pool, feedHandler *feed.Handler) {
+	handler := &handler{store, pool, feedHandler}
 
 	sr := router.PathPrefix("/v1").Subrouter()
 	sr.Use(newMiddleware(store).serve)
@@ -31,6 +32,7 @@ func Serve(router *mux.Router, store *storage.Storage, feedHandler *feed.Handler
 	sr.HandleFunc("/discover", handler.getSubscriptions).Methods("POST")
 	sr.HandleFunc("/feeds", handler.createFeed).Methods("POST")
 	sr.HandleFunc("/feeds", handler.getFeeds).Methods("GET")
+	sr.HandleFunc("/feeds/refresh", handler.refreshAllFeeds).Methods("PUT")
 	sr.HandleFunc("/feeds/{feedID}/refresh", handler.refreshFeed).Methods("PUT")
 	sr.HandleFunc("/feeds/{feedID}", handler.getFeed).Methods("GET")
 	sr.HandleFunc("/feeds/{feedID}", handler.updateFeed).Methods("PUT")
diff --git a/api/feed.go b/api/feed.go
index bcf58842..ffd3928c 100644
--- a/api/feed.go
+++ b/api/feed.go
@@ -82,6 +82,21 @@ func (h *handler) refreshFeed(w http.ResponseWriter, r *http.Request) {
 	json.NoContent(w, r)
 }
 
+func (h *handler) refreshAllFeeds(w http.ResponseWriter, r *http.Request) {
+	userID := request.UserID(r)
+	jobs, err := h.store.NewUserBatch(userID, h.store.CountFeeds(userID))
+	if err != nil {
+		json.ServerError(w, r, err)
+		return
+	}
+
+	go func() {
+		h.pool.Push(jobs)
+	}()
+
+	json.NoContent(w, r)
+}
+
 func (h *handler) updateFeed(w http.ResponseWriter, r *http.Request) {
 	feedID := request.RouteInt64Param(r, "feedID")
 	feedChanges, err := decodeFeedModificationPayload(r.Body)
diff --git a/api/handler.go b/api/handler.go
index 8585dcdb..885aa36c 100644
--- a/api/handler.go
+++ b/api/handler.go
@@ -7,9 +7,11 @@ package api // import "miniflux.app/api"
 import (
 	"miniflux.app/reader/feed"
 	"miniflux.app/storage"
+	"miniflux.app/worker"
 )
 
 type handler struct {
 	store       *storage.Storage
+	pool        *worker.Pool
 	feedHandler *feed.Handler
 }
diff --git a/service/httpd/httpd.go b/service/httpd/httpd.go
index 51399698..295e9ba9 100644
--- a/service/httpd/httpd.go
+++ b/service/httpd/httpd.go
@@ -166,7 +166,7 @@ func setupHandler(store *storage.Storage, feedHandler *feed.Handler, pool *worke
 	router.Use(middleware)
 
 	fever.Serve(router, store)
-	api.Serve(router, store, feedHandler)
+	api.Serve(router, store, pool, feedHandler)
 	ui.Serve(router, store, pool, feedHandler)
 
 	router.HandleFunc("/healthcheck", func(w http.ResponseWriter, r *http.Request) {