aboutsummaryrefslogtreecommitdiffstats
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go263
1 files changed, 263 insertions, 0 deletions
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..c8d2095
--- /dev/null
+++ b/main.go
@@ -0,0 +1,263 @@
+// xds-cli: command line tool used to control / interface X(cross) Development System.
+package main
+
+import (
+ "fmt"
+ "os"
+ "regexp"
+ "sort"
+ "strings"
+ "text/tabwriter"
+
+ "github.com/Sirupsen/logrus"
+ common "github.com/iotbzh/xds-common/golib"
+ socketio_client "github.com/sebd71/go-socket.io-client"
+ "github.com/urfave/cli"
+)
+
+var appAuthors = []cli.Author{
+ cli.Author{Name: "Sebastien Douheret", Email: "sebastien@iot.bzh"},
+}
+
+// AppName name of this application
+var AppName = "xds-cli"
+
+// AppNativeName native command name that this application can overload
+var AppNativeName = "cli"
+
+// AppVersion Version of this application
+// (set by Makefile)
+var AppVersion = "?.?.?"
+
+// AppSubVersion is the git tag id added to version string
+// Should be set by compilation -ldflags "-X main.AppSubVersion=xxx"
+// (set by Makefile)
+var AppSubVersion = "unknown-dev"
+
+// Application details
+const (
+ appCopyright = "Apache-2.0"
+ defaultLogLevel = "error"
+)
+
+// Log Global variable that hold logger
+var Log = logrus.New()
+
+// HTTPCli Global variable that hold HTTP Client
+var HTTPCli *common.HTTPClient
+
+// IOsk Global variable that hold SocketIo client
+var IOsk *socketio_client.Client
+
+// exitError exists this program with the specified error
+func exitError(code int, f string, a ...interface{}) {
+ err := fmt.Sprintf(f, a...)
+ fmt.Fprintf(os.Stderr, err+"\n")
+ os.Exit(code)
+}
+
+// main
+func main() {
+
+ // Allow to set app name from cli (useful for debugging)
+ if AppName == "" {
+ AppName = os.Getenv("XDS_APPNAME")
+ }
+ if AppName == "" {
+ panic("Invalid setup, AppName not define !")
+ }
+ if AppNativeName == "" {
+ AppNativeName = AppName[4:]
+ }
+ appUsage := fmt.Sprintf("command line tool for X(cross) Development System.")
+ appDescription := fmt.Sprintf("%s utility for X(cross) Development System\n", AppName)
+ /* SEB UPDATE DOC
+ appDescription += `
+ xds-cli configuration is driven either by environment variables or by command line
+ options or using a config file knowning that the following priority order is used:
+ 1. use option value (for example use project ID set by --id option),
+ 2. else use variable 'XDS_xxx' (for example 'XDS_PROJECT_ID' variable) when a
+ config file is specified with '--config|-c' option,
+ 3. else use 'XDS_xxx' (for example 'XDS_PROJECT_ID') environment variable.
+ `
+ */
+ // Create a new App instance
+ app := cli.NewApp()
+ app.Name = AppName
+ app.Usage = appUsage
+ app.Version = AppVersion + " (" + AppSubVersion + ")"
+ app.Authors = appAuthors
+ app.Copyright = appCopyright
+ app.Metadata = make(map[string]interface{})
+ app.Metadata["version"] = AppVersion
+ app.Metadata["git-tag"] = AppSubVersion
+ app.Metadata["logger"] = Log
+
+ // Create env vars help
+ dynDesc := "\nENVIRONMENT VARIABLES:"
+ for _, f := range app.Flags {
+ var env, usage string
+ switch f.(type) {
+ case cli.StringFlag:
+ fs := f.(cli.StringFlag)
+ env = fs.EnvVar
+ usage = fs.Usage
+ case cli.BoolFlag:
+ fb := f.(cli.BoolFlag)
+ env = fb.EnvVar
+ usage = fb.Usage
+ default:
+ exitError(1, "Un-implemented option type")
+ }
+ if env != "" {
+ dynDesc += fmt.Sprintf("\n %s \t\t %s", env, usage)
+ }
+ }
+ app.Description = appDescription + dynDesc
+
+ // Declare global flags
+ app.Flags = []cli.Flag{
+ cli.StringFlag{
+ Name: "config, c",
+ EnvVar: "XDS_CONFIG",
+ Usage: "env config file to source on startup",
+ },
+ cli.StringFlag{
+ Name: "log, l",
+ EnvVar: "XDS_LOGLEVEL",
+ Usage: "logging level (supported levels: panic, fatal, error, warn, info, debug)",
+ Value: defaultLogLevel,
+ },
+ cli.StringFlag{
+ Name: "url",
+ EnvVar: "XDS_SERVER_URL",
+ Value: "localhost:8000",
+ Usage: "remote XDS server url",
+ },
+ cli.BoolFlag{
+ Name: "timestamp, ts",
+ EnvVar: "XDS_TIMESTAMP",
+ Usage: "prefix output with timestamp",
+ },
+ }
+
+ // Declare commands
+ app.Commands = []cli.Command{}
+
+ initCmdProjects(&app.Commands)
+ initCmdSdks(&app.Commands)
+ initCmdExec(&app.Commands)
+ initCmdMisc(&app.Commands)
+
+ sort.Sort(cli.FlagsByName(app.Flags))
+ sort.Sort(cli.CommandsByName(app.Commands))
+
+ app.Before = func(ctx *cli.Context) error {
+ var err error
+ loglevel := ctx.String("log")
+ // Set logger level and formatter
+ if Log.Level, err = logrus.ParseLevel(loglevel); err != nil {
+ msg := fmt.Sprintf("Invalid log level : \"%v\"\n", loglevel)
+ return cli.NewExitError(msg, 1)
+ }
+ Log.Formatter = &logrus.TextFormatter{}
+
+ Log.Infof("%s version: %s", AppName, app.Version)
+ // SEB Add again Log.Debugf("Environment: %v", os.Environ())
+
+ if err = XdsConnInit(ctx); err != nil {
+ // Directly call HandleExitCoder to avoid to print help (ShowAppHelp)
+ // Note that this function wil never return and program will exit
+ cli.HandleExitCoder(err)
+ }
+
+ return nil
+ }
+
+ // Close HTTP client and WS connection on exit
+ defer func() {
+ XdsConnClose()
+ }()
+
+ app.Run(os.Args)
+}
+
+// XdsConnInit Initialized HTTP and WebSocket connection to XDS agent
+func XdsConnInit(ctx *cli.Context) error {
+ var err error
+
+ // Define HTTP and WS url
+ baseURL := ctx.String("url")
+ if !strings.HasPrefix(ctx.String("url"), "http://") {
+ baseURL = "http://" + ctx.String("url")
+ }
+
+ // Create HTTP client
+ Log.Debugln("Connect HTTP client on ", baseURL)
+ conf := common.HTTPClientConfig{
+ URLPrefix: "/api/v1",
+ HeaderClientKeyName: "Xds-Agent-Sid",
+ CsrfDisable: true,
+ LogOut: Log.Out,
+ LogPrefix: "XDSAGENT: ",
+ LogLevel: common.HTTPLogLevelWarning,
+ }
+
+ HTTPCli, err = common.HTTPNewClient(baseURL, conf)
+ if err != nil {
+ errmsg := err.Error()
+ if m, err := regexp.MatchString("Get http.?://", errmsg); m && err == nil {
+ i := strings.LastIndex(errmsg, ":")
+ errmsg = "Cannot connection to " + baseURL + errmsg[i:]
+ }
+ return cli.NewExitError(errmsg, 1)
+ }
+ HTTPCli.SetLogLevel(ctx.String("loglevel"))
+
+ // Create io Websocket client
+ Log.Debugln("Connecting IO.socket client on ", baseURL)
+
+ opts := &socketio_client.Options{
+ Transport: "websocket",
+ Header: make(map[string][]string),
+ }
+ opts.Header["XDS-AGENT-SID"] = []string{HTTPCli.GetClientID()}
+
+ IOsk, err = socketio_client.NewClient(baseURL, opts)
+ if err != nil {
+ return cli.NewExitError("IO.socket connection error: "+err.Error(), 1)
+ }
+
+ IOsk.On("error", func(err error) {
+ fmt.Println("ERROR Websocket: ", err.Error())
+ })
+
+ ctx.App.Metadata["httpCli"] = HTTPCli
+ ctx.App.Metadata["ioskCli"] = IOsk
+
+ return nil
+}
+
+// XdsConnClose Terminate connection to XDS agent
+func XdsConnClose() {
+ Log.Debugf("Closing HTTP client session...")
+ /* TODO
+ if httpCli, ok := app.Metadata["httpCli"]; ok {
+ c := httpCli.(*common.HTTPClient)
+ }
+ */
+
+ Log.Debugf("Closing WebSocket connection...")
+ /*
+ if ioskCli, ok := app.Metadata["ioskCli"]; ok {
+ c := ioskCli.(*socketio_client.Client)
+ }
+ */
+}
+
+// NewTableWriter Create a writer that inserts padding around tab-delimited
+func NewTableWriter() *tabwriter.Writer {
+ writer := new(tabwriter.Writer)
+ writer.Init(os.Stdout, 0, 8, 0, '\t', 0)
+ return writer
+}