feat: sync streams

internal/stream/enable.go

@@ -0,0 +1,87 @@
+package stream
+
+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("streams-available", name)
+	enabledConfigFilePath := nginx.GetConfPath("streams-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("%s", output)
+	}
+
+	output = nginx.Reload()
+	if nginx.GetLogLevel(output) > nginx.Warn {
+		return fmt.Errorf("%s", 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().
+				SetHeader("X-Node-Secret", node.Token).
+				Post(fmt.Sprintf("/api/streams/%s/enable", name))
+			if err != nil {
+				notification.Error("Enable Remote Stream Error", err.Error())
+				return
+			}
+			if resp.StatusCode() != http.StatusOK {
+				notification.Error("Enable Remote Stream Error", NewSyncResult(node.Name, name, resp).String())
+				return
+			}
+			notification.Success("Enable Remote Stream Success", NewSyncResult(node.Name, name, resp).String())
+		}()
+	}
+
+	wg.Wait()
+}