feat: deploy config to remote nodes #359

api/config/add.go

@@ -1,65 +1,97 @@
 package config
 
 import (
-	"github.com/0xJacky/Nginx-UI/api"
-	"github.com/0xJacky/Nginx-UI/internal/config"
-	"github.com/0xJacky/Nginx-UI/internal/helper"
-	"github.com/0xJacky/Nginx-UI/internal/nginx"
-	"github.com/gin-gonic/gin"
-	"github.com/sashabaranov/go-openai"
-	"net/http"
-	"os"
-	"time"
+    "github.com/0xJacky/Nginx-UI/api"
+    "github.com/0xJacky/Nginx-UI/internal/config"
+    "github.com/0xJacky/Nginx-UI/internal/helper"
+    "github.com/0xJacky/Nginx-UI/internal/nginx"
+    "github.com/0xJacky/Nginx-UI/model"
+    "github.com/0xJacky/Nginx-UI/query"
+    "github.com/gin-gonic/gin"
+    "github.com/sashabaranov/go-openai"
+    "net/http"
+    "os"
+    "path/filepath"
+    "time"
 )
 
 func AddConfig(c *gin.Context) {
-	var json struct {
-		Name        string `json:"name" binding:"required"`
-		NewFilepath string `json:"new_filepath" binding:"required"`
-		Content     string `json:"content"`
-		Overwrite   bool   `json:"overwrite"`
-	}
+    var json struct {
+        Name        string `json:"name" binding:"required"`
+        NewFilepath string `json:"new_filepath" binding:"required"`
+        Content     string `json:"content"`
+        Overwrite   bool   `json:"overwrite"`
+        SyncNodeIds []int  `json:"sync_node_ids"`
+    }
 
-	if !api.BindAndValid(c, &json) {
-		return
-	}
+    if !api.BindAndValid(c, &json) {
+        return
+    }
 
-	name := json.Name
-	content := json.Content
-	path := json.NewFilepath
-	if !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
-		c.JSON(http.StatusForbidden, gin.H{
-			"message": "new filepath is not under the nginx conf path",
-		})
-		return
-	}
+    name := json.Name
+    content := json.Content
+    path := json.NewFilepath
+    if !helper.IsUnderDirectory(path, nginx.GetConfPath()) {
+        c.JSON(http.StatusForbidden, gin.H{
+            "message": "new filepath is not under the nginx conf path",
+        })
+        return
+    }
 
-	if !json.Overwrite && helper.FileExists(path) {
-		c.JSON(http.StatusNotAcceptable, gin.H{
-			"message": "File exists",
-		})
-		return
-	}
+    if !json.Overwrite && helper.FileExists(path) {
+        c.JSON(http.StatusNotAcceptable, gin.H{
+            "message": "File exists",
+        })
+        return
+    }
 
-	err := os.WriteFile(path, []byte(content), 0644)
-	if err != nil {
-		api.ErrHandler(c, err)
-		return
-	}
+    // check if the dir exists, if not, use mkdirAll to create the dir
+    dir := filepath.Dir(path)
+    if !helper.FileExists(dir) {
+        err := os.MkdirAll(dir, 0755)
+        if err != nil {
+            api.ErrHandler(c, err)
+            return
+        }
+    }
 
-	output := nginx.Reload()
-	if nginx.GetLogLevel(output) >= nginx.Warn {
-		c.JSON(http.StatusInternalServerError, gin.H{
-			"message": output,
-		})
-		return
-	}
+    err := os.WriteFile(path, []byte(content), 0644)
+    if err != nil {
+        api.ErrHandler(c, err)
+        return
+    }
 
-	c.JSON(http.StatusOK, config.Config{
-		Name:            name,
-		Content:         content,
-		ChatGPTMessages: make([]openai.ChatCompletionMessage, 0),
-		FilePath:        path,
-		ModifiedAt:      time.Now(),
-	})
+    output := nginx.Reload()
+    if nginx.GetLogLevel(output) >= nginx.Warn {
+        c.JSON(http.StatusInternalServerError, gin.H{
+            "message": output,
+        })
+        return
+    }
+
+    q := query.Config
+    _, err = q.Where(q.Filepath.Eq(path)).Delete()
+    if err != nil {
+        api.ErrHandler(c, err)
+        return
+    }
+
+    err = q.Create(&model.Config{
+        Name:          name,
+        Filepath:      path,
+        SyncNodeIds:   json.SyncNodeIds,
+        SyncOverwrite: json.Overwrite,
+    })
+    if err != nil {
+        api.ErrHandler(c, err)
+        return
+    }
+
+    c.JSON(http.StatusOK, config.Config{
+        Name:            name,
+        Content:         content,
+        ChatGPTMessages: make([]openai.ChatCompletionMessage, 0),
+        FilePath:        path,
+        ModifiedAt:      time.Now(),
+    })
 }