package cli import ( "context" "errors" "fmt" "io" "io/ioutil" "log" "os" "path/filepath" "time" "github.com/docker/docker/api/types" "github.com/docker/docker/client" "github.com/strongdm/comply/internal/config" "github.com/strongdm/comply/internal/plugin/github" "github.com/urfave/cli" ) // Version is set by the build system. const Version = "0.0.0-development" // Main should be invoked by the main function in the main package. func Main() { err := newApp().Run(os.Args) if err != nil { log.Fatal(err) } } func newApp() *cli.App { app := cli.NewApp() app.Name = "comply" app.HideVersion = true app.Version = Version app.Usage = "policy compliance toolkit" app.Commands = []cli.Command{ initCommand, } app.Commands = append(app.Commands, beforeCommand(buildCommand, projectMustExist)) app.Commands = append(app.Commands, beforeCommand(schedulerCommand, projectMustExist)) app.Commands = append(app.Commands, beforeCommand(serveCommand, projectMustExist)) app.Commands = append(app.Commands, beforeCommand(syncCommand, projectMustExist)) app.Commands = append(app.Commands, beforeCommand(todoCommand, projectMustExist)) // Plugins github.Register() return app } func beforeCommand(c cli.Command, bf ...cli.BeforeFunc) cli.Command { c.Before = beforeAll(bf...) return c } func beforeAll(bf ...cli.BeforeFunc) cli.BeforeFunc { return func(c *cli.Context) error { for _, f := range bf { if err := f(c); err != nil { return err } } return nil } } func feedbackError(message string) error { return errors.New(fmt.Sprintf("\n\nERROR\n=====\n%s\n", message)) } func projectMustExist(c *cli.Context) error { _, err := ioutil.ReadFile(filepath.Join(config.ProjectRoot(), "comply.yml")) if err != nil { return feedbackError("command must be run from the root of a valid comply project (comply.yml must exist; have you run `comply init`?)") } return nil } func dockerMustExist(c *cli.Context) error { dockerErr := fmt.Errorf("Docker must be available in order to run `%s`", c.Command.Name) ctx := context.Background() cli, err := client.NewEnvClient() if err != nil { return dockerErr } r, err := cli.ImagePull(ctx, "strongdm/pandoc:latest", types.ImagePullOptions{}) if err != nil { return dockerErr } defer r.Close() done := make(chan struct{}) defer close(done) go func() { // if docker IO takes more than N seconds, notify user we're (likely) downloading the pandoc image longishPull := time.After(time.Second * 6) select { case <-longishPull: fmt.Println("Downloading strongdm/pandoc image (this may take sometime) ...") case <-done: // in this case, the docker pull was quick -- suggesting we already have the container } }() // hold function open until all docker IO is complete io.Copy(ioutil.Discard, r) return nil }