System.CommandLine版CSRebot

Posted 2022-07-04 dotNET跨平台

之前自己实现过一个CSRebot命令行工具，现在用System.CommandLine来实现，就规范和省事多了，虽然System.CommandLine还没有正式发布，但它的实现思路还是很不错的。

下面的代码只简单实现了MSSQL库生成C#体类的功能，其他库或其他语言的实体类生成，大家可以自己实现一下，以熟悉System.CommandLine的使用。

直接上代码了：

using Microsoft.Data.SqlClient;
using System.CommandLine;
using System.CommandLine.Binding;
using System.Data;
using System.Text;
using System;


//创建根命令
var rootCommand = new RootCommand("这是一款C#开发辅助工具，CSRebot");
rootCommand.SetHandler(() =>

    Console.WriteLine("老桂欢迎您使用CSRebot!");
    Console.ForegroundColor = ConsoleColor.Green;
    Console.WriteLine("\\r\\n   ___________ ____       __          __ \\r\\n  / ____/ ___// __ \\\\___  / /_  ____  / /_\\r\\n / /    \\\\__ \\\\/ /_/ / _ \\\\/ __ \\\\/ __ \\\\/ __/\\r\\n/ /___ ___/ / _, _/  __/ /_/ / /_/ / /_  \\r\\n\\\\____//____/_/ |_|\\\\___/_.___/\\\\____/\\\\__/  \\r\\n                                         \\r\\n");
    Console.ResetColor();
    Console.WriteLine("help command：csrebot -h");
    Console.WriteLine();
);
//创建子命令 show
var dbtoCommand = new Command("dbto", "从数据库生成");


#region 选项 language
//创建子命令选项 language 别名 lan
var languageOption = new Option<string>(name: "--language", description: "从数据库生成实体类的语言")

    IsRequired = true,
.FromAmong("c#", "C#", "csharp", "CSharp", "go", "GO", "java", "JAVA");
languageOption.AddAlias("-lan");
languageOption.AddAlias("-l");
//添加language选项到dbto命令中
dbtoCommand.AddOption(languageOption);
#endregion


#region connestionString
//创建子命令选项 connestionString 别名 constr
var connectionStringOption = new Option<string>(name: "--connectionstring", description: "数据库连接字符串")

    IsRequired = true,
;
connectionStringOption.AddAlias("-constr");


//添加language选项到dbto命令中
dbtoCommand.AddOption(connectionStringOption);
#endregion


#region dbtype
//创建子命令选项 connestionString 别名 constr
var dbTypeOption = new Option<string>(name: "--dbtype", description: "数据库类型")

    IsRequired = true,
.FromAmong("mssql", "mysql", "pgsql"); ;
dbTypeOption.AddAlias("-dbt");
dbTypeOption.AddAlias("-t");
//添加language选项到dbto命令中
dbtoCommand.AddOption(dbTypeOption);
#endregion


//设置命令dbto执行的动作，这是带上language参数
dbtoCommand.SetHandler(async (string language, string connectionstring, string dbType) =>

    switch (language.ToLower())
    
        case "c#":
        case "csharp":
            await DBToCSharpAsync(connectionstring, dbType);
            break;
        case "go":
            break;
        case "java":
            break;
        default:
            break;


    
, languageOption, connectionStringOption, dbTypeOption);


//添加命令dbto到 根命令中
rootCommand.Add(dbtoCommand);
await rootCommand.InvokeAsync(args);




static async Task DBToCSharpAsync(string connectionString, string dbType)

    switch (dbType.ToLower())
    
        case "mssql":
            await MSSQLToCSharpAsync(connectionString);
            break;


        default:
            break;
    



static async Task MSSQLToCSharpAsync(string connectionString)

    using var con = new SqlConnection(connectionString);
    using var cmd = new SqlCommand("select name from sysobjects where xtype='U'", con);
    await con.OpenAsync();
    using var reader = await cmd.ExecuteReaderAsync();
    var tableNames = new List<string>();
    while (reader.Read())
    
        tableNames.Add(reader.GetString(0));
    
    await reader.CloseAsync();


    var dbPath = $"Environment.CurrentDirectory\\\\con.Database";
    if (!Directory.Exists(dbPath))
    
        Directory.CreateDirectory(dbPath);
    
    foreach (var tableName in tableNames)
    
        using var fileCmd = new SqlCommand("SELECT syscolumns.name,systypes.name as typename FROM syscolumns, systypes WHERE syscolumns.xusertype = systypes.xusertype AND syscolumns.id = object_id(@tablename)", con);
        fileCmd.Parameters.Add("tablename", SqlDbType.VarChar).Value = tableName;
        using var fileReader = await fileCmd.ExecuteReaderAsync();
        var tablefieldses = new List<dynamic>();
        while (fileReader.Read())
        
            tablefieldses.Add(new  name = fileReader.GetString(0), typename = fileReader.GetString(1) );
        
        fileReader.Close();
        await fileReader.DisposeAsync();


        var csBuilder = new StringBuilder();
        csBuilder.AppendLine($"public class tableName");
        csBuilder.AppendLine("");
        foreach (var tablefields in tablefieldses)
        
            var typeName = TypeMap.MSSQLToCSharp[tablefields.typename];
            var name = tablefields.name;
            csBuilder.AppendLine($"   public typeName nameget;set;");
        
        csBuilder.AppendLine("");
        File.WriteAllText($"dbPath\\\\tableName.cs", csBuilder.ToString(), Encoding.UTF8);
    



static class TypeMap

    internal static Dictionary<string, string> MSSQLToCSharp => new Dictionary<string, string>
    
        "bigint","long",
        "binary","byte[]",
        "bit","bool",
        "char","string",
        "date","DateTime",
        "datetime","DateTime",
        "datetime2","DateTime",
        "datetimeoffset","DateTimeOffset" ,
        "decimal","decimal" ,
        "float","double" ,
        "image","byte[]" ,
        "int","int" ,
        "money","decimal" ,
        "nchar","string" ,
        "ntext","string" ,
        "numeric","decimal" ,
        "nvarchar","string" ,
        "real","float" ,
        "rowversion","byte[]" ,
        "smalldatetime","DateTime" ,
        "smallint","short" ,
        "smallmoney","decimal" ,
        "sql_variant","object",
        "text","string" ,
        "time","TimeSpan" ,
        "timestamp","byte[]" ,
        "tinyint","byte" ,
        "uniqueidentifier","Guid" ,
        "varbinary","byte[]" ,
        "varchar","string" ,
    ;


    internal static Dictionary<string, string> MySQLToCSharp => new Dictionary<string, string>
    
        "","" 
    ;
    internal static Dictionary<string, string> PgSQLToCSharp => new Dictionary<string, string>
    
        "","" 
    ;

安装CSRebot

完成后，选中项目并打包，然后进入项目所有的文件夹执行工具安装命令

dotnet tool install -g --add-source ./nupkg CSRebot

测试

csrebot dbto -l C# -constr 'server=localhost;database=库名;uid=用户名;pwd=密码;encrypt=true;trustservercertificate=true' -t mssql