fix: Service status acquisition compatible with snap (#9169)

Author: lan-yonghui

agent/utils/systemctl/systemctl.go

@@ -1,6 +1,7 @@
 package systemctl
 
 import (
+	"bytes"
 	"fmt"
 	"github.com/pkg/errors"
 	"os/exec"
@@ -16,31 +17,65 @@ func RunSystemCtl(args ...string) (string, error) {
 	return string(output), nil
 }
 
+func isSnapService(serviceName string) bool {
+	cmd := exec.Command("snap", "services")
+	var out bytes.Buffer
+	cmd.Stdout = &out
+	if err := cmd.Run(); err != nil {
+		return false
+	}
+	return strings.Contains(out.String(), serviceName)
+}
+
+func isSnapServiceActive(serviceName string) bool {
+	cmd := exec.Command("snap", "services")
+	var out bytes.Buffer
+	cmd.Stdout = &out
+	if err := cmd.Run(); err != nil {
+		return false
+	}
+	lines := strings.Split(out.String(), "\n")
+	for _, line := range lines {
+		if strings.Contains(line, serviceName) && strings.Contains(line, "active") {
+			return true
+		}
+	}
+	return false
+}
+
 func IsActive(serviceName string) (bool, error) {
 	out, err := RunSystemCtl("is-active", serviceName)
-	if err != nil {
-		return false, err
+	if err == nil {
+		return strings.TrimSpace(out) == "active", nil
+	}
+
+	if isSnapServiceActive(serviceName) {
+		return true, nil
 	}
-	return out == "active\n", nil
+	return false, fmt.Errorf("service %s is not active: %v", serviceName, err)
 }
 
 func IsEnable(serviceName string) (bool, error) {
 	out, err := RunSystemCtl("is-enabled", serviceName)
-	if err != nil {
-		return false, err
+	if err == nil {
+		return strings.TrimSpace(out) == "enabled", nil
+	}
+
+	if isSnapServiceActive(serviceName) {
+		return true, nil
 	}
-	return out == "enabled\n", nil
+	return false, fmt.Errorf("failed to determine if service %s is enabled: %v", serviceName, err)
 }
 
 func IsExist(serviceName string) (bool, error) {
 	out, err := RunSystemCtl("is-enabled", serviceName)
-	if err != nil {
-		if strings.Contains(out, "disabled") {
-			return true, nil
-		}
-		return false, nil
+	if err == nil || strings.Contains(out, "disabled") {
+		return true, nil
+	}
+	if isSnapService(serviceName) {
+		return true, nil
 	}
-	return true, nil
+	return false, nil
 }
 
 func handlerErr(out string, err error) error {
@@ -55,10 +90,27 @@ func handlerErr(out string, err error) error {
 
 func Restart(serviceName string) error {
 	out, err := RunSystemCtl("restart", serviceName)
+	if err == nil {
+		return nil
+	}
+	if isSnapService(serviceName) {
+		cmd := exec.Command("snap", "restart", serviceName)
+		output, snapErr := cmd.CombinedOutput()
+		return handlerErr(string(output), snapErr)
+	}
 	return handlerErr(out, err)
 }
 
 func Operate(operate, serviceName string) error {
 	out, err := RunSystemCtl(operate, serviceName)
+	if err == nil {
+		return nil
+	}
+
+	if isSnapService(serviceName) && (operate == "start" || operate == "stop" || operate == "restart") {
+		cmd := exec.Command("snap", operate, serviceName)
+		output, snapErr := cmd.CombinedOutput()
+		return handlerErr(string(output), snapErr)
+	}
 	return handlerErr(out, err)
 }