rise.go

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/codegangsta/cli"
	"github.com/franela/goreq"

	"github.com/nitrous-io/rise-cli-go/cli/collab"
	"github.com/nitrous-io/rise-cli-go/cli/common"
	"github.com/nitrous-io/rise-cli-go/cli/configuration"
	"github.com/nitrous-io/rise-cli-go/cli/deploy"
	"github.com/nitrous-io/rise-cli-go/cli/domains"
	"github.com/nitrous-io/rise-cli-go/cli/env"
	"github.com/nitrous-io/rise-cli-go/cli/initcmd"
	"github.com/nitrous-io/rise-cli-go/cli/login"
	"github.com/nitrous-io/rise-cli-go/cli/logout"
	"github.com/nitrous-io/rise-cli-go/cli/password"
	"github.com/nitrous-io/rise-cli-go/cli/projects"
	"github.com/nitrous-io/rise-cli-go/cli/protect"
	"github.com/nitrous-io/rise-cli-go/cli/repo"
	"github.com/nitrous-io/rise-cli-go/cli/rollback"
	"github.com/nitrous-io/rise-cli-go/cli/signup"
	"github.com/nitrous-io/rise-cli-go/cli/ssl"
	"github.com/nitrous-io/rise-cli-go/cli/versions"
	"github.com/nitrous-io/rise-cli-go/config"
	"github.com/nitrous-io/rise-cli-go/pkg/readline"
	"github.com/nitrous-io/rise-cli-go/tr"
	"github.com/nitrous-io/rise-cli-go/tui"

	log "github.com/Sirupsen/logrus"
)

func init() {
	cli.AppHelpTemplate = `NAME:
   {{.Name}} - {{.Usage}}

USAGE:
   {{if .UsageText}}{{.UsageText}}{{else}}{{.HelpName}} {{if .Flags}}[global options]{{end}}{{if .Commands}} command [command options]{{end}} {{if .ArgsUsage}}{{.ArgsUsage}}{{else}}[arguments...]{{end}}{{end}}
   {{if .Version}}
VERSION:
   {{.Version}}
   {{end}}{{if len .Authors}}
AUTHOR(S):
   {{range .Authors}}{{ . }}{{end}}
   {{end}}{{if .Commands}}
COMMANDS:
   {{range .Commands}}{{if .Usage}}{{join .Names ", "}}{{ "\t" }}{{.Usage}}
   {{end}}{{end}}{{end}}{{if .Flags}}
GLOBAL OPTIONS:
   {{range .Flags}}{{.}}
   {{end}}{{end}}{{if .Copyright }}
COPYRIGHT:
   {{.Copyright}}
   {{end}}
`
}

func main() {
	log.SetFormatter(&tui.Formatter{})
	log.SetOutput(tui.Out)
	log.SetLevel(log.InfoLevel)
	readline.Output = tui.Out

	common.CheckForUpdates()

	// Set Goreq's connect timeout to 10s globally (its default is 1s which
	// can be too short).
	goreq.SetConnectTimeout(10 * time.Second)

	app := cli.NewApp()
	app.Version = config.Version
	app.Usage = tr.T("rise_cli_desc")

	app.Commands = []cli.Command{
		{
			Name:   "signup",
			Usage:  tr.T("signup_desc"),
			Action: signup.Signup,
		},
		{
			Name:   "login",
			Usage:  tr.T("login_desc"),
			Action: login.Login,
		},
		{
			Name:   "logout",
			Usage:  tr.T("logout_desc"),
			Action: logout.Logout,
		},
		{
			Name: "password",
			Subcommands: []cli.Command{
				{
					Name:   "change",
					Usage:  tr.T("password_change_desc"),
					Action: password.Change,
				},
				{
					Name:   "reset",
					Usage:  tr.T("password_reset_desc"),
					Action: password.Reset,
					Flags: []cli.Flag{
						cli.BoolFlag{
							Name:  "continue, c",
							Usage: tr.T("password_reset_continue"),
						},
					},
				},
			},
		},
		{
			Name:   "password.change",
			Usage:  tr.T("password_change_desc"),
			Action: password.Change,
		},
		{
			Name:   "password.reset",
			Usage:  tr.T("password_reset_desc"),
			Action: password.Reset,
			Flags: []cli.Flag{
				cli.BoolFlag{
					Name:  "continue, c",
					Usage: tr.T("password_reset_continue"),
				},
			},
		},
		{
			Name:   "init",
			Usage:  tr.T("init_desc"),
			Action: initcmd.Init,
		},
		{
			Name:   "config",
			Usage:  tr.T("config_desc"),
			Action: configuration.Update,
		},
		{
			Name:    "publish",
			Aliases: []string{"deploy"},
			Usage:   tr.T("publish_desc"),
			Action:  deploy.Deploy,
			Flags: []cli.Flag{
				cli.BoolFlag{
					Name:  "verbose, v",
					Usage: tr.T("publish_verbose"),
				},
			},
		},
		{
			Name:   "domains",
			Usage:  tr.T("domains_desc"),
			Action: domains.List,
			Subcommands: []cli.Command{
				{
					Name:      "add",
					Usage:     tr.T("domains_add_desc"),
					Action:    domains.Add,
					ArgsUsage: fmt.Sprintf(tr.T("domains_add_args"), config.DefaultDomain),
				},
				{
					Name:      "rm",
					Usage:     tr.T("domains_rm_desc"),
					Action:    domains.Remove,
					ArgsUsage: fmt.Sprintf(tr.T("domains_rm_args"), config.DefaultDomain),
				},
			},
		},
		{
			Name:      "domains.add",
			Usage:     tr.T("domains_add_desc"),
			Action:    domains.Add,
			ArgsUsage: fmt.Sprintf(tr.T("domains_add_args"), config.DefaultDomain),
		},
		{
			Name:      "domains.rm",
			Usage:     tr.T("domains_rm_desc"),
			Action:    domains.Remove,
			ArgsUsage: fmt.Sprintf(tr.T("domains_rm_args"), config.DefaultDomain),
		},
		{
			Name:   "projects",
			Usage:  tr.T("projects_desc"),
			Action: projects.List,
			Subcommands: []cli.Command{
				{
					Name:   "rm",
					Usage:  tr.T("projects_add_desc"),
					Action: projects.Remove,
					Flags: []cli.Flag{
						cli.BoolFlag{
							Name:  "force, f",
							Usage: tr.T("projects_rm_force"),
						},
					},
				},
			},
		},
		{
			Name:   "projects.rm",
			Usage:  tr.T("projects_rm_desc"),
			Action: projects.Remove,
			Flags: []cli.Flag{
				cli.BoolFlag{
					Name:  "force, f",
					Usage: tr.T("projects_rm_force"),
				},
			},
		},
		{
			Name:   "collab",
			Usage:  tr.T("collab_desc"),
			Action: collab.List,
			Subcommands: []cli.Command{
				{
					Name:   "add",
					Usage:  tr.T("collab_add_desc"),
					Action: collab.Add,
				},
				{
					Name:   "rm",
					Usage:  tr.T("collab_rm_desc"),
					Action: collab.Remove,
				},
			},
		},
		{
			Name:   "collab.add",
			Usage:  tr.T("collab_add_desc"),
			Action: collab.Add,
		},
		{
			Name:   "collab.rm",
			Usage:  tr.T("collab_rm_desc"),
			Action: collab.Remove,
		},
		{
			Name:      "rollback",
			Usage:     tr.T("rollback_desc"),
			ArgsUsage: tr.T("rollback_args"),
			Action:    rollback.Rollback,
		},
		{
			Name:   "versions",
			Usage:  tr.T("versions_desc"),
			Action: versions.Versions,
		},
		{
			Name:      "ssl.info",
			Usage:     tr.T("ssl_info_desc"),
			Action:    ssl.Info,
			ArgsUsage: tr.T("ssl_info_args"),
		},
		{
			Name:      "ssl.set",
			Usage:     tr.T("ssl_set_desc"),
			Action:    ssl.Set,
			ArgsUsage: tr.T("ssl_set_args"),
		},
		{
			Name:      "ssl.rm",
			Usage:     tr.T("ssl_rm_desc"),
			Action:    ssl.Delete,
			ArgsUsage: tr.T("ssl_rm_args"),
		},
		{
			Name:      "ssl.force",
			Usage:     tr.T("ssl_force_desc"),
			Action:    ssl.Force,
			ArgsUsage: tr.T("ssl_force_args"),
		},
		{
			Name:      "ssl.letsencrypt",
			Usage:     tr.T("ssl_letsencrypt_desc"),
			Action:    ssl.Letsencrypt,
			ArgsUsage: tr.T("ssl_letsencrypt_args"),
		},
		{
			Name: "ssl",
			Subcommands: []cli.Command{
				{
					Name:      "info",
					Usage:     tr.T("ssl_info_desc"),
					Action:    ssl.Info,
					ArgsUsage: tr.T("ssl_info_args"),
				},
				{
					Name:      "set",
					Usage:     tr.T("ssl_set_desc"),
					Action:    ssl.Set,
					ArgsUsage: tr.T("ssl_set_args"),
				},
				{
					Name:      "rm",
					Usage:     tr.T("ssl_rm_desc"),
					Action:    ssl.Delete,
					ArgsUsage: tr.T("ssl_rm_args"),
				},
				{
					Name:      "force",
					Usage:     tr.T("ssl_force_desc"),
					Action:    ssl.Force,
					ArgsUsage: tr.T("ssl_force_args"),
				},
				{
					Name:      "letsencrypt",
					Usage:     tr.T("ssl_letsencrypt_desc"),
					Action:    ssl.Letsencrypt,
					ArgsUsage: tr.T("ssl_letsencrypt_args"),
				},
			},
		},
		{
			Name:      "reinit",
			Usage:     tr.T("reinit_desc"),
			Action:    initcmd.ReInit,
			ArgsUsage: tr.T("reinit_args"),
		},
		{
			Name:      "protect",
			Usage:     tr.T("protect_desc"),
			Action:    protect.Protect,
			ArgsUsage: tr.T("protect_args"),
		},
		{
			Name:   "unprotect",
			Usage:  tr.T("unprotect_desc"),
			Action: protect.Unprotect,
		},
		{
			Name:   "env",
			Usage:  tr.T("env_list_desc"),
			Action: env.List,
			Subcommands: []cli.Command{
				{
					Name:      "set",
					Usage:     tr.T("env_set_desc"),
					Action:    env.Add,
					ArgsUsage: tr.T("env_set_args"),
				},
				{
					Name:      "rm",
					Usage:     tr.T("env_rm_desc"),
					Action:    env.Delete,
					ArgsUsage: tr.T("env_rm_args"),
				},
			},
		},
		{
			Name:      "env.set",
			Usage:     tr.T("env_set_desc"),
			Action:    env.Add,
			ArgsUsage: tr.T("env_set_args"),
		},
		{
			Name:      "env.rm",
			Usage:     tr.T("env_rm_desc"),
			Action:    env.Delete,
			ArgsUsage: tr.T("env_rm_args"),
		},
		{
			Name:   "repo.link",
			Usage:  tr.T("repo_link_desc"),
			Action: repo.Link,
		},
		{
			Name:   "repo.unlink",
			Usage:  tr.T("repo_unlink_desc"),
			Action: repo.Unlink,
		},
		{
			Name:   "repo.info",
			Usage:  tr.T("repo_info_desc"),
			Action: repo.Info,
		},
		{
			Name: "repo",
			Subcommands: []cli.Command{
				{
					Name:   "link",
					Usage:  tr.T("repo_link_desc"),
					Action: repo.Link,
				},
				{
					Name:   "unlink",
					Usage:  tr.T("repo_unlink_desc"),
					Action: repo.Unlink,
				},
				{
					Name:   "info",
					Usage:  tr.T("repo_info_desc"),
					Action: repo.Info,
				},
			},
		},
	}

	app.Run(os.Args)
}