re-add autostart and symlink setup for systemd. Adds the ability to run ssui with -setupautostart to install and configure a systemd service that runs the latest SSUI, even if the executable changes in updates

Co-authored-by: akirilov <atanas_kirilov@hotmail.com>

Author: JacksonTheMaster

.gitignore

@@ -38,7 +38,8 @@ UIMod/config/config.json
 C:/custom/file.txt
 UIMod/config/customdetections.json
 winhttp.dll
-autostart*
+./autostart.sh
+./autostart.ps1
 __debug_bin*
 frontend/node_modules
 frontend/dist

src/core/loader/cmdargs.go

@@ -3,12 +3,15 @@ package loader
 import (
 	"flag"
 	"fmt"
+	"os"
+	"runtime"
 	"strings"
 	"time"
 
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/config"
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/core/security"
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/logger"
+	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/setup/autostart"
 )
 
 // Define flags matching the config variable names
@@ -22,6 +25,7 @@ var devModeFlag bool
 var skipSteamCMDFlag bool
 var sanityCheckFlag bool
 var overrideAdvertisedIpFlag string
+var autoStartSSUIOnBoot bool
 
 // ParseFlags parses command-line arguments ONCE at startup (called from func main)
 func ParseFlags() {
@@ -41,6 +45,7 @@ func ParseFlags() {
 	flag.BoolVar(&skipSteamCMDFlag, "NoSteamCMD", false, "Skips SteamCMD installation")
 	flag.BoolVar(&sanityCheckFlag, "NoSanityCheck", false, "Skips the sanity check. Not recommended.")
 	flag.StringVar(&overrideAdvertisedIpFlag, "OverrideAdvertisedIp", "", "Override the advertised server IP (to allow server advertisement if you are behind a reverse proxy)")
+	flag.BoolVar(&autoStartSSUIOnBoot, "setupautostart", false, "Setup Auto-start SSUI on boot")
 
 	// Parse command-line flags
 	flag.Parse()
@@ -62,6 +67,20 @@ func HandleFlags() {
 		config.SetSkipSteamCMD(true)
 	}
 
+	if autoStartSSUIOnBoot {
+		if runtime.GOOS != "linux" {
+			logger.Core.Error("Autostart is only supported on Linux. Exiting in 5 seconds...")
+			time.Sleep(5 * time.Second)
+			os.Exit(0)
+		}
+		err := autostart.Initialize()
+		if err != nil {
+			logger.Core.Error("Failed to initialize autostart: " + err.Error())
+		}
+		time.Sleep(5 * time.Second)
+		os.Exit(0)
+	}
+
 	if backendEndpointPortFlag != "" && backendEndpointPortFlag != "8443" {
 		oldPort := config.GetSSUIWebPort()
 		config.SetSSUIWebPort(backendEndpointPortFlag)

src/setup/autostart/autostart.go

@@ -0,0 +1,24 @@
+package autostart
+
+import (
+	"fmt"
+)
+
+func Initialize() error {
+
+	err := SetupAutostartScripts()
+	if err != nil {
+		return fmt.Errorf("failed to setup autostart script, cannot proceed with autostart setup: %w", err)
+	}
+
+	err = SetupBinarySymlink()
+	if err != nil {
+		return fmt.Errorf("failed to create symlink for autostart: %w", err)
+	}
+
+	err = SetupService()
+	if err != nil {
+		return fmt.Errorf("failed to setup service: %w", err)
+	}
+	return nil
+}

src/setup/autostartscripts.go src/setup/autostart/autostartscripts.gosrc/setup/autostartscripts.go renamed to src/setup/autostart/autostartscripts.go

@@ -1,6 +1,7 @@
-package setup
+package autostart
 
 import (
+	"fmt"
 	"io"
 	"io/fs"
 	"os"
@@ -9,40 +10,58 @@ import (
 	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/config"
 )
 
-func SetupAutostartScripts() {
+func SetupAutostartScripts() error {
 	scriptFS, err := fs.Sub(config.V1UIFS, "UIMod/onboard_bundled/scripts")
 	if err != nil {
-		return
+		return err
 	}
 
 	if runtime.GOOS == "windows" {
 		script, err := scriptFS.Open("autostart.ps1")
 		if err != nil {
-			return
+			return err
 		}
 		defer script.Close()
 		data, err := io.ReadAll(script)
 		if err != nil {
-			return
+			return err
 		}
+
+		_, err = os.Stat("autostart.ps1")
+		if err == nil {
+			err = os.Remove("autostart.ps1")
+			if err != nil {
+				return fmt.Errorf("failed to remove autostart.ps1: %w", err)
+			}
+		}
+
 		err = os.WriteFile("autostart.ps1", data, 0755)
 		if err != nil {
-			return
+			return err
 		}
 	}
 	if runtime.GOOS == "linux" {
 		script, err := scriptFS.Open("autostart.sh")
 		if err != nil {
-			return
+			return err
 		}
 		defer script.Close()
 		data, err := io.ReadAll(script)
 		if err != nil {
-			return
+			return err
 		}
-		err = os.WriteFile("autostart.service", data, 0755)
+		_, err = os.Stat("autostart.sh")
+		if err == nil {
+			err = os.Remove("autostart.sh")
+			if err != nil {
+				return fmt.Errorf("failed to remove autostart.sh: %w", err)
+			}
+		}
+
+		err = os.WriteFile("autostart.sh", data, 0755)
 		if err != nil {
-			return
+			return err
 		}
 	}
+	return nil
 }

src/setup/autostart/service-lin.go

@@ -0,0 +1,72 @@
+//go:build linux
+
+package autostart
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"time"
+
+	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/logger"
+)
+
+func SetupService() error {
+	// run ./autostart.sh, which will ask for elevation when needed and setup the service
+
+	//stat /etc/systemd/system/ssui.service and it it exists tell the user that the service is already installed
+	_, err := os.Stat("/etc/systemd/system/stationeersserverui.service")
+	if err == nil {
+		logger.Main.Error("Autostart service is already installed. Exiting in 5 seconds...")
+		time.Sleep(5 * time.Second)
+		os.Exit(1)
+	}
+
+	exec.Command("chmod", "+x", "autostart.sh").Run()
+	exec.Command("./autostart.sh", "install").Run()
+
+	err = exec.Command("chmod", "+x", "autostart.sh").Run()
+	if err != nil {
+		return fmt.Errorf("failed to chmod autostart.sh: %w", err)
+	}
+	cmd := exec.Command("./autostart.sh")
+	err = cmd.Run()
+	if err != nil {
+		if exitErr, ok := err.(*exec.ExitError); ok {
+			// handle exit codes in a switch statement
+			switch exitErr.ExitCode() {
+			case 10:
+				return fmt.Errorf("autostart.sh failed: For security reasons, it is not recommended to run this service as a root user")
+			case 2:
+				return fmt.Errorf("autostart.sh Error: systemd is not available on this system")
+			case 3:
+				return fmt.Errorf("autostart.sh Error: Could not determine base directory")
+			case 4:
+				return fmt.Errorf("autostart.sh Failed to stop ssui.service")
+			case 5:
+				return fmt.Errorf("autostart.sh Error: Failed to create service file")
+			case 6:
+				return fmt.Errorf("autostart.sh Error: Failed to reload systemd daemon")
+			case 7:
+				return fmt.Errorf("autostart.sh Error: Failed to enable ssui.service")
+			case 8:
+				return fmt.Errorf("autostart.sh Error: Failed to start ssui.service")
+			default:
+				return fmt.Errorf("autostart.sh failed with exit code %d: %w", exitErr.ExitCode(), err)
+			}
+		}
+
+		return fmt.Errorf("failed to run autostart.sh: %w", err)
+	}
+	// stat for the file to check if it exists, if it does, remove it
+	_, err = os.Stat("autostart.sh")
+	if err == nil {
+		err = os.Remove("autostart.sh")
+		if err != nil {
+			return fmt.Errorf("failed to remove autostart.sh: %w", err)
+		}
+	}
+	logger.Main.Info("Autostart setup complete. Restart SSUI without the --setupautostart flag to start normally. Exiting in 5 seconds...")
+	return nil
+
+}

src/setup/autostart/service-win.go

@@ -0,0 +1,8 @@
+//go:build windows
+
+package autostart
+
+func SetupService() error {
+	//TODO: Windows implementation
+	return nil
+}

src/setup/autostart/symlink-linux.go

@@ -0,0 +1,88 @@
+package autostart
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"strings"
+
+	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/config"
+	"github.com/JacksonTheMaster/StationeersServerUI/v5/src/logger"
+)
+
+func SetupBinarySymlink() error {
+	if runtime.GOOS != "linux" {
+		return fmt.Errorf("not on linux")
+	}
+	// Check if we are inside a docker container, if so, don't try to create a symlink
+	if IsInsideContainer() {
+		logger.Install.Info("Inside a container, skipping symlink creation.")
+		return fmt.Errorf("inside a container")
+	}
+
+	if err := updateSymlink(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func updateSymlink() error {
+	// Get the absolute path of the current executable
+	executablePath, err := os.Executable()
+	if err != nil {
+		return err
+	}
+
+	// Get the directory of the executable
+	executableDir := filepath.Dir(executablePath)
+
+	// Construct the target filename (e.g., StationeersServerControl5.0.x86_64)
+	currentExecutableName := "StationeersServerControlv" + config.GetVersion() + ".x86_64"
+
+	// Construct the full path to the target executable
+	targetPath := filepath.Join(executableDir, currentExecutableName)
+
+	symlinkPath := filepath.Join(executableDir, "StationeersServerUI.lnk")
+
+	// Remove existing symlink if it exists
+	if err := os.Remove(symlinkPath); err != nil && !os.IsNotExist(err) {
+		return err
+	}
+
+	// Create the symlink
+	return os.Symlink(targetPath, symlinkPath)
+}
+
+func IsInsideContainer() bool {
+	// Check .dockerenv file (Docker-specific)
+	if _, err := os.Stat("/.dockerenv"); err == nil {
+		return true
+	}
+
+	// Check cgroup (works for Docker and other container runtimes)
+	return isContainerFromCGroup()
+}
+
+func isContainerFromCGroup() bool {
+	file, err := os.Open("/proc/1/cgroup")
+	if err != nil {
+		return false
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := scanner.Text()
+		// Check for various container runtime indicators
+		if strings.Contains(line, "docker") ||
+			strings.Contains(line, "containerd") ||
+			strings.Contains(line, "kubepods") ||
+			strings.Contains(line, "crio") ||
+			strings.Contains(line, "libpod") {
+			return true
+		}
+	}
+	return false
+}