init

2020-08-19 16:04:21 -07:00
commit 8482272d45
336 changed files with 197309 additions and 0 deletions
@@ -0,0 +1,121 @@
+package main
+
+import (
+	"fmt"
+	"github.com/analogj/scrutiny/collector/pkg/collector"
+	"github.com/analogj/scrutiny/collector/pkg/version"
+	"github.com/sirupsen/logrus"
+	"log"
+	"os"
+	"time"
+
+	utils "github.com/analogj/go-util/utils"
+	"github.com/fatih/color"
+	"github.com/urfave/cli/v2"
+)
+
+var goos string
+var goarch string
+
+func main() {
+
+	cli.CommandHelpTemplate = `NAME:
+   {{.HelpName}} - {{.Usage}}
+USAGE:
+   {{if .UsageText}}{{.UsageText}}{{else}}{{.HelpName}} {{if .ArgsUsage}}{{.ArgsUsage}}{{else}}[arguments...]{{end}}{{end}}{{if .Category}}
+CATEGORY:
+   {{.Category}}{{end}}{{if .Description}}
+DESCRIPTION:
+   {{.Description}}{{end}}{{if .VisibleFlags}}
+OPTIONS:
+   {{range .VisibleFlags}}{{.}}
+   {{end}}{{end}}
+`
+
+	app := &cli.App{
+		Name:     "scrutiny-collector-selftest",
+		Usage:    "smartctl self-test data collector for scrutiny",
+		Version:  version.VERSION,
+		Compiled: time.Now(),
+		Authors: []*cli.Author{
+			{
+				Name:  "Jason Kulatunga",
+				Email: "jason@thesparktree.com",
+			},
+		},
+		Before: func(c *cli.Context) error {
+
+			collectorSelfTest := "AnalogJ/scrutiny/selftest"
+
+			var versionInfo string
+			if len(goos) > 0 && len(goarch) > 0 {
+				versionInfo = fmt.Sprintf("%s.%s-%s", goos, goarch, version.VERSION)
+			} else {
+				versionInfo = fmt.Sprintf("dev-%s", version.VERSION)
+			}
+
+			subtitle := collectorSelfTest + utils.LeftPad2Len(versionInfo, " ", 65-len(collectorSelfTest))
+
+			color.New(color.FgGreen).Fprintf(c.App.Writer, fmt.Sprintf(utils.StripIndent(
+				`
+			 ___   ___  ____  __  __  ____  ____  _  _  _  _
+			/ __) / __)(  _ \(  )(  )(_  _)(_  _)( \( )( \/ )
+			\__ \( (__  )   / )(__)(   )(   _)(_  )  (  \  /
+			(___/ \___)(_)\_)(______) (__) (____)(_)\_) (__)
+			%s
+
+			`), subtitle))
+
+			return nil
+		},
+
+		Commands: []*cli.Command{
+			{
+				Name:  "run",
+				Usage: "Run the scrutiny self-test data collector",
+				Action: func(c *cli.Context) error {
+
+					collectorLogger := logrus.WithFields(logrus.Fields{
+						"type": "selftest",
+					})
+
+					if c.Bool("debug") {
+						logrus.SetLevel(logrus.DebugLevel)
+					} else {
+						logrus.SetLevel(logrus.InfoLevel)
+					}
+
+					stCollector, err := collector.CreateSelfTestCollector(
+						collectorLogger,
+						c.String("api-endpoint"),
+					)
+
+					if err != nil {
+						return err
+					}
+
+					return stCollector.Run()
+				},
+
+				Flags: []cli.Flag{
+					&cli.StringFlag{
+						Name:  "api-endpoint",
+						Usage: "The api server endpoint",
+						Value: "http://localhost:8080",
+					},
+
+					&cli.BoolFlag{
+						Name:  "debug",
+						Usage: "Enable debug logging",
+					},
+				},
+			},
+		},
+	}
+
+	err := app.Run(os.Args)
+	if err != nil {
+		log.Fatal(color.HiRedString("ERROR: %v", err))
+	}
+
+}