feat(connection): enhance Snowflake connection with token support and MFA caching

Author: flarco

core/dbio/connection/connection.go

@@ -749,7 +749,7 @@ func (c *Connection) setURL() (err error) {
 		// template = "snowflake://{username}:{password}@{host}.snowflakecomputing.com:443/{database}?schema={schema}&warehouse={warehouse}"
 		setIfMissing("username", c.Data["user"])
 		setIfMissing("host", c.Data["account"])
-		setIfMissing("password", "") // make password optional, especially when using a private key
+		setIfMissing("password", "") // make password optional, especially when using a private key or token
 		c.Data["host"] = strings.ReplaceAll(cast.ToString(c.Data["host"]), ".snowflakecomputing.com", "")
 		template = "snowflake://{username}:{password}@{host}.snowflakecomputing.com:443/{database}?"
 		if _, ok := c.Data["warehouse"]; ok {
@@ -767,6 +767,16 @@ func (c *Connection) setURL() (err error) {
 		if _, ok := c.Data["passcode"]; ok {
 			template = template + "&passcode={passcode}"
 		}
+		if _, ok := c.Data["token"]; ok {
+			template = template + "&token={token}"
+		}
+		// Enable MFA token caching by default when using username_password_mfa authenticator
+		if strings.EqualFold(cast.ToString(c.Data["authenticator"]), "username_password_mfa") {
+			setIfMissing("clientRequestMfaToken", "true")
+		}
+		if _, ok := c.Data["clientRequestMfaToken"]; ok {
+			template = template + "&clientRequestMfaToken={clientRequestMfaToken}"
+		}
 	case dbio.TypeDbDatabricks:
 		setIfMissing("token", c.Data["password"])
 		setIfMissing("port", c.Type.DefPort())

core/dbio/database/database_snowflake.go

@@ -55,6 +55,13 @@ func (conn *SnowflakeConn) Init() error {
 		)
 	}
 
+	if strings.EqualFold(conn.GetProp("authenticator"), "programmatic_access_token") &&
+		conn.GetProp("token") == "" {
+		return g.Error(
+			"did not provide property `token` with authenticator=programmatic_access_token. See https://docs.slingdata.io/connections/database-connections/snowflake",
+		)
+	}
+
 	if m := conn.GetProp("copy_method"); m != "" {
 		conn.CopyMethod = conn.GetProp("copy_method")
 	}