From 2936a2c859355d632f01f709041def08b96aa976 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Wed, 3 Sep 2025 23:04:38 +0000 Subject: [PATCH 1/3] Initial plan From e9b19ecd1a6a06d12567927e933aad4589d61c91 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Wed, 3 Sep 2025 23:08:57 +0000 Subject: [PATCH 2/3] Add Snowflake database support implementation Co-authored-by: alex-wolf-ps <33110774+alex-wolf-ps@users.noreply.github.com> --- DBChatPro.UI/Components/Pages/ConnectDb.razor | 1 + DBChatPro.UI/Program.cs | 1 + DBChatPro.UI/SampleConnectionStrings.md | 3 + DbChatPro.Core/DbChatPro.Core.csproj | 1 + .../Services/DatabaseManagerService.cs | 7 +- .../Services/SnowflakeDatabaseService.cs | 118 ++++++++++++++++++ 6 files changed, 130 insertions(+), 1 deletion(-) create mode 100644 DbChatPro.Core/Services/SnowflakeDatabaseService.cs diff --git a/DBChatPro.UI/Components/Pages/ConnectDb.razor b/DBChatPro.UI/Components/Pages/ConnectDb.razor index ef4a32f..2c80978 100644 --- a/DBChatPro.UI/Components/Pages/ConnectDb.razor +++ b/DBChatPro.UI/Components/Pages/ConnectDb.razor @@ -32,6 +32,7 @@ MYSQL POSTGRESQL ORACLE + SNOWFLAKE diff --git a/DBChatPro.UI/Program.cs b/DBChatPro.UI/Program.cs index 67eebd9..ea894b2 100644 --- a/DBChatPro.UI/Program.cs +++ b/DBChatPro.UI/Program.cs @@ -24,6 +24,7 @@ builder.Services.AddScoped(); builder.Services.AddScoped(); builder.Services.AddScoped(); +builder.Services.AddScoped(); if (!string.IsNullOrEmpty(builder.Configuration["AWS:Profile"])) { diff --git a/DBChatPro.UI/SampleConnectionStrings.md b/DBChatPro.UI/SampleConnectionStrings.md index 679d78b..98346b3 100644 --- a/DBChatPro.UI/SampleConnectionStrings.md +++ b/DBChatPro.UI/SampleConnectionStrings.md @@ -9,3 +9,6 @@ Host=127.0.0.1;Port=5432;Database=sakila;Username=sakila;Password=p_ssW0rd; User Id=;Password=;Data Source=:/; + + +account=;user=;password=;db=;schema=;warehouse=; diff --git a/DbChatPro.Core/DbChatPro.Core.csproj b/DbChatPro.Core/DbChatPro.Core.csproj index d4bedc8..4789d17 100644 --- a/DbChatPro.Core/DbChatPro.Core.csproj +++ b/DbChatPro.Core/DbChatPro.Core.csproj @@ -29,5 +29,6 @@ + diff --git a/DbChatPro.Core/Services/DatabaseManagerService.cs b/DbChatPro.Core/Services/DatabaseManagerService.cs index 48770ba..591d21f 100644 --- a/DbChatPro.Core/Services/DatabaseManagerService.cs +++ b/DbChatPro.Core/Services/DatabaseManagerService.cs @@ -11,7 +11,8 @@ public class DatabaseManagerService( MySqlDatabaseService mySqlDb, SqlServerDatabaseService msSqlDb, PostgresDatabaseService postgresDb, - OracleDatabaseService oracleDb) : IDatabaseService + OracleDatabaseService oracleDb, + SnowflakeDatabaseService snowflakeDb) : IDatabaseService { public async Task>> GetDataTable(AIConnection conn, string sqlQuery) { @@ -25,6 +26,8 @@ public async Task>> GetDataTable(AIConnection conn, string sql return await postgresDb.GetDataTable(conn, sqlQuery); case "ORACLE": return await oracleDb.GetDataTable(conn, sqlQuery); + case "SNOWFLAKE": + return await snowflakeDb.GetDataTable(conn, sqlQuery); } return null; @@ -42,6 +45,8 @@ public async Task GenerateSchema(AIConnection conn) return await postgresDb.GenerateSchema(conn); case "ORACLE": return await oracleDb.GenerateSchema(conn); + case "SNOWFLAKE": + return await snowflakeDb.GenerateSchema(conn); } return new() { SchemaStructured = new List(), SchemaRaw = new List() }; diff --git a/DbChatPro.Core/Services/SnowflakeDatabaseService.cs b/DbChatPro.Core/Services/SnowflakeDatabaseService.cs new file mode 100644 index 0000000..82358a5 --- /dev/null +++ b/DbChatPro.Core/Services/SnowflakeDatabaseService.cs @@ -0,0 +1,118 @@ +// Generated by AI - Snowflake database service implementation +using DBChatPro.Models; +using Snowflake.Data.Client; + +namespace DBChatPro +{ + public class SnowflakeDatabaseService : IDatabaseService + { + public async Task>> GetDataTable(AIConnection conn, string sqlQuery) + { + var rows = new List>(); + using var connection = new SnowflakeDbConnection(conn.ConnectionString); + + await connection.OpenAsync(); + + using var command = new SnowflakeDbCommand(sqlQuery, connection); + using var reader = await command.ExecuteReaderAsync(); + + int count = 0; + bool headersAdded = false; + while (await reader.ReadAsync()) + { + var cols = new List(); + var headerCols = new List(); + if (!headersAdded) + { + for (int i = 0; i < reader.FieldCount; i++) + { + headerCols.Add(reader.GetName(i).ToString()); + } + headersAdded = true; + rows.Add(headerCols); + } + + for (int i = 0; i <= reader.FieldCount - 1; i++) + { + try + { + cols.Add(reader.GetValue(i).ToString()); + } + catch + { + cols.Add("DataTypeConversionError"); + } + } + rows.Add(cols); + } + + return rows; + } + + public async Task GenerateSchema(AIConnection conn) + { + var dbSchema = new DatabaseSchema() { SchemaRaw = new List(), SchemaStructured = new List() }; + List> rows = new(); + + // Parse connection string for database and schema + var pairs = conn.ConnectionString.Split(";"); + var database = pairs.Where(x => x.ToUpper().Contains("DB") || x.ToUpper().Contains("DATABASE")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; + var schema = pairs.Where(x => x.ToUpper().Contains("SCHEMA")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; + + // Snowflake schema query using INFORMATION_SCHEMA + string sqlQuery = @"SELECT + table_name, + column_name + FROM + information_schema.columns + WHERE + table_catalog = ? + AND table_schema = ? + ORDER BY + table_name, + column_name"; + + using var connection = new SnowflakeDbConnection(conn.ConnectionString); + await connection.OpenAsync(); + + using var command = new SnowflakeDbCommand(sqlQuery, connection); + command.Parameters.Add(new SnowflakeDbParameter("1", database)); + command.Parameters.Add(new SnowflakeDbParameter("2", schema)); + + using var reader = await command.ExecuteReaderAsync(); + while (await reader.ReadAsync()) + { + rows.Add(new KeyValuePair(reader.GetString(0), reader.GetString(1))); + } + + var groups = rows.GroupBy(x => x.Key); + + foreach (var group in groups) + { + dbSchema.SchemaStructured.Add(new TableSchema() { TableName = group.Key, Columns = group.Select(x => x.Value).ToList() }); + } + + var textLines = new List(); + + foreach (var table in dbSchema.SchemaStructured) + { + var schemaLine = $"- {table.TableName} ("; + + foreach (var column in table.Columns) + { + schemaLine += column + ", "; + } + + schemaLine += ")"; + schemaLine = schemaLine.Replace(", )", " )"); + + Console.WriteLine(schemaLine); + textLines.Add(schemaLine); + } + + dbSchema.SchemaRaw = textLines; + + return dbSchema; + } + } +} \ No newline at end of file From 5b4f515eef319845d52ccbe5b94b33271acde0ce Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Wed, 3 Sep 2025 23:10:44 +0000 Subject: [PATCH 3/3] Improve Snowflake connection string parsing logic Co-authored-by: alex-wolf-ps <33110774+alex-wolf-ps@users.noreply.github.com> --- DbChatPro.Core/Services/SnowflakeDatabaseService.cs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/DbChatPro.Core/Services/SnowflakeDatabaseService.cs b/DbChatPro.Core/Services/SnowflakeDatabaseService.cs index 82358a5..a73f88f 100644 --- a/DbChatPro.Core/Services/SnowflakeDatabaseService.cs +++ b/DbChatPro.Core/Services/SnowflakeDatabaseService.cs @@ -56,8 +56,9 @@ public async Task GenerateSchema(AIConnection conn) // Parse connection string for database and schema var pairs = conn.ConnectionString.Split(";"); - var database = pairs.Where(x => x.ToUpper().Contains("DB") || x.ToUpper().Contains("DATABASE")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; - var schema = pairs.Where(x => x.ToUpper().Contains("SCHEMA")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; + var database = pairs.Where(x => x.ToUpper().Contains("DB=") || x.ToUpper().Contains("DATABASE=")).FirstOrDefault()?.Split("=").Last() ?? + pairs.Where(x => x.ToUpper().Contains("DB") && x.Contains("=")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; + var schema = pairs.Where(x => x.ToUpper().Contains("SCHEMA=")).FirstOrDefault()?.Split("=").Last() ?? "PUBLIC"; // Snowflake schema query using INFORMATION_SCHEMA string sqlQuery = @"SELECT