infra: add support TF_DATA_DIR

xynova · Jul 12, 2019 · 12d5fc1 · 12d5fc1
1 parent 6decc82
commit 12d5fc1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/terraform_utils/provider_wrapper/provider.go b/terraform_utils/provider_wrapper/provider.go
@@ -115,7 +115,11 @@ func (p *ProviderWrapper) Refresh(info *terraform.InstanceInfo, state *terraform
 }
 
 func (p *ProviderWrapper) initProvider() error {
-	pluginPath := command.DefaultDataDir + string(os.PathSeparator) + "plugins" + string(os.PathSeparator) + runtime.GOOS + "_" + runtime.GOARCH
+	defaultDataDir := os.Getenv("TF_DATA_DIR")
+	if defaultDataDir == "" {
+		defaultDataDir = command.DefaultDataDir
+	}
+	pluginPath := defaultDataDir + string(os.PathSeparator) + "plugins" + string(os.PathSeparator) + runtime.GOOS + "_" + runtime.GOARCH
 	files, err := ioutil.ReadDir(pluginPath)
 	if err != nil {
 		pluginPath = os.Getenv("HOME") + string(os.PathSeparator) + "." + command.DefaultPluginVendorDir