104 lines
1.8 KiB
Go
104 lines
1.8 KiB
Go
package actions
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"unknown.com/gokill/internal"
|
|
)
|
|
|
|
type Command struct {
|
|
Command string `json:"command"`
|
|
ActionChan ActionResultChan
|
|
}
|
|
|
|
func (c Command) DryExecute() {
|
|
fmt.Printf("Test Executing Command:\n%s ", c.Command)
|
|
c.ActionChan <- nil
|
|
}
|
|
|
|
func (c Command) splitCommandString() (string, []string, error) {
|
|
splitted := strings.Fields(c.Command)
|
|
|
|
if len(splitted) == 0 {
|
|
return "", nil, fmt.Errorf("Command is empty")
|
|
}
|
|
|
|
if len(splitted) == 1 {
|
|
return splitted[0], []string(nil), nil
|
|
}
|
|
|
|
return splitted[0], splitted[1:], nil
|
|
}
|
|
|
|
func (c Command) Execute() {
|
|
command, args, err := c.splitCommandString()
|
|
fmt.Println("Executing command: ", c.Command)
|
|
|
|
if err != nil {
|
|
c.ActionChan <- err
|
|
return
|
|
}
|
|
|
|
cmd := exec.Command(command, args...)
|
|
|
|
stdout, err := cmd.Output()
|
|
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
}
|
|
|
|
fmt.Println(string(stdout[:]))
|
|
|
|
c.ActionChan <- nil
|
|
}
|
|
|
|
func CreateCommand(config internal.ActionConfig, c ActionResultChan) (Command, error) {
|
|
result := Command{}
|
|
|
|
err := json.Unmarshal(config.Options, &result)
|
|
|
|
if err != nil {
|
|
return Command{}, err
|
|
}
|
|
|
|
if result.Command == "" {
|
|
return Command{}, internal.OptionMissingError{"command"}
|
|
}
|
|
|
|
result.ActionChan = c
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (cc Command) Create(config internal.ActionConfig, c ActionResultChan) (Action, error) {
|
|
return CreateCommand(config, c)
|
|
}
|
|
|
|
func (p Command) GetName() string {
|
|
return "Command"
|
|
}
|
|
|
|
func (p Command) GetDescription() string {
|
|
return "Invoces given command using exec."
|
|
}
|
|
|
|
func (p Command) GetExample() string {
|
|
return `
|
|
{
|
|
"type": "Command",
|
|
"options": {
|
|
"command": "srm /path/to/file"
|
|
}
|
|
}
|
|
`
|
|
}
|
|
|
|
func (p Command) GetOptions() []internal.ConfigOption {
|
|
return []internal.ConfigOption{
|
|
{"command", "string", "command to execute", ""},
|
|
}
|
|
}
|