fix for windows support #119
Description
Line 98 in 1d8f8ea
package props
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"math"
"os"
"os/exec"
"os/user"
"runtime"
"strconv"
"syscall"
"time"
nats "github.com/nats-io/nats.go"
"github.com/taigrr/log-socket/log"
"github.com/gogrlx/grlx/v2/types"
)
var nc *nats.Conn
func RegisterNatsConn(conn *nats.Conn) {
nc = conn
}
func FCache(target types.KeyManager, cmdRun types.CmdRun) (types.CmdRun, error) {
topic := "grlx.sprouts." + target.SproutID + ".props.cache"
var results types.CmdRun
b, _ := json.Marshal(cmdRun)
msg, err := nc.Request(topic, b, time.Second*15+cmdRun.Timeout)
if err != nil {
return results, err
}
err = json.Unmarshal(msg.Data, &results)
return results, err
}
func FGet(target types.KeyManager, cmdRun types.CmdRun) (types.CmdRun, error) {
topic := "grlx.sprouts." + target.SproutID + ".props.get"
var results types.CmdRun
b, _ := json.Marshal(cmdRun)
msg, err := nc.Request(topic, b, time.Second*15+cmdRun.Timeout)
if err != nil {
return results, err
}
err = json.Unmarshal(msg.Data, &results)
return results, err
}
func FSet(target types.KeyManager, cmdRun types.CmdRun) (types.CmdRun, error) {
topic := "grlx.sprouts." + target.SproutID + ".props.set"
var results types.CmdRun
b, _ := json.Marshal(cmdRun)
msg, err := nc.Request(topic, b, time.Second*15+cmdRun.Timeout)
if err != nil {
return results, err
}
err = json.Unmarshal(msg.Data, &results)
return results, err
}
func SRun(cmd types.CmdRun) (types.CmdRun, error) {
ctx, cancel := context.WithTimeout(context.Background(), cmd.Timeout)
defer cancel()
envMutex.Lock()
osPath := os.Getenv("PATH")
newPath := ""
if val, ok := cmd.Env["PATH"]; cmd.Path == "" && (!ok || (ok && val == "")) {
_, err := exec.LookPath(cmd.Command)
if err != nil {
envMutex.Unlock()
cmd.Error = err
return cmd, err
}
} else {
if cmd.Path != "" {
newPath += cmd.Path + string(os.PathListSeparator)
}
if ok && val != "" {
newPath += val + string(os.PathListSeparator)
}
}
os.Setenv("PATH", newPath+osPath)
command := exec.CommandContext(ctx, cmd.Command, cmd.Args...)
os.Setenv("PATH", osPath)
env := os.Environ()
envMutex.Unlock()
for key, val := range cmd.Env {
env = append(env, key+"="+val)
}
command.Env = env
var uid uint32
// TODO fix for windows support
if cmd.RunAs != "" && runtime.GOOS != "windows" {
u, err := user.Lookup(cmd.RunAs)
if err != nil {
return cmd, err
}
uid64, err := strconv.Atoi(u.Uid)
if err != nil {
return cmd, err
}
if uid64 > math.MaxInt32 {
return cmd, fmt.Errorf("UID %d is invalid", uid64)
}
uid = uint32(uid64)
command.SysProcAttr = &syscall.SysProcAttr{}
command.SysProcAttr.Credential = &syscall.Credential{Uid: uid}
}
command.Dir = cmd.CWD
// TODO replace os.Stdout/err here with writes to websocket to get live returnable data
var stdoutBuf, stderrBuf bytes.Buffer
command.Stdout = io.MultiWriter(&stdoutBuf) //, os.Stdout)
command.Stderr = io.MultiWriter(&stderrBuf) //, os.Stderr)
timer := time.Now()
err := command.Run()
cmd.Duration = time.Since(timer)
if err != nil {
log.Errorf("cmd.Run() failed with %s\n", err)
}
cmd.Stdout = stdoutBuf.String()
cmd.Stderr = stderrBuf.String()
cmd.ErrCode = command.ProcessState.ExitCode()
return cmd, err
}