Merge pull request #2 from apollo1220/stats-request

Add stats request

controllers/podcast.go

@@ -645,3 +645,33 @@ func UpdateSetting(c *gin.Context) {
 	}
 
 }
+
+func GetStats(c *gin.Context) {
+	var items []db.PodcastItem
+
+	if err := db.GetAllPodcastItems(&items); err != nil {
+		fmt.Println(err.Error())
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request"})
+	}
+
+	var podcasts []db.Podcast
+	if err := db.GetAllPodcasts(&podcasts, ""); err != nil {
+		fmt.Println(err.Error())
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request"})
+	}
+
+	downloadedCount := 0
+	listenedCount := 0
+	podcastCount := len(podcasts)
+	for _, i := range items {
+		if i.DownloadStatus == db.Downloaded {
+			downloadedCount++
+		}
+		if i.IsPlayed {
+			listenedCount++
+		}
+	}
+
+	c.JSON(200, gin.H{"downloaded": downloadedCount, "listened": listenedCount, "podcasts": podcastCount})
+
+}

main.go

@@ -181,6 +181,8 @@ func main() {
 	router.POST("/podcasts/:id/tags/:tagId", controllers.AddTagToPodcast)
 	router.DELETE("/podcasts/:id/tags/:tagId", controllers.RemoveTagFromPodcast)
 
+	router.GET("/stats", controllers.GetStats)
+
 	router.GET("/add", controllers.AddPage)
 	router.GET("/search", controllers.Search)
 	router.GET("/", controllers.HomePage)