using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using System.Xml.Linq; using RazorEngine; using RazorEngine.Templating; namespace SugarCodeGeneration.Codes { /// /// 生成所需要的代码 /// public class Methods { public static string GetCurrentProjectPath { get { return Environment.CurrentDirectory.Replace(@"\bin\Debug", ""); } } public static string GetSlnPath { get { var path = Directory.GetParent(GetCurrentProjectPath).FullName; return path; } } public static void AddCsproj(string classPath, string projectName) { CreateProject(projectName); var classDirectory = Methods.GetSlnPath + "\\" + projectName + "\\" + classPath.TrimStart('\\'); if (FileHelper.IsExistDirectory(classDirectory) == false) { FileHelper.CreateDirectory(classDirectory); } var files = Directory.GetFiles(classDirectory).ToList().Select(it => classPath + "\\" + Path.GetFileName(it)); var xmlPath = GetSlnPath + @"\" + projectName + @"\" + projectName + ".csproj"; var xml = File.ReadAllText(xmlPath, System.Text.Encoding.UTF8); var firstLine = System.IO.File.ReadLines(xmlPath, System.Text.Encoding.UTF8).First(); var newXml = xml.Replace(firstLine, "").TrimStart('\r').TrimStart('\n'); XDocument xe = XDocument.Load(xmlPath); var itemGroup = xe.Root.Elements().Where(it => it.Name.LocalName == "ItemGroup" && it.Elements().Any(y => y.Name.LocalName == "Compile")).First(); var compieList = itemGroup.Elements().ToList(); var noAddFiles = files.Where(it => !compieList.Any(f => it.Equals(f.Attribute("Include").Value, StringComparison.CurrentCultureIgnoreCase))).ToList(); if (noAddFiles.Any()) { foreach (var item in noAddFiles) { var addItem = new XElement("Compile", new XAttribute("Include", item)); itemGroup.AddFirst(addItem); } } newXml = xe.ToString().Replace("xmlns=\"\"", ""); xe = XDocument.Parse(newXml); xe.Save(xmlPath); } public static void CreateBLL(string templatePath, string savePath, List tables) { string template = System.IO.File.ReadAllText(templatePath); //从文件中读出模板内容 string templateKey = "bll"; //取个名字 foreach (var item in tables) { BLLParameter model = new BLLParameter() { Name = item }; var result = Engine.Razor.RunCompile(template, templateKey, model.GetType(), model); var cp = savePath + "\\" + item + "Manager.cs"; if (FileHelper.IsExistFile(cp) == false) FileHelper.CreateFile(cp, result, System.Text.Encoding.UTF8); } } public static void CreateDbContext(string templatePath, string savePath, object model) { string template = System.IO.File.ReadAllText(templatePath); //从文件中读出模板内容 string templateKey = "dbcontext"; //取个名字 var result = Engine.Razor.RunCompile(template, templateKey, model.GetType(), model); FileHelper.CreateFile(savePath, result, System.Text.Encoding.UTF8); } public static void CreateProject(string name) { var templatePath = GetCurrentProjectPath + "/Template/Project.txt"; string project = System.IO.File.ReadAllText(templatePath).Replace("@pid", Guid.NewGuid().ToString()); //从文件中读出模板内容 var projectPath = GetSlnPath + "\\" + name + "\\" + name + ".csproj"; var projectDic = GetSlnPath + "\\" + name + "\\"; var binDic = GetSlnPath + "\\" + name + "\\bin"; if (!FileHelper.IsExistFile(projectPath)) { if (!FileHelper.IsExistDirectory(projectDic)) { FileHelper.CreateDirectory(projectDic); } if (!FileHelper.IsExistDirectory(binDic)) { FileHelper.CreateDirectory(binDic); } FileHelper.CreateFile(projectPath,project,System.Text.Encoding.UTF8); } } } }