From a7a8322f133365e34c2a7e5e19cc557d2bde18ba Mon Sep 17 00:00:00 2001
From: EnumDev <enumdev@enumerated.dev>
Date: Mon, 10 Mar 2025 15:06:56 +0200
Subject: [PATCH] Add crash_on_safe_exit option to service files

---
 cmd/esvm/main.go | 55 +++++++++++++++++++++++++++---------------------
 1 file changed, 31 insertions(+), 24 deletions(-)

diff --git a/cmd/esvm/main.go b/cmd/esvm/main.go
index 4998af4..56d1423 100644
--- a/cmd/esvm/main.go
+++ b/cmd/esvm/main.go
@@ -30,17 +30,18 @@ const (
 )
 
 type EnitService struct {
-	Name           string   `yaml:"name"`
-	Description    string   `yaml:"description,omitempty"`
-	Dependencies   []string `yaml:"dependencies,omitempty"`
-	Type           string   `yaml:"type"`
-	StartCmd       string   `yaml:"start_cmd"`
-	ExitMethod     string   `yaml:"exit_method"`
-	StopCmd        string   `yaml:"stop_cmd,omitempty"`
-	Restart        string   `yaml:"restart,omitempty"`
-	ServiceRunPath string
-	restartCount   int
-	stopChannel    chan bool
+	Name            string   `yaml:"name"`
+	Description     string   `yaml:"description,omitempty"`
+	Dependencies    []string `yaml:"dependencies,omitempty"`
+	Type            string   `yaml:"type"`
+	StartCmd        string   `yaml:"start_cmd"`
+	ExitMethod      string   `yaml:"exit_method"`
+	CrashOnSafeExit bool     `yaml:"crash_on_safe_exit"`
+	StopCmd         string   `yaml:"stop_cmd,omitempty"`
+	Restart         string   `yaml:"restart,omitempty"`
+	ServiceRunPath  string
+	restartCount    int
+	stopChannel     chan bool
 }
 
 // Build-time variables
@@ -139,17 +140,18 @@ func Init() {
 			}
 
 			service := EnitService{
-				Name:           "",
-				Description:    "",
-				Dependencies:   make([]string, 0),
-				Type:           "",
-				StartCmd:       "",
-				ExitMethod:     "",
-				StopCmd:        "",
-				Restart:        "",
-				ServiceRunPath: "",
-				restartCount:   0,
-				stopChannel:    make(chan bool),
+				Name:            "",
+				Description:     "",
+				Dependencies:    make([]string, 0),
+				Type:            "",
+				StartCmd:        "",
+				ExitMethod:      "",
+				StopCmd:         "",
+				Restart:         "",
+				CrashOnSafeExit: true,
+				ServiceRunPath:  "",
+				restartCount:    0,
+				stopChannel:     make(chan bool),
 			}
 			if err := yaml.Unmarshal(bytes, &service); err != nil {
 				logger.Printf("Could not read service file at %s!\n", path.Join(serviceConfigDir, "services", entry.Name()))
@@ -367,8 +369,13 @@ func (service *EnitService) StartService() error {
 				_ = service.setCurrentState(EnitServiceCompleted)
 				return
 			}
-			logger.Printf("Service (%s) has crashed!\n", service.Name)
-			_ = service.setCurrentState(EnitServiceCrashed)
+			if !service.CrashOnSafeExit {
+				logger.Printf("Service (%s) has exited\n", service.Name)
+				_ = service.setCurrentState(EnitServiceStopped)
+			} else {
+				logger.Printf("Service (%s) has crashed!\n", service.Name)
+				_ = service.setCurrentState(EnitServiceCrashed)
+			}
 
 			if service.Restart == "always" {
 				_ = service.StartService()