more abstraction and simple dashboard for processes

supervisor/process.go

@@ -0,0 +1,93 @@
+package supervisor
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/exec"
+	"strings"
+	"time"
+)
+
+type Process struct {
+	Name             string             `json:"name"`
+	Description      string             `json:"description,omitempty"`
+	ExecutePath      string             `json:"executePath"`
+	WorkingDirectory string             `json:"workingDirectory,omitempty"`
+	StartDelay       int                `json:"startDelay,omitempty"`
+	Priority         int                `json:"priority,omitempty"`
+	Arguments        []string           `json:"arguments,omitempty"`
+	process          *exec.Cmd          `json:"-"`
+	cancel           context.CancelFunc `json:"-"`
+	Running          bool               `json:"-"`
+}
+
+var Init bool
+
+func (p *Process) Start() error {
+	if p.Running {
+		return fmt.Errorf("process %s already running", p.Name)
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	p.cancel = cancel
+
+	var args []string
+	for _, arg := range p.Arguments {
+		fields := strings.Fields(arg)
+		args = append(args, fields...)
+	}
+
+	cmd := exec.CommandContext(ctx, p.ExecutePath, args...)
+	cmd.Dir = p.WorkingDirectory
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	p.process = cmd
+
+	p.Running = true
+
+	if !Init {
+		time.Sleep(time.Duration(p.StartDelay) * time.Millisecond)
+		Init = true
+	}
+
+	if err := cmd.Start(); err != nil {
+		p.Running = false
+		p.process = nil
+		p.cancel = nil
+		return fmt.Errorf("failed to start process %s: %w", p.Name, err)
+	}
+
+	go func() {
+		err := cmd.Wait()
+		if err != nil {
+		} else {
+			p.Running = false
+			p.process = nil
+			p.cancel = nil
+		}
+		p.Running = false
+	}()
+
+	return nil
+}
+
+func (p *Process) Stop() error {
+	if !p.Running {
+		return fmt.Errorf("process %s is not running", p.Name)
+	}
+
+	// Cancel the context
+	p.cancel()
+
+	err := p.process.Process.Kill()
+	if err != nil {
+		return fmt.Errorf("failed to kill process %s: %w", p.Name, err)
+	}
+
+	p.Running = false
+	p.process = nil
+	p.cancel = nil
+	return nil
+}

supervisor/supervisor.go

@@ -0,0 +1,58 @@
+package supervisor
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"sort"
+)
+
+type Supervisor struct {
+	Processes []*Process `json:"processes"`
+}
+
+func ReadTemplate(path string) (supervisor Supervisor, err error) {
+	content, err := os.ReadFile(path)
+	if err != nil {
+		return
+	}
+
+	err = json.Unmarshal(content, &supervisor)
+	return
+}
+
+func (s *Supervisor) StartProcesses() error {
+	sort.Slice(s.Processes, func(i, j int) bool {
+		return s.Processes[i].Priority < s.Processes[j].Priority
+	})
+
+	for _, p := range s.Processes {
+		p.Start()
+	}
+	return nil
+}
+
+func (s *Supervisor) GetProcesses() (processes []Process) {
+	for _, p := range s.Processes {
+		processes = append(processes, *p)
+	}
+	return
+}
+
+func (s *Supervisor) StartProcessByName(name string) error {
+	for _, p := range s.Processes {
+		if p.Name == name {
+			return p.Start()
+		}
+	}
+	return fmt.Errorf("process %s not found", name)
+}
+
+func (s *Supervisor) StopProcessByName(name string) error {
+	for _, p := range s.Processes {
+		if p.Name == name {
+			return p.Stop()
+		}
+	}
+	return fmt.Errorf("process %s not found", name)
+}