aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--controller/controller.go35
-rw-r--r--db/db.go2
-rw-r--r--db/model.go35
-rw-r--r--routes/routes.go18
4 files changed, 64 insertions, 26 deletions
diff --git a/controller/controller.go b/controller/controller.go
index ffa89ce..252be73 100644
--- a/controller/controller.go
+++ b/controller/controller.go
@@ -10,6 +10,10 @@ import (
"github.com/gin-gonic/gin"
)
+type Env struct {
+ Entries db.EntryModel
+}
+
func spawnWorker(link string, output string) {
work.Submit(func(ctx context.Context) error {
yt.RunYtDlpProcess(link, output)
@@ -17,38 +21,35 @@ func spawnWorker(link string, output string) {
})
}
-func GetEntries(c *gin.Context) {
- var entries []db.YdlEntry
- db.DB.Find(&entries)
+func (e Env) GetEntries(c *gin.Context) {
+ entries := e.Entries.All()
c.HTML(http.StatusOK, "index", entries)
}
-func GetEntry(c *gin.Context) {
- var entry db.YdlEntry
+func (e *Env) GetEntry(c *gin.Context) {
id := c.Param("id")
- where := "id = " + id
- db.DB.Where(where).FirstOrInit(&entry)
+ entry := e.Entries.Find(id)
c.HTML(http.StatusOK, "entry", entry)
}
-func UpdateEntry(c *gin.Context) {
- var entry db.YdlEntry
+func (e *Env) UpdateEntry(c *gin.Context) {
+ var entry db.Entry
c.ShouldBind(&entry)
- db.DB.Save(&entry)
- c.HTML(http.StatusOK, "entry", entry)
+ e.Entries.Update(entry)
+ c.Redirect(http.StatusFound, "/")
}
-func CreateEntry(c *gin.Context) {
- var entry db.YdlEntry
+func (e *Env) CreateEntry(c *gin.Context) {
+ var entry db.Entry
c.ShouldBind(&entry)
- db.DB.Create(&entry)
+ e.Entries.Create(entry)
spawnWorker(entry.Link, entry.OutputFolder)
c.Redirect(http.StatusFound, "/")
}
-func DeleteEntry(c *gin.Context) {
- var entry db.YdlEntry
+func (e *Env) DeleteEntry(c *gin.Context) {
+ var entry db.Entry
id := c.Param("id")
- db.DB.Delete(&entry, id)
+ e.Entries.Delete(id)
c.HTML(http.StatusOK, "entry", entry)
}
diff --git a/db/db.go b/db/db.go
index 128dcc8..5c904fd 100644
--- a/db/db.go
+++ b/db/db.go
@@ -17,5 +17,5 @@ func ConnectDb() {
panic("failed to connect to the database.")
}
- DB.AutoMigrate(&YdlEntry{})
+ DB.AutoMigrate(&Entry{})
}
diff --git a/db/model.go b/db/model.go
index 6f35cd0..4b814f9 100644
--- a/db/model.go
+++ b/db/model.go
@@ -1,11 +1,42 @@
package db
-import "gorm.io/gorm"
+import (
+ "gorm.io/gorm"
+)
-type YdlEntry struct {
+type Entry struct {
gorm.Model
Title string
Link string
Format string
OutputFolder string
}
+
+type EntryModel struct {
+ DB *gorm.DB
+}
+
+func (m EntryModel) Find(id string) Entry {
+ var entry Entry
+ where := "id = " + id
+ m.DB.Where(where).FirstOrInit(&entry)
+ return entry
+}
+
+func (m EntryModel) All() []Entry {
+ var entries []Entry
+ m.DB.Find(&entries)
+ return entries
+}
+
+func (m EntryModel) Create(entry Entry) {
+ m.DB.Create(&entry)
+}
+
+func (m EntryModel) Update(entry Entry) {
+ m.DB.Save(&entry)
+}
+
+func (m EntryModel) Delete(id string) {
+ m.DB.Delete(&Entry{}, id)
+}
diff --git a/routes/routes.go b/routes/routes.go
index d78acc7..a960277 100644
--- a/routes/routes.go
+++ b/routes/routes.go
@@ -2,17 +2,23 @@ package routes
import (
"git.sr.ht/~gabrielgio/midr/controller"
+ "git.sr.ht/~gabrielgio/midr/db"
"github.com/gin-gonic/gin"
)
func HandleRequests() {
+
+ env := &controller.Env{
+ Entries: db.EntryModel{DB: db.DB},
+ }
+
r := gin.Default()
r.LoadHTMLGlob("templates/*")
- r.GET("/", controller.GetEntries)
- r.GET("entries/", controller.GetEntry)
- r.POST("entries/", controller.CreateEntry)
- r.GET("entries/:id", controller.GetEntry)
- r.POST("entries/:id", controller.UpdateEntry)
- r.DELETE("entries/:id", controller.DeleteEntry)
+ r.GET("/", env.GetEntries)
+ r.GET("entries/", env.GetEntry)
+ r.POST("entries/", env.CreateEntry)
+ r.GET("entries/:id", env.GetEntry)
+ r.POST("entries/:id", env.UpdateEntry)
+ r.DELETE("entries/:id", env.DeleteEntry)
r.Run(":8000")
}