From 8cccf6f4f0b5037442cdac1ce26308514a075ab9 Mon Sep 17 00:00:00 2001 From: BroodjeAap Date: Mon, 10 Oct 2022 19:20:04 +0000 Subject: [PATCH] added export/import of filters+connections --- main.go | 101 ++++++++++++++++++++++++++++++++++++++ models.go | 5 ++ templates/watch/edit.html | 34 ++++++++++++- 3 files changed, 138 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 7a1a348..3814fbb 100644 --- a/main.go +++ b/main.go @@ -3,6 +3,7 @@ package main import ( "encoding/json" "fmt" + "io/ioutil" "log" "net/http" "path/filepath" @@ -72,6 +73,8 @@ func (web *Web) initRouter() { web.router.POST("/watch/create", web.watchCreatePost) web.router.POST("/watch/update", web.watchUpdate) web.router.POST("/watch/delete", web.deleteWatch) + web.router.GET("/watch/export/:id", web.exportWatch) + web.router.POST("/watch/import/:id", web.importWatch) web.router.GET("/cache/view", web.cacheView) web.router.POST("/cache/clear", web.cacheClear) @@ -357,6 +360,104 @@ func (web *Web) cacheClear(c *gin.Context) { c.Redirect(http.StatusSeeOther, "/cache/view") } +func (web *Web) exportWatch(c *gin.Context) { + watchID := c.Param("id") + export := WatchExport{} + var watch Watch + web.db.Model(&Watch{}).Find(&watch, watchID) + web.db.Model(&Filter{}).Find(&export.Filters, "watch_id = ?", watchID) + web.db.Model(&FilterConnection{}).Find(&export.Connections, "watch_id = ?", watchID) + + c.Header("Content-Disposition", "attachment; filename="+watch.Name+".json") + c.Header("Content-Type", c.Request.Header.Get("Content-Type")) + + c.JSON(http.StatusOK, export) +} + +func (web *Web) importWatch(c *gin.Context) { + watchID, err := strconv.Atoi(c.Param("id")) + if err != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + + file, err := c.FormFile("json") + + if err != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + + openedFile, err := file.Open() + if err != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + jsn, _ := ioutil.ReadAll(openedFile) + + export := WatchExport{} + + if err := json.Unmarshal(jsn, &export); err != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + + // stop/delete cronjobs running for this watch + var cronFilters []Filter + web.db.Model(&Filter{}).Where("watch_id = ? AND type = 'cron'", watchID).Find(&cronFilters) + for _, filter := range cronFilters { + entry, exist := web.cronWatch[filter.ID] + if exist { + web.cron.Remove(entry.ID) + delete(web.cronWatch, filter.ID) + } + } + + filterMap := make(map[uint]*Filter) + for i := range export.Filters { + filter := &export.Filters[i] + filterMap[filter.ID] = filter + filter.ID = 0 + filter.WatchID = uint(watchID) + if filter.Type == "cron" { + entryID, err := web.cron.AddFunc(filter.Var1, func() { triggerSchedule(filter.WatchID, web) }) + if err != nil { + log.Println("Could not start job for Watch: ", filter.WatchID) + continue + } + log.Println("Started CronJob for WatchID", filter.WatchID, "with schedule:", filter.Var1) + web.cronWatch[filter.ID] = web.cron.Entry(entryID) + } + } + web.db.Delete(&Filter{}, "watch_id = ?", watchID) + + if len(export.Filters) > 0 { + tx := web.db.Create(&export.Filters) + if tx.Error != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + } + + web.db.Delete(&FilterConnection{}, "watch_id = ?", watchID) + for i := range export.Connections { + connection := &export.Connections[i] + connection.ID = 0 + connection.WatchID = uint(watchID) + connection.OutputID = filterMap[connection.OutputID].ID + connection.InputID = filterMap[connection.InputID].ID + } + if len(export.Connections) > 0 { + tx := web.db.Create(&export.Connections) + if tx.Error != nil { + c.AbortWithError(http.StatusBadRequest, err) + return + } + } + + c.Redirect(http.StatusSeeOther, fmt.Sprintf("/watch/edit/%d", watchID)) +} + func main() { viper.SetConfigName("config") viper.SetConfigType("yaml") diff --git a/models.go b/models.go index bf09507..1411a76 100644 --- a/models.go +++ b/models.go @@ -53,3 +53,8 @@ type FilterOutput struct { Value string `yaml:"filter_output_value" json:"filter_output_value"` Time time.Time `yaml:"filter_output_time" json:"filter_output_time"` } + +type WatchExport struct { + Filters []Filter `json:"filters"` + Connections []FilterConnection `json:"connections"` +} diff --git a/templates/watch/edit.html b/templates/watch/edit.html index c19bbdc..56f3f7f 100644 --- a/templates/watch/edit.html +++ b/templates/watch/edit.html @@ -29,6 +29,16 @@ Save Watch + + + Export Watch + + + + +
@@ -167,7 +177,27 @@
- + {{ end }}