chore: support for kubernetes runner (#96)

Signed-off-by: Bo-Yi Wu <[email protected]>
appleboy · Jul 1, 2020 · ee03588 · ee03588
1 parent 153fe2c
commit ee03588
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -16,7 +16,7 @@ var (
 func main() {
 	// Load env-file if it exists first
 	if filename, found := os.LookupEnv("PLUGIN_ENV_FILE"); found {
-		_ = godotenv.Load(filename)
+		godotenv.Load(filename)
 	}
 
 	app := cli.NewApp()
@@ -251,6 +251,10 @@ func main() {
 		},
 	}
 
+	if _, err := os.Stat("/run/drone/env"); err == nil {
+		godotenv.Overload("/run/drone/env")
+	}
+
 	if err := app.Run(os.Args); err != nil {
 		log.Fatal(err)
 	}