iot/projects/IoTNode/DbConfig.cs

using Application.Domain.Entities;
using Infrastructure.Application.Entites.Settings;
using Infrastructure.Data;
using Infrastructure.Extensions;
using Infrastructure.Security;
using IoT.Shared;
using Microsoft.AspNetCore.Hosting;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Hosting;
using System;
using System.Collections.Generic;
using System.Linq;

namespace IoTNode
{
    public class DbConfig : IDbConfig
    {
        private readonly IWebHostEnvironment _env;
        private readonly IConfiguration _cfg;
        private readonly IEncryptionService _encryptionService;

        public DbConfig(IWebHostEnvironment env, IConfiguration cfg, IEncryptionService encryptionService)
        {
            this._env = env;
            this._cfg = cfg;
            this._encryptionService = encryptionService;
        }

        public void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
        }

        public void OnModelCreating(ModelBuilder modelBuilder)
        {
            IoTSharedDbConfig.OnModelCreating(modelBuilder);
            modelBuilder.Ignore<NodeCategory>();
            modelBuilder.Ignore<NodeCategoryNode>();
            modelBuilder.Ignore<NodeCategoryNode>();
            modelBuilder.Ignore<NodeCategoryNode>();
            modelBuilder.Ignore<Organ>();
            modelBuilder.Ignore<Organ>();
            modelBuilder.Ignore<OrganUser>();
            modelBuilder.Ignore<OrganUser>();
            modelBuilder.Ignore<OrganUser>();
            modelBuilder.Ignore<OrganNode>();
            modelBuilder.Ignore<OrganNode>();
            modelBuilder.Ignore<OrganNode>();
            modelBuilder.Ignore<OrganNode>();
            modelBuilder.Ignore<OrganScene>();
            modelBuilder.Ignore<OrganSceneCommand>();
            modelBuilder.Ignore<OrganSceneCommand>();
            modelBuilder.Ignore<OrganSceneTimer>();
            modelBuilder.Ignore<OrganSceneTigger>();
            modelBuilder.Ignore<OrganSceneTigger>();
        }

        public void Seed(DbContext db)
        {
            IoTSharedDbConfig.Seed(db);
            var saRole = new Role { Name = "超级管理员", IsReadOnly = true };
            var adminRole = new Role { Name = "管理员", IsReadOnly = true };

            var skips = new string[] { "添加Api", "修改Api", "添加分类", "修改分类", "添加参数", "修改参数", "添加产品", "修改产品", "添加节点", "添加权限", "修改权限", "添加权限分类", "修改权限分类", "添加设备" };

            foreach (var item in db.Set<Permission>())
            {
                saRole.RolePermissions.Add(new RolePermission { Permission = item, IsReadOnly = true });
                if (!item.Name.Contains("删除") && !skips.Contains(item.Name))
                {
                    adminRole.RolePermissions.Add(new RolePermission { Permission = item, IsReadOnly = true });
                }
            }

            var securityStam = "123456";
            db.Set<User>().Add(new User
            {
                UserName = "super",
                SecurityStamp = securityStam,
                PasswordHash = _encryptionService.CreatePasswordHash("123456", securityStam),
                Email = "super@test.com",
                UserRoles = new List<UserRole> { new UserRole { Role = saRole } }
            });
            db.Set<User>().Add(new User
            {
                UserName = "admin",
                SecurityStamp = securityStam,
                PasswordHash = _encryptionService.CreatePasswordHash("123456", securityStam),
                Email = "admin@test.com",
                UserRoles = new List<UserRole> { new UserRole { Role = adminRole } }
            });
            db.SaveChanges();
            //
            var set = db.Set<Setting>();
            set.Add(new Setting { Name = "name", Value = this._cfg.GetValue("seed:NodeName", "节点"), Type = SettingType.Text });
            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 });
            //
            var macAddress = Helper.Instance.GetMacAddress();
            set.Add(new Setting { Name = "sn", Value = macAddress, Type = SettingType.Text });
            set.Add(new Setting
            {
                Name = "code",
                Value = this._env.IsDevelopment() ? Helper.Instance.MacEncrypt(Helper.Instance.GetMacAddress()) : "根据mac生成的授权码",
                Type = SettingType.Text
            });
            //
            var host = this._cfg.GetValue("seed:host", "localhost");
            var stream = this._cfg.GetValue("seed:stream", "localhost");
            var delay = this._cfg.GetValue("seed:delay", "100");
            set.Add(new Setting { Name = "debug", Value = "false", Type = SettingType.Text });
            set.Add(new Setting { Name = "notify:enabled", Value = "true", Type = SettingType.Text });
            set.Add(new Setting { Name = "notify:host", Value = $"http://{host}/IoTCenter", Type = SettingType.Text });
            set.Add(new Setting { Name = "timer.seconds", Value = "180", Type = SettingType.Text });
            set.Add(new Setting { Name = "onvif.timer", Value = "1", Type = SettingType.Text });
            set.Add(new Setting { Name = "onvif.speed", Value = "0.2", Type = SettingType.Text });
            set.Add(new Setting { Name = "camera.usr", Value = "admin", Type = SettingType.Text });
            set.Add(new Setting { Name = "camera.pwd", Value = "dsideal123", Type = SettingType.Text });
            set.Add(new Setting { Name = "stream.rtmp", Value = stream, Type = SettingType.Text });
            set.Add(new Setting { Name = "ffmpeg.args", Value = " -y -threads {0} -rtsp_transport tcp -use_wallclock_as_timestamps 1 -stimeout 3000000 -i \"{1}\" -fflags +genpts -c copy -f flv \"{2}\"", Type = SettingType.Text });
            set.Add(new Setting { Name = "fbee.writelist", Value = "", Type = SettingType.Text });
            set.Add(new Setting { Name = "camera.writelist", Value = "", Type = SettingType.Text });
            set.Add(new Setting { Name = "delay", Value = delay, Type = SettingType.Text });
            db.SaveChanges();
            var organ = this._cfg.GetValue("seed:organ", "默认机构");
            db.Set<Node>().Add(new Node
            {
                Id = $"nodeid-{macAddress}".ToGuid(),
                Name = $"{organ}-节点-{macAddress}",
                Number = macAddress,
                IsOnline = true,
                Version = Helper.Instance.GetVersion(),
                Image = "/images/classroom.svg"
            });
            db.SaveChanges();
        }
    }
}