using System.Collections.Generic; using System.Linq; using System.Reflection; using Microsoft.EntityFrameworkCore; using Oqtane.Extensions; using Oqtane.Models; using Oqtane.Shared; namespace Oqtane.Repository { public class PageModuleRepository : IPageModuleRepository { private readonly IDbContextFactory _dbContextFactory; private readonly IModuleDefinitionRepository _moduleDefinitions; private readonly IModuleRepository _modules; private readonly IPermissionRepository _permissions; private readonly ISettingRepository _settings; public PageModuleRepository(IDbContextFactory dbContextFactory, IModuleDefinitionRepository moduleDefinitions, IModuleRepository modules, IPermissionRepository permissions, ISettingRepository settings) { _dbContextFactory = dbContextFactory; _moduleDefinitions = moduleDefinitions; _modules = modules; _permissions = permissions; _settings = settings; } public IEnumerable GetPageModules(int siteId) { using var db = _dbContextFactory.CreateDbContext(); var pagemodules = db.PageModule .Include(item => item.Module) // eager load module .Include(item => item.Page) // eager load page .Where(item => item.Module.SiteId == siteId).ToList(); if (pagemodules.Any()) { var moduledefinitions = _moduleDefinitions.GetModuleDefinitions(siteId).ToList(); var permissions = _permissions.GetPermissions(siteId, EntityNames.Module).ToList(); for (int index = 0; index < pagemodules.Count; index++) { pagemodules[index] = GetPageModule(pagemodules[index], moduledefinitions, permissions); } } return pagemodules; } public PageModule AddPageModule(PageModule pageModule) { using var db = _dbContextFactory.CreateDbContext(); db.PageModule.Add(pageModule); db.SaveChanges(); return pageModule; } public PageModule UpdatePageModule(PageModule pageModule) { using var db = _dbContextFactory.CreateDbContext(); db.Entry(pageModule).State = EntityState.Modified; db.SaveChanges(); return pageModule; } public PageModule GetPageModule(int pageModuleId) { return GetPageModule(pageModuleId, true); } public PageModule GetPageModule(int pageModuleId, bool tracking) { using var db = _dbContextFactory.CreateDbContext(); PageModule pagemodule; if (tracking) { pagemodule = db.PageModule .Include(item => item.Module) // eager load module .Include(item => item.Page) // eager load page .FirstOrDefault(item => item.PageModuleId == pageModuleId); } else { pagemodule = db.PageModule.AsNoTracking() .Include(item => item.Module) // eager load module .Include(item => item.Page) // eager load page .FirstOrDefault(item => item.PageModuleId == pageModuleId); } if (pagemodule != null) { var moduledefinitions = _moduleDefinitions.GetModuleDefinitions(pagemodule.Module.SiteId).ToList(); var permissions = _permissions.GetPermissions(pagemodule.Module.SiteId, EntityNames.Module).ToList(); pagemodule = GetPageModule(pagemodule, moduledefinitions, permissions); } return pagemodule; } public PageModule GetPageModule(int pageId, int moduleId) { using var db = _dbContextFactory.CreateDbContext(); var pagemodule = db.PageModule .Include(item => item.Module) // eager load module .Include(item => item.Page) // eager load page .SingleOrDefault(item => item.PageId == pageId && item.ModuleId == moduleId); if (pagemodule != null) { var moduledefinitions = _moduleDefinitions.GetModuleDefinitions(pagemodule.Module.SiteId).ToList(); var permissions = _permissions.GetPermissions(pagemodule.Module.SiteId, EntityNames.Module).ToList(); pagemodule = GetPageModule(pagemodule, moduledefinitions, permissions); } return pagemodule; } public void DeletePageModule(int pageModuleId) { using var db = _dbContextFactory.CreateDbContext(); var pageModule = db.PageModule .Include(item => item.Module) // eager load module .Include(item => item.Page) // eager load page .SingleOrDefault(item => item.PageModuleId == pageModuleId); _settings.DeleteSettings(EntityNames.PageModule, pageModuleId); db.PageModule.Remove(pageModule); db.SaveChanges(); // check if there are any remaining module instances in the site var pageModules = GetPageModules(pageModule.Module.SiteId); if (!pageModules.Any(item => item.ModuleId == pageModule.ModuleId)) { _modules.DeleteModule(pageModule.ModuleId); } } private PageModule GetPageModule(PageModule pageModule, List moduleDefinitions, List modulePermissions) { var permissions = modulePermissions.Where(item => item.EntityId == pageModule.ModuleId).ToList(); // moduledefinition permissionnames can specify permissions for other entities (ie. API permissions) pageModule.Module.ModuleDefinition = moduleDefinitions.Find(item => item.ModuleDefinitionName == pageModule.Module.ModuleDefinitionName); if (pageModule.Module.ModuleDefinition != null && !string.IsNullOrEmpty(pageModule.Module.ModuleDefinition.PermissionNames) && pageModule.Module.ModuleDefinition.PermissionNames.Contains(":")) { foreach (var permissionname in pageModule.Module.ModuleDefinition.PermissionNames.Split(",", System.StringSplitOptions.RemoveEmptyEntries)) { if (permissionname.Contains(":")) { // moduledefinition permissionnames can be in the form of "EntityName:PermissionName:Roles" var segments = permissionname.Split(':'); if (segments.Length == 3 && segments[0] != EntityNames.Module) { permissions.AddRange(_permissions.GetPermissions(pageModule.Module.SiteId, segments[0], segments[1]).Where(item => item.EntityId == -1)); } } } } pageModule.Module.PermissionList = permissions?.ToList(); return pageModule; } } }