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) == 5 {
		host, owner, repo = parseRepo(args[0])
	} else {
		host = "unknown"
		owner = "unknown"
		repo = "unknown"
		branch = "unknown"
		sha = "unknown"
	}

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