feat(self-check): check stream if stream loaded #1028

internal/nginx/modules.go

@@ -0,0 +1,44 @@
+package nginx
+
+import (
+	"regexp"
+	"strings"
+)
+
+const (
+	ModuleStream = "stream_module"
+)
+
+func GetModules() (modules []string) {
+	out := getNginxV()
+	
+	// Regular expression to find modules in nginx -V output
+	r := regexp.MustCompile(`--with-([a-zA-Z0-9_-]+)(_module)?`)
+	
+	// Find all matches
+	matches := r.FindAllStringSubmatch(out, -1)
+	
+	// Extract module names from matches
+	for _, match := range matches {
+		module := match[1]
+		// If the module doesn't end with "_module", add it
+		if !strings.HasSuffix(module, "_module") {
+			module = module + "_module"
+		}
+		modules = append(modules, module)
+	}
+
+	return modules
+}
+
+func IsModuleLoaded(module string) bool {
+	modules := GetModules()
+	
+	for _, m := range modules {
+		if m == module {
+			return true
+		}
+	}
+	
+	return false
+}

internal/self_check/tasks.go

@@ -5,6 +5,7 @@ import (
 	"github.com/0xJacky/Nginx-UI/internal/translation"
 	"github.com/elliotchance/orderedmap/v3"
 	"github.com/uozi-tech/cosy"
+	"github.com/0xJacky/Nginx-UI/internal/nginx"
 )
 
 type Task struct {
@@ -44,15 +45,6 @@ var selfCheckTasks = []*Task{
 		CheckFunc: CheckSitesDirectory,
 		FixFunc:   FixSitesDirectory,
 	},
-	{
-		Key:  "Directory-Streams",
-		Name: translation.C("Streams directory exists"),
-		Description: translation.C("Check if the " +
-			"streams-available and streams-enabled directories are " +
-			"under the nginx configuration directory"),
-		CheckFunc: CheckStreamDirectory,
-		FixFunc:   FixStreamDirectory,
-	},
 	{
 		Key:  "NginxConf-Sites-Enabled",
 		Name: translation.C("Nginx.conf includes sites-enabled directory"),
@@ -61,14 +53,6 @@ var selfCheckTasks = []*Task{
 		CheckFunc: CheckNginxConfIncludeSites,
 		FixFunc:   FixNginxConfIncludeSites,
 	},
-	{
-		Key:  "NginxConf-Streams-Enabled",
-		Name: translation.C("Nginx.conf includes streams-enabled directory"),
-		Description: translation.C("Check if the nginx.conf includes the " +
-			"streams-enabled directory"),
-		CheckFunc: CheckNginxConfIncludeStreams,
-		FixFunc:   FixNginxConfIncludeStreams,
-	},
 	{
 		Key:  "NginxConf-ConfD",
 		Name: translation.C("Nginx.conf includes conf.d directory"),
@@ -112,8 +96,23 @@ var selfCheckTasks = []*Task{
 var selfCheckTaskMap = orderedmap.NewOrderedMap[string, *Task]()
 
 func init() {
-	for _, task := range selfCheckTasks {
-		selfCheckTaskMap.Set(task.Key, task)
+	if nginx.IsModuleLoaded(nginx.ModuleStream) {
+		selfCheckTasks = append(selfCheckTasks, &Task{
+			Key:  "Directory-Streams",
+			Name: translation.C("Streams directory exists"),
+			Description: translation.C("Check if the " +
+				"streams-available and streams-enabled directories are " +
+				"under the nginx configuration directory"),
+			CheckFunc: CheckStreamDirectory,
+			FixFunc:   FixStreamDirectory,
+		}, &Task{
+			Key:  "NginxConf-Streams-Enabled",
+			Name: translation.C("Nginx.conf includes streams-enabled directory"),
+			Description: translation.C("Check if the nginx.conf includes the " +
+			"streams-enabled directory"),
+			CheckFunc: CheckNginxConfIncludeStreams,
+			FixFunc:   FixNginxConfIncludeStreams,
+		})
 	}
 	if helper.InNginxUIOfficialDocker() {
 		selfCheckTasks = append(selfCheckTasks, &Task{
@@ -124,4 +123,8 @@ func init() {
 			CheckFunc: CheckDockerSocket,
 		})
 	}
+
+	for _, task := range selfCheckTasks {
+		selfCheckTaskMap.Set(task.Key, task)
+	}
 }