add timeout connexion and add check on open connect

.vscode/settings.json

@@ -3,25 +3,21 @@
   //-------- Editor configuration --------
   "editor.insertSpaces": true,
   "editor.tabSize": 4,
-
   //-------- Files configuration --------
   "files.autoGuessEncoding": false,
   "files.insertFinalNewline": true,
   "files.trimTrailingWhitespace": true,
   "search.exclude": {
     "**/Tests/Data*": true
   },
-
   //-------- PowerShell configuration --------
   "powershell.codeFormatting.alignPropertyValuePairs": true,
   "powershell.codeFormatting.preset": "Allman",
   "powershell.scriptAnalysis.settingsPath": "./ScriptAnalyzerSettings.psd1",
-
   //-------- Language configuration --------
   "[json]": {
     "editor.tabSize": 2
   },
-
   "[xml]": {
     "editor.tabSize": 2
   },

Docs/en-US/Invoke-SnowSql.md

@@ -14,7 +14,7 @@ Invokes a Snowflake SQL statement
 
 ### QueryConnection (Default)
 ```
-Invoke-SnowSql [-Connection <Object>] [-Query <String[]>] [-WhatIf] [-Confirm] [<CommonParameters>]
+Invoke-SnowSql [-Connection <Object>] [-Query <String[]>] [-Timeout <Int>] [-WhatIf] [-Confirm] [<CommonParameters>]
 ```
 
 ### PathCred
@@ -25,7 +25,7 @@ Invoke-SnowSql -Endpoint <String> -Credential <PSCredential> [-Path <String>] [-
 
 ### QueryCred
 ```
-Invoke-SnowSql -Endpoint <String> -Credential <PSCredential> [-Query <String[]>] [-WhatIf] [-Confirm]
+Invoke-SnowSql -Endpoint <String> -Credential <PSCredential> [-Query <String[]>] [-Timeout <Int>] [-WhatIf] [-Confirm]
  [<CommonParameters>]
 ```
 
@@ -123,6 +123,21 @@ Accept pipeline input: False
 Accept wildcard characters: False
 ```
 
+### -Timeout
+Snowflake timeout for connexion
+
+```yaml
+Type: Integer
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: 2
+Default value: 10
+Accept pipeline input: False
+Accept wildcard characters: False
+```
+
 ### -Confirm
 Prompts you for confirmation before running the cmdlet.
 

Docs/en-US/Open-SnowSqlConnection.md

@@ -13,7 +13,7 @@ Opens a connection to Snowflake
 ## SYNTAX
 
 ```
-Open-SnowSqlConnection [-Endpoint] <String> [-Credential] <PSCredential> [-WhatIf] [-Confirm]
+Open-SnowSqlConnection [-Endpoint] <String> [-Credential] <PSCredential> [-Timeout] <Int> [-WhatIf] [-Confirm]
  [<CommonParameters>]
 ```
 
@@ -27,6 +27,11 @@ Establishes a few important environment values for connecting to snowflake
 Open-SnowSqlConnection -Endpoint contoso.east-us-2.azure -Credential (Get-Credential)
 ```
 
+### EXAMPLE 2
+```
+Open-SnowSqlConnection -Endpoint contoso.east-us-2.azure -Credential (Get-Credential) -Timeout 20
+```
+
 ## PARAMETERS
 
 ### -Credential
@@ -59,6 +64,21 @@ Accept pipeline input: False
 Accept wildcard characters: False
 ```
 
+### -Timeout
+Snowflake timeout for connexion
+
+```yaml
+Type: Integer
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: 2
+Default value: 10
+Accept pipeline input: False
+Accept wildcard characters: False
+```
+
 ### -Confirm
 Prompts you for confirmation before running the cmdlet.
 

SnowSQL/public/Invoke-SnowSql.ps1

@@ -44,6 +44,12 @@ function Invoke-SnowSql
         [string[]]
         $Query = '!help',
 
+        # Timeout query
+        [Parameter(ParameterSetName = 'QueryCred')]
+        [Parameter(ParameterSetName = 'QueryConnection')]
+        [int]
+        $timeout = 10,
+
         # SnowSql script file to execute
         [Parameter(ParameterSetName = 'PathCred')]
         [Parameter(ParameterSetName = 'PathConnection')]
@@ -56,16 +62,16 @@ function Invoke-SnowSql
         try
         {
             $snowSql = Get-Command snowsql -ErrorAction Stop |
-                Select-Object -First 1 -ExpandProperty Source
+            Select-Object -First 1 -ExpandProperty Source
         }
         catch
         {
             Write-Error "Could not find [snowsql.exe] on local system. Install snowsql.exe and make it available in the %path%. Install instructions can be found here [https://docs.snowflake.net/manuals/user-guide/snowsql-install-config.html]" -ErrorAction Stop
         }
 
-        if($PSCmdlet.ParameterSetName -match 'Connection$')
+        if ($PSCmdlet.ParameterSetName -match 'Connection$')
         {
-            if( -not $Connection )
+            if ( -not $Connection )
             {
                 $Connection = Get-SnowSqlConnection
             }
@@ -100,6 +106,10 @@ function Invoke-SnowSql
             {
                 '--option', 'log_level=DEBUG'
             }
+            if ($timeout)
+            {
+                '--option', $('login_timeout=' + $timeout)
+            }
             if ($Path)
             {
                 '--filename', $Path
@@ -113,8 +123,15 @@ function Invoke-SnowSql
         Write-Debug ("Executing [& '$snowsql' $snowSqlParam]" -f $snowsql)
         if ($PSCmdlet.ShouldProcess("Execute SnowSql on [$Endpoint] as [$($Credential.UserName)]. Use -Debug to see full command"))
         {
-            $env:SNOWSQL_PWD = $Credential.GetNetworkCredential().password
-            $results = & $snowsql @snowSqlParam | ConvertFrom-Csv
+            $env:SNOWSQL_PWD = $Credential.GetNetworkCredential().Password
+            try
+            {
+                $results = & $snowsql @snowSqlParam | ConvertFrom-Csv
+            }
+            catch
+            {
+                $results = $null
+            }
             $env:SNOWSQL_PWD = ""
             Write-Verbose "LastExitCode[$LastExitCode]"
         }

SnowSQL/public/Open-SnowSqlConnection.ps1

@@ -15,7 +15,7 @@ function Open-SnowSqlConnection
     #>
 
     [OutputType('SnowSql.Connection')]
-    [Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSShouldProcess", "", Justification="Implemented in Invoke-SnowSql")]
+    [Diagnostics.CodeAnalysis.SuppressMessageAttribute("PSShouldProcess", "", Justification = "Implemented in Invoke-SnowSql")]
     [cmdletbinding(SupportsShouldProcess)]
     param(
         # Snowflake endpoint (ex: contoso.east-us-2.azure)
@@ -26,8 +26,19 @@ function Open-SnowSqlConnection
         # Credential for snowflake endpoint
         [Parameter(Mandatory)]
         [PSCredential]
-        $Credential
+        $Credential,
+
+        # Timeout connexion
+        [int]
+        $Timeout = 10
     )
+    begging
+    {
+        if ($Endpoint -match '(http[s]?)(:\/\/)([^\s,]+)')
+        {
+            Write-Error ("Snowflake endpoint must not be a URL {0}" -f $Endpoint) -ErrorAction Stop
+        }
+    }
 
     end
     {
@@ -42,10 +53,16 @@ function Open-SnowSqlConnection
             Query       = '!help'
             ErrorAction = 'Stop'
             Connection  = $SnowSqlConnection
+            timeout     = $Timeout
+        }
+        $Result = Invoke-SnowSql @invokeSnowSqlSplat
+        if (-not $Result)
+        {
+            Write-Error ("Unable to connect to SnowSql endpoint {0}" -f $Endpoint) -ErrorAction Stop
+        }
+        else
+        {
+            return $Script:SnowSqlConnection
         }
-        $null = Invoke-SnowSql @invokeSnowSqlSplat
-        $Script:SnowSqlConnection = $SnowSqlConnection
-
-        return $Script:SnowSqlConnection
     }
 }