Handle private registries w/ port

Fixes #2

Author: jwilder

docker_client.go

@@ -29,11 +29,10 @@ func splitDockerImage(img string) (string, string, string) {
 		repository = img[index:]
 	}
 
-	if strings.Contains(img, ":") {
-		separator := strings.Index(img, ":")
-		repository = img[index:separator]
-		index = separator + 1
-		tag = img[index:]
+	if strings.Contains(repository, ":") {
+		separator := strings.Index(repository, ":")
+		tag = repository[separator+1:]
+		repository = repository[0:separator]
 	}
 
 	return registry, repository, tag

docker-gen_test.go docker_client_test.go

@@ -97,3 +97,28 @@ func TestSplitDockerImageWithRepositoryAndTag(t *testing.T) {
 	}
 
 }
+
+func TestSplitDockerImageWithLocalRepositoryAndTag(t *testing.T) {
+	registry, repository, tag := splitDockerImage("localhost:8888/ubuntu:12.04")
+
+	if registry != "localhost:8888" {
+		t.Fatalf("registry does not match: expected %s got %s", "localhost:8888", registry)
+	}
+
+	if repository != "ubuntu" {
+		t.Fatalf("repository does not match: expected %s got %s", "ubuntu", repository)
+	}
+
+	if tag != "12.04" {
+		t.Fatalf("tag does not match: expected %s got %s", "12.04", tag)
+	}
+	dockerImage := DockerImage{
+		Registry:   registry,
+		Repository: repository,
+		Tag:        tag,
+	}
+	if "localhost:8888/ubuntu:12.04" != dockerImage.String() {
+		t.Fail()
+	}
+
+}