feat: support JSON data type in MySQL

Author: SockworkOrange

CodeGenerator/Generators/CsprojGen.cs

@@ -12,6 +12,7 @@ internal class CsprojGen(string outputDirectory, string projectName, string name
     private const string DefaultMysqlConnectorVersion = "2.4.0";
     private const string DefaultSqliteVersion = "9.0.0";
     private const string DefaultCsvHelperVersion = "33.0.1";
+    private const string DefaultSystemTextJsonVersion = "9.0.6";
 
     public File GenerateFile()
     {
@@ -48,18 +49,28 @@ string GetPackageReferences()
         {
             var optionalDapperPackageReference = options.UseDapper
                 ? Environment.NewLine + $"""        <PackageReference Include="Dapper" Version="{GetDapperVersion(options)}"/>"""
-                : "";
-            var optionalCsvHelper = options.DriverName == DriverName.MySqlConnector
+                : string.Empty;
+            var optionalCsvHelper = options.DriverName is DriverName.MySqlConnector
                 ? Environment.NewLine + $"""        <PackageReference Include="CsvHelper" Version="{DefaultCsvHelperVersion}"/>"""
-                : "";
+                : string.Empty;
+            var optionalSystemTextJson = IsSystemTextJsonNeeded()
+                ? Environment.NewLine + $"""        <PackageReference Include="System.Text.Json" Version="{DefaultSystemTextJsonVersion}"/>"""
+                : string.Empty;
             return $"""
                 <ItemGroup>
-                    <PackageReference Include="{options.DriverName.ToName()}" Version="{GetDriverVersion(options)}"/>{optionalDapperPackageReference}{optionalCsvHelper}
+                    <PackageReference Include="{options.DriverName.ToName()}" Version="{GetDriverVersion(options)}"/>{optionalDapperPackageReference}{optionalCsvHelper}{optionalSystemTextJson}
                 </ItemGroup>
             """;
         }
     }
 
+    private bool IsSystemTextJsonNeeded()
+    {
+        if (options.DotnetFramework.IsDotnetCore())
+            return false;
+        return options.DriverName is DriverName.MySqlConnector or DriverName.Npgsql;
+    }
+
     private static string GetDriverVersion(Options options)
     {
         if (string.IsNullOrEmpty(options.OverrideDriverVersion))

CodeGenerator/Generators/ModelsGen.cs

@@ -23,14 +23,15 @@ public File GenerateFile(
         Dictionary<string, Dictionary<string, Table>> tables,
         Dictionary<string, Dictionary<string, Enum>> enums)
     {
-        var usingDirectives = dbDriver.GetUsingDirectivesForModels();
+        var usingDirectives = dbDriver
+            .GetUsingDirectivesForModels()
+            .Select(x => UsingDirective(ParseName(x)))
+            .ToArray();
         var dataclassModels = GenerateDataClasses(tables);
         var enumModels = GenerateEnums(enums);
+        var models = dataclassModels.Concat(enumModels).ToArray();
 
-        var root = RootGen.CompilationRootGen(
-            IdentifierName(namespaceName),
-            usingDirectives,
-            dataclassModels.Concat(enumModels).ToArray());
+        var root = RootGen.CompilationRootGen(IdentifierName(namespaceName), usingDirectives, models);
         root = root.AddCommentOnTop(Consts.AutoGeneratedComment);
 
         return new File

CodeGenerator/Generators/QueriesGen.cs

@@ -49,7 +49,10 @@ private static CompilationUnitSyntax AddResharperDisables(CompilationUnitSyntax
     private (UsingDirectiveSyntax[], MemberDeclarationSyntax) GenerateClass(IEnumerable<Query> queries,
         string className)
     {
-        var usingDirectives = dbDriver.GetUsingDirectivesForQueries();
+        var usingDirectives = dbDriver
+            .GetUsingDirectivesForQueries()
+            .Select(x => UsingDirective(ParseName(x)))
+            .ToArray();
         var classMembers = queries.SelectMany(GetMembersForSingleQuery);
         return (usingDirectives, GetClassDeclaration(className, classMembers));
     }

CodeGenerator/Generators/UtilsGen.cs

@@ -1,5 +1,6 @@
 using Microsoft.CodeAnalysis.CSharp.Syntax;
 using SqlcGenCsharp.Drivers;
+using System.Linq;
 using static Microsoft.CodeAnalysis.CSharp.SyntaxFactory;
 using File = Plugin.File;
 
@@ -19,8 +20,12 @@ internal class UtilsGen(DbDriver dbDriver, string namespaceName)
         if (classImpl == null)
             return null;
 
+        var usingDirectives = dbDriver
+            .GetUsingDirectivesForUtils()
+            .Select(x => UsingDirective(ParseName(x)))
+            .ToArray();
         var root = RootGen.CompilationRootGen(
-            IdentifierName(NamespaceName), dbDriver.GetUsingDirectivesForUtils(), [classImpl]);
+            IdentifierName(NamespaceName), usingDirectives, [classImpl]);
         root = root.AddCommentOnTop(Consts.AutoGeneratedComment);
 
         return new File

Drivers/ColumnMapping.cs

@@ -6,9 +6,15 @@ namespace SqlcGenCsharp.Drivers;
 public record DbTypeInfo(int? Length = null, string? NpgsqlTypeOverride = null);
 
 public class ColumnMapping(
-    Dictionary<string, DbTypeInfo> dbTypes, Func<int, string> readerFn, Func<int, string>? readerArrayFn = null)
+    Dictionary<string, DbTypeInfo> dbTypes,
+    Func<int, string> readerFn,
+    Func<int, string>? readerArrayFn = null,
+    string? usingDirective = null,
+    Func<string, bool, bool, string>? writerFn = null)
 {
     public Func<int, string> ReaderFn { get; } = readerFn;
     public Func<int, string>? ReaderArrayFn { get; } = readerArrayFn;
+    public string? UsingDirective { get; } = usingDirective;
+    public Func<string, bool, bool, string>? WriterFn { get; } = writerFn;
     public Dictionary<string, DbTypeInfo> DbTypes { get; } = dbTypes;
 }