private static DbManager CreateDbManager(GenTablesOptions options) { switch (options.ConnectionType) { case ConnectionType.MsSql: return(MsSqlDbManager.Create(options)); case ConnectionType.MySql: return(MySqlDbManager.Create(options.ConnectionString)); case ConnectionType.PgSql: return(PgSqlDbManager.Create(options.ConnectionString)); default: throw new SqExpressCodeGenException("Unknown connection type: " + options.ConnectionType); } }
public static async Task RunGenTablesOptions(GenTablesOptions options) { ILogger logger = new DefaultLogger(Console.Out, options.Verbosity); logger.LogMinimal("Table proxy classes generation is running..."); string directory = EnsureDirectory(options.OutputDir, logger, "Output", true); if (string.IsNullOrEmpty(options.ConnectionString)) { throw new SqExpressCodeGenException("Connection string cannot be empty"); } logger.LogNormal("Checking existing code..."); IReadOnlyDictionary <TableRef, ClassDeclarationSyntax> existingCode = ExistingCodeExplorer.FindTableDescriptors(directory); if (logger.IsNormalOrHigher) { logger.LogNormal(existingCode.Count > 0 ? $"Found {existingCode.Count} already existing table descriptor classes." : "No table descriptor classes found."); } var sqlManager = CreateDbManager(options); logger.LogNormal("Connecting to database..."); var connectionTest = await sqlManager.TryOpenConnection(); if (!string.IsNullOrEmpty(connectionTest)) { throw new SqExpressCodeGenException(connectionTest); } logger.LogNormal("Success!"); var tables = await sqlManager.SelectTables(); if (logger.IsNormalOrHigher) { logger.LogNormal(tables.Count > 0 ? $"Found {tables.Count} tables." : "No tables found in the database."); if (logger.IsDetailed) { foreach (var tableModel in tables) { Console.WriteLine($"{tableModel.DbName} ({tableModel.Name})"); foreach (var tableModelColumn in tableModel.Columns) { Console.WriteLine($"- {tableModelColumn.DbName.Name} {tableModelColumn.ColumnType.GetType().Name}{(tableModelColumn.Pk.HasValue ? " (PK)":null)}{(tableModelColumn.Fk != null ? $" (FK: {string.Join(';', tableModelColumn.Fk.Select(f=>f.ToString()))})" : null)}"); } } } } logger.LogNormal("Code generation..."); IReadOnlyDictionary <TableRef, TableModel> tableMap = tables.ToDictionary(t => t.DbName); var tableClassGenerator = new TableClassGenerator(tableMap, options.Namespace, existingCode); foreach (var table in tables) { string filePath = Path.Combine(directory, $"{table.Name}.cs"); if (logger.IsDetailed) { logger.LogDetailed($"{table.DbName} to \"{filePath}\"."); } var text = tableClassGenerator.Generate(table, out var existing).ToFullString(); await File.WriteAllTextAsync(filePath, text); if (logger.IsDetailed) { logger.LogDetailed(existing ? "Existing file updated." : "New file created."); } } var allTablePath = Path.Combine(directory, "AllTables.cs"); if (logger.IsDetailed) { logger.LogDetailed($"AllTables to \"{allTablePath}\"."); } await File.WriteAllTextAsync(allTablePath, TableListClassGenerator.Generate(allTablePath, tables, options.Namespace, options.TableClassPrefix).ToFullString()); logger.LogMinimal("Table proxy classes generation successfully completed!"); }