main.go

   1 /*
   2  * Copyright (C) 2017-2018 "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  * xds-cli: command line tool used to control / interface X(cross) Development System.
  19  */
  20
  21 package main
  22
  23 import (
  24         "fmt"
  25         "os"
  26         "regexp"
  27         "sort"
  28         "strings"
  29         "text/tabwriter"
  30
  31         "gerrit.automotivelinux.org/gerrit/src/xds/xds-agent.git/lib/xaapiv1"
  32         common "gerrit.automotivelinux.org/gerrit/src/xds/xds-common.git/golib"
  33         "github.com/Sirupsen/logrus"
  34
  35         "github.com/joho/godotenv"
  36         socketio_client "github.com/sebd71/go-socket.io-client"
  37         "github.com/urfave/cli"
  38 )
  39
  40 var appAuthors = []cli.Author{
  41         cli.Author{Name: "Sebastien Douheret", Email: "sebastien@iot.bzh"},
  42 }
  43
  44 // AppName name of this application
  45 var AppName = "xds-cli"
  46
  47 // AppNativeName native command name that this application can overload
  48 var AppNativeName = "cli"
  49
  50 // AppVersion Version of this application
  51 // (set by Makefile)
  52 var AppVersion = "?.?.?"
  53
  54 // AppSubVersion is the git tag id added to version string
  55 // Should be set by compilation -ldflags "-X main.AppSubVersion=xxx"
  56 // (set by Makefile)
  57 var AppSubVersion = "unknown-dev"
  58
  59 // Application details
  60 const (
  61         appCopyright    = "Copyright (C) 2017-2018 IoT.bzh - Apache-2.0"
  62         defaultLogLevel = "error"
  63 )
  64
  65 // Log Global variable that hold logger
  66 var Log = logrus.New()
  67
  68 // EnvConfFileMap Global variable that hold environment vars loaded from config file
  69 var EnvConfFileMap map[string]string
  70
  71 // HTTPCli Global variable that hold HTTP Client
  72 var HTTPCli *common.HTTPClient
  73
  74 // IOsk Global variable that hold SocketIo client
  75 var IOsk *socketio_client.Client
  76
  77 // exitError exists this program with the specified error
  78 func exitError(code int, f string, a ...interface{}) {
  79         earlyDisplay()
  80         err := fmt.Sprintf(f, a...)
  81         fmt.Fprintf(os.Stderr, err+"\n")
  82         os.Exit(code)
  83 }
  84
  85 // earlyDebug Used to log info before logger has been initialized
  86 var earlyDebug []string
  87
  88 func earlyPrintf(format string, args ...interface{}) {
  89         earlyDebug = append(earlyDebug, fmt.Sprintf(format, args...))
  90 }
  91
  92 func earlyDisplay() {
  93         for _, str := range earlyDebug {
  94                 Log.Infof("%s", str)
  95         }
  96         earlyDebug = []string{}
  97 }
  98
  99 // main
 100 func main() {
 101
 102         // Allow to set app name from cli (useful for debugging)
 103         if AppName == "" {
 104                 AppName = os.Getenv("XDS_APPNAME")
 105         }
 106         if AppName == "" {
 107                 panic("Invalid setup, AppName not define !")
 108         }
 109         if AppNativeName == "" {
 110                 AppNativeName = AppName[4:]
 111         }
 112         appUsage := fmt.Sprintf("command line tool for X(cross) Development System.")
 113         appDescription := fmt.Sprintf("%s utility for X(cross) Development System\n", AppName)
 114         appDescription += `
 115     Setting of global options is driven either by environment variables or by command
 116     line options or using a config file knowning that the following priority order is used:
 117       1. use option value (for example --url option),
 118       2. else use variable 'XDS_xxx' (for example 'XDS_AGENT_URL' variable) when a
 119          config file is specified with '--config|-c' option,
 120       3. else use 'XDS_xxx' (for example 'XDS_AGENT_URL') environment variable.
 121
 122     Examples:
 123     # Get help of 'projects' sub-command
 124     ` + AppName + ` projects --help
 125
 126     # List all SDKs
 127     ` + AppName + ` sdks ls
 128
 129     # Add a new project
 130     ` + AppName + ` prj add --label="myProject" --type=cs --path=$HOME/xds-workspace/myProject
 131 `
 132
 133         // Create a new App instance
 134         app := cli.NewApp()
 135         app.Name = AppName
 136         app.Usage = appUsage
 137         app.Version = AppVersion + " (" + AppSubVersion + ")"
 138         app.Authors = appAuthors
 139         app.Copyright = appCopyright
 140         app.Metadata = make(map[string]interface{})
 141         app.Metadata["version"] = AppVersion
 142         app.Metadata["git-tag"] = AppSubVersion
 143         app.Metadata["logger"] = Log
 144         app.EnableBashCompletion = true
 145
 146         // Create env vars help
 147         dynDesc := "\nENVIRONMENT VARIABLES:"
 148         for _, f := range app.Flags {
 149                 var env, usage string
 150                 switch f.(type) {
 151                 case cli.StringFlag:
 152                         fs := f.(cli.StringFlag)
 153                         env = fs.EnvVar
 154                         usage = fs.Usage
 155                 case cli.BoolFlag:
 156                         fb := f.(cli.BoolFlag)
 157                         env = fb.EnvVar
 158                         usage = fb.Usage
 159                 default:
 160                         exitError(1, "Un-implemented option type")
 161                 }
 162                 if env != "" {
 163                         dynDesc += fmt.Sprintf("\n %s \t\t %s", env, usage)
 164                 }
 165         }
 166         app.Description = appDescription + dynDesc
 167
 168         // Declare global flags
 169         app.Flags = []cli.Flag{
 170                 cli.StringFlag{
 171                         Name:   "config, c",
 172                         EnvVar: "XDS_CONFIG",
 173                         Usage:  "env config file to source on startup",
 174                 },
 175                 cli.StringFlag{
 176                         Name:   "log, l",
 177                         EnvVar: "XDS_LOGLEVEL",
 178                         Usage:  "logging level (supported levels: panic, fatal, error, warn, info, debug)",
 179                         Value:  defaultLogLevel,
 180                 },
 181                 cli.StringFlag{
 182                         Name:   "logfile",
 183                         Value:  "stderr",
 184                         Usage:  "filename where logs will be redirected (default stderr)\n\t",
 185                         EnvVar: "XDS_LOGFILENAME",
 186                 },
 187                 cli.StringFlag{
 188                         Name:   "url, u",
 189                         EnvVar: "XDS_AGENT_URL",
 190                         Value:  "localhost:8800",
 191                         Usage:  "local XDS agent url",
 192                 },
 193                 cli.StringFlag{
 194                         Name:   "url-server, us",
 195                         EnvVar: "XDS_SERVER_URL",
 196                         Value:  "",
 197                         Usage:  "overwrite remote XDS server url (default value set in xds-agent-config.json file)",
 198                 },
 199                 cli.BoolFlag{
 200                         Name:   "timestamp, ts",
 201                         EnvVar: "XDS_TIMESTAMP",
 202                         Usage:  "prefix output with timestamp",
 203                 },
 204         }
 205
 206         // Declare commands
 207         app.Commands = []cli.Command{}
 208
 209         initCmdProjects(&app.Commands)
 210         initCmdSdks(&app.Commands)
 211         initCmdExec(&app.Commands)
 212         initCmdTargets(&app.Commands)
 213         initCmdMisc(&app.Commands)
 214
 215         // Add --config option to all commands to support --config option either before or after command verb
 216         // IOW support following both syntaxes:
 217         //   xds-cli exec --config myPrj.conf ...
 218         //   xds-cli --config myPrj.conf exec ...
 219         for i, cmd := range app.Commands {
 220                 if len(cmd.Flags) > 0 {
 221                         app.Commands[i].Flags = append(cmd.Flags, cli.StringFlag{Hidden: true, Name: "config, c"})
 222                 }
 223                 for j, subCmd := range cmd.Subcommands {
 224                         app.Commands[i].Subcommands[j].Flags = append(subCmd.Flags, cli.StringFlag{Hidden: true, Name: "config, c"})
 225                 }
 226         }
 227
 228         sort.Sort(cli.FlagsByName(app.Flags))
 229         sort.Sort(cli.CommandsByName(app.Commands))
 230
 231         // Early and manual processing of --config option in order to set XDS_xxx
 232         // variables before parsing of option by app cli
 233         confFile := os.Getenv("XDS_CONFIG")
 234         for idx, a := range os.Args[1:] {
 235                 if a == "-c" || a == "--config" || a == "-config" {
 236                         confFile = os.Args[idx+2]
 237                         break
 238                 }
 239         }
 240
 241         // Load config file if requested
 242         if confFile != "" {
 243                 earlyPrintf("confFile detected: %v", confFile)
 244                 confFile, err := common.ResolveEnvVar(confFile)
 245                 if err != nil {
 246                         exitError(1, "Error while resolving confFile: %v", err)
 247                 }
 248                 earlyPrintf("Resolved confFile: %v", confFile)
 249                 if !common.Exists(confFile) {
 250                         exitError(1, "Error env config file not found")
 251                 }
 252                 // Load config file variables that will overwrite env variables
 253                 err = godotenv.Overload(confFile)
 254                 if err != nil {
 255                         exitError(1, "Error loading env config file "+confFile)
 256                 }
 257
 258                 // Keep confFile settings in a map
 259                 EnvConfFileMap, err = godotenv.Read(confFile)
 260                 if err != nil {
 261                         exitError(1, "Error reading env config file "+confFile)
 262                 }
 263                 earlyPrintf("EnvConfFileMap: %v", EnvConfFileMap)
 264         }
 265
 266         app.Before = func(ctx *cli.Context) error {
 267                 var err error
 268
 269                 // Don't init anything when no argument or help option is set
 270                 if ctx.NArg() == 0 {
 271                         return nil
 272                 }
 273                 for _, a := range ctx.Args() {
 274                         switch a {
 275                         case "-h", "--h", "-help", "--help":
 276                                 return nil
 277                         }
 278                 }
 279
 280                 loglevel := ctx.String("log")
 281                 // Set logger level and formatter
 282                 if Log.Level, err = logrus.ParseLevel(loglevel); err != nil {
 283                         msg := fmt.Sprintf("Invalid log level : \"%v\"\n", loglevel)
 284                         return cli.NewExitError(msg, 1)
 285                 }
 286                 Log.Formatter = &logrus.TextFormatter{}
 287
 288                 if ctx.String("logfile") != "stderr" {
 289                         logFile, _ := common.ResolveEnvVar(ctx.String("logfile"))
 290                         fdL, err := os.OpenFile(logFile, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
 291                         if err != nil {
 292                                 msgErr := fmt.Sprintf("Cannot create log file %s", logFile)
 293                                 return cli.NewExitError(msgErr, 1)
 294                         }
 295                         Log.Infof("Logging to file: %s", logFile)
 296                         Log.Out = fdL
 297                 }
 298
 299                 Log.Infof("%s version: %s", AppName, app.Version)
 300                 earlyDisplay()
 301                 Log.Debugf("\nEnvironment: %v\n", os.Environ())
 302
 303                 if err = XdsConnInit(ctx); err != nil {
 304                         // Directly call HandleExitCoder to avoid to print help (ShowAppHelp)
 305                         // Note that this function wil never return and program will exit
 306                         cli.HandleExitCoder(err)
 307                 }
 308
 309                 return nil
 310         }
 311
 312         // Close HTTP client and WS connection on exit
 313         defer func() {
 314                 XdsConnClose()
 315         }()
 316
 317         // Start signals monitoring routine
 318         MonitorSignals()
 319
 320         // Run the cli app
 321         app.Run(os.Args)
 322 }
 323
 324 // XdsConnInit Initialized HTTP and WebSocket connection to XDS agent
 325 func XdsConnInit(ctx *cli.Context) error {
 326         var err error
 327
 328         // Define HTTP and WS url
 329         agentURL := ctx.String("url")
 330         serverURL := ctx.String("url-server")
 331
 332         // Allow to only set port number
 333         if match, _ := regexp.MatchString("^([0-9]+)$", agentURL); match {
 334                 agentURL = "http://localhost:" + ctx.String("url")
 335         }
 336         if match, _ := regexp.MatchString("^([0-9]+)$", serverURL); match {
 337                 serverURL = "http://localhost:" + ctx.String("url-server")
 338         }
 339         // Add http prefix if missing
 340         if agentURL != "" && !strings.HasPrefix(agentURL, "http://") {
 341                 agentURL = "http://" + agentURL
 342         }
 343         if serverURL != "" && !strings.HasPrefix(serverURL, "http://") {
 344                 serverURL = "http://" + serverURL
 345         }
 346
 347         lvl := common.HTTPLogLevelWarning
 348         if Log.Level == logrus.DebugLevel {
 349                 lvl = common.HTTPLogLevelDebug
 350         }
 351
 352         // Create HTTP client
 353         Log.Debugln("Connect HTTP client on ", agentURL)
 354         conf := common.HTTPClientConfig{
 355                 URLPrefix:           "/api/v1",
 356                 HeaderClientKeyName: "Xds-Agent-Sid",
 357                 CsrfDisable:         true,
 358                 LogOut:              Log.Out,
 359                 LogPrefix:           "XDSAGENT: ",
 360                 LogLevel:            lvl,
 361         }
 362
 363         HTTPCli, err = common.HTTPNewClient(agentURL, conf)
 364         if err != nil {
 365                 errmsg := err.Error()
 366                 m, err := regexp.MatchString("Get http.?://", errmsg)
 367                 if (m && err == nil) || strings.Contains(errmsg, "Failed to get device ID") {
 368                         i := strings.LastIndex(errmsg, ":")
 369                         newErr := "Cannot connection to " + agentURL
 370                         if i > 0 {
 371                                 newErr += " (" + strings.TrimSpace(errmsg[i+1:]) + ")"
 372                         } else {
 373                                 newErr += " (" + strings.TrimSpace(errmsg) + ")"
 374                         }
 375                         errmsg = newErr
 376                 }
 377                 return cli.NewExitError(errmsg, 1)
 378         }
 379         HTTPCli.SetLogLevel(ctx.String("loglevel"))
 380         Log.Infoln("HTTP session ID : ", HTTPCli.GetClientID())
 381
 382         // Create io Websocket client
 383         Log.Debugln("Connecting IO.socket client on ", agentURL)
 384
 385         opts := &socketio_client.Options{
 386                 Transport: "websocket",
 387                 Header:    make(map[string][]string),
 388         }
 389         opts.Header["XDS-AGENT-SID"] = []string{HTTPCli.GetClientID()}
 390
 391         IOsk, err = socketio_client.NewClient(agentURL, opts)
 392         if err != nil {
 393                 return cli.NewExitError("IO.socket connection error: "+err.Error(), 1)
 394         }
 395
 396         IOsk.On("error", func(err error) {
 397                 fmt.Println("ERROR Websocket: ", err.Error())
 398         })
 399
 400         ctx.App.Metadata["httpCli"] = HTTPCli
 401         ctx.App.Metadata["ioskCli"] = IOsk
 402
 403         // Display version in logs (debug helpers)
 404         ver := xaapiv1.XDSVersion{}
 405         if err := XdsVersionGet(&ver); err != nil {
 406                 return cli.NewExitError("ERROR while retrieving XDS version: "+err.Error(), 1)
 407         }
 408         Log.Infof("XDS Agent/Server version: %v", ver)
 409
 410         // Get current config and update connection to server when needed
 411         xdsConf := xaapiv1.APIConfig{}
 412         if err := XdsConfigGet(&xdsConf); err != nil {
 413                 return cli.NewExitError("ERROR while getting XDS config: "+err.Error(), 1)
 414         }
 415         if len(xdsConf.Servers) < 1 {
 416                 return cli.NewExitError("No XDS Server connected", 1)
 417         }
 418         svrCfg := xdsConf.Servers[XdsServerIndexGet()]
 419         if (serverURL != "" && svrCfg.URL != serverURL) || !svrCfg.Connected {
 420                 Log.Infof("Update XDS Server config: serverURL=%v, svrCfg=%v", serverURL, svrCfg)
 421                 if serverURL != "" {
 422                         svrCfg.URL = serverURL
 423                 }
 424                 svrCfg.ConnRetry = 10
 425                 if err := XdsConfigSet(xdsConf); err != nil {
 426                         return cli.NewExitError("ERROR while updating XDS server URL: "+err.Error(), 1)
 427                 }
 428         }
 429
 430         return nil
 431 }
 432
 433 // XdsConnClose Terminate connection to XDS agent
 434 func XdsConnClose() {
 435         Log.Debugf("Closing HTTP client session...")
 436         /* TODO
 437         if httpCli, ok := app.Metadata["httpCli"]; ok {
 438                 c := httpCli.(*common.HTTPClient)
 439         }
 440         */
 441
 442         Log.Debugf("Closing WebSocket connection...")
 443         /*
 444                 if ioskCli, ok := app.Metadata["ioskCli"]; ok {
 445                         c := ioskCli.(*socketio_client.Client)
 446                 }
 447         */
 448 }
 449
 450 // NewTableWriter Create a writer that inserts padding around tab-delimited
 451 func NewTableWriter() *tabwriter.Writer {
 452         writer := new(tabwriter.Writer)
 453         writer.Init(os.Stdout, 0, 8, 0, '\t', 0)
 454         return writer
 455 }