Allow connections via SSH agent

docs/data-sources/file.md

@@ -64,6 +64,7 @@ Required:
 
 Optional:
 
+- **agent** (Boolean) Use a local SSH agent to login to the remote host. Defaults to `false`.
 - **password** (String, Sensitive) The pasword for the user on the remote host.
 - **port** (Number) The ssh port on the remote host. Defaults to `22`.
 - **private_key** (String, Sensitive) The private key used to login to the remote host.
@@ -77,6 +78,7 @@ Optional:
 
 Read-Only:
 
+- **agent** (Boolean)
 - **host** (String)
 - **password** (String)
 - **port** (Number)

docs/index.md

@@ -65,6 +65,7 @@ Required:
 
 Optional:
 
+- **agent** (Boolean) Use a local SSH agent to login to the remote host. Defaults to `false`.
 - **password** (String, Sensitive) The pasword for the user on the remote host.
 - **port** (Number) The ssh port on the remote host. Defaults to `22`.
 - **private_key** (String, Sensitive) The private key used to login to the remote host.

docs/resources/file.md

@@ -71,6 +71,7 @@ Required:
 
 Optional:
 
+- **agent** (Boolean) Use a local SSH agent to login to the remote host. Defaults to `false`.
 - **password** (String, Sensitive) The pasword for the user on the remote host.
 - **port** (Number) The ssh port on the remote host. Defaults to `22`.
 - **private_key** (String, Sensitive) The private key used to login to the remote host.
@@ -84,6 +85,7 @@ Optional:
 
 Read-Only:
 
+- **agent** (Boolean)
 - **host** (String)
 - **password** (String)
 - **port** (Number)

internal/provider/connection.go

@@ -3,10 +3,12 @@ package provider
 import (
 	"fmt"
 	"io/ioutil"
+	"net"
 	"os"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"golang.org/x/crypto/ssh"
+	"golang.org/x/crypto/ssh/agent"
 )
 
 var connectionSchemaResource = &schema.Resource{
@@ -56,6 +58,13 @@ var connectionSchemaResource = &schema.Resource{
 			Optional:    true,
 			Description: "The name of the local environment variable containing the private key used to login to the remote host.",
 		},
+		"agent": {
+			Type:        schema.TypeBool,
+			Optional:    true,
+			Default:     false,
+			Description: "Use a local SSH agent to login to the remote host.",
+		},
+
 	},
 }
 
@@ -107,6 +116,15 @@ func ConnectionFromResourceData(d *schema.ResourceData) (string, *ssh.ClientConf
 		clientConfig.Auth = append(clientConfig.Auth, ssh.PublicKeys(signer))
 	}
 
+	enableAgent, ok := d.GetOk("result_conn.0.agent")
+	if ok && enableAgent.(bool) {
+		connection, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK"))
+		if err != nil {
+			return "", nil, fmt.Errorf("couldn't connect to SSH agent: %s", err.Error())
+		}
+		clientConfig.Auth = append(clientConfig.Auth, ssh.PublicKeysCallback(agent.NewClient(connection).Signers))
+	}
+
 	host := fmt.Sprintf("%s:%d", d.Get("result_conn.0.host").(string), d.Get("result_conn.0.port").(int))
 	return host, &clientConfig, nil
 }