Merge pull request #8 from daviddyball/master

Private Repository Support

pkg/plugin/publish/docker.go

@@ -3,7 +3,6 @@ package publish
 import (
 	"fmt"
 	"strconv"
-	"strings"
 
 	"github.com/drone/drone/pkg/build/buildfile"
 	"github.com/drone/drone/pkg/build/repo"
@@ -15,19 +14,28 @@ type Docker struct {
 	Dockerfile string `yaml:"docker_file"`
 
 	// Connection information for the docker server that will build the image
-	Server string `yaml:"docker_server"`
+	DockerServer string `yaml:"docker_server"`
-	Port   int    `yaml:"docker_port"`
+	DockerServerPort   int	`yaml:"docker_port"`
 	// The Docker client version to download. This must match the docker version on the server
 	DockerVersion string `yaml:"docker_version"`
 
-	RepoBaseName string `yaml:"repo_base_name"`
+	// Optional Arguments to allow finer-grained control of registry
+	// endpoints
+	RegistryLoginUrl string `yaml:"registry_login_url"`
+	ImageName string `yaml:"image_name"`
+	RegistryLogin bool `yaml:"registry_login"`
 
 	// Authentication credentials for index.docker.io
 	Username string `yaml:"username"`
 	Password string `yaml:"password"`
 	Email	string `yaml:"email"`
 
-	Branch string `yaml:"branch,omitempty"`
+	// Keep the build on the Docker host after pushing?
+	KeepBuild bool `yaml:"keep_build"`
+	// Do we want to override "latest" automatically with this build?
+	PushLatest bool `yaml:"push_latest"`
+	CustomTag string `yaml:"custom_tag"`
+	Branch string `yaml:"branch"`
 }
 
 // Write adds commands to the buildfile to do the following:
@@ -36,12 +44,19 @@ type Docker struct {
 // 3. Push that docker image to index.docker.io.
 // 4. Delete the docker image on the server it was build on so we conserve disk space.
 func (d *Docker) Write(f *buildfile.Buildfile, r *repo.Repo) {
-	if len(d.Email) == 0 || len(d.Server) == 0 || d.Port == 0 || len(d.DockerVersion) == 0 ||
+	if len(d.DockerServer) == 0 || d.DockerServerPort == 0 || len(d.DockerVersion) == 0 ||
-		len(d.RepoBaseName) == 0 || len(d.Username) == 0 || len(d.Password) == 0 {
+		len(d.ImageName) == 0 {
-		f.WriteCmdSilent(`echo "Docker Plugin: Missing argument(s)"`)
+		f.WriteCmdSilent(`echo -e "Docker Plugin: Missing argument(s)"\n\n`)
+		if len(d.DockerServer) == 0  { f.WriteCmdSilent(`echo -e "\tdocker_server not defined in yaml`) }
+		if d.DockerServerPort == 0   { f.WriteCmdSilent(`echo -e "\tdocker_port not defined in yaml`) }
+		if len(d.DockerVersion) == 0 { f.WriteCmdSilent(`echo -e "\tdocker_version not defined in yaml`) }
+		if len(d.ImageName) == 0     { f.WriteCmdSilent(`echo -e "\timage_name not defined in yaml`) }
 		return
 	}
 
+	// Ensure correct apt-get has the https method-driver as per (http://askubuntu.com/questions/165676/)
+	f.WriteCmd("sudo apt-get install apt-transport-https")
+
 	// Install Docker on the container
 	f.WriteCmd("sudo sh -c \"echo deb https://get.docker.io/ubuntu docker main\\ > " +
 		"/etc/apt/sources.list.d/docker.list\"")
@@ -50,27 +65,52 @@ func (d *Docker) Write(f *buildfile.Buildfile, r *repo.Repo) {
 	f.WriteCmd("sudo apt-get update")
 	f.WriteCmd("sudo apt-get --yes install lxc-docker-" + d.DockerVersion)
 
-	dockerServerUrl := d.Server + ":" + strconv.Itoa(d.Port)
+	// Format our Build Server Endpoint
-	splitRepoName := strings.Split(r.Name, "/")
+	dockerServerUrl := d.DockerServer + ":" + strconv.Itoa(d.DockerServerPort)
-	dockerRepo := d.RepoBaseName + "/" + splitRepoName[len(splitRepoName)-1]
 
 	dockerPath := "."
 	if len(d.Dockerfile) != 0 {
 		dockerPath = fmt.Sprintf("- < %s", d.Dockerfile)
 	}
 
-	// Run the command commands to build and deploy the image. Note that we both create a new image
+	// Run the command commands to build and deploy the image.
-	// tagged with the git hash as well as update the "latest" image.
+	// Are we setting a custom tag, or do we use the git hash?
-	f.WriteCmd(fmt.Sprintf("docker -H %s build -t %s %s", dockerServerUrl, dockerRepo, dockerPath))
+	imageTag := ""
-	f.WriteCmd(fmt.Sprintf("docker -H %s build -t %s:$(git rev-parse --short HEAD) %s",
+	if len(d.CustomTag) > 0 {
-		dockerServerUrl, dockerRepo, dockerPath))
+		imageTag = d.CustomTag
+	} else {
+		imageTag = "$(git rev-parse --short HEAD)"
+	}
+	f.WriteCmd(fmt.Sprintf("docker -H %s build -t %s:%s %s", dockerServerUrl, d.ImageName, imageTag, dockerPath))
 
-	// Login and push to index.docker.io
+	// Login?
+	if d.RegistryLogin == true {
+		// Are we logging in to a custom Registry?
+		if len(d.RegistryLoginUrl) > 0 {
+			f.WriteCmdSilent(fmt.Sprintf("docker -H %s login -u %s -p %s -e %s %s",
+				dockerServerUrl, d.Username, d.Password, d.Email, d.RegistryLoginUrl))
+		} else {
+			// Assume index.docker.io
 			f.WriteCmdSilent(fmt.Sprintf("docker -H %s login -u %s -p %s -e %s",
 				dockerServerUrl, d.Username, d.Password, d.Email))
-	f.WriteCmd(fmt.Sprintf("docker -H %s push %s", dockerServerUrl, dockerRepo))
+		}
+	}
+
+	// Are we overriding the "latest" tag?
+	if d.PushLatest {
+		f.WriteCmd(fmt.Sprintf("docker -H %s tag %s:%s %s:latest",
+			dockerServerUrl, d.ImageName, imageTag, d.ImageName))
+	}
+
+	f.WriteCmd(fmt.Sprintf("docker -H %s push %s", dockerServerUrl, d.ImageName))
 
 	// Delete the image from the docker server we built on.
-	f.WriteCmd(fmt.Sprintf("docker -H %s rmi %s:$(git rev-parse --short HEAD)",
+	if ! d.KeepBuild {
-		dockerServerUrl, dockerRepo))
+		f.WriteCmd(fmt.Sprintf("docker -H %s rmi %s:%s",
+			dockerServerUrl, d.ImageName, imageTag))
+		if d.PushLatest {
+			f.WriteCmd(fmt.Sprintf("docker -H %s rmi %s:latest",
+				dockerServerUrl, d.ImageName))
+		}
+	}
 }

pkg/plugin/publish/docker_test.go

@@ -24,6 +24,140 @@ func setUpWithDrone(input string) (string, error) {
 	return bf.String(), err
 }
 
+// Private Registry Test (no auth)
+var privateRegistryNoAuthYaml = `
+publish:
+  docker:
+    dockerfile: file_path
+    docker_server: server
+    docker_port: 1000
+    docker_version: 1.0
+    registry_login: false
+    image_name: registry/image
+`
+func TestPrivateRegistryNoAuth(t *testing.T) {
+	response, err := setUpWithDrone(privateRegistryNoAuthYaml)
+	t.Log(privateRegistryNoAuthYaml)
+	if err != nil {
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
+	}
+	if !strings.Contains(response, "docker -H server:1000 build -t registry/image:$(git rev-parse --short HEAD)") {
+		t.Fatalf("Response: " + response + " doesn't contain registry in image-names: expected registry/image\n\n")
+	}
+}
+
+// Private Registry Test (with auth)
+var privateRegistryAuthYaml = `
+publish:
+  docker:
+    dockerfile: file_path
+    docker_server: server
+    docker_port: 1000
+    docker_version: 1.0
+    registry_login_url: https://registry:8000/v1/
+    registry_login: true
+    username: username
+    password: password
+    email: email@example.com
+    image_name: registry/image
+`
+func TestPrivateRegistryAuth(t *testing.T) {
+	response, err := setUpWithDrone(privateRegistryAuthYaml)
+	t.Log(privateRegistryAuthYaml)
+	if err != nil {
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
+	}
+	if !strings.Contains(response, "docker -H server:1000 login -u username -p password -e email@example.com https://registry:8000/v1/") {
+		t.Log("\n\n\n\ndocker -H server:1000 login -u username -p xxxxxxxx -e email@example.com https://registry:8000/v1/\n\n\n\n")
+		t.Fatalf("Response: " + response + " doesn't contain private registry login\n\n")
+	}
+	if !strings.Contains(response, "docker -H server:1000 build -t registry/image:$(git rev-parse --short HEAD) .") {
+		t.Log("docker -H server:1000 build -t registry/image:$(git rev-parse --short HEAD) .")
+		t.Fatalf("Response: " + response + " doesn't contain registry in image-names\n\n")
+	}
+}
+
+// Override "latest" Test
+var overrideLatestTagYaml = `
+publish:
+  docker:
+    docker_server: server
+    docker_port: 1000
+    docker_version: 1.0
+    username: username
+    password: password
+    email: email@example.com
+    image_name: username/image
+    push_latest: true
+`
+func TestOverrideLatestTag(t *testing.T) {
+	response, err := setUpWithDrone(overrideLatestTagYaml)
+	t.Log(overrideLatestTagYaml)
+	if err != nil {
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
+	}
+	if !strings.Contains(response, "docker -H server:1000 build -t username/image:$(git rev-parse --short HEAD) .") {
+		t.Fatalf("Response: " + response + " doesn't contain the git-ref tagged image\n\n")
+	}
+	if !strings.Contains(response, "docker -H server:1000 tag username/image:$(git rev-parse --short HEAD) username/image:latest") {
+		t.Fatalf("Response: " + response + " doesn't contain 'latest' tag command\n\n")
+	}
+}
+
+// Keep builds Test
+var keepBuildsYaml = `
+publish:
+  docker:
+    docker_server: server
+    docker_port: 1000
+    docker_version: 1.0
+    keep_build: true
+    username: username
+    password: password
+    email: email@example.com
+    image_name: image
+`
+func TestKeepBuilds(t *testing.T) {
+	response, err := setUpWithDrone(keepBuildsYaml)
+	t.Log(keepBuildsYaml)
+	if err != nil {
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
+	}
+	if strings.Contains(response, "docker -H server:1000 rmi") {
+		t.Fatalf("Response: " + response + " incorrectly instructs the docker server to remove the builds when it shouldn't\n\n")
+	}
+}
+
+// Custom Tag test
+var customTagYaml = `
+publish:
+  docker:
+    docker_server: server
+    docker_port: 1000
+    docker_version: 1.0
+    custom_tag: release-0.1
+    username: username
+    password: password
+    email: email@example.com
+    image_name: username/image
+`
+func TestCustomTag(t *testing.T) {
+	response, err := setUpWithDrone(customTagYaml)
+	t.Log(customTagYaml)
+	if err != nil {
+		t.Fatalf("Can't unmarshal script: %s\n", err.Error())
+	}
+	if strings.Contains(response, "$(git rev-parse --short HEAD)") {
+		t.Fatalf("Response: " + response + " is tagging images from git-refs when it should use a custom tag\n\n")
+	}
+	if !strings.Contains(response, "docker -H server:1000 build -t username/image:release-0.1") {
+		t.Fatalf("Response: " + response + " isn't tagging images using our custom tag\n\n")
+	}
+	if !strings.Contains(response, "docker -H server:1000 push username/image"){
+		t.Fatalf("Response: " + response + " doesn't push the custom tagged image\n\n")
+	}
+}
+
 var missingFieldsYaml = `
 publish:
   docker:
@@ -32,11 +166,12 @@ publish:
 
 func TestMissingFields(t *testing.T) {
 	response, err := setUpWithDrone(missingFieldsYaml)
+	t.Log(missingFieldsYaml)
 	if err != nil {
-        t.Fatalf("Can't unmarshal script: %s", err.Error())
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
 	}
-    if !strings.Contains(response, "echo \"Docker Plugin: Missing argument(s)") {
+	if !strings.Contains(response, "Missing argument(s)") {
-        t.Fatalf("Response: " + response + " didn't contain missing arguments warning")
+		t.Fatalf("Response: " + response + " didn't contain missing arguments warning\n\n")
 	}
 }
 
@@ -47,34 +182,36 @@ publish:
     docker_server: server
     docker_port: 1000
     docker_version: 1.0
-    repo_base_name: base_repo
     username: user
     password: password
     email: email
+    image_name: user/image
+    push_latest: true
+    registry_login: true
 `
 
 func TestValidYaml(t *testing.T) {
 	response, err := setUpWithDrone(validYaml)
+	t.Log(validYaml)
 	if err != nil {
-        t.Fatalf("Can't unmarshal script: %s", err.Error())
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
 	}
 
-    if !strings.Contains(response, "docker -H server:1000 build -t base_repo/name - <") {
+	if !strings.Contains(response, "docker -H server:1000 tag user/image:$(git rev-parse --short HEAD) user/image:latest") {
-        t.Fatalf("Response: " + response + " doesn't contain build command for latest")
+		t.Fatalf("Response: " + response + " doesn't contain tag command for latest\n\n")
 	}
-    if !strings.Contains(response, "docker -H server:1000 build -t base_repo/name" +
+	if !strings.Contains(response, "docker -H server:1000 build -t user/image:$(git rev-parse --short HEAD) - <") {
-        ":$(git rev-parse --short HEAD)") {
+		t.Fatalf("Response: " + response + "doesn't contain build command for commit hash\n\n")
-        t.Fatalf("Response: " + response + "doesn't contain build command for commit hash")
 	}
 	if !strings.Contains(response, "docker -H server:1000 login -u user -p password -e email") {
-        t.Fatalf("Response: " + response + " doesn't contain login command")
+		t.Fatalf("Response: " + response + " doesn't contain login command\n\n")
 	}
-    if !strings.Contains(response, "docker -H server:1000 push base_repo/name") {
+	if !strings.Contains(response, "docker -H server:1000 push user/image") {
-        t.Fatalf("Response: " + response + " doesn't contain push command")
+		t.Fatalf("Response: " + response + " doesn't contain push command\n\n")
 	}
-    if !strings.Contains(response, "docker -H server:1000 rmi base_repo/name:" +
+	if !strings.Contains(response, "docker -H server:1000 rmi user/image:" +
 		"$(git rev-parse --short HEAD)") {
-        t.Fatalf("Response: " + response + " doesn't contain remove image command")
+		t.Fatalf("Response: " + response + " doesn't contain remove image command\n\n")
 	}
 }
 
@@ -84,7 +221,7 @@ publish:
     docker_server: server
     docker_port: 1000
     docker_version: 1.0
-    repo_base_name: base_repo
+    image_name: user/image
     username: user
     password: password
     email: email
@@ -92,11 +229,12 @@ publish:
 
 func TestWithoutDockerFile(t *testing.T) {
 	response, err := setUpWithDrone(withoutDockerFileYaml)
+	t.Log(withoutDockerFileYaml)
 	if err != nil {
-		t.Fatalf("Can't unmarshal script: %s", err.Error())
+		t.Fatalf("Can't unmarshal script: %s\n\n", err.Error())
 	}
 
-	if !strings.Contains(response, "docker -H server:1000 build -t base_repo/name .") {
+	if !strings.Contains(response, "docker -H server:1000 build -t user/image:$(git rev-parse --short HEAD) .") {
-		t.Fatalf("Response: " + response + " doesn't contain build command")
+		t.Fatalf("Response: " + response + " doesn't contain build command\n\n")
 	}
 }