add proxy arguments for provider

Author: Thierno BARRY

mysql/provider.go

@@ -4,6 +4,8 @@ import (
 	"database/sql"
 	"fmt"
 	"net"
+	"net/url"
+	"regexp"
 	"strings"
 	"time"
 
@@ -53,6 +55,12 @@ func Provider() terraform.ResourceProvider {
 				DefaultFunc: schema.EnvDefaultFunc("MYSQL_PASSWORD", nil),
 			},
 
+			"proxy": {
+				Type:         schema.TypeString,
+				Optional:     true,
+				ValidateFunc: validation.StringMatch(regexp.MustCompile("^socks5h?://.*:\\d+$"), "The proxy URL is not a valid socks url."),
+			},
+
 			"tls": {
 				Type:        schema.TypeString,
 				Optional:    true,
@@ -105,7 +113,11 @@ func providerConfigure(d *schema.ResourceData) (interface{}, error) {
 		AllowNativePasswords: true,
 	}
 
-	dialer := proxy.FromEnvironment()
+	dialer, err := getDialerFromArgOrFromEnv(d)
+	if err != nil {
+		return nil, err
+	}
+
 	mysql.RegisterDial("tcp", func(network string) (net.Conn, error) {
 		return dialer.Dial("tcp", network)
 	})
@@ -119,6 +131,26 @@ func providerConfigure(d *schema.ResourceData) (interface{}, error) {
 
 var identQuoteReplacer = strings.NewReplacer("`", "``")
 
+func getDialerFromArgOrFromEnv(d *schema.ResourceData) (proxy.Dialer, error) {
+	proxyFromEnv := proxy.FromEnvironment()
+	proxyArg := d.Get("proxy").(string)
+
+	if len(proxyArg) > 0 {
+		proxyURL, err := url.Parse(proxyArg)
+		if err != nil {
+			return nil, err
+		}
+		proxy, err := proxy.FromURL(proxyURL, proxyFromEnv)
+		if err != nil {
+			return nil, err
+		}
+
+		return proxy, nil
+	}
+
+	return proxyFromEnv, nil
+}
+
 func quoteIdentifier(in string) string {
 	return fmt.Sprintf("`%s`", identQuoteReplacer.Replace(in))
 }