You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
2.7 KiB
121 lines
2.7 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/akhilrex/podgrab/controllers"
|
|
"github.com/akhilrex/podgrab/db"
|
|
"github.com/akhilrex/podgrab/service"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/jasonlvhit/gocron"
|
|
_ "github.com/joho/godotenv/autoload"
|
|
)
|
|
|
|
func main() {
|
|
|
|
// os.Remove("./podgrab.db")
|
|
|
|
var err error
|
|
db.DB, err = db.Init()
|
|
if err != nil {
|
|
fmt.Println("statuse: ", err)
|
|
}
|
|
db.Migrate()
|
|
|
|
r := gin.Default()
|
|
dataPath := os.Getenv("DATA")
|
|
//r.Static("/assets", "./assets")
|
|
r.Static("/assets", dataPath)
|
|
r.LoadHTMLGlob("client/*")
|
|
|
|
r.GET("/podcasts", controllers.AddPodcast)
|
|
r.POST("/podcasts", controllers.GetAllPodcasts)
|
|
r.GET("/podcasts/:id", controllers.GetPodcastById)
|
|
r.GET("/podcasts/:id/items", controllers.GetPodcastItemsByPodcastId)
|
|
|
|
r.GET("/podcastitems", controllers.GetAllPodcastItems)
|
|
r.GET("/podcastitems/:id", controllers.GetPodcastItemById)
|
|
|
|
r.GET("/ping", func(c *gin.Context) {
|
|
|
|
data, err := service.AddPodcast(c.Query("url"))
|
|
go service.AddPodcastItems(&data)
|
|
//data, err := db.GetAllPodcastItemsToBeDownloaded()
|
|
if err == nil {
|
|
c.JSON(200, data)
|
|
} else {
|
|
c.JSON(http.StatusInternalServerError, err.Error())
|
|
}
|
|
})
|
|
r.GET("/pong", func(c *gin.Context) {
|
|
|
|
data, err := db.GetAllPodcastItemsToBeDownloaded()
|
|
|
|
for _, item := range *data {
|
|
url, _ := service.Download(item.FileURL, item.Title, item.Podcast.Title)
|
|
service.SetPodcastItemAsDownloaded(item.ID, url)
|
|
}
|
|
|
|
if err == nil {
|
|
c.JSON(200, data)
|
|
} else {
|
|
c.JSON(http.StatusInternalServerError, err.Error())
|
|
}
|
|
})
|
|
|
|
r.GET("/", func(c *gin.Context) {
|
|
//var podcasts []db.Podcast
|
|
podcasts := service.GetAllPodcasts()
|
|
c.HTML(http.StatusOK, "index.html", gin.H{"title": "Podgrab", "podcasts": podcasts})
|
|
})
|
|
r.POST(
|
|
"/", func(c *gin.Context) {
|
|
var addPodcastData controllers.AddPodcastData
|
|
err := c.ShouldBind(&addPodcastData)
|
|
|
|
if err == nil {
|
|
|
|
_, err = service.AddPodcast(addPodcastData.Url)
|
|
if err == nil {
|
|
c.Redirect(http.StatusFound, "/")
|
|
|
|
} else {
|
|
|
|
c.JSON(http.StatusBadRequest, err)
|
|
|
|
}
|
|
} else {
|
|
// fmt.Println(err.Error())
|
|
c.JSON(http.StatusBadRequest, err)
|
|
}
|
|
|
|
})
|
|
|
|
go assetEnv()
|
|
go intiCron()
|
|
|
|
r.Run() // listen and serve on 0.0.0.0:8080 (for windows "localhost:8080")
|
|
|
|
}
|
|
|
|
func intiCron() {
|
|
checkFrequency, err := strconv.Atoi(os.Getenv("CHECK_FREQUENCY"))
|
|
if err != nil {
|
|
checkFrequency = 10
|
|
log.Print(err)
|
|
}
|
|
gocron.Every(uint64(checkFrequency)).Hours().Do(service.DownloadMissingEpisodes)
|
|
gocron.Every(uint64(checkFrequency)).Hours().Do(service.RefreshEpisodes)
|
|
<-gocron.Start()
|
|
}
|
|
|
|
func assetEnv() {
|
|
log.Println("Config Dir: ", os.Getenv("CONFIG"))
|
|
log.Println("Assets Dir: ", os.Getenv("DATA"))
|
|
log.Println("Check Frequency (hrs): ", os.Getenv("CHECK_FREQUENCY"))
|
|
}
|