HTCloud/HT.Cloud.Web/Areas/SystemManage/Controllers/CodeGeneratorController.cs

using Microsoft.AspNetCore.Mvc;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Threading.Tasks;
using System.Web;
using HT.Cloud.Code;
using HT.Cloud.CodeGenerator;
using HT.Cloud.Domain;
using HT.Cloud.Service;

namespace HT.Cloud.Web.Areas.SystemManage.Controllers
{
	[Area("SystemManage")]
	public class CodeGeneratorController : BaseController
	{
		public DatabaseTableService _service { get; set; }
		private readonly ISqlSugarClient _context;

		public CodeGeneratorController(ISqlSugarClient context)
		{
			_context = context;
		}

		#region 视图功能

		[HttpGet]
		public virtual ActionResult AddForm()
		{
			return View();
		}

		[HttpGet]
		public virtual ActionResult RuleForm()
		{
			return View();
		}

		[HttpGet]
		public virtual ActionResult EntityCode()
		{
			return View();
		}

		#endregion 视图功能

		#region 获取数据

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetTablePageListJson(Pagination pagination, string keyword, string dbNumber)
		{
			//导出全部页使用
			if (pagination.rows == 0 && pagination.page == 0)
			{
				pagination.rows = 99999999;
				pagination.page = 1;
			}
			List<DbTableInfo> data = _service.GetTablePageList(keyword, dbNumber, pagination);
			return Success(pagination.records, data);
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetDbNumberListJson()
		{
			List<dynamic> data = _service.GetDbNumberListJson();
			return Content(data.ToJson());
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetTableJson(string keyword, string dbNumber)
		{
			List<DbTableInfo> data = _service.GetTableList(keyword, dbNumber);
			return Content(data.ToJson());
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetTableSelectJson(string keyword, string dbNumber)
		{
			List<DbTableInfo> data = _service.GetTableList(keyword, dbNumber);
			List<object> list = new List<object>();
			foreach (var item in data)
			{
				list.Add(new { name = item.Name, value = item.Name });
			}
			return Content(list.ToJson());
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetTableFieldJson(string keyValue, string keyword, string dbNumber)
		{
			List<DbColumnInfo> data = _service.GetTableFieldList(keyValue, dbNumber);
			if (!string.IsNullOrEmpty(keyword))
			{
				data = data.Where(a => a.DbColumnName.Contains(keyword)).ToList();
			}
			return Success(data.Count, data);
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetTableFieldSelectJson(string keyValue, string dbNumber)
		{
			List<DbColumnInfo> data = _service.GetTableFieldList(keyValue, dbNumber);
			List<object> list = new List<object>();
			foreach (var item in data)
			{
				list.Add(new { text = item.DbColumnName, id = item.DbColumnName });
			}
			return Content(list.ToJson());
		}

		[HttpGet]
		[HandlerAjaxOnly]
		public ActionResult GetBaseConfigJson(string keyValue, string dbNumber)
		{
			BaseConfigModel data = new BaseConfigModel();

			string tableDescription = string.Empty;
			List<DbColumnInfo> tDataTableField = _service.GetTableFieldList(keyValue, dbNumber);

			var columnList = tDataTableField.Where(p => !BaseField.BaseFieldList.Contains(p.DbColumnName) && !p.IsPrimarykey).Select(p => new { p.DbColumnName, p.ColumnDescription }).ToList();
			List<ColumnField> dic = new List<ColumnField>();
			foreach (var item in columnList)
			{
				ColumnField field = new ColumnField();
				field.field = item.DbColumnName;
				field.title = string.IsNullOrEmpty(item.ColumnDescription) ? item.DbColumnName : item.ColumnDescription;
				field.isFilter = true;
				field.isAotuWidth = false;
				field.isSorted = true;
				field.isShow = true;
				field.width = 100;
				dic.Add(field);
			}
			string serverPath = GlobalContext.HostingEnvironment.ContentRootPath;
			data = new SingleTableTemplate(_context).GetBaseConfig(serverPath, _logService.currentuser.UserName, keyValue, tableDescription, dic);
			return Content(data.ToJson());
		}

		#endregion 获取数据

		#region 提交数据

		[HttpPost]
		[HandlerAjaxOnly]
		[IgnoreAntiforgeryToken]
		public ActionResult CodePreviewJson(BaseConfigModel baseConfig, string dbNumber)
		{
			try
			{
				List<DbColumnInfo> list = _service.GetTableFieldList(baseConfig.TableName, dbNumber);
				SingleTableTemplate template = new SingleTableTemplate(_context);
				string idcolumn = list.FirstOrDefault(a => a.IsPrimarykey == true)?.DbColumnName;
				Dictionary<string, string> dic = new Dictionary<string, string>();
				baseConfig.PageIndex.ButtonList = Extensions.removeNull(baseConfig.PageIndex.ButtonList);
				baseConfig.PageIndex.ColumnList = baseConfig.PageIndex.ColumnList.Where(a => a.field != "").ToList();
				baseConfig.PageForm.FieldList.Remove("");
				string idType = "string";
				//构造虚拟参数
				foreach (var item in baseConfig.PageIndex.ColumnList)
				{
					if (!list.Where(a => a.DbColumnName == item.field).Any())
					{
						DbColumnInfo temp = new DbColumnInfo();
						temp.IsPrimarykey = false;
						temp.DbColumnName = item.field;
						temp.ColumnDescription = item.title;
						temp.IsIdentity = false;
						temp.IsNullable = true;
						list.Add(temp);
					}
				}
				DataTable dt = DataTableHelper.ListToDataTable(list);  // 用DataTable类型，避免依赖
				var tableinfo = _service.repository.Db.DbMaintenance.GetColumnInfosByTableName(baseConfig.TableName, false);
				string codeEntity = template.BuildEntity(baseConfig, dt, idcolumn);
				string codeService = template.BuildService(baseConfig, dt, idcolumn, idType);
				string codeController = template.BuildController(baseConfig, idcolumn, idType);
				string codeIndex = template.BuildIndex(baseConfig, idcolumn);
				string codeForm = template.BuildForm(baseConfig);
				string codeDetails = template.BuildDetails(baseConfig);
				string codeMenu = template.BuildMenu(baseConfig, idcolumn);
				var json = new
				{
					CodeEntity = HttpUtility.HtmlEncode(codeEntity),
					CodeService = HttpUtility.HtmlEncode(codeService),
					CodeController = HttpUtility.HtmlEncode(codeController),
					CodeIndex = HttpUtility.HtmlEncode(codeIndex),
					CodeForm = HttpUtility.HtmlEncode(codeForm),
					CodeDetails = HttpUtility.HtmlEncode(codeDetails),
					CodeMenu = HttpUtility.HtmlEncode(codeMenu)
				};
				return Success("操作成功", json);
			}
			catch (System.Exception ex)
			{
				return Error(ex.Message);
			}
		}

		[HttpPost]
		[HandlerAjaxOnly]
		public async Task<ActionResult> CodeGenerateJson(BaseConfigModel baseConfig, string Code)
		{
			try
			{
				if (!GlobalContext.SystemConfig.Debug)
				{
					throw new System.Exception("请在本地开发模式时使用代码生成");
				}
				else
				{
					SingleTableTemplate template = new SingleTableTemplate(_context);
					await template.CreateCode(baseConfig, HttpUtility.UrlDecode(Code));
				}
				return await Success("操作成功。", "", "");
			}
			catch (System.Exception ex)
			{
				return await Error(ex.Message, "", "");
			}
		}

		[HttpPost]
		[HandlerAjaxOnly]
		public async Task<ActionResult> EntityCodeGenerateJson(BaseConfigModel baseConfig, string keyValue, string dbNumber)
		{
			try
			{
				if (!GlobalContext.SystemConfig.Debug)
				{
					throw new System.Exception("请在本地开发模式时使用代码生成");
				}
				else
				{
					List<DbColumnInfo> list = _service.GetTableFieldList(baseConfig.TableName, dbNumber);
					SingleTableTemplate template = new SingleTableTemplate(_context);
					DataTable dt = DataTableHelper.ListToDataTable(list);  // 用DataTable类型，避免依赖
					string idcolumn = list.FirstOrDefault(a => a.IsPrimarykey == true)?.DbColumnName;
					string codeEntity = template.BuildEntity(baseConfig, dt, idcolumn, true);
					await template.EntityCreateCode(baseConfig, codeEntity);
				}
				return await Success("操作成功。", "", "");
			}
			catch (System.Exception ex)
			{
				return await Error(ex.Message, "", "");
			}
		}

		#endregion 提交数据
	}
}