From f140ca382801937af7bef9577a9c8e348230bddd Mon Sep 17 00:00:00 2001
From: TomVasile <43349666+TomVasile@users.noreply.github.com>
Date: Tue, 21 Apr 2020 21:17:04 +0000
Subject: [PATCH] Introduce optional flag for dist directory

---
 main.go   | 7 +++++++
 plugin.go | 3 ++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 490572b..ffa028d 100644
--- a/main.go
+++ b/main.go
@@ -52,6 +52,12 @@ func main() {
 			Usage:  "skip build and only upload pre-build packages",
 			EnvVar: "PLUGIN_SKIP_BUILD",
 		},
+		cli.StringFlag{
+			Name:	"dist_dir",
+			Usage:  "used when distribution directory is not in build root",
+			Value:  "dist/",
+			EnvVar: "PLUGIN_DIST_DIR",
+		},
 	}
 
 	app.Run(os.Args)
@@ -65,6 +71,7 @@ func run(c *cli.Context) {
 		SetupFile:     c.String("setupfile"),
 		Distributions: c.StringSlice("distributions"),
 		SkipBuild:     c.Bool("skip_build"),
+		DistDir:       c.String("dist_dir"),
 	}
 
 	if err := plugin.Exec(); err != nil {
diff --git a/plugin.go b/plugin.go
index c98f2b0..8fe36fa 100644
--- a/plugin.go
+++ b/plugin.go
@@ -16,6 +16,7 @@ type Plugin struct {
 	SetupFile     string
 	Distributions []string
 	SkipBuild     bool
+	DistDir       string
 }
 
 func (p Plugin) buildCommand() *exec.Cmd {
@@ -41,7 +42,7 @@ func (p Plugin) uploadCommand() *exec.Cmd {
 	args = append(args, p.Username)
 	args = append(args, "--password")
 	args = append(args, p.Password)
-	args = append(args, filepath.Join(filepath.Dir(p.SetupFile), "dist/*"))
+	args = append(args, filepath.Join(filepath.Dir(p.DistDir), "/*"))
 
 	return exec.Command("twine", args...)
 }