77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
// Package executor exec commands
|
|
package executor
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
var localVars = make(map[string]string)
|
|
|
|
func BuildExecutor(oldStateTerminal *term.State, fd *int) func(string) {
|
|
return func(line string) {
|
|
line = strings.TrimSpace(line)
|
|
|
|
if strings.Contains(line, "=") && !strings.HasPrefix(line, "export ") {
|
|
parts := strings.SplitN(line, "=", 2)
|
|
key := strings.TrimSpace(parts[0])
|
|
value := strings.TrimSpace(parts[1])
|
|
|
|
localVars[key] = value
|
|
return
|
|
}
|
|
|
|
line = os.Expand(line, GetVariable)
|
|
|
|
if line == "" {
|
|
return
|
|
}
|
|
|
|
args := strings.Fields(line)
|
|
command := args[0]
|
|
commandArgs := args[1:]
|
|
|
|
if command == "exit" {
|
|
fmt.Println("Bye Bye my small deer :3")
|
|
if oldStateTerminal != nil {
|
|
_ = term.Restore(*fd, oldStateTerminal)
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
if command == "cd" {
|
|
if len(commandArgs) < 1 {
|
|
fmt.Fprintln(os.Stderr, "cd: argument was skipped")
|
|
return
|
|
}
|
|
if err := os.Chdir(commandArgs[0]); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
}
|
|
return
|
|
}
|
|
|
|
cmd := exec.Command(command, commandArgs...)
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "Execution error: %v\n", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetVariable(key string) string {
|
|
if val, exists := localVars[key]; exists {
|
|
return val
|
|
}
|
|
if val := os.Getenv(key); val != "" {
|
|
return val
|
|
}
|
|
return ""
|
|
}
|