From ca95f0f93cf4a25f08bd9b961b2042efbca72a9d Mon Sep 17 00:00:00 2001
From: Maximo Cuadros <mcuadros@gmail.com>
Date: Thu, 10 Nov 2022 21:49:30 +0100
Subject: [PATCH] PluginManager, download directly, quick fix for fixed
 versions

---
 .github/workflows/test.yml      |  1 -
 starlark/types/provider_test.go |  2 +-
 terraform/plugins.go            | 83 +++++++++++++++++++++++++++++++++
 3 files changed, 84 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index ef856ac..256db2f 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -26,7 +26,6 @@ jobs:
 
     - name: Allow skip provisioner tests
       uses: allenevans/set-env@v2.0.0
-      if: matrix.os != 'ubuntu-latest'
       with:
         ALLOW_PROVISIONER_SKIP: 1
         
diff --git a/starlark/types/provider_test.go b/starlark/types/provider_test.go
index 7870cb3..a13842a 100644
--- a/starlark/types/provider_test.go
+++ b/starlark/types/provider_test.go
@@ -36,7 +36,7 @@ func TestProvider(t *testing.T) {
 }
 
 func TestProvisioner(t *testing.T) {
-	if stdos.Getenv("ALLOW_PROVISIONER_SKIP") != "" && !terraform.IsTerraformBinaryAvailable() {
+	if stdos.Getenv("ALLOW_PROVISIONER_SKIP") != "" {
 		t.Skip("terraform binary now available in $PATH")
 	}
 
diff --git a/terraform/plugins.go b/terraform/plugins.go
index c583f0b..a8c33ba 100644
--- a/terraform/plugins.go
+++ b/terraform/plugins.go
@@ -1,9 +1,15 @@
 package terraform
 
 import (
+	"archive/zip"
 	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
 	"os"
 	"os/exec"
+	"path/filepath"
+	"runtime"
 	"strings"
 
 	"github.com/hashicorp/go-hclog"
@@ -28,6 +34,11 @@ type PluginManager struct {
 func (m *PluginManager) Provider(provider, version string, forceLocal bool) (*plugin.Client, discovery.PluginMeta, error) {
 	meta, ok := m.getLocal("provider", provider, version)
 	if !ok && !forceLocal {
+		meta, ok, _ = m.getProviderRemoteDirectDownload(provider, version)
+		if ok {
+			return client(meta), meta, nil
+		}
+
 		var err error
 		meta, _, err = m.getProviderRemote(provider, version)
 		if err != nil {
@@ -87,6 +98,77 @@ func client(m discovery.PluginMeta) *plugin.Client {
 	})
 }
 
+const releaseTemplateURL = "https://releases.hashicorp.com/terraform-provider-%s/%s/terraform-provider-%[1]s_%[2]s_%s_%s.zip"
+
+func (m *PluginManager) getProviderRemoteDirectDownload(provider, v string) (discovery.PluginMeta, bool, error) {
+	url := fmt.Sprintf(releaseTemplateURL, provider, v, runtime.GOOS, runtime.GOARCH)
+	if err := m.downloadURL(url); err != nil {
+		return discovery.PluginMeta{}, false, err
+	}
+
+	meta, ok := m.getLocal("provider", provider, v)
+	return meta, ok, nil
+}
+
+func (m *PluginManager) downloadURL(url string) error {
+	resp, err := http.Get(url)
+	if err != nil {
+		return fmt.Errorf("error downloading %s file: %w", url, err)
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("invalid URL: %s", url)
+	}
+
+	defer resp.Body.Close()
+	file, err := ioutil.TempFile("", "ascode")
+	if err != nil {
+		return err
+
+	}
+
+	if _, err := io.Copy(file, resp.Body); err != nil {
+		return fmt.Errorf("error downloading %s file: %w", url, err)
+	}
+
+	file.Close()
+	defer os.Remove(file.Name())
+
+	archive, err := zip.OpenReader(file.Name())
+	if err != nil {
+		panic(err)
+	}
+
+	defer archive.Close()
+
+	for _, f := range archive.File {
+		file := filepath.Join(m.Path, f.Name)
+
+		if !strings.HasPrefix(file, filepath.Clean(m.Path)+string(os.PathSeparator)) {
+			return fmt.Errorf("invalid path")
+		}
+
+		output, err := os.OpenFile(file, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
+		if err != nil {
+			return err
+		}
+
+		r, err := f.Open()
+		if err != nil {
+			return err
+		}
+
+		if _, err := io.Copy(output, r); err != nil {
+			return err
+		}
+
+		output.Close()
+		r.Close()
+	}
+
+	return nil
+}
+
 const defaultVersionContraint = "> 0"
 
 func (m *PluginManager) getProviderRemote(provider, v string) (discovery.PluginMeta, bool, error) {
@@ -94,6 +176,7 @@ func (m *PluginManager) getProviderRemote(provider, v string) (discovery.PluginM
 		v = defaultVersionContraint
 	}
 
+	m.getProviderRemoteDirectDownload(provider, v)
 	installer := &discovery.ProviderInstaller{
 		Dir:                   m.Path,
 		PluginProtocolVersion: discovery.PluginInstallProtocolVersion,