mirror of
https://github.com/strongdm/comply
synced 2024-11-16 21:04:54 +00:00
119 lines
2.8 KiB
Go
119 lines
2.8 KiB
Go
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.
|
|
var Version = ""
|
|
|
|
// 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"
|
|
if Version == "" {
|
|
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
|
|
}
|