feat: 工具升级

This commit is contained in:
chenchun
2024-06-05 18:38:43 +08:00
parent 19983283f2
commit 9cc5d76888

View File

@@ -0,0 +1,118 @@
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Yi.Abp.Tool.Commands
{
public class AddModuleCommand : ICommand
{
public List<string> CommandStrs => new List<string> { "add-module" };
public async Task InvokerAsync(Dictionary<string, string> options, string[] args)
{
//只有一个add
if (args.Length <= 1)
{
throw new UserFriendlyException("命令错误add-module命令后必须添加 模块名");
}
//需要添加名称
var moduleName = args[1];
options.TryGetValue("modulePath", out var modulePath);
if (string.IsNullOrEmpty(modulePath))
{
throw new UserFriendlyException("命令错误,添加模块,必须指定模块路径");
}
var slnPath = string.Empty;
options.TryGetValue("s", out var slnPath1);
options.TryGetValue("solution", out var slnPath2);
slnPath = string.IsNullOrEmpty(slnPath1) ? slnPath2 : slnPath1;
if (string.IsNullOrEmpty(slnPath))
{
slnPath = "./";
}
CheckFirstSlnPath(slnPath);
var dotnetSlnCommandPart1 = $"dotnet sln \"{slnPath}\" add \"{modulePath}\\{moduleName}.";
var dotnetSlnCommandPart2 = new List<string>() { "Application", "Application.Contracts", "Domain", "Domain.Shared", "SqlSugarCore" };
var paths = dotnetSlnCommandPart2.Select(x => $@"{modulePath}\{moduleName}." + x).ToArray();
CheckPathExist(paths);
var cmdCommands = dotnetSlnCommandPart2.Select(x => dotnetSlnCommandPart1 + x+"\"").ToArray();
StartCmd(cmdCommands);
await Console.Out.WriteLineAsync("恭喜~模块添加成功!");
return Task.CompletedTask;
}
/// <summary>
/// 获取一个sln解决方案多个将报错
/// </summary>
/// <returns></returns>
private string CheckFirstSlnPath(string slnPath)
{
string[] slnFiles = Directory.GetFiles(slnPath, "*.sln");
if (slnFiles.Length > 1)
{
throw new UserFriendlyException("当前目录包含多个sln解决方案请只保留一个");
}
if (slnFiles.Length == 0)
{
throw new UserFriendlyException("当前目录未找到sln解决方案请检查");
}
return slnFiles[0];
}
/// <summary>
/// 执行cmd命令
/// </summary>
/// <param name="cmdCommands"></param>
private void StartCmd(params string[] cmdCommands)
{
ProcessStartInfo psi = new ProcessStartInfo
{
FileName = "cmd.exe",
Arguments = $"/c {string.Join("&", cmdCommands)}",
RedirectStandardInput = true,
RedirectStandardOutput = true,
RedirectStandardError = true,
CreateNoWindow = true,
UseShellExecute = false
};
Process proc = new Process
{
StartInfo = psi
};
proc.Start();
string output = proc.StandardOutput.ReadToEnd();
Console.WriteLine(output);
proc.WaitForExit();
}
/// <summary>
/// 检查路径
/// </summary>
/// <param name="paths"></param>
/// <exception cref="UserFriendlyException"></exception>
private void CheckPathExist(string[] paths)
{
foreach (string path in paths)
{
if (!Directory.Exists(path))
{
throw new UserFriendlyException($"路径错误,请检查你的路径,找不到:{path}");
}
}
}
}
}