nginx-ui/internal/nginx/nginx.go

package nginx

import (
	"os"
	"os/exec"
	"strings"
	"sync"
	"time"

	"github.com/0xJacky/Nginx-UI/settings"
)

var (
	mutex      sync.Mutex
	lastOutput string
)

func TestConf() (out string) {
	mutex.Lock()
	defer mutex.Unlock()
	if settings.NginxSettings.TestConfigCmd != "" {
		out = execShell(settings.NginxSettings.TestConfigCmd)

		return
	}

	out = execCommand("nginx", "-t")

	return
}

func Reload() (out string) {
	mutex.Lock()
	defer mutex.Unlock()
	if settings.NginxSettings.ReloadCmd != "" {
		out = execShell(settings.NginxSettings.ReloadCmd)
		return
	}

	out = execCommand("nginx", "-s", "reload")

	return
}

func Restart() {
	mutex.Lock()
	defer mutex.Unlock()

	// fix(docker): nginx restart always output network error
	time.Sleep(500 * time.Millisecond)

	if settings.NginxSettings.RestartCmd != "" {
		lastOutput = execShell(settings.NginxSettings.RestartCmd)

		return
	}

	pidPath := GetPIDPath()
	daemon := GetSbinPath()

	lastOutput = execCommand("start-stop-daemon", "--stop", "--quiet", "--oknodo", "--retry=TERM/30/KILL/5", "--pidfile", pidPath)

	if daemon == "" {
		lastOutput += execCommand("nginx")

		return
	}

	lastOutput += execCommand("start-stop-daemon", "--start", "--quiet", "--pidfile", pidPath, "--exec", daemon)

	return
}

func GetLastOutput() string {
	mutex.Lock()
	defer mutex.Unlock()
	return lastOutput
}

// GetModulesPath returns the nginx modules path
func GetModulesPath() string {
	// First try to get from nginx -V output
	output := execCommand("nginx", "-V")
	if output != "" {
		// Look for --modules-path in the output
		if strings.Contains(output, "--modules-path=") {
			parts := strings.Split(output, "--modules-path=")
			if len(parts) > 1 {
				// Extract the path
				path := strings.Split(parts[1], " ")[0]
				// Remove quotes if present
				path = strings.Trim(path, "\"")
				return path
			}
		}
	}

	// Default path if not found
	return "/usr/lib/nginx/modules"
}

func execShell(cmd string) (out string) {
	bytes, err := exec.Command("/bin/sh", "-c", cmd).CombinedOutput()
	out = string(bytes)
	if err != nil {
		out += " " + err.Error()
	}
	return
}

func execCommand(name string, cmd ...string) (out string) {
	bytes, err := exec.Command(name, cmd...).CombinedOutput()
	out = string(bytes)
	if err != nil {
		out += " " + err.Error()
	}
	return
}

func IsNginxRunning() bool {
	pidPath := GetPIDPath()
	if fileInfo, err := os.Stat(pidPath); err != nil || fileInfo.Size() == 0 {
		return false
	}
	return true
}