diff --git a/cmd/drone-ecr/main.go b/cmd/drone-ecr/main.go index 0df3fac..1a949c6 100644 --- a/cmd/drone-ecr/main.go +++ b/cmd/drone-ecr/main.go @@ -21,6 +21,7 @@ const defaultRegion = "us-east-1" func main() { var ( repo = getenv("PLUGIN_REPO") + registry = getenv("PLUGIN_REGISTRY") 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") @@ -48,7 +49,12 @@ func main() { } svc := ecr.New(sess) - username, password, registry, err := getAuthInfo(svc) + username, password, defaultRegistry, err := getAuthInfo(svc) + + if registry == "" { + registry = defaultRegistry + } + if err != nil { log.Fatal(fmt.Sprintf("error getting ECR auth: %v", err)) }