From 1dec169812c7b180d65b65445f1e913fc513fd5b Mon Sep 17 00:00:00 2001
From: Brad Rydzewski <brydzews@cisco.com>
Date: Thu, 14 Aug 2014 12:36:04 -0700
Subject: [PATCH] fixed restart client

---
 client/commits.go |  2 +-
 cmd/restart.go    | 17 ++++++++++-------
 2 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/client/commits.go b/client/commits.go
index 5928987c1..347f72a28 100644
--- a/client/commits.go
+++ b/client/commits.go
@@ -31,7 +31,7 @@ func (s *CommitService) GetOutput(host, owner, name, branch, sha string) (io.Rea
 
 // POST /v1/repos/{host}/{owner}/{name}/branches/{branch}/commits/{commit}?action=rebuild
 func (s *CommitService) Rebuild(host, owner, name, branch, sha string) error {
-	var path = fmt.Sprintf("/v1/repos/%s/%s/%s/branches/%s/commits/%s", host, owner, name, branch, sha)
+	var path = fmt.Sprintf("/v1/repos/%s/%s/%s/branches/%s/commits/%s?action=rebuild", host, owner, name, branch, sha)
 	return s.run("POST", path, nil, nil)
 }
 
diff --git a/cmd/restart.go b/cmd/restart.go
index e8657ccad..727af2545 100644
--- a/cmd/restart.go
+++ b/cmd/restart.go
@@ -22,14 +22,17 @@ func restartCommandFunc(c *cli.Context, client *client.Client) error {
 	var host, owner, repo, branch, sha string
 	var args = c.Args()
 
-	if len(args) == 5 {
+	if len(args) != 0 {
 		host, owner, repo = parseRepo(args[0])
-	} else {
-		host = "unknown"
-		owner = "unknown"
-		repo = "unknown"
-		branch = "unknown"
-		sha = "unknown"
+	}
+
+	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)