Allow customising the snowflake driver name

Author: georgestagg

ggsql-vscode/src/connections.ts

@@ -200,7 +200,8 @@ function readSnowflakeConnections(): {
  * Build an ODBC connection string for Snowflake with the given parts.
  */
 function buildSnowflakeOdbc(parts: Record<string, string | undefined>): string {
-    let connStr = `Driver=Snowflake;Server=${parts.account}.snowflakecomputing.com`;
+    const driver = parts.driver || 'Snowflake';
+    let connStr = `Driver=${driver};Server=${parts.account}.snowflakecomputing.com`;
     if (parts.uid) {
         connStr += `;UID=${parts.uid}`;
     }
@@ -271,6 +272,8 @@ function createSnowflakeDefaultDriver(
         ];
     }
 
+    inputs.push({ id: 'driver', label: 'Driver', type: 'string', value: 'Snowflake' });
+
     return {
         driverId: 'ggsql-snowflake-default',
         metadata: {
@@ -282,7 +285,9 @@ function createSnowflakeDefaultDriver(
         generateCode: (inputs) => {
             const name =
                 inputs.find((i) => i.id === 'connection_name')?.value?.trim() || 'default';
-            return `-- @connect: odbc://Driver=Snowflake;ConnectionName=${name}`;
+            const driver =
+                inputs.find((i) => i.id === 'driver')?.value?.trim() || 'Snowflake';
+            return `-- @connect: odbc://Driver=${driver};ConnectionName=${name}`;
         },
         connect: snowflakeConnect(positronApi),
     };
@@ -308,6 +313,7 @@ function createSnowflakePasswordDriver(
                 { id: 'warehouse', label: 'Warehouse', type: 'string' },
                 { id: 'database', label: 'Database', type: 'string', value: '' },
                 { id: 'schema', label: 'Schema', type: 'string', value: '' },
+                { id: 'driver', label: 'Driver', type: 'string', value: 'Snowflake' },
             ],
         } as ConnectionsDriverMetadata,
         generateCode: (inputs) => {
@@ -320,6 +326,7 @@ function createSnowflakePasswordDriver(
                 warehouse: get('warehouse'),
                 database: get('database') || undefined,
                 schema: get('schema') || undefined,
+                driver: get('driver') || undefined,
             });
         },
         connect: snowflakeConnect(positronApi),
@@ -345,6 +352,7 @@ function createSnowflakeSSODriver(
                 { id: 'warehouse', label: 'Warehouse', type: 'string' },
                 { id: 'database', label: 'Database', type: 'string', value: '' },
                 { id: 'schema', label: 'Schema', type: 'string', value: '' },
+                { id: 'driver', label: 'Driver', type: 'string', value: 'Snowflake' },
             ],
         } as ConnectionsDriverMetadata,
         generateCode: (inputs) => {
@@ -357,6 +365,7 @@ function createSnowflakeSSODriver(
                 warehouse: get('warehouse'),
                 database: get('database') || undefined,
                 schema: get('schema') || undefined,
+                driver: get('driver') || undefined,
             });
         },
         connect: snowflakeConnect(positronApi),
@@ -382,6 +391,7 @@ function createSnowflakePATDriver(
                 { id: 'warehouse', label: 'Warehouse', type: 'string' },
                 { id: 'database', label: 'Database', type: 'string', value: '' },
                 { id: 'schema', label: 'Schema', type: 'string', value: '' },
+                { id: 'driver', label: 'Driver', type: 'string', value: 'Snowflake' },
             ],
         } as ConnectionsDriverMetadata,
         generateCode: (inputs) => {
@@ -394,6 +404,7 @@ function createSnowflakePATDriver(
                 warehouse: get('warehouse'),
                 database: get('database') || undefined,
                 schema: get('schema') || undefined,
+                driver: get('driver') || undefined,
             });
         },
         connect: snowflakeConnect(positronApi),