mirror of
https://github.com/OliveTin/OliveTin
synced 2025-12-12 00:55:34 +00:00
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package executor
|
|
|
|
import (
|
|
pb "github.com/jamesread/OliveTin/gen/grpc"
|
|
acl "github.com/jamesread/OliveTin/internal/acl"
|
|
config "github.com/jamesread/OliveTin/internal/config"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"context"
|
|
"errors"
|
|
"os/exec"
|
|
"time"
|
|
)
|
|
|
|
type InternalLogEntry struct {
|
|
Datetime string
|
|
Content string
|
|
Stdout string
|
|
Stderr string
|
|
TimedOut bool
|
|
ExitCode int32
|
|
ActionTitle string
|
|
}
|
|
|
|
type Executor struct {
|
|
Logs []InternalLogEntry
|
|
}
|
|
|
|
// ExecAction executes an action.
|
|
func (e *Executor) ExecAction(cfg *config.Config, user *acl.User, actualAction *config.ActionButton) *pb.StartActionResponse {
|
|
log.WithFields(log.Fields{
|
|
"actionName": actualAction.Title,
|
|
}).Infof("StartAction")
|
|
|
|
res := execAction(cfg, actualAction)
|
|
|
|
e.Logs = append(e.Logs, *res)
|
|
|
|
return &pb.StartActionResponse{
|
|
LogEntry: &pb.LogEntry{
|
|
ActionTitle: actualAction.Title,
|
|
TimedOut: res.TimedOut,
|
|
Stderr: res.Stderr,
|
|
Stdout: res.Stdout,
|
|
ExitCode: res.ExitCode,
|
|
},
|
|
}
|
|
}
|
|
|
|
func execAction(cfg *config.Config, actualAction *config.ActionButton) *InternalLogEntry {
|
|
res := &InternalLogEntry{
|
|
Datetime: time.Now().Format("2006-01-02 15:04:05"),
|
|
TimedOut: false,
|
|
ActionTitle: actualAction.Title,
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"title": actualAction.Title,
|
|
"timeout": actualAction.Timeout,
|
|
}).Infof("Found action")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Duration(actualAction.Timeout)*time.Second)
|
|
defer cancel()
|
|
|
|
cmd := exec.CommandContext(ctx, "sh", "-c", actualAction.Shell)
|
|
stdout, stderr := cmd.Output()
|
|
|
|
res.ExitCode = int32(cmd.ProcessState.ExitCode())
|
|
res.Stdout = string(stdout)
|
|
|
|
if stderr == nil {
|
|
res.Stderr = ""
|
|
} else {
|
|
res.Stderr = stderr.Error()
|
|
}
|
|
|
|
if ctx.Err() == context.DeadlineExceeded {
|
|
res.TimedOut = true
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"stdout": res.Stdout,
|
|
"stderr": res.Stderr,
|
|
"timedOut": res.TimedOut,
|
|
"exit": res.ExitCode,
|
|
}).Infof("Finished command.")
|
|
|
|
return res
|
|
}
|
|
|
|
func sanitizeAction(action *config.ActionButton) {
|
|
if action.Timeout < 3 {
|
|
action.Timeout = 3
|
|
}
|
|
}
|
|
|
|
func FindAction(cfg *config.Config, actionTitle string) (*config.ActionButton, error) {
|
|
for _, action := range cfg.ActionButtons {
|
|
if action.Title == actionTitle {
|
|
sanitizeAction(&action)
|
|
|
|
return &action, nil
|
|
}
|
|
}
|
|
|
|
return nil, errors.New("Action not found")
|
|
}
|