System.CommandLine版CSRebot

之前自己实现过一个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} {name}{
    {get;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