improve service logs (#2)

Author: sxwebdev

.gitignore

@@ -59,3 +59,5 @@ yarn.lock
 
 # Added by goreleaser init:
 dist/
+
+gcx.dev.yaml

app.go

@@ -1,6 +1,7 @@
 package sdmanager
 
 import (
+	"context"
 	"fmt"
 	"strings"
 
@@ -9,6 +10,10 @@ import (
 
 // Создание новой модели приложения
 func NewApplication(o AppOptions) AppModel {
+	if o.ctx == nil {
+		o.ctx = context.Background()
+	}
+
 	return AppModel{
 		Mode:       ModeMainMenu,
 		MenuModel:  NewMenuModel(),
@@ -86,7 +91,7 @@ func (m AppModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 	case ModeServiceInput:
 		// Обновляем модель ввода имени сервиса
-		serviceModel, cmd, err := UpdateServiceInput(msg, m.ServiceInputModel)
+		serviceModel, cmd, err := UpdateServiceInput(m.options.ctx, msg, m.ServiceInputModel)
 		m.ServiceInputModel = serviceModel
 
 		// Обрабатываем ошибку

app_options.go

@@ -1,7 +1,10 @@
 package sdmanager
 
+import "context"
+
 type AppOptions struct {
 	serviceName string
+	ctx         context.Context
 }
 
 type AppOption func(*AppOptions)
@@ -11,3 +14,9 @@ func WithServiceName(serviceName string) AppOption {
 		o.serviceName = serviceName
 	}
 }
+
+func WithContext(ctx context.Context) AppOption {
+	return func(o *AppOptions) {
+		o.ctx = ctx
+	}
+}

cmd/sdmanager/main.go

@@ -1,8 +1,11 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/sxwebdev/sdmanager"
 )
@@ -20,7 +23,10 @@ func PrintVersion() {
 }
 
 func main() {
-	err := sdmanager.RunSystemdManager()
+	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGKILL)
+	defer cancel()
+
+	err := sdmanager.RunSystemdManager(sdmanager.WithContext(ctx))
 	if err != nil {
 		fmt.Printf("Error: %s\n", err)
 		os.Exit(1)

service.go

@@ -1,7 +1,10 @@
 package sdmanager
 
 import (
+	"bufio"
 	"bytes"
+	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"os"
@@ -303,20 +306,49 @@ func RestartService(serviceName string) (string, error) {
 }
 
 // Выполнение просмотра логов
-func ViewServiceLogs(serviceName string) (string, error) {
-	cmd := exec.Command("journalctl", "-n", "50", "-u", serviceName)
-	output, err := cmd.CombinedOutput()
-	outputStr := strings.TrimSpace(string(output))
+func ViewServiceLogs(ctx context.Context, serviceName string) error {
+	cmd := exec.CommandContext(ctx, "journalctl", "-n", "50", "-u", serviceName, "--output=json", "--no-pager")
 
+	stdout, err := cmd.StdoutPipe()
 	if err != nil {
-		// Если сервис не найден, это не критическая ошибка
-		if strings.Contains(outputStr, "No journal files") {
-			return "Логи для сервиса не найдены", nil
-		}
-		return "", fmt.Errorf("ошибка при получении логов: %w", err)
+		return fmt.Errorf("stdout pipe: %w", err)
 	}
 
-	return outputStr, nil
+	if err := cmd.Start(); err != nil {
+		return fmt.Errorf("ошибка запуска: %w", err)
+	}
+
+	done := make(chan struct{})
+	go func() {
+		scanner := bufio.NewScanner(stdout)
+
+		for scanner.Scan() {
+			jsonLine := scanner.Text()
+
+			var entry map[string]interface{}
+			if err := json.Unmarshal([]byte(jsonLine), &entry); err != nil {
+				fmt.Fprintf(os.Stderr, "ошибка парсинга JSON: %v\n", err)
+				continue
+			}
+
+			if message, ok := entry["MESSAGE"].(string); ok {
+				fmt.Println(message)
+			}
+		}
+
+		if err := scanner.Err(); err != nil {
+			fmt.Fprintf(os.Stderr, "ошибка чтения вывода: %v\n", err)
+		}
+
+		close(done)
+	}()
+
+	select {
+	case <-done:
+		return cmd.Wait()
+	case <-ctx.Done():
+		return cmd.Process.Kill()
+	}
 }
 
 // Полностью установить сервис (создать файл, reload, enable, start)

service_input.go

@@ -1,6 +1,7 @@
 package sdmanager
 
 import (
+	"context"
 	"strings"
 
 	"github.com/charmbracelet/bubbles/textinput"
@@ -40,7 +41,7 @@ func NewServiceInputModel(action string) ServiceInputModel {
 }
 
 // Обработка событий при вводе имени сервиса
-func UpdateServiceInput(msg tea.Msg, model ServiceInputModel) (ServiceInputModel, tea.Cmd, error) {
+func UpdateServiceInput(ctx context.Context, msg tea.Msg, model ServiceInputModel) (ServiceInputModel, tea.Cmd, error) {
 	switch msg := msg.(type) {
 	case tea.KeyMsg:
 		switch msg.Type {
@@ -75,7 +76,7 @@ func UpdateServiceInput(msg tea.Msg, model ServiceInputModel) (ServiceInputModel
 			case ActionRestart:
 				result, err = RestartService(serviceName)
 			case ActionViewLog:
-				result, err = ViewServiceLogs(serviceName)
+				err = ViewServiceLogs(ctx, serviceName)
 			}
 
 			if err != nil {