feat: 完成tool搭建

2024-11-05 22:12:30 +08:00
parent c944bd3b0e
commit d83db53acb
8 changed files with 352 additions and 336 deletions
--- a/Yi.Abp.Net8/tool/Yi.Abp.Tool/Commands/NewCommand.cs
+++ b/Yi.Abp.Net8/tool/Yi.Abp.Tool/Commands/NewCommand.cs
@@ -1,84 +1,90 @@
-// using System;
-// using System.Collections.Generic;
-// using System.IO.Compression;
-// using System.Linq;
-// using System.Text;
-// using System.Threading.Tasks;
-// using Microsoft.AspNetCore.Mvc;
-// using Yi.Abp.Tool.Application.Contracts;
-// using Yi.Abp.Tool.Application.Contracts.Dtos;
-//
-// namespace Yi.Abp.Tool.Commands
-// {
-//     public class NewCommand : ICommand
-//     {
-//         private readonly ITemplateGenService _templateGenService;
-//         public NewCommand(ITemplateGenService templateGenService)
-//         {
-//             _templateGenService = templateGenService;
-//         }
-//
-//         public List<string> CommandStrs => new List<string>() { "new" };
-//
-//
-//         public async Task InvokerAsync(Dictionary<string, string> options, string[] args)
-//         {
-//             var id = Guid.NewGuid().ToString("N");
-//             //只有一个new
-//             if (args.Length <= 1)
-//             {
-//                 throw new UserFriendlyException("命令错误，new命令后必须添加 名称");
-//             }
-//             string name = args[1];
-//
-//             #region 处理生成类型
-//
-//             options.TryGetValue("t", out var templateType);
-//             var zipPath = string.Empty;
-//             byte[] fileByteArray;
-//             if (templateType == "module")
-//             {
-//                 //代表模块生成
-//                 fileByteArray = await _templateGenService.CreateModuleAsync(new TemplateGenCreateInputDto
-//                 {
-//                     Name = name,
-//                 });
-//             }
-//             else
-//             {
-//                 //代表模块生成
-//                 fileByteArray = await _templateGenService.CreateProjectAsync(new TemplateGenCreateInputDto
-//                 {
-//                     Name = name,
-//                 });
-//             }
-//             zipPath = $"{id}.zip";
-//             await File.WriteAllBytesAsync(zipPath, fileByteArray);
-//
-//             #endregion
-//
-//             #region 处理解决方案文件夹
-//             //默认是当前目录
-//             var unzipDirPath = "./";
-//             //如果创建解决方案文件夹
-//             if (options.TryGetValue("csf", out _))
-//             {
-//                 var moduleName = name.ToLower().Replace(".", "-");
-//
-//                 if (Directory.Exists(moduleName))
-//                 {
-//                     throw new UserFriendlyException($"文件夹[{moduleName}]已存在，请删除后重试");
-//                 }
-//                 Directory.CreateDirectory(moduleName);
-//                 unzipDirPath = moduleName;
-//             }
-//             #endregion
-//             ZipFile.ExtractToDirectory(zipPath, unzipDirPath);
-//             //创建压缩包后删除临时目录
-//             File.Delete(zipPath);
-//
-//
-//             await Console.Out.WriteLineAsync("恭喜~模块已生成！");
-//         }
-//     }
-// }
+using System;
+using System.Collections.Generic;
+using System.IO.Compression;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.Extensions.CommandLineUtils;
+using Yi.Abp.Tool.Application.Contracts;
+using Yi.Abp.Tool.Application.Contracts.Dtos;
+
+namespace Yi.Abp.Tool.Commands
+{
+    public class NewCommand : ICommand
+    {
+        private readonly ITemplateGenService _templateGenService;
+
+        public NewCommand(ITemplateGenService templateGenService)
+        {
+            _templateGenService = templateGenService;
+        }
+
+
+        public string Command => "new";
+        public string? Description => "创建一个模板";
+
+        public void CommandLineApplication(CommandLineApplication application)
+        {
+            var templateTypeOption = application.Option("-t", "模板类型", CommandOptionType.SingleValue);
+            var csfOption = application.Option("csf", "是否创建解决方案", CommandOptionType.SingleValue);
+            var moduleNameArgument = application.Argument("moduleName", "模块名", null);
+
+
+            #region 处理生成类型
+
+            var id = Guid.NewGuid().ToString("N");
+            var zipPath = string.Empty;
+            byte[] fileByteArray;
+
+            var templateType = templateTypeOption.HasValue() ? templateTypeOption.Value() : "module";
+            if (templateType == "module")
+            {
+                //代表模块生成
+                fileByteArray = (_templateGenService.CreateModuleAsync(new TemplateGenCreateInputDto
+                {
+                    Name = moduleNameArgument.Value,
+                }).Result);
+            }
+            else
+            {
+                //代表模块生成
+                fileByteArray = _templateGenService.CreateProjectAsync(new TemplateGenCreateInputDto
+                {
+                    Name = moduleNameArgument.Value,
+                }).Result;
+            }
+
+            zipPath = $"{id}.zip";
+            File.WriteAllBytes(zipPath, fileByteArray);
+
+            #endregion
+
+            #region 处理解决方案文件夹
+
+            //默认是当前目录
+            var unzipDirPath = "./";
+            //如果创建解决方案文件夹
+            if (csfOption.HasValue())
+            {
+                var moduleName = moduleNameArgument.Value.ToLower().Replace(".", "-");
+
+                if (Directory.Exists(moduleName))
+                {
+                    throw new UserFriendlyException($"文件夹[{moduleName}]已存在，请删除后重试");
+                }
+
+                Directory.CreateDirectory(moduleName);
+                unzipDirPath = moduleName;
+            }
+
+            #endregion
+
+            ZipFile.ExtractToDirectory(zipPath, unzipDirPath);
+            //创建压缩包后删除临时目录
+            File.Delete(zipPath);
+
+            Console.WriteLine("恭喜~模块已生成！");
+        }
+    }
+}