replace govendor with dep

cmd/drone-docker-ecr/main.go

@@ -1,5 +1,134 @@
 package main
 
-func main() {
+import (
+	"encoding/base64"
+	"fmt"
+	"log"
+	"os"
+	"os/exec"
+	"strconv"
+	"strings"
 
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/awserr"
+	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/ecr"
+)
+
+const defaultRegion = "us-east-1"
+
+func main() {
+	var (
+		repo   = getenv("PLUGIN_REPO")
+		region = getenv("PLUGIN_REGION", "ECR_REGION", "AWS_REGION")
+		key    = getenv("PLUGIN_ACCESS_KEY", "ECR_ACCESS_KEY", "AWS_ACCESS_KEY_ID")
+		secret = getenv("PLUGIN_SECRET_KEY", "ECR_SECRET_KEY", "AWS_SECRET_ACCESS_KEY")
+		create = parseBoolOrDefault(false, getenv("PLUGIN_CREATE_REPOSITORY", "ECR_CREATE_REPOSITORY"))
+	)
+
+	// set the region
+	if region == "" {
+		region = defaultRegion
+	}
+
+	os.Setenv("AWS_ACCESS_KEY_ID", key)
+	os.Setenv("AWS_SECRET_ACCESS_KEY", secret)
+	os.Setenv("AWS_REGION", region)
+
+	sess, err := session.NewSession(&aws.Config{
+		Credentials: credentials.NewEnvCredentials(),
+		Region:      &region,
+	})
+
+	if err != nil {
+		log.Fatal(fmt.Sprintf("error creating aws session: %v", err))
+	}
+
+	svc := ecr.New(sess)
+	username, password, registry, err := getAuthInfo(svc)
+	if err != nil {
+		os.Exit(1)
+	}
+
+	if !strings.HasPrefix(repo, registry) {
+		repo = fmt.Sprintf("%s/%s", registry, repo)
+	}
+
+	if create {
+		err = ensureRepoExists(svc, strings.TrimPrefix(repo, registry))
+		if err != nil {
+			os.Exit(1)
+		}
+	}
+
+	os.Setenv("PLUGIN_REPO", repo)
+	os.Setenv("PLUGIN_REGISTRY", registry)
+	os.Setenv("DOCKER_USERNAME", username)
+	os.Setenv("DOCKER_PASSWORD", password)
+
+	// invoke the base docker plugin binary
+	cmd := exec.Command("drone-docker")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err = cmd.Run(); err != nil {
+		os.Exit(1)
+	}
+}
+
+func ensureRepoExists(svc *ecr.ECR, name string) (err error) {
+	input := &ecr.CreateRepositoryInput{}
+	input.SetRepositoryName(name)
+	_, err = svc.CreateRepository(input)
+	if err != nil {
+		if aerr, ok := err.(awserr.Error); ok && aerr.Code() == ecr.ErrCodeRepositoryAlreadyExistsException {
+			// eat it, we skip checking for existing to save two requests
+			err = nil
+		}
+	}
+
+	return
+}
+
+func getAuthInfo(svc *ecr.ECR) (username, password, registry string, err error) {
+	var result *ecr.GetAuthorizationTokenOutput
+	var decoded []byte
+
+	result, err = svc.GetAuthorizationToken(&ecr.GetAuthorizationTokenInput{})
+	if err != nil {
+		return
+	}
+
+	auth := result.AuthorizationData[0]
+	token := *auth.AuthorizationToken
+	decoded, err = base64.StdEncoding.DecodeString(token)
+	if err != nil {
+		return
+	}
+
+	registry = strings.TrimPrefix(*auth.ProxyEndpoint, "https://")
+	creds := strings.Split(string(decoded), ":")
+	username = creds[0]
+	password = creds[1]
+	return
+}
+
+func parseBoolOrDefault(defaultValue bool, s string) (result bool) {
+	var err error
+	result, err = strconv.ParseBool(s)
+	if err != nil {
+		result = false
+	}
+
+	return
+}
+
+func getenv(key ...string) (s string) {
+	for _, k := range key {
+		s = os.Getenv(k)
+		if s != "" {
+			return
+		}
+	}
+	return
 }