summaryrefslogtreecommitdiffstats
path: root/cmd-exec.go
diff options
context:
space:
mode:
Diffstat (limited to 'cmd-exec.go')
-rw-r--r--cmd-exec.go166
1 files changed, 166 insertions, 0 deletions
diff --git a/cmd-exec.go b/cmd-exec.go
new file mode 100644
index 0000000..612851f
--- /dev/null
+++ b/cmd-exec.go
@@ -0,0 +1,166 @@
+package main
+
+import (
+ "fmt"
+ "os"
+ "strings"
+
+ "github.com/iotbzh/xds-agent/lib/apiv1"
+ common "github.com/iotbzh/xds-common/golib"
+ "github.com/joho/godotenv"
+ "github.com/urfave/cli"
+)
+
+func initCmdExec(cmdDef *[]cli.Command) {
+ *cmdDef = append(*cmdDef, cli.Command{
+ Name: "exec",
+ Usage: "execute a command in XDS",
+ Action: exec,
+ Flags: []cli.Flag{
+ cli.StringFlag{
+ Name: "id",
+ EnvVar: "XDS_PROJECT_ID",
+ Usage: "project ID you want to build (mandatory variable)",
+ },
+ cli.StringFlag{
+ Name: "rpath",
+ EnvVar: "XDS_RPATH",
+ Usage: "relative path into project",
+ },
+ cli.StringFlag{
+ Name: "sdkid",
+ EnvVar: "XDS_SDK_ID",
+ Usage: "Cross Sdk ID to use to build project",
+ },
+ },
+ })
+}
+
+func exec(ctx *cli.Context) error {
+ prjID := ctx.String("id")
+ confFile := ctx.String("config")
+ rPath := ctx.String("rPath")
+ sdkid := ctx.String("sdkid")
+
+ // Check mandatory args
+ if prjID == "" {
+ return cli.NewExitError("project id must be set (see --id option)", 1)
+ }
+
+ // Load config file if requested
+ envMap := make(map[string]string)
+ if confFile != "" {
+ if !common.Exists(confFile) {
+ exitError(1, "Error env config file not found")
+ }
+ // Load config file variables that will overwrite env variables
+ err := godotenv.Overload(confFile)
+ if err != nil {
+ exitError(1, "Error loading env config file "+confFile)
+ }
+ envMap, err = godotenv.Read(confFile)
+ if err != nil {
+ exitError(1, "Error reading env config file "+confFile)
+ }
+ }
+
+ argsCommand := make([]string, len(ctx.Args()))
+ copy(argsCommand, ctx.Args())
+ Log.Infof("Execute: /exec %v", argsCommand)
+
+ // Log useful info for debugging
+ ver := apiv1.XDSVersion{}
+ XdsVersionGet(&ver)
+ Log.Infof("XDS version: %v", ver)
+
+ // Process Socket IO events
+ type exitResult struct {
+ error error
+ code int
+ }
+ exitChan := make(chan exitResult, 1)
+
+ IOsk.On("disconnection", func(err error) {
+ exitChan <- exitResult{err, 2}
+ })
+
+ outFunc := func(timestamp, stdout, stderr string) {
+ tm := ""
+ if ctx.Bool("WithTimestamp") {
+ tm = timestamp + "| "
+ }
+ if stdout != "" {
+ fmt.Printf("%s%s", tm, stdout)
+ }
+ if stderr != "" {
+ fmt.Fprintf(os.Stderr, "%s%s", tm, stderr)
+ }
+ }
+
+ IOsk.On(apiv1.ExecOutEvent, func(ev apiv1.ExecOutMsg) {
+ outFunc(ev.Timestamp, ev.Stdout, ev.Stderr)
+ })
+
+ IOsk.On(apiv1.ExecExitEvent, func(ev apiv1.ExecExitMsg) {
+ exitChan <- exitResult{ev.Error, ev.Code}
+ })
+
+ // Retrieve the project definition
+ prj := apiv1.ProjectConfig{}
+ if err := HTTPCli.Get("/projects/"+prjID, &prj); err != nil {
+ return cli.NewExitError(err, 1)
+ }
+
+ // Auto setup rPath if needed
+ if rPath == "" {
+ cwd, err := os.Getwd()
+ if err == nil {
+ fldRp := prj.ClientPath
+ if !strings.HasPrefix(fldRp, "/") {
+ fldRp = "/" + fldRp
+ }
+ Log.Debugf("Try to auto-setup rPath: cwd=%s ; ClientPath=%s", cwd, fldRp)
+ if sp := strings.SplitAfter(cwd, fldRp); len(sp) == 2 {
+ rPath = strings.Trim(sp[1], "/")
+ Log.Debugf("Auto-setup rPath to: '%s'", rPath)
+ }
+ }
+ }
+
+ // Build env
+ Log.Debugf("Command env: %v", envMap)
+ env := []string{}
+ for k, v := range envMap {
+ env = append(env, k+"="+v)
+ }
+
+ // Send build command
+ args := apiv1.ExecArgs{
+ ID: prjID,
+ SdkID: sdkid,
+ Cmd: strings.Trim(argsCommand[0], " "),
+ Args: argsCommand[1:],
+ Env: env,
+ RPath: rPath,
+ CmdTimeout: 60,
+ }
+
+ LogPost("POST /exec %v", args)
+ if err := HTTPCli.Post("/exec", args, nil); err != nil {
+ return cli.NewExitError(err.Error(), 1)
+ }
+
+ // Wait exit
+ select {
+ case res := <-exitChan:
+ errStr := ""
+ if res.code == 0 {
+ Log.Debugln("Exit successfully")
+ }
+ if res.error != nil {
+ Log.Debugln("Exit with ERROR: ", res.error.Error())
+ errStr = res.error.Error()
+ }
+ return cli.NewExitError(errStr, res.code)
+ }
+}