main.go

package main

import (
	"context"
	"fmt"
	"github.com/sqkam/systemdx/color"

	"time"

	"sync"

	"go.uber.org/automaxprocs/maxprocs"
	"gopkg.in/natefinch/lumberjack.v2"
	"os"
	"os/exec"
	"os/signal"
	"path/filepath"
	"strings"
)

var ListenConfigChan chan struct{}
var wg = &sync.WaitGroup{}
var cancelFuncs []func()

func main() {
	_, _ = maxprocs.Set(maxprocs.Logger(func(string, ...any) {}))
	go func() {
		for {
			for _, val := range cancelFuncs {
				val()
			}
			wg.Wait()
			//time.Sleep(time.Second)
			cancelFuncs = nil
			conf := InitConfig()
			for _, val := range conf.Units {
				val := val
				if !val.Disable {
					go run(val.Exec, val.WorkDir)
				}
			}
			<-ListenConfigChan
		}
	}()

	quit := make(chan os.Signal)
	signal.Notify(quit, os.Interrupt, os.Kill)
	<-quit
	for _, val := range cancelFuncs {
		val()
	}
	wg.Wait()
	fmt.Printf("%v\n", "exit")
	time.Sleep(time.Second * 1)

}

func run(cmdRaw, workDir string) {
	wg.Add(1)
	cmdString, args := splitCmd(cmdRaw)
	cmdName := filepath.Base(cmdString)
	absWorkDir, _ := filepath.Abs(workDir)
	absCwd, _ := filepath.Abs(".")

	err := os.MkdirAll("./log", 0644)
	if err != nil {
		fmt.Printf("[systemdd] err |%s %s %s|\n", color.Yellow, err.Error(), color.Reset)
		return
	}
	runningCtx, cancelFunc := context.WithCancel(context.Background())
	cancelFuncs = append(cancelFuncs, cancelFunc)
	cmd := exec.Command(cmdString, args...)
	cmd.Dir = absWorkDir
	if workDir == "" && filepath.IsAbs(cmdString) {
		fmt.Printf("[systemdd] err |%s %s: %s %s|\n", color.Yellow, cmdString, "workDir empty , attempting to use cmd path", color.Reset)
		cmd.Dir = filepath.Dir(cmdString)
	}

	logFileName := filepath.Join("./log", cmdName+".log")
	lumberjackLogger := &lumberjack.Logger{
		Filename:   logFileName,
		MaxSize:    20,   // A file can be up to 20M.
		MaxBackups: 5,    // Save up to 5 files at the same time.
		MaxAge:     10,   // A file can exist for a maximum of 10 days.
		Compress:   true, // Compress with gzip.
	}

	cmd.Stdout = lumberjackLogger
	cmd.Stderr = lumberjackLogger
	err = cmd.Start()
	if err != nil {
		fmt.Printf("[systemdd] err |%s %s %s|\n", color.Yellow, "start failed attempting to use default path", color.Reset)
		cmd.Dir = absCwd
		err = cmd.Start()
		if err != nil {
			fmt.Printf("[systemdd] err |%s %s %s|\n", color.Yellow, "start failed"+cmdRaw, color.Reset)
			return
		}
		fmt.Printf("[systemdd] success |%s %s %s|\n", color.Blue, "using default path start success", color.Reset)
	}
	fmt.Printf("[systemdd] started |%s %s is running,work_dir:%s %s|\n", color.Magenta, cmdName, cmd.Dir, color.Reset)

	go func() {
		select {
		case <-runningCtx.Done():
			if cmd.ProcessState == nil {
				fmt.Printf("[systemdd] killing |%s %s %s|\n", color.Cyan, cmdName, color.Reset)
				cmd.Process.Kill()
			}
		}
	}()

	err = cmd.Wait()
	var errMessage string
	if err == nil {
		errMessage = "nil"
	} else {
		errMessage = err.Error()
	}
	fmt.Printf("[systemdd] finished |%s %s %s| errorMessage: %s %s %s\n", color.Green, cmdRaw, color.Reset, color.Red, errMessage, color.Reset)
	wg.Done()
}

func splitCmd(cmdRaw string) (cmd string, args []string) {
	cmdRaw = strings.Trim(cmdRaw, " ")
	data := make([]byte, 0, len(cmdRaw))
	j := false

	for _, d := range []byte(cmdRaw) {
		if d >= 'A' && d <= 'z' || d == []byte("-")[0] || d == []byte("'")[0] || d == []byte("\"")[0] || d == []byte("/")[0] {
			if j {
				data = append(data, ' ')

				j = false
			}
			data = append(data, d)

		} else {
			j = true
		}
	}
	split := strings.Split(string(data), " ")
	return split[0], split[1:]
}