287 lines
12 KiB
C#
287 lines
12 KiB
C#
using CloudBuilder.CodeGenerator.Config;
|
||
using CloudBuilder.CodeGenerator.Data;
|
||
using CloudBuilder.CodeGenerator.Utility;
|
||
using CloudBuilder.Core.Application.Builders;
|
||
using CloudBuilder.Core.DatabaseAccessor.Entity;
|
||
using CloudBuilder.Core.DependencyInjection.Form;
|
||
using CloudBuilder.Core.Extensions;
|
||
using CloudBuilder.Core.Service;
|
||
using CloudBuilder.Core.Utility;
|
||
using Krypton.Toolkit;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.VisualStudio.OLE.Interop;
|
||
using RazorEngineCore;
|
||
using System.Data;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
using System.Windows.Forms.Design;
|
||
|
||
namespace CloudBuilder.CodeGenerator
|
||
{
|
||
public partial class OperationUserControl : UserControl, IUserControl
|
||
{
|
||
private readonly IApplicationService app;
|
||
private readonly TableControl tableControl;
|
||
private ConfigUserControl configUserControl;
|
||
private readonly TemplateUserControl templateUserControl;
|
||
private readonly OutputUserControl outputUserControl;
|
||
private ToolInitConfig toolInitConfig;
|
||
public OperationUserControl(IApplicationService app,
|
||
[FromKeyedServices("TableControl")] TableControl tableControl,
|
||
[FromKeyedServices("TemplateUserControl")] TemplateUserControl templateUserControl,
|
||
[FromKeyedServices("OutputUserControl")] OutputUserControl outputUserControl)
|
||
{
|
||
InitializeComponent();
|
||
this.app = app;
|
||
this.tableControl = tableControl;
|
||
|
||
this.templateUserControl = templateUserControl;
|
||
this.outputUserControl = outputUserControl;
|
||
InitUserControl();
|
||
}
|
||
|
||
public void InitUserControl()
|
||
{
|
||
assemblyComboBox.DisplayMember = "Text";
|
||
assemblyComboBox.ValueMember = "Value";
|
||
|
||
toolInitConfig = ToolInitConfig.GetInstance();
|
||
|
||
List<ComboBoxItem> l = new List<ComboBoxItem>();
|
||
List<string> types = new List<string>();
|
||
foreach (NamespaceObject nso in toolInitConfig.NamespaceObjects)
|
||
{
|
||
if (l.Select(x => x.Text).Contains(nso.Assembly)) continue;
|
||
l.Add(CreateNewComboBoxItem(nso.Assembly, nso.Assembly));
|
||
}
|
||
|
||
DataTable dtbl = DataTableConvertor.ToDataTable<ComboBoxItem>(l);
|
||
assemblyComboBox.DataSource = dtbl;
|
||
assemblyComboBox.SelectedValue = toolInitConfig.CurrentAssembly;
|
||
|
||
InitkryptonCheckedListBox1(toolInitConfig.CurrentAssembly);
|
||
}
|
||
|
||
private void InitkryptonCheckedListBox1(string currentAssembly)
|
||
{
|
||
if (string.IsNullOrEmpty(currentAssembly)) return;
|
||
int i = 0;
|
||
kryptonCheckedListBox1.Items.Clear();
|
||
foreach (NamespaceObject nso in toolInitConfig.NamespaceObjects.Where(x => x.Assembly == currentAssembly).ToList())
|
||
{
|
||
if (nso.TemplateType.Contains("Entity") && nso.TemplateType != "Entity") continue;
|
||
|
||
if (nso.TemplateType.Contains("TypeScript")) continue;
|
||
|
||
kryptonCheckedListBox1.Items.Add(CreateNewItem(nso.TemplateType));
|
||
|
||
if (!nso.TemplateType.Contains("Entity"))
|
||
{
|
||
i++;
|
||
continue;
|
||
}
|
||
|
||
kryptonCheckedListBox1.SetItemCheckState(i++, CheckState.Checked);
|
||
}
|
||
}
|
||
|
||
private void outputFileBut_Click(object sender, EventArgs e)
|
||
{
|
||
ModelData model = GetModelData();
|
||
if (model.sysTableSchemas == null || model.sysTableSchemas.Length == 0)
|
||
{
|
||
MessageBox.Show("请选择要生成的数据表", "提示");
|
||
return;
|
||
}
|
||
|
||
for (var i = 0; i < kryptonCheckedListBox1.Items.Count; i++)
|
||
{
|
||
if (kryptonCheckedListBox1.GetItemCheckState(i) == CheckState.Checked)
|
||
{
|
||
KryptonListItem a = (KryptonListItem)kryptonCheckedListBox1.Items[i];
|
||
|
||
WriteAllText(model, a.ShortText);
|
||
}
|
||
}
|
||
|
||
MessageBox.Show("生成成功", "提示");
|
||
}
|
||
|
||
private void WriteAllText(ModelData model, string templateType)
|
||
{
|
||
NamespaceObject namespaceObject = toolInitConfig.NamespaceObjects.Where(x => x.Assembly == assemblyComboBox.Text.Trim() && x.TemplateType == templateType).FirstOrDefault();
|
||
|
||
if (namespaceObject == null) return;
|
||
|
||
string fileName = Path.Combine(toolInitConfig.TemplatePath, namespaceObject.TemplateFileName);
|
||
if (!File.Exists(fileName)) return;
|
||
|
||
if (!Path.Exists(namespaceObject.OutFilePath)) return;
|
||
|
||
var tp = File.ReadAllText(fileName);
|
||
|
||
string outputFileName = Path.Combine(namespaceObject.OutFilePath, namespaceObject.FileNamePrefix + model.sysTableSchemas[0].TableName.StringToProerty() + namespaceObject.FileNameSuffix);
|
||
|
||
if (templateType == "RegistEntity")
|
||
{
|
||
outputFileName = Path.Combine(namespaceObject.OutFilePath, namespaceObject.Assembly.Replace(".", "") + namespaceObject.FileNameSuffix);
|
||
|
||
//文件不存在，先初始化文件
|
||
if (!File.Exists(outputFileName)) File.WriteAllText(outputFileName, RazorEngineHelper.CompiledTemplate(tp, model));
|
||
|
||
string allTxt = File.ReadAllText(outputFileName);
|
||
|
||
tp = File.ReadAllText(Path.Combine(toolInitConfig.TemplatePath, namespaceObject.TemplateFileName.Replace(".cshtml", ".Key.cshtml")));
|
||
var temp = RazorEngineHelper.CompiledTemplate(tp, model);
|
||
|
||
if (allTxt.Contains("<" + model.sysTableSchemas[0].TableName.StringToProerty() + "Entity>()")) return;
|
||
|
||
int end = allTxt.IndexOf("//***END***", StringComparison.OrdinalIgnoreCase);
|
||
|
||
allTxt = allTxt.Substring(0, end) + temp + allTxt.Substring(end, allTxt.Length - end);
|
||
|
||
File.WriteAllText(outputFileName, allTxt);
|
||
return;
|
||
}
|
||
else if (templateType == "Permission")
|
||
{
|
||
outputFileName = Path.Combine(namespaceObject.OutFilePath, namespaceObject.Assembly.Split('.')[namespaceObject.Assembly.Split('.').Length - 1] + namespaceObject.FileNameSuffix);
|
||
|
||
//文件存在不覆盖,插入
|
||
if (File.Exists(outputFileName))
|
||
{
|
||
string allTxt = File.ReadAllText(outputFileName);
|
||
|
||
tp = File.ReadAllText(Path.Combine(toolInitConfig.TemplatePath, namespaceObject.TemplateFileName));
|
||
var temp = RazorEngineHelper.CompiledTemplate(tp, model);
|
||
|
||
if (allTxt.Contains(model.sysTableSchemas[0].TableName.StringToProerty())) return;
|
||
|
||
int end = allTxt.IndexOf("//***END***", StringComparison.OrdinalIgnoreCase);
|
||
|
||
allTxt = allTxt.Substring(0, end) + temp + allTxt.Substring(end, allTxt.Length - end);
|
||
|
||
File.WriteAllText(outputFileName, allTxt);
|
||
return;
|
||
}
|
||
else
|
||
{
|
||
tp = File.ReadAllText(Path.Combine(toolInitConfig.TemplatePath, namespaceObject.TemplateFileName.Replace(".cshtml", ".auto.cshtml")));
|
||
var temp = RazorEngineHelper.CompiledTemplate(tp, model);
|
||
File.WriteAllText(outputFileName, temp);
|
||
}
|
||
}
|
||
//Entity可以直接輸出，其他要文件不存在才生成
|
||
if (templateType == "Entity" || templateType == "IEntityKey" || !File.Exists(outputFileName))
|
||
File.WriteAllText(outputFileName, RazorEngineHelper.CompiledTemplate(tp, model));
|
||
|
||
if ((model.KeyCount >= 0) && templateType == "Entity")
|
||
{
|
||
if (model.KeyCount > 1)
|
||
WriteAllText(model, "IEntityKey");
|
||
WriteAllText(model, "RegistEntity");
|
||
}
|
||
else if (templateType == "Service")
|
||
{
|
||
WriteAllText(model, "Permission");
|
||
}
|
||
}
|
||
|
||
private void templateBut_Click(object sender, EventArgs e)
|
||
{
|
||
try
|
||
{
|
||
ModelData modelData = GetModelData();
|
||
|
||
((OutputUserControl)outputUserControl).contextTxt.Text = RazorEngineHelper.CompiledTemplate(((TemplateUserControl)templateUserControl).contextTxt.Text, modelData);
|
||
|
||
MessageBox.Show("生成成功", "提示");
|
||
|
||
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
MessageBox.Show(ex.Message, "出错");
|
||
}
|
||
}
|
||
|
||
private ModelData GetModelData()
|
||
{
|
||
configUserControl = app.ServiceProvider.GetKeyedService<ConfigUserControl>("ConfigUserControl");
|
||
ModelData modelData = new ModelData();
|
||
modelData.sysTableSchemas = ((TableControl)tableControl).GetDataSource();
|
||
modelData.NamespaceObject = toolInitConfig.NamespaceObjects.Where(x => x.TemplateType == configUserControl.GetTemplateType() && x.Assembly == configUserControl.GetAssembly()).FirstOrDefault();
|
||
if (modelData.sysTableSchemas != null && modelData.sysTableSchemas.Length > 0)
|
||
{
|
||
modelData.KeyCount = modelData.sysTableSchemas.Where(x => x.Pk > 0).Count();
|
||
if (modelData.KeyCount == 1)
|
||
{
|
||
modelData.KeyType = modelData.sysTableSchemas.Where(x => x.Pk > 0).FirstOrDefault().TypeOfName;
|
||
}
|
||
else if (modelData.KeyCount > 1)
|
||
{
|
||
modelData.KeyType = $"{modelData.sysTableSchemas[0].TableName.StringToProerty()}EntityKey";
|
||
}
|
||
else
|
||
{
|
||
modelData.KeyType = "string";
|
||
}
|
||
}
|
||
List<Assembly> assemblies = new List<Assembly>();
|
||
if (toolInitConfig.AssemblyReferences != null)
|
||
{
|
||
foreach (var reference in toolInitConfig.AssemblyReferences)
|
||
{
|
||
if (string.IsNullOrEmpty(reference.AssemblyFileName)) continue;
|
||
|
||
if (!File.Exists(reference.AssemblyFileName)) continue;
|
||
|
||
assemblies.Add(Assembly.LoadFrom(reference.AssemblyFileName));
|
||
}
|
||
}
|
||
if (configUserControl.GetTemplateType() == "Controller" || configUserControl.GetTemplateType() == "TypeScriptService")
|
||
modelData.ClassInfos = RazorEngineHelper.PackIServiceClassInfo(assemblies.Where(x => Path.GetFileNameWithoutExtension(x.ManifestModule.Name) == modelData.NamespaceObject.Assembly).FirstOrDefault());
|
||
else
|
||
if (configUserControl.GetTemplateType() == "TypeScript")
|
||
modelData.ClassInfos = RazorEngineHelper.PackTypeScriptClassInfo(assemblies.Where(x => Path.GetFileNameWithoutExtension(x.ManifestModule.Name) == modelData.NamespaceObject.Assembly).FirstOrDefault());
|
||
|
||
return modelData;
|
||
}
|
||
|
||
|
||
public class ComboBoxItem
|
||
{
|
||
public string Text { get; set; }
|
||
public string Value { get; set; }
|
||
}
|
||
|
||
|
||
private ComboBoxItem CreateNewComboBoxItem(string text, string value)
|
||
{
|
||
ComboBoxItem item = new ComboBoxItem();
|
||
item.Text = text;
|
||
item.Value = value;
|
||
|
||
return item;
|
||
}
|
||
|
||
private object CreateNewItem(string text)
|
||
{
|
||
KryptonListItem item = new KryptonListItem();
|
||
item.ShortText = text;
|
||
|
||
return item;
|
||
}
|
||
|
||
private void assemblyComboBox_SelectionChangeCommitted(object sender, EventArgs e)
|
||
{
|
||
((MainFunctionView)(this.ParentForm)).SetConfigUserControl(assemblyComboBox.SelectedValue);
|
||
}
|
||
|
||
private void assemblyComboBox_SelectedValueChanged(object sender, EventArgs e)
|
||
{
|
||
InitkryptonCheckedListBox1((string)assemblyComboBox.SelectedValue);
|
||
}
|
||
}
|
||
}
|