images: replacing the dependency on acceptance.AzureProvider

azurerm/internal/acceptance/ssh/commands.go

@@ -0,0 +1,8 @@
+package ssh
+
+const (
+	// LinuxAgentDeprovisionCommand is the command needed to "deprovision" the (Windows)Azure VM Agent
+	// this clears out caches, resets the vm agent configuration and generally "generalizes" this VM
+	// and is vaguely akin to `sysprep` on Windows.
+	LinuxAgentDeprovisionCommand = "sudo waagent -verbose -deprovision+user -force"
+)

azurerm/internal/acceptance/ssh/run.go

@@ -0,0 +1,61 @@
+package ssh
+
+import (
+	"bytes"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+	"golang.org/x/crypto/ssh"
+)
+
+type Runner struct {
+	Hostname      string
+	Port          int
+	Username      string
+	Password      string
+	CommandsToRun []string
+}
+
+func (r Runner) Run() error {
+	if err := resource.Retry(5*time.Minute, r.tryRun); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (r Runner) tryRun() *resource.RetryError {
+	config := &ssh.ClientConfig{
+		User: r.Username,
+		Auth: []ssh.AuthMethod{
+			ssh.Password(r.Password),
+		},
+		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+	}
+
+	hostAddress := fmt.Sprintf("%s:%d", r.Hostname, r.Port)
+	log.Printf("[INFO] SSHing to %q...", hostAddress)
+	client, err := ssh.Dial("tcp", hostAddress, config)
+	if err != nil {
+		return resource.RetryableError(fmt.Errorf("connecting to host: %+v", err))
+	}
+
+	session, err := client.NewSession()
+	if err != nil {
+		return resource.RetryableError(fmt.Errorf("creating session: %+v", err))
+	}
+	defer session.Close()
+
+	for _, cmd := range r.CommandsToRun {
+		log.Printf("[DEBUG] Running %q..", cmd)
+		var b bytes.Buffer
+		session.Stdout = &b
+		if err := session.Run(cmd); err != nil {
+			return resource.NonRetryableError(fmt.Errorf("failure running command %q: %+v", cmd, err))
+		}
+	}
+
+	return nil
+}