init.go

package cmd

import (
	"fmt"
	"github.com/fatih/color"
	"strings"
	"time"

	"github.com/briandowns/spinner"
	"github.com/mitchellh/cli"
	"github.com/roots/trellis-cli/trellis"
)

type InitCommand struct {
	UI      cli.Ui
	Trellis *trellis.Trellis
}

func (c *InitCommand) Run(args []string) int {
	if err := c.Trellis.LoadProject(); err != nil {
		c.UI.Error(err.Error())
		return 1
	}

	commandArgumentValidator := &CommandArgumentValidator{required: 0, optional: 0}
	commandArgumentErr := commandArgumentValidator.validate(args)
	if commandArgumentErr != nil {
		c.UI.Error(commandArgumentErr.Error())
		c.UI.Output(c.Help())
		return 1
	}

	if err := c.Trellis.CreateConfigDir(); err != nil {
		c.UI.Error(err.Error())
		return 1
	}

	if ok, _ := c.Trellis.Virtualenv.Installed(); !ok {
		c.UI.Info("virtualenv not found")
		s := spinner.New(spinner.CharSets[14], 100*time.Millisecond)
		s.Suffix = " Installing virtualenv..."
		s.Start()
		c.Trellis.Virtualenv.Install()
		s.Stop()
		c.UI.Info(color.GreenString("✓ virtualenv installed"))
	}

	c.UI.Info(fmt.Sprintf("Creating virtualenv in %s", c.Trellis.Virtualenv.Path))

	err := c.Trellis.Virtualenv.Create()
	if err != nil {
		c.UI.Error(fmt.Sprintf("Error creating virtualenv: %s", err))
		return 1
	}

	c.UI.Info(color.GreenString("✓ Virtualenv created"))

	pip := CommandExecWithStderrOnly("pip", []string{"install", "-r", "requirements.txt"}, c.UI)

	s := spinner.New(spinner.CharSets[14], 100*time.Millisecond)
	s.Suffix = " Installing pip dependencies (pip install -r requirements.txt) ..."
	s.Start()

	err = pip.Run()
	s.Stop()

	if err != nil {
		return 1
	}

	c.UI.Info(color.GreenString("✓ Dependencies installed"))

	return 0
}

func (c *InitCommand) Synopsis() string {
	return "Initializes an existing Trellis project"
}

func (c *InitCommand) Help() string {
	helpText := `
Usage: trellis init [options]

Initializes an existing Trellis project to be managed by trellis-cli.
The initialization process does two things:

1. installs virtualenv if necessary (see below for details)
2. creates a virtual environment specific to the project to manage dependencies
3. installs dependencies via pip (specified by requirements.txt in your Trellis project)

trellis-cli will attempt to use an already installed method to manage virtualenvs
and only fallback to installing virtualenv if necessary:

1. if python3 is installed, use built-in virtualenv feature
2. use virtualenv command if available
3. finally install virtualenv at $HOME/.trellis/virtualenv

To learn more about virtual environments, see https://docs.python.org/3/tutorial/venv.html

This command is idempotent meaning it can be run multiple times without side-effects.

  $ trellis init

Options:
  -h, --help  show this help
`

	return strings.TrimSpace(helpText)
}