// WTM默认页面 Wtm buidin page using Microsoft.EntityFrameworkCore; using System; using System.Collections.Generic; using System.ComponentModel.DataAnnotations; using System.Linq; using WalkingTec.Mvvm.Core; using WalkingTec.Mvvm.Core.Extensions; namespace WalkingTec.Mvvm.Mvc.Admin.ViewModels.FrameworkMenuVMs { public class FrameworkMenuVM2 : BaseCRUDVM { [Display(Name = "_Admin.Action")] public List SelectedActionIDs { get; set; } [Display(Name = "_Admin.Module")] public string SelectedModule { get; set; } [Display(Name = "_Admin.AllowedRole")] public List SelectedRolesCodes { get; set; } public FrameworkMenuVM2() { SelectedRolesCodes = new List(); } protected override void InitVM() { SelectedRolesCodes.AddRange(DC.Set().Where(x => x.MenuItemId == Entity.ID && x.RoleCode != null && x.Allowed == true).Select(x => x.RoleCode).ToList()); var data = DC.Set().ToList(); var topMenu = data.Where(x => x.ParentId == null).ToList().FlatTree(x => x.DisplayOrder); var modules = Wtm.GlobaInfo.AllModule; if (Entity.Url != null && Entity.IsInside == true) { SelectedModule = modules.Where(x => x.IsApi == true && (x.FullName == Entity.ClassName)).FirstOrDefault()?.FullName; if (SelectedModule != null) { var urls = modules.Where(x => x.FullName == SelectedModule && x.IsApi == true).SelectMany(x => x.Actions).Where(x => x.IgnorePrivillege == false).Select(x => x.Url).ToList(); SelectedActionIDs = DC.Set().Where(x => urls.Contains(x.Url) && x.IsInside == true && x.FolderOnly == false).Select(x => x.MethodName).ToList(); } else { SelectedModule = Entity.Url; } } } public override void Validate() { if (Entity.IsInside == true && Entity.FolderOnly == false) { if (string.IsNullOrEmpty(SelectedModule) == true) { MSD.AddModelError("SelectedModule", Localizer["Validate.{0}required", Localizer["_Admin.Module"]]); } else { var modules = Wtm.GlobaInfo.AllModule; var test = DC.Set().Where(x => x.Url != null && x.Url.ToLower() == this.Entity.Url.ToLower() && x.ID != Entity.ID).FirstOrDefault(); if (test != null) { MSD.AddModelError(" error", Localizer["_Admin.ModuleHasSet"]); } } } base.Validate(); } public override void DoEdit(bool updateAllFields = false) { if (Entity.IsInside == false) { if (Entity.Url != null && Entity.Url != "") { if (string.IsNullOrEmpty(Entity.Domain) == true) { if (Entity.Url.ToLower().StartsWith("http://") == false && Entity.Url.ToLower().StartsWith("https://") == false && Entity.Url.StartsWith("@") == false) { Entity.Url = "http://" + Entity.Url; } } else { if (Entity.Url.StartsWith("/") == false) { Entity.Url = "/" + Entity.Url; } } Entity.Url = Entity.Url.TrimEnd('/'); } } else { if (string.IsNullOrEmpty(SelectedModule) == false && Entity.FolderOnly == false) { var modules = Wtm.GlobaInfo.AllModule; var ndc = DC.ReCreate(); var actionsInDB = DC.Set().AsNoTracking().Where(x => x.ParentId == Entity.ID).ToList(); var mo = modules.Where(x => x.FullName == this.SelectedModule && x.IsApi == true).FirstOrDefault(); if (mo != null) { Entity.ModuleName = mo.ModuleName; Entity.ClassName = mo.FullName; Entity.MethodName = null; var otherActions = mo.Actions; int order = 1; Entity.Children = new List(); foreach (var action in otherActions) { if (SelectedActionIDs != null && SelectedActionIDs.Contains(action.MethodName)) { Guid aid = action.ID; var adb = actionsInDB.Where(x => x.Url.ToLower() == action.Url.ToLower()).FirstOrDefault(); if (adb != null) { aid = adb.ID; } FrameworkMenu menu = new FrameworkMenu(); menu.FolderOnly = false; menu.IsPublic = Entity.IsPublic; menu.Parent = Entity; menu.ShowOnMenu = false; menu.DisplayOrder = order++; menu.Privileges = new List(); menu.IsInside = true; menu.Domain = Entity.Domain; menu.PageName = action.ActionDes?.Description ?? action.ActionName; menu.ModuleName = action.Module.ModuleName; menu.ActionName = action.ActionDes?.Description ?? action.ActionName; menu.Url = action.Url; menu.ClassName = action.Module.FullName; menu.MethodName = action.MethodName; menu.ID = aid; Entity.Children.Add(menu); } } } else { Entity.ModuleName = ""; Entity.ClassName = ""; Entity.MethodName = ""; } } else { Entity.Children = null; Entity.Url = null; } } if (FC.ContainsKey("Entity.Children") == false) { FC.Add("Entity.Children", 0); FC.Add("Entity.Children[0].IsPublic", 0); FC.Add("Entity.Children[0].PageName", 0); FC.Add("Entity.Children[0].ModuleName", 0); FC.Add("Entity.Children[0].ActionName", 0); FC.Add("Entity.Children[0].ClassName", 0); FC.Add("Entity.Children[0].MethodName", 0); FC.Add("Entity.Children[0].Url", 0); } base.DoEdit(updateAllFields); List guids = new List(); guids.Add(Entity.ID); if (Entity.Children != null) { guids.AddRange(Entity.Children?.Select(x => x.ID).ToList()); } AddPrivilege(guids); } public override void DoAdd() { if (Entity.IsInside == false) { if (Entity.Url != null && Entity.Url != "") { if (string.IsNullOrEmpty(Entity.Domain) == true) { if (Entity.Url.ToLower().StartsWith("http://") == false && Entity.Url.ToLower().StartsWith("https://") == false && Entity.Url.StartsWith("@") == false) { Entity.Url = "http://" + Entity.Url; } } else { if (Entity.Url.StartsWith("/") == false) { Entity.Url = "/" + Entity.Url; } } Entity.Url = Entity.Url.TrimEnd('/'); } } else { if (string.IsNullOrEmpty(SelectedModule) == false && Entity.FolderOnly == false) { var modules = Wtm.GlobaInfo.AllModule; var mo = modules.Where(x => x.FullName == this.SelectedModule && x.IsApi == true).FirstOrDefault(); if (mo != null) { Entity.ModuleName = mo.ModuleName; Entity.ClassName = mo.FullName; Entity.MethodName = null; var otherActions = mo.Actions; int order = 1; Entity.Children = new List(); foreach (var action in otherActions) { if (SelectedActionIDs != null && SelectedActionIDs.Contains(action.MethodName)) { FrameworkMenu menu = new FrameworkMenu(); menu.FolderOnly = false; menu.IsPublic = Entity.IsPublic; menu.Parent = Entity; menu.ShowOnMenu = false; menu.DisplayOrder = order++; menu.Privileges = new List(); menu.IsInside = true; menu.Domain = Entity.Domain; menu.PageName = action.ActionDes?.Description ?? action.ActionName; menu.ModuleName = action.Module.ModuleName; menu.ActionName = action.ActionDes?.Description ?? action.ActionName; menu.Url = action.Url; menu.ClassName = action.Module.FullName; menu.MethodName = action.MethodName; Entity.Children.Add(menu); } } } } else { Entity.Children = null; Entity.Url = null; } } base.DoAdd(); List guids = new List(); guids.Add(Entity.ID); if (Entity.Children != null) { guids.AddRange(Entity.Children?.Select(x => x.ID).ToList()); } AddPrivilege(guids); } public void AddPrivilege(List menuids) { var admin = DC.Set().Where(x => x.RoleCode == "001").SingleOrDefault(); if (admin != null && SelectedRolesCodes.Contains(admin.RoleCode) == false) { SelectedRolesCodes.Add(admin.RoleCode); } foreach (var menuid in menuids) { if (SelectedRolesCodes != null) { foreach (var code in SelectedRolesCodes) { FunctionPrivilege fp = new FunctionPrivilege(); fp.MenuItemId = menuid; fp.RoleCode = code; fp.Allowed = true; DC.Set().Add(fp); } } } DC.SaveChanges(); } public override void DoDelete() { try { DC.CascadeDelete(Entity); DC.SaveChanges(); } catch { } } } }