using System; using System.Linq; using System.Text; using ZR.CodeGenerator.Model; using ZR.Model.System.Generate; namespace ZR.CodeGenerator { /// /// 代码生成模板 /// public class CodeGenerateTemplate { /// /// 表Model属性 /// /// /// public static string GetModelTemplate(GenTableColumn tbColumn) { StringBuilder sbModel = new StringBuilder(); sbModel.AppendLine(" /// "); sbModel.AppendLine($" /// 描述 :{tbColumn.ColumnComment}"); sbModel.AppendLine($" /// 空值 :{!tbColumn.IsRequired}"); sbModel.AppendLine(" /// "); if (tbColumn.IsPk || tbColumn.IsIncrement) { sbModel.AppendLine($" [SqlSugar.SugarColumn(IsPrimaryKey = {tbColumn.IsPk.ToString().ToLower()}, IsIdentity = {tbColumn.IsIncrement.ToString().ToLower()})]"); } sbModel.AppendLine($" public {tbColumn.CsharpType}{(CodeGeneratorTool.GetModelRequired(tbColumn))} {tbColumn.CsharpField} {{ get; set; }}"); return sbModel.ToString(); } /// /// 增改Dto /// /// /// public static string GetDtoProperty(GenTableColumn tbColumn) { string InputDtoContent = ""; if (GenConstants.inputDtoNoField.Any(f => f.ToLower().Contains(tbColumn.CsharpField.ToLower()))) { return InputDtoContent; } else if (tbColumn.IsInsert || tbColumn.IsEdit || tbColumn.IsPk || tbColumn.IsIncrement) { InputDtoContent += $" public {tbColumn.CsharpType}{CodeGeneratorTool.GetModelRequired(tbColumn)} {tbColumn.CsharpField} {{ get; set; }}\r\n"; } return InputDtoContent; } /// /// 查询Dto属性 /// /// /// 替换字符对象 /// public static void GetQueryDtoProperty(GenTableColumn tbColumn, ReplaceDto replaceDto) { string QueryDtoContent = ""; if (tbColumn.IsQuery) { QueryDtoContent += $" public {tbColumn.CsharpType} {tbColumn.CsharpField} {{ get; set; }}\r\n"; //字符串类型表达式 if (tbColumn.CsharpType == GenConstants.TYPE_STRING) { replaceDto.QueryCondition += $" predicate = predicate.AndIF(!string.IsNullOrEmpty(parm.{tbColumn.CsharpField}), {QueryExp(tbColumn.CsharpField, tbColumn.QueryType)};\n"; } //int类型表达式 if (CodeGeneratorTool.IsNumber(tbColumn.CsharpType)) { replaceDto.QueryCondition += $" predicate = predicate.AndIF(parm.{tbColumn.CsharpField} > 0, {QueryExp(tbColumn.CsharpField, tbColumn.QueryType)};\n"; } //时间类型 if (tbColumn.CsharpType == GenConstants.TYPE_DATE) { replaceDto.QueryCondition += $" predicate = predicate.AndIF(parm.BeginTime != null, it => it.{tbColumn.CsharpField} >= parm.BeginTime);\n"; replaceDto.QueryCondition += $" predicate = predicate.AndIF(parm.EndTime != null, it => it.{tbColumn.CsharpField} <= parm.EndTime);\n"; } } replaceDto.QueryProperty += QueryDtoContent; } #region vue 模板 /// /// 生成vuejs模板,目前只有上传文件方法 /// /// /// /// public static void TplVueJsMethod(GenTableColumn dbFieldInfo, ReplaceDto replaceDto) { string columnName = dbFieldInfo.ColumnName; var sb = new StringBuilder(); if (dbFieldInfo.HtmlType.Equals(GenConstants.HTML_IMAGE_UPLOAD)) { sb.AppendLine($" //文件上传成功方法"); sb.AppendLine($" handleUpload{dbFieldInfo.CsharpField}Success(res, file) {{"); sb.AppendLine($" this.form.{columnName} = res.data;"); sb.AppendLine($" // this.form.{columnName} = URL.createObjectURL(file.raw);"); sb.AppendLine($" // this.$refs.upload.clearFiles();"); sb.AppendLine($" }},"); replaceDto.VueBeforeUpload = TplJsBeforeUpload(); replaceDto.VueUploadUrl = TplJsUploadUrl(); } //有下拉框选项初列表查询数据 if ((dbFieldInfo.HtmlType == GenConstants.HTML_SELECT || dbFieldInfo.HtmlType == GenConstants.HTML_RADIO) && !string.IsNullOrEmpty(dbFieldInfo.DictType)) { sb.AppendLine(@$" // {dbFieldInfo.ColumnComment}字典翻译"); sb.AppendLine($" {columnName}Format(row, column) {{"); sb.AppendLine(@$" return this.selectDictLabel(this.{columnName}Options, row.{columnName});"); sb.AppendLine(@" },"); } replaceDto.VueJsMethod += sb.ToString(); } /// /// Vue rules /// /// /// public static string TplFormRules(GenTableColumn dbFieldInfo) { StringBuilder sbRule = new StringBuilder(); //Rule 规则验证 if (!dbFieldInfo.IsPk && !dbFieldInfo.IsIncrement && dbFieldInfo.IsRequired) { sbRule.AppendLine($" {dbFieldInfo.ColumnName}: [{{ required: true, message: '请输入{dbFieldInfo.ColumnComment}', trigger: \"blur\"}}],"); } else if (CodeGeneratorTool.IsNumber(dbFieldInfo.ColumnType) && dbFieldInfo.IsRequired) { sbRule.AppendLine($" {dbFieldInfo.ColumnName}: [{{ type: 'number', message: '{dbFieldInfo.ColumnName}必须为数字值', trigger: \"blur\"}}],"); } return sbRule.ToString(); } /// /// Vue 添加修改表单 /// /// /// public static string TplVueFormContent(GenTableColumn dbFieldInfo) { string columnName = dbFieldInfo.ColumnName; string labelName = CodeGeneratorTool.GetLabelName(dbFieldInfo.ColumnComment, columnName); string labelDisabled = dbFieldInfo.IsPk ? ":disabled=\"true\"" : ""; string placeHolder = dbFieldInfo.IsIncrement ? "" : $"请输入{labelName}"; StringBuilder sb = new StringBuilder(); if (GenConstants.inputDtoNoField.Any(f => f.ToLower().Contains(dbFieldInfo.CsharpField.ToLower()))) { return sb.ToString(); } if (!dbFieldInfo.IsInsert || !dbFieldInfo.IsEdit) { return sb.ToString(); } if (dbFieldInfo.HtmlType == GenConstants.HTML_INPUT_NUMBER) { sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_DATETIME) { //时间 sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_IMAGE_UPLOAD) { //图片 sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_RADIO && !string.IsNullOrEmpty(dbFieldInfo.DictType)) { string value = CodeGeneratorTool.IsNumber(dbFieldInfo.CsharpType) ? "parseInt(item.dictValue)" : "item.dictValue"; sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine($" {{{{item.dictLabel}}}}"); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_RADIO) { sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_TEXTAREA) { sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_EDITOR) { sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_SELECT && !string.IsNullOrEmpty(dbFieldInfo.DictType)) { string value = CodeGeneratorTool.IsNumber(dbFieldInfo.CsharpType) ? "parseInt(item.dictValue)" : "item.dictValue"; sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); } else { string inputNumTxt = CodeGeneratorTool.IsNumber(dbFieldInfo.CsharpType) ? ".number" : ""; sb.AppendLine(" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } return sb.ToString(); } /// /// Vue 查询表单 /// /// /// public static string TplQueryFormHtml(GenTableColumn dbFieldInfo) { StringBuilder sb = new(); string labelName = CodeGeneratorTool.GetLabelName(dbFieldInfo.ColumnComment, dbFieldInfo.ColumnName); if (!dbFieldInfo.IsQuery || dbFieldInfo.HtmlType == GenConstants.HTML_FILE_UPLOAD) return sb.ToString(); if (dbFieldInfo.HtmlType == GenConstants.HTML_DATETIME) { sb.AppendLine(" "); sb.AppendLine(" "); sb.AppendLine(" "); } else if (dbFieldInfo.HtmlType == GenConstants.HTML_SELECT && !string.IsNullOrEmpty(dbFieldInfo.DictType)) { //string value = CodeGeneratorTool.IsNumber(dbFieldInfo.CsharpType) ? "parseInt(item.dictValue)" : "item.dictValue"; sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } else { string inputNumTxt = CodeGeneratorTool.IsNumber(dbFieldInfo.CsharpType) ? ".number" : ""; sb.AppendLine($" "); sb.AppendLine($" "); sb.AppendLine(" "); } return sb.ToString(); } /// /// Vue 查询列表 /// /// /// public static string TplTableColumn(GenTableColumn dbFieldInfo) { string columnName = dbFieldInfo.ColumnName; string label = CodeGeneratorTool.GetLabelName(dbFieldInfo.ColumnComment, columnName); string showToolTip = dbFieldInfo.CsharpType == "string" ? ":show-overflow-tooltip=\"true\"" : ""; string formatter = !string.IsNullOrEmpty(dbFieldInfo.DictType) ? $" :formatter=\"{columnName}Format\"" : ""; StringBuilder sb = new StringBuilder(); if (dbFieldInfo.IsList && dbFieldInfo.HtmlType.Equals(GenConstants.HTML_IMAGE_UPLOAD)) { sb.AppendLine($" "); sb.AppendLine(" "); sb.AppendLine(" "); } //else if (dbFieldInfo.HtmlType.Equals(GenConstants.HTML_RADIO)) //{ // vueViewListContent += $" \r\n"; // vueViewListContent += " \r\n"; // vueViewListContent += " \r\n"; //} else if (dbFieldInfo.IsList) { sb.AppendLine($" "); } return sb.ToString(); } /// /// 文件上传前方法判断 /// /// public static string TplJsBeforeUpload() { StringBuilder sb = new StringBuilder(); sb.AppendLine(@" //文件上传前判断方法"); sb.AppendLine(@" beforeFileUpload(file) {"); sb.AppendLine(@" const isJPG = file.type === ""image/jpeg"";"); sb.AppendLine(@" const isLt2M = file.size / 1024 / 1024 < 2;"); sb.AppendLine(@" if (!isJPG) {"); sb.AppendLine(@" this.msgError(""上传图片只能是 JPG 格式!"");"); sb.AppendLine(@" }"); sb.AppendLine(@" if (!isLt2M) {"); sb.AppendLine(@" this.msgError(""上传图片大小不能超过 2MB!"");"); sb.AppendLine(@" }"); sb.AppendLine(@" return isJPG && isLt2M;"); sb.AppendLine(@" },"); return sb.ToString(); } public static string TplJsUploadUrl() { StringBuilder sb = new StringBuilder(); sb.AppendLine(@" //文件上传前判断方法"); sb.AppendLine(@" uploadUrl: process.env.VUE_APP_BASE_API + ""upload/SaveFile"","); return sb.ToString(); } #endregion public static string QueryExp(string propertyName, string queryType) { if (queryType.Equals("EQ")) { return $"m => m.{ propertyName} == parm.{propertyName})"; } if (queryType.Equals("GTE")) { return $"m => m.{ propertyName} >= parm.{propertyName})"; } if (queryType.Equals("GT")) { return $"m => m.{ propertyName} > parm.{propertyName})"; } if (queryType.Equals("LT")) { return $"m => m.{ propertyName} < parm.{propertyName})"; } if (queryType.Equals("LTE")) { return $"m => m.{ propertyName} <= parm.{propertyName})"; } if (queryType.Equals("NE")) { return $"m => m.{ propertyName} != parm.{propertyName})"; } if (queryType.Equals("LIKE")) { return $"m => m.{ propertyName}.Contains(parm.{propertyName}))"; } return ""; } } }