using Infrastructure.Application.Entites.Settings; using Infrastructure.Data; using Infrastructure.Domain; using Infrastructure.Events; using Infrastructure.Extensions; using Infrastructure.Security; using IoT.Shared; using IoT.Shared.Application.Domain.Entities; using Microsoft.EntityFrameworkCore; using Microsoft.Extensions.Configuration; using Microsoft.Extensions.Hosting; using Microsoft.Extensions.Logging; using System; using System.Collections.Generic; using System.Linq; using System.Reflection; namespace Platform.Data { public class PlatformDbContext : EfDbContext, IDbConfig { private readonly IEncryptionService _encryptionService; public PlatformDbContext(DbContextOptions options, ILogger logger, IHostEnvironment env, IConfiguration cfg, IEventPublisher publisher, IEncryptionService encryptionService) : base(options, logger, env, cfg, publisher) { this._encryptionService = encryptionService; } protected override void OnModelCreating(ModelBuilder modelBuilder) { IoTSharedDbConfig.OnModelCreating(modelBuilder); //Setting modelBuilder.Entity().HasIndex(o => o.Name).IsUnique(); //User modelBuilder.Entity().Property(o => o.UserName).IsRequired(); modelBuilder.Entity().HasIndex(o => o.UserName).IsUnique(); modelBuilder.Entity().HasIndex(o => o.Email).IsUnique(); modelBuilder.Entity().HasIndex(o => o.PhoneNumber).IsUnique(); //OrganRole modelBuilder.Entity().Property(o => o.Name).IsRequired(); modelBuilder.Entity().Property(o => o.Number).IsRequired(); modelBuilder.Entity().HasIndex(o => new { o.OrganId, o.Number }).IsUnique(); modelBuilder.Entity().HasOne(o => o.Organ).WithMany(o => o.Roles).HasForeignKey(o => o.OrganId).OnDelete(DeleteBehavior.Cascade); //Permission modelBuilder.Entity().Property(o => o.Name).IsRequired(); modelBuilder.Entity().Property(o => o.Number).IsRequired(); modelBuilder.Entity().HasIndex(o => o.Number).IsUnique(); modelBuilder.Entity().HasOne(o => o.Category).WithMany(o => o.Permissions).HasForeignKey(o => o.CategoryId).OnDelete(DeleteBehavior.SetNull); //OrganUserRole modelBuilder.Entity().HasOne(o => o.OrganUser).WithMany(o => o.UserRoles).HasForeignKey(o => o.OrganUserId); modelBuilder.Entity().HasOne(o => o.OrganRole).WithMany(o => o.UserRoles).HasForeignKey(o => o.OrganRoleId); modelBuilder.Entity().HasIndex(o => new { o.OrganUserId, o.OrganRoleId }).IsUnique(); //OrganRolePermission modelBuilder.Entity().HasOne(o => o.Role).WithMany(o => o.RolePermissions).HasForeignKey(o => o.RoleId); modelBuilder.Entity().HasOne(o => o.Permission).WithMany(o => o.RolePermissions).HasForeignKey(o => o.PermissionId); modelBuilder.Entity().HasIndex(o => new { o.RoleId, o.PermissionId }).IsUnique(); //Site modelBuilder.Entity().Property(o => o.Name).IsRequired(); modelBuilder.Entity().HasIndex(o => o.Name).IsUnique(); //Area modelBuilder.Entity().HasIndex(o => o.Number).IsUnique(); //Organ modelBuilder.Entity().HasIndex(o => o.Name).IsUnique(); modelBuilder.Entity().HasIndex(o => o.Number).IsUnique(); modelBuilder.Entity().HasOne(o => o.Area).WithMany(o => o.Organs).HasForeignKey(o => o.AreaId).OnDelete(DeleteBehavior.SetNull); //Building modelBuilder.Entity().Property(o => o.OrganId).IsRequired(); modelBuilder.Entity().HasIndex(o => new { o.OrganId, o.ParentId, o.Number }).IsUnique(); modelBuilder.Entity().HasOne(o => o.Organ).WithMany(o => o.Buildings).HasForeignKey(o => o.OrganId); //OrganUser modelBuilder.Entity().HasOne(o => o.Organ).WithMany(o => o.OrganUsers).HasForeignKey(o => o.OrganId); modelBuilder.Entity().HasOne(o => o.User).WithMany(o => o.OrganUsers).HasForeignKey(o => o.UserId); modelBuilder.Entity().HasIndex(o => new { o.OrganId, o.UserId }).IsUnique(); //Department modelBuilder.Entity().HasIndex(o => new { o.OrganId, o.Number }).IsUnique(); //IoTCenter //IoTProductCategory modelBuilder.Entity().HasIndex(o => o.Number).IsUnique(); modelBuilder.Entity().HasOne(o => o.Category).WithMany(o => o.Products).HasForeignKey(o => o.CategoryId).OnDelete(DeleteBehavior.SetNull); //IoTGateway modelBuilder.Entity().HasOne(o => o.Building).WithMany(o => o.IoTGateways).HasForeignKey(o => o.BuildingId).OnDelete(DeleteBehavior.SetNull); //IoTProductCategoryIoTProduct //modelBuilder.Entity().HasOne(o => o.Category).WithMany(o => o.CategoryProducts).HasForeignKey(o => o.CategoryId); //modelBuilder.Entity().HasOne(o => o.Product).WithMany().HasForeignKey(o => o.ProductId); //modelBuilder.Entity().HasIndex(o => o.ProductId).IsUnique();//通过唯一约束,保证一个商品只有一个分类 //IoTCommand modelBuilder.Entity().HasOne(o => o.Api).WithMany().HasForeignKey(o => o.ApiId); //IoTScene modelBuilder.Entity().HasOne(o => o.Building).WithMany(o => o.Scenes).HasForeignKey(o => o.BuildingId); //IoTSceneIoTCommand modelBuilder.Entity().HasOne(o => o.IoTScene).WithMany(o => o.IoTCommands).HasForeignKey(o => o.IoTSceneId); modelBuilder.Entity().HasOne(o => o.IoTCommand).WithMany().HasForeignKey(o => o.IoTCommandId); //IoTTimer modelBuilder.Entity().HasOne(o => o.IoTScene).WithMany(o => o.IoTTimers).HasForeignKey(o => o.IoTSceneId); //IoTTigger modelBuilder.Entity().HasOne(o => o.IoTScene).WithMany(o => o.IoTTiggers).HasForeignKey(o => o.IoTSceneId); modelBuilder.Entity().HasOne(o => o.Data).WithMany().HasForeignKey(o => o.DataId); //LiveRecord modelBuilder.Entity().Property(o => o.DeviceNumber).IsRequired(); base.OnModelCreating(modelBuilder); } public void Seed(DbContext db) { var root = db.Set().Add(new IoTProductCategory { Name = "产品分类", Number = "root" }).Entity; db.Set().Add(new IoTProductCategory { ParentId = root.Id, Number = "00", Name = "网关", Image = "/images/gateway.svg" }); db.Set().Add(new IoTProductCategory { ParentId = root.Id, Number = "10", Name = "安防", Image = "/images/safe.svg" }); db.Set().Add(new IoTProductCategory { ParentId = root.Id, Number = "20", Name = "电器", Image = "/images/socket.svg" }); db.Set().Add(new IoTProductCategory { ParentId = root.Id, Number = "30", Name = "照明", Image = "/images/light.svg" }); db.Set().Add(new IoTProductCategory { ParentId = root.Id, Number = "40", Name = "监测", Image = "/images/monitor.svg" }); db.SaveChanges(); db.Set().FirstOrDefault(o => o.Number == "root").Update(); db.SaveChanges(); var permissionCategoryRoot = new PermissionCategory { Name = "权限分类", Number = "root" }; db.Set().Add(permissionCategoryRoot); db.SaveChanges(); foreach (var item in db.Model.GetEntityTypes()) { var group = item.ClrType.GetCustomAttribute(); var categoryGroup = db.Set().FirstOrDefault(o => o.Number == group.Number); if (categoryGroup == null) { categoryGroup = db.Set().Add(new PermissionCategory { Parent = permissionCategoryRoot, Name = group.Name, Number = group.Number }).Entity; db.Set().Add(categoryGroup); db.SaveChanges(); } var type = item.ClrType; var name = type.GetDisplayName(); var number = type.Name; var category = new PermissionCategory { Parent = categoryGroup, Name = name, Number = number }; var scopeAttribute = type.GetCustomAttribute(); if (scopeAttribute != null) { var scope = scopeAttribute.Scope; //平台权限 if (scope.HasFlag(ScopeType.PlatformRead)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Platform, Name = $"查看平台{name}", Number = $"Read-Platform-{number}" }); } if (scope.HasFlag(ScopeType.PlatformAdd)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Platform, Name = $"添加平台{name}", Number = $"Add-Platform-{number}" }); } if (scope.HasFlag(ScopeType.PlatformEdit)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Platform, Name = $"修改平台{name}", Number = $"Edit-Platform-{number}" }); } if (scope.HasFlag(ScopeType.PlatformDelete)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Platform, Name = $"删除平台{name}", Number = $"Delete-Platform-{number}" }); } //机构权限 if (scope.HasFlag(ScopeType.OrganRead)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Organ, Name = $"查看机构{name}", Number = $"Read-Organ-{number}" }); } if (scope.HasFlag(ScopeType.OrganAdd)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Organ, Name = $"添加机构{name}", Number = $"Add-Organ-{number}" }); } if (scope.HasFlag(ScopeType.OrganEdit)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Organ, Name = $"修改机构{name}", Number = $"Edit-Organ-{number}" }); } if (scope.HasFlag(ScopeType.OrganDelete)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.Organ, Name = $"删除机构{name}", Number = $"Delete-Organ-{number}" }); } //个人权限 if (scope.HasFlag(ScopeType.UserRead)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.User, Name = $"查看个人{name}", Number = $"Read-User-{number}" }); } if (scope.HasFlag(ScopeType.UserAdd)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.User, Name = $"添加个人{name}", Number = $"Add-User-{number}" }); } if (scope.HasFlag(ScopeType.UserEdit)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.User, Name = $"修改个人{name}", Number = $"Edit-User-{number}" }); } if (scope.HasFlag(ScopeType.UserDelete)) { category.Permissions.Add(new Permission { IsReadOnly = true, Type = PermissionType.User, Name = $"删除个人{name}", Number = $"Delete-User-{number}" }); } } db.Set().Add(category); db.SaveChanges(); } db.Set().FirstOrDefault(o => o.Number == "root").Update(); db.SaveChanges(); //初始化区域 var areaRoot = new Area { Name = "区域", Number = "root", Children = new List { new Area{ Type = AreaType.Province, Name="吉林省", Number="Jilin", Children = new List { new Area{ Type= AreaType.City, Name="长春市", Number="changchun", Children = new List { new Area { Type= AreaType.Township, Name="南关区", Number="nanguan" } } } } } } }; db.Set().Add(areaRoot); db.SaveChanges(); //初始化机构 var organRoot = new Organ { Name = "机构", Number = "root", Children = new List { new Organ { Area = db.Set().FirstOrDefault(o => o.Number == "changchun"), Type= OrganType.Type5, IsReadOnly=true, Name="默认教育局", Number="jiaoyuju", Children = new List { new Organ { Area = db.Set().FirstOrDefault(o => o.Number == "nanguan"), Type= OrganType.Type5, Name="默认学校", Number="xuexiao" }, new Organ { Area = db.Set().FirstOrDefault(o => o.Number == "nanguan"), Type= OrganType.Type1, Name="其他学校", Number="qitaxuexiao" } } } } }; db.Set().Add(organRoot); db.SaveChanges(); //初始化建筑 var organId = db.Set().FirstOrDefault(o => o.Number == "xuexiao").Id; var buildingRoot = new Building { Name = "建筑", Number = "root", OrganId = organRoot.Id, Children = new List { new Building { OrganId=organId, Type= BuildingType.Building, Name="默认建筑", Number="1", Children = new List { new Building { OrganId=organId, Type= BuildingType.Layer, Name="1楼", Number="11", Children=new List { new Building{ OrganId=organId, Type= BuildingType.Room, Name="101室", Number="101" }, new Building{ OrganId=organId, Type= BuildingType.Room, Name="102室", Number="102" } } }, new Building { OrganId=organId, Type= BuildingType.Layer, Name="2楼", Number="12", Children=new List { new Building{ OrganId=organId, Type= BuildingType.Room, Name="201室", Number="201" }, new Building{ OrganId=organId, Type= BuildingType.Room, Name="202室", Number="202" } } } } } } }; db.Set().Add(buildingRoot); db.SaveChanges(); //更新Left和Right编号 db.Set().FirstOrDefault(o => o.Number == "root").Update(); db.Set().FirstOrDefault(o => o.Number == "root").Update(); db.Set().FirstOrDefault(o => o.Number == "root").Update(); db.SaveChanges(); var defaultOrganId = db.Set().FirstOrDefault(o => o.IsReadOnly).Id; var superRole = new OrganRole { OrganId = defaultOrganId, Number = "super", Name = "超级管理员", IsReadOnly = true }; var adminRole = new OrganRole { OrganId = defaultOrganId, Number = "admin", Name = "管理员" }; var organRole = new OrganRole { OrganId = defaultOrganId, Number = "organ", Name = "机构管理员" }; //初始化角色 var skips = new string[] { //"添加Api", "修改Api", "添加分类", "修改分类", "添加参数", "修改参数", "添加产品", "修改产品", "添加节点", "添加权限", "修改权限", "添加权限分类", "修改权限分类", "添加设备" }; foreach (var item in db.Set()) { superRole.RolePermissions.Add(new OrganRolePermission { IsReadOnly = true, Permission = item }); if (!item.Name.Contains("删除") && !skips.Contains(item.Name)) { adminRole.RolePermissions.Add(new OrganRolePermission { Permission = item }); if (item.Type == PermissionType.Organ) { organRole.RolePermissions.Add(new OrganRolePermission { Permission = item }); } } } db.Set().AddRange(superRole, adminRole, organRole); db.SaveChanges(); var securityStam = "123456"; db.Set().Add(new User { UserName = "super", SecurityStamp = securityStam, PasswordHash = _encryptionService.CreatePasswordHash("123456", securityStam), PasswordConfirmed = true, Email = "super@test.com", EmailConfirmed = true, NickName = "超级管理员", OrganUsers = new List { new OrganUser { IsReadOnly=true, Type= OrganUserType.Type1, OrganId=defaultOrganId, UserRoles=new List { new OrganUserRole{ IsReadOnly=true, OrganRoleId=superRole.Id } } }, new OrganUser { Type= OrganUserType.Type1, OrganId=db.Set().FirstOrDefault(o=>o.Name=="默认学校").Id, } } }); db.Set().Add(new User { UserName = "admin", SecurityStamp = securityStam, PasswordHash = _encryptionService.CreatePasswordHash("123456", securityStam), PasswordConfirmed = true, Email = "admin@test.com", EmailConfirmed = true, NickName = "管理员", OrganUsers = new List { new OrganUser { Type= OrganUserType.Type1, OrganId=defaultOrganId, UserRoles=new List { new OrganUserRole{ OrganRoleId=adminRole.Id} } } } }); db.Set().Add(new User { UserName = "organ", SecurityStamp = securityStam, PasswordHash = _encryptionService.CreatePasswordHash("123456", securityStam), PasswordConfirmed = true, Email = "organ@test.com", EmailConfirmed = true, NickName = "机构管理员", OrganUsers = new List { new OrganUser { Type= OrganUserType.Type1, OrganId=defaultOrganId, UserRoles=new List { new OrganUserRole{ OrganRoleId=organRole.Id} } } } }); db.SaveChanges(); // var set = db.Set(); set.Add(new Setting { Name = "name", Value = "物联平台", Type = SettingType.String }); set.Add(new Setting { Name = "logo", Value = "/images/logo.png", Type = SettingType.ImageUrl }); set.Add(new Setting { Name = "copyright", Value = "Copyright © {0} Company. All rights reserved", Type = SettingType.Html }); set.Add(new Setting { Name = "iosAppUrl", Value = "itms-services://?action=download-manifest&url=https://iot.edusoa.com/IoTCenter/Info.plist", Type = SettingType.String }); // var macAddress = Helper.Instance.GetMacAddress(); set.Add(new Setting { Name = "sn", Value = macAddress, Type = SettingType.String }); set.Add(new Setting { Name = "id", Value = macAddress.Md5(), Type = SettingType.String }); set.Add(new Setting { Name = "code", Value = "根据id生成的授权码", Type = SettingType.String }); set.Add(new Setting { Name = "delay", Value = "500", Type = SettingType.String }); db.SaveChanges(); // set.Add(new Setting { Name = "AccessTokenTimeout", Value = "1440", Type = SettingType.String }); set.Add(new Setting { Name = "RefreshTokenTimeout", Value = "518400", Type = SettingType.String }); set.Add(new Setting { Name = "MaxFailedAccessAttemptsBeforeLockout", Value = "5", Type = SettingType.String }); set.Add(new Setting { Name = "DefaultAccountLockoutMinutes", Value = "10", Type = SettingType.String }); set.Add(new Setting { Name = "CaptchaSeconds", Value = "60", Type = SettingType.String }); set.Add(new Setting { Name = "RegisterDisabled", Value = "false", Type = SettingType.String }); // set.Add(new Setting { Name = "email:host", Value = "nbaxp.com", Type = SettingType.String }); set.Add(new Setting { Name = "email:port", Value = "25", Type = SettingType.String }); set.Add(new Setting { Name = "email:user", Value = "admin@nbaxp.com", Type = SettingType.String }); set.Add(new Setting { Name = "email:password", Value = "123456", Type = SettingType.String }); set.Add(new Setting { Name = "sms", Value = "false", Type = SettingType.String }); set.Add(new Setting { Name = "sms:url", Value = "https://api.netease.im/sms/sendcode.action", Type = SettingType.String }); set.Add(new Setting { Name = "sms:key", Value = "123456", Type = SettingType.String }); set.Add(new Setting { Name = "sms:secret", Value = "123456", Type = SettingType.String }); db.SaveChanges(); } } }