ModelsGen.cs

using Microsoft.CodeAnalysis.CSharp.Syntax;
using Plugin;
using SqlcGenCsharp.Drivers;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using static Microsoft.CodeAnalysis.CSharp.SyntaxFactory;


namespace SqlcGenCsharp.Generators;

internal class ModelsGen(DbDriver dbDriver, string namespaceName)
{
    private const string ClassName = "Models";

    private RootGen RootGen { get; } = new(dbDriver.Options);

    private DataClassesGen DataClassesGen { get; } = new(dbDriver);


    private EnumsGen EnumsGen { get; } = new(dbDriver);

    public File GenerateFile(
        Dictionary<string, Dictionary<string, Table>> tables,
        Dictionary<string, Dictionary<string, Enum>> enums)
    {
        var usingDirectives = dbDriver
            .GetUsingDirectivesForModels()
            .OrderBy(x => x)
            .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, models);
        root = root.AddCommentOnTop(Consts.AutoGeneratedComment);

        return new File
        {
            Name = $"{ClassName}.cs",
            Contents = root.ToByteString()
        };
    }

    private MemberDeclarationSyntax[] GenerateDataClasses(Dictionary<string, Dictionary<string, Table>> tables)
    {
        return (
            from schemaTables in tables
            from table in schemaTables.Value
            let className = table.Value.Rel.Name.ToModelName(table.Value.Rel.Schema, dbDriver.DefaultSchema)
            select DataClassesGen.Generate(className, null, table.Value.Columns, dbDriver.Options, null)
        ).ToArray();
    }

    private MemberDeclarationSyntax[] GenerateEnums(Dictionary<string, Dictionary<string, Enum>> enums)
    {
        return enums.SelectMany(s =>
        {
            return s.Value.SelectMany(e =>
            {
                var enumName = e.Value.Name.ToModelName(s.Key, dbDriver.DefaultSchema);
                return EnumsGen.Generate(enumName, e.Value.Vals);
            });
        }).ToArray();
    }
}