package main

import (
	"github.com/codegangsta/cli"
	"github.com/drone/drone/client"
)

// NewRestartCommand returns the CLI command for "restart".
func NewRestartCommand() cli.Command {
	return cli.Command{
		Name:  "restart",
		Usage: "restarts a build on the server",
		Flags: []cli.Flag{},
		Action: func(c *cli.Context) {
			handle(c, restartCommandFunc)
		},
	}
}

// restartCommandFunc executes the "restart" command.
func restartCommandFunc(c *cli.Context, client *client.Client) error {
	var host, owner, repo, branch, sha string
	var args = c.Args()

	if len(args) != 0 {
		host, owner, repo = parseRepo(args[0])
	}

	switch len(args) {
	case 2:
		branch = "master"
		sha = args[1]
	case 3, 4, 5:
		branch = args[1]
		sha = args[2]
	}

	return client.Commits.Rebuild(host, owner, repo, branch, sha)
}