feat(site): sync operation

internal/helper/file.go

@@ -2,8 +2,17 @@ package helper
 
 import "os"
 
-func FileExists(filename string) bool {
-	_, err := os.Stat(filename)
+func FileExists(filepath string) bool {
+	_, err := os.Stat(filepath)
+	if os.IsNotExist(err) {
+		return false
+	}
+
+	return true
+}
+
+func SymbolLinkExists(filepath string) bool {
+	_, err := os.Lstat(filepath)
 	if os.IsNotExist(err) {
 		return false
 	}

internal/site/delete.go

@@ -0,0 +1,85 @@
+package site
+
+import (
+	"fmt"
+	"github.com/0xJacky/Nginx-UI/internal/helper"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/internal/notification"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/go-resty/resty/v2"
+	"github.com/uozi-tech/cosy/logger"
+	"net/http"
+	"os"
+	"runtime"
+	"sync"
+)
+
+// Delete deletes a site by removing the file in sites-available
+func Delete(name string) (err error) {
+	availablePath := nginx.GetConfPath("sites-available", name)
+
+	s := query.Site
+	_, err = s.Where(s.Path.Eq(availablePath)).Unscoped().Delete(&model.Site{})
+	if err != nil {
+		return
+	}
+
+	enabledPath := nginx.GetConfPath("sites-enabled", name)
+
+	if !helper.FileExists(availablePath) {
+		return fmt.Errorf("site not found")
+	}
+
+	if helper.FileExists(enabledPath) {
+		return fmt.Errorf("site is enabled")
+	}
+
+	certModel := model.Cert{Filename: name}
+	_ = certModel.Remove()
+
+	err = os.Remove(availablePath)
+	if err != nil {
+		return
+	}
+
+	go syncDelete(name)
+
+	return
+}
+
+func syncDelete(name string) {
+	nodes := getSyncNodes(name)
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(nodes))
+
+	for _, node := range nodes {
+		go func() {
+			defer func() {
+				if err := recover(); err != nil {
+					buf := make([]byte, 1024)
+					runtime.Stack(buf, false)
+					logger.Error(err)
+				}
+			}()
+			defer wg.Done()
+
+			client := resty.New()
+			client.SetBaseURL(node.URL)
+			resp, err := client.R().
+				Delete(fmt.Sprintf("/api/sites/%s", name))
+			if err != nil {
+				notification.Error("Delete Remote Site Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Delete Remote Site Error", string(resp.Body()))
+				return
+			}
+			notification.Success("Delete Remote Site Success", string(resp.Body()))
+		}()
+	}
+
+	wg.Wait()
+}

internal/site/disable.go

@@ -0,0 +1,80 @@
+package site
+
+import (
+	"fmt"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/internal/notification"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/go-resty/resty/v2"
+	"github.com/uozi-tech/cosy/logger"
+	"net/http"
+	"os"
+	"runtime"
+	"sync"
+)
+
+// Disable disables a site by removing the symlink in sites-enabled
+func Disable(name string) (err error) {
+	enabledConfigFilePath := nginx.GetConfPath("sites-enabled", name)
+	_, err = os.Stat(enabledConfigFilePath)
+	if err != nil {
+		return
+	}
+
+	err = os.Remove(enabledConfigFilePath)
+	if err != nil {
+		return
+	}
+
+	// delete auto cert record
+	certModel := model.Cert{Filename: name}
+	err = certModel.Remove()
+	if err != nil {
+		return
+	}
+
+	output := nginx.Reload()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		return fmt.Errorf(output)
+	}
+
+	go syncDisable(name)
+
+	return
+}
+
+func syncDisable(name string) {
+	nodes := getSyncNodes(name)
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(nodes))
+
+	for _, node := range nodes {
+		go func() {
+			defer func() {
+				if err := recover(); err != nil {
+					buf := make([]byte, 1024)
+					runtime.Stack(buf, false)
+					logger.Error(err)
+				}
+			}()
+			defer wg.Done()
+
+			client := resty.New()
+			client.SetBaseURL(node.URL)
+			resp, err := client.R().
+				Post(fmt.Sprintf("/api/sites/%s/disable", name))
+			if err != nil {
+				notification.Error("Disable Remote Site Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Disable Remote Site Error", string(resp.Body()))
+				return
+			}
+			notification.Success("Disable Remote Site Success", string(resp.Body()))
+		}()
+	}
+
+	wg.Wait()
+}

internal/site/duplicate.go

@@ -0,0 +1,24 @@
+package site
+
+import (
+	"github.com/0xJacky/Nginx-UI/internal/helper"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/pkg/errors"
+)
+
+// Duplicate duplicates a site by copying the file
+func Duplicate(src, dst string) (err error) {
+	src = nginx.GetConfPath("sites-available", src)
+	dst = nginx.GetConfPath("sites-available", dst)
+
+	if helper.FileExists(dst) {
+		return errors.New("file exists")
+	}
+
+	_, err = helper.CopyFile(src, dst)
+	if err != nil {
+		return
+	}
+
+	return
+}

internal/site/enable.go

@@ -0,0 +1,86 @@
+package site
+
+import (
+	"fmt"
+	"github.com/0xJacky/Nginx-UI/internal/helper"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/internal/notification"
+	"github.com/go-resty/resty/v2"
+	"github.com/uozi-tech/cosy/logger"
+	"net/http"
+	"os"
+	"runtime"
+	"sync"
+)
+
+// Enable enables a site by creating a symlink in sites-enabled
+func Enable(name string) (err error) {
+	configFilePath := nginx.GetConfPath("sites-available", name)
+	enabledConfigFilePath := nginx.GetConfPath("sites-enabled", name)
+
+	_, err = os.Stat(configFilePath)
+	if err != nil {
+		return
+	}
+
+	if helper.FileExists(enabledConfigFilePath) {
+		return
+	}
+
+	err = os.Symlink(configFilePath, enabledConfigFilePath)
+	if err != nil {
+		return
+	}
+
+	// Test nginx config, if not pass, then disable the site.
+	output := nginx.TestConf()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		_ = os.Remove(enabledConfigFilePath)
+		return fmt.Errorf(output)
+	}
+
+	output = nginx.Reload()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		return fmt.Errorf(output)
+	}
+
+	go syncEnable(name)
+
+	return
+}
+
+func syncEnable(name string) {
+	nodes := getSyncNodes(name)
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(nodes))
+
+	for _, node := range nodes {
+		go func() {
+			defer func() {
+				if err := recover(); err != nil {
+					buf := make([]byte, 1024)
+					runtime.Stack(buf, false)
+					logger.Error(err)
+				}
+			}()
+			defer wg.Done()
+
+			client := resty.New()
+			client.SetBaseURL(node.URL)
+			resp, err := client.R().
+				Post(fmt.Sprintf("/api/sites/%s/enable", name))
+			if err != nil {
+				notification.Error("Enable Remote Site Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Enable Remote Site Error", string(resp.Body()))
+				return
+			}
+			notification.Success("Enable Remote Site Success", string(resp.Body()))
+		}()
+	}
+
+	wg.Wait()
+}

internal/site/rename.go

@@ -0,0 +1,103 @@
+package site
+
+import (
+	"fmt"
+	"github.com/0xJacky/Nginx-UI/internal/helper"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/internal/notification"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/go-resty/resty/v2"
+	"github.com/uozi-tech/cosy/logger"
+	"net/http"
+	"os"
+	"runtime"
+	"sync"
+)
+
+func Rename(oldName string, newName string) (err error) {
+	oldPath := nginx.GetConfPath("sites-available", oldName)
+	newPath := nginx.GetConfPath("sites-available", newName)
+
+	if oldPath == newPath {
+		return
+	}
+
+	// check if dst file exists, do not rename
+	if helper.FileExists(newPath) {
+		return fmt.Errorf("file exists")
+	}
+
+	s := query.Site
+	_, _ = s.Where(s.Path.Eq(oldPath)).Update(s.Path, newPath)
+
+	err = os.Rename(oldPath, newPath)
+	if err != nil {
+		return
+	}
+
+	// recreate a soft link
+	oldEnabledConfigFilePath := nginx.GetConfPath("sites-enabled", oldName)
+	if helper.SymbolLinkExists(oldEnabledConfigFilePath) {
+		_ = os.Remove(oldEnabledConfigFilePath)
+		newEnabledConfigFilePath := nginx.GetConfPath("sites-enabled", newName)
+		err = os.Symlink(newPath, newEnabledConfigFilePath)
+		if err != nil {
+			return
+		}
+	}
+
+	// test nginx configuration
+	output := nginx.TestConf()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		return fmt.Errorf(output)
+	}
+
+	// reload nginx
+	output = nginx.Reload()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		return fmt.Errorf(output)
+	}
+
+	go syncRename(oldName, newName)
+
+	return
+}
+
+func syncRename(oldName, newName string) {
+	nodes := getSyncNodes(newName)
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(nodes))
+
+	for _, node := range nodes {
+		go func() {
+			defer func() {
+				if err := recover(); err != nil {
+					buf := make([]byte, 1024)
+					runtime.Stack(buf, false)
+					logger.Error(err)
+				}
+			}()
+			defer wg.Done()
+
+			client := resty.New()
+			client.SetBaseURL(node.URL)
+			resp, err := client.R().
+				SetBody(map[string]string{
+					"new_name": newName,
+				}).
+				Post(fmt.Sprintf("/api/sites/%s/rename", oldName))
+			if err != nil {
+				notification.Error("Rename Remote Site Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Rename Remote Site Error", string(resp.Body()))
+				return
+			}
+			notification.Success("Rename Remote Site Success", string(resp.Body()))
+		}()
+	}
+
+	wg.Wait()
+}

internal/site/save.go

@@ -0,0 +1,100 @@
+package site
+
+import (
+	"fmt"
+	"github.com/0xJacky/Nginx-UI/internal/helper"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/internal/notification"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/go-resty/resty/v2"
+	"github.com/uozi-tech/cosy/logger"
+	"net/http"
+	"os"
+	"runtime"
+	"sync"
+)
+
+// Save saves a site configuration file
+func Save(name string, content string, overwrite bool, siteCategoryId uint64, syncNodeIds []uint64) (err error) {
+	path := nginx.GetConfPath("sites-available", name)
+	if !overwrite && helper.FileExists(path) {
+		return fmt.Errorf("file exists")
+	}
+
+	err = os.WriteFile(path, []byte(content), 0644)
+	if err != nil {
+		return
+	}
+
+	enabledConfigFilePath := nginx.GetConfPath("sites-enabled", name)
+	if helper.FileExists(enabledConfigFilePath) {
+		// Test nginx configuration
+		output := nginx.TestConf()
+
+		if nginx.GetLogLevel(output) > nginx.Warn {
+			return fmt.Errorf(output)
+		}
+
+		output = nginx.Reload()
+
+		if nginx.GetLogLevel(output) > nginx.Warn {
+			return fmt.Errorf(output)
+		}
+	}
+
+	s := query.Site
+	_, err = s.Where(s.Path.Eq(path)).
+		Select(s.SiteCategoryID, s.SyncNodeIDs).
+		Updates(&model.Site{
+			SiteCategoryID: siteCategoryId,
+			SyncNodeIDs:    syncNodeIds,
+		})
+	if err != nil {
+		return
+	}
+
+	go syncSave(name, content)
+
+	return
+}
+
+func syncSave(name string, content string) {
+	nodes := getSyncNodes(name)
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(nodes))
+
+	for _, node := range nodes {
+		go func() {
+			defer func() {
+				if err := recover(); err != nil {
+					buf := make([]byte, 1024)
+					runtime.Stack(buf, false)
+					logger.Error(err)
+				}
+			}()
+			defer wg.Done()
+
+			client := resty.New()
+			client.SetBaseURL(node.URL)
+			resp, err := client.R().
+				SetBody(map[string]interface{}{
+					"content":   content,
+					"overwrite": true,
+				}).
+				Post(fmt.Sprintf("/api/sites/%s", name))
+			if err != nil {
+				notification.Error("Save Remote Site Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Save Remote Site Error", string(resp.Body()))
+				return
+			}
+			notification.Success("Save Remote Site Success", string(resp.Body()))
+		}()
+	}
+
+	wg.Wait()
+}

internal/site/sync.go

@@ -0,0 +1,35 @@
+package site
+
+import (
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
+	"github.com/0xJacky/Nginx-UI/model"
+	"github.com/0xJacky/Nginx-UI/query"
+	"github.com/samber/lo"
+	"github.com/uozi-tech/cosy/logger"
+)
+
+func getSyncNodes(name string) (nodes []*model.Environment) {
+	configFilePath := nginx.GetConfPath("sites-available", name)
+	s := query.Site
+	site, err := s.Where(s.Path.Eq(configFilePath)).
+		Preload(s.SiteCategory).First()
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+
+	syncNodeIds := site.SyncNodeIDs
+	// inherit sync node ids from site category
+	if site.SiteCategory != nil {
+		syncNodeIds = append(syncNodeIds, site.SiteCategory.SyncNodeIds...)
+	}
+	syncNodeIds = lo.Uniq(syncNodeIds)
+
+	e := query.Environment
+	nodes, err = e.Where(e.ID.In(syncNodeIds...)).Find()
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+	return
+}