mirror of
				https://github.com/Kareadita/Kavita.git
				synced 2025-10-31 02:27:04 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			91 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
			
		
		
	
	
			91 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C#
		
	
	
	
	
	
| using System.Collections.Generic;
 | |
| using System.IO;
 | |
| using System.Linq;
 | |
| using System.Reflection;
 | |
| using System.Threading.Tasks;
 | |
| using API.Constants;
 | |
| using API.Entities;
 | |
| using API.Entities.Enums;
 | |
| using API.Services;
 | |
| using Kavita.Common;
 | |
| using Microsoft.AspNetCore.Identity;
 | |
| using Microsoft.EntityFrameworkCore;
 | |
| 
 | |
| namespace API.Data
 | |
| {
 | |
|     public static class Seed
 | |
|     {
 | |
|         public static async Task SeedRoles(RoleManager<AppRole> roleManager)
 | |
|         {
 | |
|             var roles = typeof(PolicyConstants)
 | |
|                 .GetFields(BindingFlags.Public | BindingFlags.Static)
 | |
|                 .Where(f => f.FieldType == typeof(string))
 | |
|                 .ToDictionary(f => f.Name,
 | |
|                     f => (string) f.GetValue(null)).Values
 | |
|                 .Select(policyName => new AppRole() {Name = policyName})
 | |
|                 .ToList();
 | |
| 
 | |
|             foreach (var role in roles)
 | |
|             {
 | |
|                 var exists = await roleManager.RoleExistsAsync(role.Name);
 | |
|                 if (!exists)
 | |
|                 {
 | |
|                     await roleManager.CreateAsync(role);
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         public static async Task SeedSettings(DataContext context)
 | |
|         {
 | |
|             await context.Database.EnsureCreatedAsync();
 | |
| 
 | |
|             IList<ServerSetting> defaultSettings = new List<ServerSetting>()
 | |
|             {
 | |
|                 new() {Key = ServerSettingKey.CacheDirectory, Value = CacheService.CacheDirectory},
 | |
|                 new () {Key = ServerSettingKey.TaskScan, Value = "daily"},
 | |
|                 new () {Key = ServerSettingKey.LoggingLevel, Value = "Information"}, // Not used from DB, but DB is sync with appSettings.json
 | |
|                 new () {Key = ServerSettingKey.TaskBackup, Value = "weekly"},
 | |
|                 new () {Key = ServerSettingKey.BackupDirectory, Value = Path.GetFullPath(Path.Join(Directory.GetCurrentDirectory(), "backups/"))},
 | |
|                 new () {Key = ServerSettingKey.Port, Value = "5000"}, // Not used from DB, but DB is sync with appSettings.json
 | |
|                 new () {Key = ServerSettingKey.AllowStatCollection, Value = "true"},
 | |
|             };
 | |
| 
 | |
|             foreach (var defaultSetting in defaultSettings)
 | |
|             {
 | |
|                 var existing = context.ServerSetting.FirstOrDefault(s => s.Key == defaultSetting.Key);
 | |
|                 if (existing == null)
 | |
|                 {
 | |
|                     await context.ServerSetting.AddAsync(defaultSetting);
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             await context.SaveChangesAsync();
 | |
|             
 | |
|             // Port and LoggingLevel are managed in appSettings.json. Update the DB values to match
 | |
|             context.ServerSetting.FirstOrDefault(s => s.Key == ServerSettingKey.Port).Value =
 | |
|                 Configuration.Port + string.Empty;
 | |
|             context.ServerSetting.FirstOrDefault(s => s.Key == ServerSettingKey.LoggingLevel).Value =
 | |
|                 Configuration.LogLevel + string.Empty;
 | |
| 
 | |
|             await context.SaveChangesAsync();
 | |
| 
 | |
|         }
 | |
| 
 | |
|         public static async Task SeedSeriesMetadata(DataContext context)
 | |
|         {
 | |
|             await context.Database.EnsureCreatedAsync();
 | |
|             
 | |
|             context.Database.EnsureCreated();
 | |
|             var series = await context.Series
 | |
|                 .Include(s => s.Metadata).ToListAsync();
 | |
|                 
 | |
|             foreach (var s in series)
 | |
|             {
 | |
|                 s.Metadata ??= new SeriesMetadata();
 | |
|             }
 | |
| 
 | |
|             await context.SaveChangesAsync();
 | |
|         }
 | |
|     }
 | |
| }
 |