cmd-sdks.go

   1 /*
   2  * Copyright (C) 2017 "IoT.bzh"
   3  * Author Sebastien Douheret <sebastien@iot.bzh>
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *   http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 package main
  20
  21 import (
  22         "fmt"
  23         "os"
  24         "regexp"
  25
  26         "github.com/iotbzh/xds-agent/lib/xaapiv1"
  27         "github.com/urfave/cli"
  28 )
  29
  30 func initCmdSdks(cmdDef *[]cli.Command) {
  31         *cmdDef = append(*cmdDef, cli.Command{
  32                 Name:     "sdks",
  33                 Aliases:  []string{"sdk"},
  34                 HideHelp: true,
  35                 Usage:    "SDKs commands group",
  36                 Subcommands: []cli.Command{
  37                         {
  38                                 Name:   "get",
  39                                 Usage:  "Get a property of a SDK",
  40                                 Action: sdksGet,
  41                                 Flags: []cli.Flag{
  42                                         cli.StringFlag{
  43                                                 Name:   "id",
  44                                                 Usage:  "sdk id",
  45                                                 EnvVar: "XDS_SDK_ID",
  46                                         },
  47                                 },
  48                         },
  49                         {
  50                                 Name:    "list",
  51                                 Aliases: []string{"ls"},
  52                                 Usage:   "List installed SDKs",
  53                                 Action:  sdksList,
  54                                 Flags: []cli.Flag{
  55                                         cli.BoolFlag{
  56                                                 Name:  "all, a",
  57                                                 Usage: "display all existing sdks (installed + downloadable)",
  58                                         },
  59                                         cli.StringFlag{
  60                                                 Name:  "filter, f",
  61                                                 Usage: "regexp to filter output (filtering done only on ID, Name, Version and Arch fields) ",
  62                                         },
  63                                         cli.BoolFlag{
  64                                                 Name:  "verbose, v",
  65                                                 Usage: "display verbose output",
  66                                         },
  67                                 },
  68                         },
  69                         {
  70                                 Name:    "install",
  71                                 Aliases: []string{"i"},
  72                                 Usage:   "Install a SDK",
  73                                 Action:  sdksInstall,
  74                                 Flags: []cli.Flag{
  75                                         cli.StringFlag{
  76                                                 Name:   "id",
  77                                                 Usage:  "sdk id to install",
  78                                                 EnvVar: "XDS_SDK_ID",
  79                                         },
  80                                         cli.StringFlag{
  81                                                 Name:  "file, f",
  82                                                 Usage: "use this file to install SDK",
  83                                         },
  84                                         cli.BoolFlag{
  85                                                 Name:  "force",
  86                                                 Usage: "force SDK installation when already installed",
  87                                         },
  88                                 },
  89                         },
  90                         {
  91                                 Name:    "uninstall",
  92                                 Aliases: []string{"rm"},
  93                                 Usage:   "UnInstall an existing SDK",
  94                                 Action:  sdksUnInstall,
  95                                 Flags: []cli.Flag{
  96                                         cli.StringFlag{
  97                                                 Name:   "id",
  98                                                 Usage:  "sdk id to un-install",
  99                                                 EnvVar: "XDS_SDK_ID",
 100                                         },
 101                                 },
 102                         },
 103                         {
 104                                 Name:    "abort",
 105                                 Aliases: []string{"a"},
 106                                 Usage:   "Abort an install action",
 107                                 Action:  sdksAbort,
 108                                 Flags: []cli.Flag{
 109                                         cli.StringFlag{
 110                                                 Name:   "id",
 111                                                 Usage:  "sdk id to which abort action",
 112                                                 EnvVar: "XDS_SDK_ID",
 113                                         },
 114                                 },
 115                         },
 116                 },
 117         })
 118 }
 119
 120 func sdksList(ctx *cli.Context) error {
 121         // Get SDKs list
 122         sdks := []xaapiv1.SDK{}
 123         if err := _sdksListGet(&sdks); err != nil {
 124                 return cli.NewExitError(err.Error(), 1)
 125         }
 126
 127         _displaySdks(sdks, ctx.Bool("verbose"), ctx.Bool("all"), ctx.String("filter"))
 128         return nil
 129 }
 130
 131 func sdksGet(ctx *cli.Context) error {
 132         id := GetID(ctx)
 133         if id == "" {
 134                 return cli.NewExitError("id parameter or option must be set", 1)
 135         }
 136         sdks := xaapiv1.SDK{}
 137         url := XdsServerComputeURL("/sdks/" + id)
 138         if err := HTTPCli.Get(url, &sdks); err != nil {
 139                 return cli.NewExitError(err.Error(), 1)
 140         }
 141
 142         _displaySdks([]xaapiv1.SDK{sdks}, true, true, "")
 143         return nil
 144 }
 145
 146 func _displaySdks(sdks []xaapiv1.SDK, verbose bool, all bool, filter string) {
 147         // Display result
 148         first := true
 149         writer := NewTableWriter()
 150         for _, s := range sdks {
 151                 if s.Status != xaapiv1.SdkStatusInstalled && !all {
 152                         continue
 153                 }
 154                 if filter != "" {
 155                         re := regexp.MustCompile(filter)
 156                         if !(re.MatchString(s.ID) || re.MatchString(s.Name) ||
 157                                 re.MatchString(s.Profile) || re.MatchString(s.Arch) ||
 158                                 re.MatchString(s.Version)) {
 159                                 continue
 160                         }
 161                 }
 162
 163                 if verbose {
 164                         if !first {
 165                                 fmt.Fprintln(writer)
 166                         }
 167                         fmt.Fprintln(writer, "ID\t"+s.ID)
 168                         fmt.Fprintln(writer, "Name\t"+s.Name)
 169                         fmt.Fprintln(writer, "Description\t"+s.Description)
 170                         fmt.Fprintln(writer, "Profile\t"+s.Profile)
 171                         fmt.Fprintln(writer, "Arch\t"+s.Arch)
 172                         fmt.Fprintln(writer, "Version\t"+s.Version)
 173                         fmt.Fprintln(writer, "Status\t"+s.Status)
 174                         fmt.Fprintln(writer, "Path\t"+s.Path)
 175                         fmt.Fprintln(writer, "Url\t"+s.URL)
 176
 177                 } else {
 178                         if first {
 179                                 if all {
 180                                         fmt.Fprintf(writer, "List of available SDKs: \n")
 181                                 } else {
 182                                 fmt.Fprintf(writer, "List of installed SDKs: \n")
 183                                 }
 184                                 fmt.Fprintf(writer, "  ID\tNAME\tSTATUS\tVERSION\tARCH\n")
 185                         }
 186                         fmt.Fprintf(writer, "  %s\t%s\t%s\t%s\t%s\n", s.ID[:8], s.Name, s.Status, s.Version, s.Arch)
 187                 }
 188                 first = false
 189         }
 190         writer.Flush()
 191 }
 192
 193 func _sdksListGet(sdks *[]xaapiv1.SDK) error {
 194         url := XdsServerComputeURL("/sdks")
 195         if err := HTTPCli.Get(url, &sdks); err != nil {
 196                 return err
 197         }
 198         Log.Debugf("Result of %s: %v", url, sdks)
 199
 200         return nil
 201 }
 202
 203 func sdksInstall(ctx *cli.Context) error {
 204         id := GetID(ctx)
 205         if id == "" {
 206                 return cli.NewExitError("id parameter or option must be set", 1)
 207         }
 208
 209         // Process Socket IO events
 210         type exitResult struct {
 211                 error string
 212                 code  int
 213         }
 214         exitChan := make(chan exitResult, 1)
 215
 216         IOsk.On("disconnection", func(err error) {
 217                 Log.Debugf("WS disconnection event with err: %v\n", err)
 218                 errMsg := ""
 219                 if err != nil {
 220                         errMsg = err.Error()
 221                 }
 222                 exitChan <- exitResult{errMsg, 2}
 223         })
 224
 225         IOsk.On(xaapiv1.EVTSDKInstall, func(ev xaapiv1.EventMsg) {
 226                 sdkEvt, _ := ev.DecodeSDKMsg()
 227
 228                 if sdkEvt.Stdout != "" {
 229                         fmt.Printf("%s", sdkEvt.Stdout)
 230                 }
 231                 if sdkEvt.Stderr != "" {
 232                         fmt.Fprintf(os.Stderr, "%s", sdkEvt.Stderr)
 233                 }
 234
 235                 if sdkEvt.Exited {
 236                         exitChan <- exitResult{sdkEvt.Error, sdkEvt.Code}
 237                 }
 238         })
 239
 240         evReg := xaapiv1.EventRegisterArgs{Name: xaapiv1.EVTSDKInstall}
 241         if err := HTTPCli.Post("/events/register", &evReg, nil); err != nil {
 242                 return cli.NewExitError(err, 1)
 243         }
 244
 245         file := ctx.String("file")
 246         force := ctx.Bool("force")
 247         url := XdsServerComputeURL("/sdks")
 248         sdks := xaapiv1.SDKInstallArgs{ID: id, Filename: file, Force: force}
 249         newSdk := xaapiv1.SDK{}
 250         if err := HTTPCli.Post(url, &sdks, &newSdk); err != nil {
 251                 return cli.NewExitError(err, 1)
 252         }
 253         Log.Debugf("Result of %s: %v", url, newSdk)
 254         fmt.Printf("Installation of '%s' SDK (id %v) successfully started.\n", newSdk.Name, newSdk.ID)
 255
 256
 257         // Wait exit
 258         select {
 259         case res := <-exitChan:
 260                 if res.code == 0 {
 261                         Log.Debugln("Exit successfully")
 262                 }
 263                 if res.error != "" {
 264                         Log.Debugln("Exit with ERROR: ", res.error)
 265                 }
 266                 return cli.NewExitError(res.error, res.code)
 267         }
 268 }
 269
 270 func sdksUnInstall(ctx *cli.Context) error {
 271         id := GetID(ctx)
 272         if id == "" {
 273                 return cli.NewExitError("id parameter or option must be set", 1)
 274         }
 275
 276         return fmt.Errorf("not supported yet")
 277 }
 278
 279 func sdksAbort(ctx *cli.Context) error {
 280         id := GetID(ctx)
 281         if id == "" {
 282                 return cli.NewExitError("id parameter or option must be set", 1)
 283         }
 284
 285         sdks := xaapiv1.SDKInstallArgs{ID: id}
 286         newSdk := xaapiv1.SDK{}
 287         url := XdsServerComputeURL("/sdks/abortinstall")
 288         if err := HTTPCli.Post(url, &sdks, &newSdk); err != nil {
 289                 return cli.NewExitError(err, 1)
 290         }
 291
 292         Log.Debugf("Result of %s: %v", url, newSdk)
 293         return nil
 294 }