Merge pull request #192 from secrethub/feature/demo-init

Add demo init command

Author: SimonBarendse

internals/demo/command.go

@@ -7,19 +7,23 @@ import (
 
 // Command is a command to run the secrethub example app.
 type Command struct {
-	io ui.IO
+	io        ui.IO
+	newClient newClientFunc
 }
 
 // NewCommand creates a new example app command.
-func NewCommand(io ui.IO) *Command {
+func NewCommand(io ui.IO, newClient newClientFunc) *Command {
 	return &Command{
-		io: io,
+		io:        io,
+		newClient: newClient,
 	}
 }
 
 // Register registers the command, arguments and flags on the provided Registerer.
 func (cmd *Command) Register(r command.Registerer) {
-	clause := r.Command("demo", "Runs the secrethub demo app as used in different guides.")
+	clause := r.Command("demo", "Manage the demo application.")
+	clause.Hidden()
 
+	NewInitCommand(cmd.io, cmd.newClient).Register(clause)
 	NewServeCommand(cmd.io).Register(clause)
 }

internals/demo/init.go

@@ -0,0 +1,91 @@
+package demo
+
+import (
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"fmt"
+
+	"github.com/secrethub/secrethub-cli/internals/cli/ui"
+	"github.com/secrethub/secrethub-cli/internals/secrethub/command"
+
+	"github.com/secrethub/secrethub-go/internals/api"
+	"github.com/secrethub/secrethub-go/pkg/secrethub"
+	"github.com/secrethub/secrethub-go/pkg/secretpath"
+)
+
+type newClientFunc func() (secrethub.ClientInterface, error)
+
+const defaultDemoRepo = "demo"
+
+type InitCommand struct {
+	repo api.RepoPath
+
+	io        ui.IO
+	newClient newClientFunc
+}
+
+func NewInitCommand(io ui.IO, newClient newClientFunc) *InitCommand {
+	return &InitCommand{
+		io:        io,
+		newClient: newClient,
+	}
+}
+
+// Register registers the command, arguments and flags on the provided Registerer.
+func (cmd *InitCommand) Register(r command.Registerer) {
+	clause := r.Command("init", "Create the secrets necessary to connect with the demo application.")
+	clause.HelpLong("demo init creates a repository with the username and password needed to connect to the demo API.")
+
+	clause.Flag("repo", "The path of the repository to create. Defaults to a "+defaultDemoRepo+" repo in your personal namespace.").SetValue(&cmd.repo)
+
+	command.BindAction(clause, cmd.Run)
+}
+
+// Run handles the command with the options as specified in the command.
+func (cmd *InitCommand) Run() error {
+	client, err := cmd.newClient()
+	if err != nil {
+		return err
+	}
+
+	var repoPath string
+	var username string
+	if cmd.repo == "" {
+		me, err := client.Me().GetUser()
+		if err != nil {
+			return err
+		}
+		username = me.Username
+		repoPath = secretpath.Join(me.Username, defaultDemoRepo)
+	} else {
+		username = secretpath.Namespace(cmd.repo.Value())
+		repoPath = cmd.repo.Value()
+	}
+
+	_, err = client.Repos().Create(repoPath)
+	if err == api.ErrRepoAlreadyExists && cmd.repo == "" {
+		return fmt.Errorf("demo repo %s already exists, use --repo to specify another repo to use", repoPath)
+	} else if err != nil {
+		return err
+	}
+
+	usernamePath := secretpath.Join(repoPath, "username")
+	_, err = client.Secrets().Write(usernamePath, []byte(username))
+	if err != nil {
+		return err
+	}
+
+	h := hmac.New(sha256.New, []byte("this-is-no-good-way-to-generate-a-password-that-is-why-we-only-use-it-for-demo-purposes"))
+	password := base64.RawStdEncoding.EncodeToString(h.Sum([]byte(username)))[:20]
+
+	passwordPath := secretpath.Join(repoPath, "password")
+	_, err = client.Secrets().Write(passwordPath, []byte(password))
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("Created the following secrets:\n%s\n%s\n", usernamePath, passwordPath)
+
+	return nil
+}

internals/secrethub/app.go

@@ -184,5 +184,5 @@ func (app *App) registerCommands() {
 	NewClearClipboardCommand().Register(app.cli)
 	NewKeyringClearCommand().Register(app.cli)
 
-	demo.NewCommand(app.io).Register(app.cli)
+	demo.NewCommand(app.io, app.clientFactory.NewClient).Register(app.cli)
 }