using System; using System.Collections.Generic; using System.Linq; using System.Reflection; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Caching.Memory; using Oqtane.Extensions; using Oqtane.Models; using Oqtane.Modules; using Oqtane.Shared; namespace Oqtane.Repository { public class ModuleDefinitionRepository : IModuleDefinitionRepository { private MasterDBContext _db; private readonly IMemoryCache _cache; private readonly IPermissionRepository _permissions; private List _moduleDefinitions; // lazy load public ModuleDefinitionRepository(MasterDBContext context, IMemoryCache cache, IPermissionRepository permissions) { _db = context; _cache = cache; _permissions = permissions; } public IEnumerable GetModuleDefinitions(int siteId) { return LoadModuleDefinitions(siteId); } public ModuleDefinition GetModuleDefinition(int moduleDefinitionId, int siteId) { List moduledefinitions = LoadModuleDefinitions(siteId); return moduledefinitions.Find(item => item.ModuleDefinitionId == moduleDefinitionId); } public void UpdateModuleDefinition(ModuleDefinition moduleDefinition) { _db.Entry(moduleDefinition).State = EntityState.Modified; _db.SaveChanges(); _permissions.UpdatePermissions(moduleDefinition.SiteId, EntityNames.ModuleDefinition, moduleDefinition.ModuleDefinitionId, moduleDefinition.Permissions); _cache.Remove("moduledefinitions:" + moduleDefinition.SiteId.ToString()); } public void DeleteModuleDefinition(int moduleDefinitionId, int siteId) { ModuleDefinition moduleDefinition = _db.ModuleDefinition.Find(moduleDefinitionId); _permissions.DeletePermissions(siteId, EntityNames.ModuleDefinition, moduleDefinitionId); _db.ModuleDefinition.Remove(moduleDefinition); _db.SaveChanges(); } public List LoadModuleDefinitions(int siteId) { // get module definitions for site List moduleDefinitions = _cache.GetOrCreate("moduledefinitions:" + siteId.ToString(), entry => { entry.SlidingExpiration = TimeSpan.FromMinutes(30); return LoadSiteModuleDefinitions(siteId); }); return moduleDefinitions; } private List LoadSiteModuleDefinitions(int siteId) { if (_moduleDefinitions == null) { // get module assemblies _moduleDefinitions = LoadModuleDefinitionsFromAssemblies(); } List moduleDefinitions = _moduleDefinitions; // get module definition permissions for site List permissions = _permissions.GetPermissions(siteId, EntityNames.ModuleDefinition).ToList(); // get module definitions in database List moduledefs = _db.ModuleDefinition.ToList(); // sync module assemblies with database foreach (ModuleDefinition moduledefinition in moduleDefinitions) { ModuleDefinition moduledef = moduledefs.Where(item => item.ModuleDefinitionName == moduledefinition.ModuleDefinitionName).FirstOrDefault(); if (moduledef == null) { // new module definition moduledef = new ModuleDefinition { ModuleDefinitionName = moduledefinition.ModuleDefinitionName }; _db.ModuleDefinition.Add(moduledef); _db.SaveChanges(); _permissions.UpdatePermissions(siteId, EntityNames.ModuleDefinition, moduledef.ModuleDefinitionId, moduledefinition.Permissions); } else { // existing module definition if (!string.IsNullOrEmpty(moduledef.Name)) { moduledefinition.Name = moduledef.Name; } if (!string.IsNullOrEmpty(moduledef.Description)) { moduledefinition.Description = moduledef.Description; } if (!string.IsNullOrEmpty(moduledef.Categories)) { moduledefinition.Categories = moduledef.Categories; } if (permissions.Count == 0) { _permissions.UpdatePermissions(siteId, EntityNames.ModuleDefinition, moduledef.ModuleDefinitionId, moduledefinition.Permissions); } else { moduledefinition.Permissions = _permissions.EncodePermissions(permissions.Where(item => item.EntityId == moduledef.ModuleDefinitionId)); } // remove module definition from list as it is already synced moduledefs.Remove(moduledef); } moduledefinition.ModuleDefinitionId = moduledef.ModuleDefinitionId; moduledefinition.SiteId = siteId; moduledefinition.CreatedBy = moduledef.CreatedBy; moduledefinition.CreatedOn = moduledef.CreatedOn; moduledefinition.ModifiedBy = moduledef.ModifiedBy; moduledefinition.ModifiedOn = moduledef.ModifiedOn; } // any remaining module definitions are orphans foreach (ModuleDefinition moduledefinition in moduledefs) { _permissions.DeletePermissions(siteId, EntityNames.ModuleDefinition, moduledefinition.ModuleDefinitionId); _db.ModuleDefinition.Remove(moduledefinition); // delete _db.SaveChanges(); } return moduleDefinitions; } private List LoadModuleDefinitionsFromAssemblies() { List moduleDefinitions = new List(); // iterate through Oqtane module assemblies Assembly[] assemblies = AppDomain.CurrentDomain.GetAssemblies() .Where(item => item.FullName.StartsWith("Oqtane.") || item.FullName.Contains(".Module.")).ToArray(); foreach (Assembly assembly in assemblies) { moduleDefinitions = LoadModuleDefinitionsFromAssembly(moduleDefinitions, assembly); } return moduleDefinitions; } private List LoadModuleDefinitionsFromAssembly(List moduledefinitions, Assembly assembly) { ModuleDefinition moduledefinition; Type[] modulecontroltypes = assembly.GetTypes().Where(item => item.GetInterfaces().Contains(typeof(IModuleControl))).ToArray(); foreach (Type modulecontroltype in modulecontroltypes) { if (modulecontroltype.Name != "ModuleBase" && !modulecontroltype.Namespace.EndsWith(".Controls")) { string[] typename = modulecontroltype.AssemblyQualifiedName.Split(',').Select(item => item.Trim()).ToList().ToArray(); string[] segments = typename[0].Split('.'); Array.Resize(ref segments, segments.Length - 1); string moduleType = string.Join(".", segments); string qualifiedModuleType = moduleType + ", " + typename[1]; int index = moduledefinitions.FindIndex(item => item.ModuleDefinitionName == qualifiedModuleType); if (index == -1) { // determine if this module implements IModule Type moduletype = assembly .GetTypes() .Where(item => item.Namespace != null) .Where(item => item.Namespace.StartsWith(moduleType)) .FirstOrDefault(item => item.GetInterfaces().Contains(typeof(IModule))); if (moduletype != null) { var moduleobject = Activator.CreateInstance(moduletype); moduledefinition = (ModuleDefinition)moduletype.GetProperty("ModuleDefinition").GetValue(moduleobject); } else { moduledefinition = new ModuleDefinition { Name = moduleType.Substring(moduleType.LastIndexOf(".") + 1), Description = "Manage " + moduleType.Substring(moduleType.LastIndexOf(".") + 1), Categories = ((qualifiedModuleType.StartsWith("Oqtane.Modules.Admin.")) ? "Admin" : ""), Version = new Version(1, 0, 0).ToString() }; } // set internal properties moduledefinition.ModuleDefinitionName = qualifiedModuleType; moduledefinition.ControlTypeTemplate = moduleType + "." + Constants.ActionToken + ", " + typename[1]; moduledefinition.AssemblyName = assembly.FullName.Split(",")[0]; if (string.IsNullOrEmpty(moduledefinition.Categories)) { moduledefinition.Categories = "Common"; } if (moduledefinition.Categories == "Admin") { moduledefinition.Permissions = new List { new Permission(PermissionNames.Utilize, Constants.AdminRole, true) }.EncodePermissions(); } else { moduledefinition.Permissions = new List { new Permission(PermissionNames.Utilize, Constants.AdminRole, true), new Permission(PermissionNames.Utilize, Constants.RegisteredRole, true) }.EncodePermissions(); } moduledefinitions.Add(moduledefinition); index = moduledefinitions.FindIndex(item => item.ModuleDefinitionName == qualifiedModuleType); } moduledefinition = moduledefinitions[index]; // actions var modulecontrolobject = Activator.CreateInstance(modulecontroltype); string actions = (string)modulecontroltype.GetProperty("Actions").GetValue(modulecontrolobject); if (actions != "") { foreach (string action in actions.Split(',')) { moduledefinition.ControlTypeRoutes += (action + "=" + modulecontroltype.FullName + ", " + typename[1] + ";"); } } moduledefinitions[index] = moduledefinition; } } return moduledefinitions; } } }