panel/app/http/controllers/plugins/supervisor_controller.go

package plugins

import (
	"strconv"
	"strings"

	"github.com/goravel/framework/contracts/http"
	"panel/pkg/tools"

	"panel/app/http/controllers"
)

type SupervisorController struct {
	ServiceName string
}

func NewSupervisorController() *SupervisorController {
	var serviceName string
	if tools.IsRHEL() {
		serviceName = "supervisord"
	} else {
		serviceName = "supervisor"
	}

	return &SupervisorController{
		ServiceName: serviceName,
	}
}

// Status 状态
func (r *SupervisorController) Status(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	status, err := tools.ServiceStatus(r.ServiceName)
	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "获取Supervisor状态失败")
	}

	return controllers.Success(ctx, status)
}

// Start 启动
func (r *SupervisorController) Start(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	if err := tools.ServiceStart(r.ServiceName); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "启动Supervisor失败")
	}

	return controllers.Success(ctx, nil)
}

// Stop 停止
func (r *SupervisorController) Stop(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	if err := tools.ServiceStop(r.ServiceName); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "停止Supervisor失败")
	}

	return controllers.Success(ctx, nil)
}

// Restart 重启
func (r *SupervisorController) Restart(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	if err := tools.ServiceRestart(r.ServiceName); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "重启Supervisor失败")
	}

	return controllers.Success(ctx, nil)
}

// Reload 重载
func (r *SupervisorController) Reload(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	if err := tools.ServiceReload(r.ServiceName); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "重载Supervisor失败")
	}

	return controllers.Success(ctx, nil)
}

// Log 日志
func (r *SupervisorController) Log(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	log, err := tools.Exec(`tail -n 200 /var/log/supervisor/supervisord.log`)
	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, log)
	}

	return controllers.Success(ctx, log)
}

// ClearLog 清空日志
func (r *SupervisorController) ClearLog(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	if out, err := tools.Exec(`echo "" > /var/log/supervisor/supervisord.log`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// Config 获取配置
func (r *SupervisorController) Config(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	var config string
	if tools.IsRHEL() {
		config = tools.Read(`/etc/supervisord.conf`)
	} else {
		config = tools.Read(`/etc/supervisor/supervisord.conf`)
	}

	return controllers.Success(ctx, config)
}

// SaveConfig 保存配置
func (r *SupervisorController) SaveConfig(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	config := ctx.Request().Input("config")
	var err error
	if tools.IsRHEL() {
		err = tools.Write(`/etc/supervisord.conf`, config, 0644)
	} else {
		err = tools.Write(`/etc/supervisor/supervisord.conf`, config, 0644)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, err.Error())
	}

	return r.Restart(ctx)
}

// Processes 进程列表
func (r *SupervisorController) Processes(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	page := ctx.Request().QueryInt("page", 1)
	limit := ctx.Request().QueryInt("limit", 10)

	type process struct {
		Name   string `json:"name"`
		Status string `json:"status"`
		Pid    string `json:"pid"`
		Uptime string `json:"uptime"`
	}

	out, err := tools.Exec(`supervisorctl status | awk '{print $1}'`)
	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	var processList []process
	for _, line := range strings.Split(out, "\n") {
		if len(line) == 0 {
			continue
		}

		var p process
		p.Name = line
		if status, err := tools.Exec(`supervisorctl status ` + line + ` | awk '{print $2}'`); err == nil {
			p.Status = status
		}
		if p.Status == "RUNNING" {
			pid, _ := tools.Exec(`supervisorctl status ` + line + ` | awk '{print $4}'`)
			p.Pid = strings.ReplaceAll(pid, ",", "")
			uptime, _ := tools.Exec(`supervisorctl status ` + line + ` | awk '{print $6}'`)
			p.Uptime = uptime
		} else {
			p.Pid = "-"
			p.Uptime = "-"
		}
		processList = append(processList, p)
	}

	startIndex := (page - 1) * limit
	endIndex := page * limit
	if startIndex > len(processList) {
		return controllers.Success(ctx, http.Json{
			"total": 0,
			"items": []process{},
		})
	}
	if endIndex > len(processList) {
		endIndex = len(processList)
	}
	pagedProcessList := processList[startIndex:endIndex]
	if pagedProcessList == nil {
		pagedProcessList = []process{}
	}

	return controllers.Success(ctx, http.Json{
		"total": len(processList),
		"items": pagedProcessList,
	})
}

// StartProcess 启动进程
func (r *SupervisorController) StartProcess(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	if out, err := tools.Exec(`supervisorctl start ` + process); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// StopProcess 停止进程
func (r *SupervisorController) StopProcess(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	if out, err := tools.Exec(`supervisorctl stop ` + process); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// RestartProcess 重启进程
func (r *SupervisorController) RestartProcess(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	if out, err := tools.Exec(`supervisorctl restart ` + process); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// ProcessLog 进程日志
func (r *SupervisorController) ProcessLog(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	var logPath string
	var err error
	if tools.IsRHEL() {
		logPath, err = tools.Exec(`cat '/etc/supervisord.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
	} else {
		logPath, err = tools.Exec(`cat '/etc/supervisor/conf.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "无法从进程"+process+"的配置文件中获取日志路径")
	}

	log, err := tools.Exec(`tail -n 200 ` + logPath)
	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, log)
	}

	return controllers.Success(ctx, log)
}

// ClearProcessLog 清空进程日志
func (r *SupervisorController) ClearProcessLog(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	var logPath string
	var err error
	if tools.IsRHEL() {
		logPath, err = tools.Exec(`cat '/etc/supervisord.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
	} else {
		logPath, err = tools.Exec(`cat '/etc/supervisor/conf.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "无法从进程"+process+"的配置文件中获取日志路径")
	}

	if out, err := tools.Exec(`echo "" > ` + logPath); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// ProcessConfig 获取进程配置
func (r *SupervisorController) ProcessConfig(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Query("process")
	var config string
	if tools.IsRHEL() {
		config = tools.Read(`/etc/supervisord.d/` + process + `.conf`)
	} else {
		config = tools.Read(`/etc/supervisor/conf.d/` + process + `.conf`)
	}

	return controllers.Success(ctx, config)
}

// SaveProcessConfig 保存进程配置
func (r *SupervisorController) SaveProcessConfig(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	config := ctx.Request().Input("config")
	var err error
	if tools.IsRHEL() {
		err = tools.Write(`/etc/supervisord.d/`+process+`.conf`, config, 0644)
	} else {
		err = tools.Write(`/etc/supervisor/conf.d/`+process+`.conf`, config, 0644)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusUnprocessableEntity, err.Error())
	}

	if out, err := tools.Exec(`supervisorctl reread`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}
	if out, err := tools.Exec(`supervisorctl update`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}
	if out, err := tools.Exec(`supervisorctl start ` + process); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// AddProcess 添加进程
func (r *SupervisorController) AddProcess(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	validator, err := ctx.Request().Validate(map[string]string{
		"name":    "required|alpha_dash",
		"user":    "required|alpha_dash",
		"path":    "required",
		"command": "required",
		"num":     "required",
	})
	if err != nil {
		return controllers.Error(ctx, http.StatusUnprocessableEntity, err.Error())
	}
	if validator.Fails() {
		return controllers.Error(ctx, http.StatusUnprocessableEntity, validator.Errors().One())
	}

	name := ctx.Request().Input("name")
	user := ctx.Request().Input("user")
	path := ctx.Request().Input("path")
	command := ctx.Request().Input("command")
	num := ctx.Request().InputInt("num", 1)
	config := `[program:` + name + `]
command=` + command + `
process_name=%(program_name)s
directory=` + path + `
autostart=true
autorestart=true
user=` + user + `
numprocs=` + strconv.Itoa(num) + `
redirect_stderr=true
stdout_logfile=/var/log/supervisor/` + name + `.log
stdout_logfile_maxbytes=2MB
`
	if tools.IsRHEL() {
		err = tools.Write(`/etc/supervisord.d/`+name+`.conf`, config, 0644)
	} else {
		err = tools.Write(`/etc/supervisor/conf.d/`+name+`.conf`, config, 0644)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusUnprocessableEntity, err.Error())
	}

	if out, err := tools.Exec(`supervisorctl reread`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}
	if out, err := tools.Exec(`supervisorctl update`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}
	if out, err := tools.Exec(`supervisorctl start ` + name); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}

// DeleteProcess 删除进程
func (r *SupervisorController) DeleteProcess(ctx http.Context) http.Response {
	check := controllers.Check(ctx, "supervisor")
	if check != nil {
		return check
	}

	process := ctx.Request().Input("process")
	if out, err := tools.Exec(`supervisorctl stop ` + process); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	var logPath string
	var err error
	if tools.IsRHEL() {
		logPath, err = tools.Exec(`cat '/etc/supervisord.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
		tools.Remove(`/etc/supervisord.d/` + process + `.conf`)
	} else {
		logPath, err = tools.Exec(`cat '/etc/supervisor/conf.d/` + process + `.conf' | grep stdout_logfile= | awk -F "=" '{print $2}'`)
		tools.Remove(`/etc/supervisor/conf.d/` + process + `.conf`)
	}

	if err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, "无法从进程"+process+"的配置文件中获取日志路径")
	}

	tools.Remove(logPath)
	if out, err := tools.Exec(`supervisorctl reread`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}
	if out, err := tools.Exec(`supervisorctl update`); err != nil {
		return controllers.Error(ctx, http.StatusInternalServerError, out)
	}

	return controllers.Success(ctx, nil)
}