LANCommander/LANCommander.Server.Services/StorageLocationService.cs

using AutoMapper;
using LANCommander.SDK.Enums;
using LANCommander.Server.Data;
using LANCommander.Server.Data.Models;
using Microsoft.AspNetCore.Http;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using ZiggyCreatures.Caching.Fusion;

namespace LANCommander.Server.Services
{
    public sealed class StorageLocationService(
        ILogger<StorageLocationService> logger,
        SettingsProvider<Settings.Settings> settingsProvider,
        IFusionCache cache,
        IMapper mapper,
        IHttpContextAccessor httpContextAccessor,
        IDbContextFactory<DatabaseContext> contextFactory) : BaseDatabaseService<StorageLocation>(logger, settingsProvider, cache, mapper, httpContextAccessor, contextFactory)
    {
        public override async Task<StorageLocation> AddAsync(StorageLocation entity)
        {
            return await base.AddAsync(entity, async context =>
            {
                await context.UpdateRelationshipAsync(sl => sl.Archives);
                await context.UpdateRelationshipAsync(sl => sl.GameSaves);
                await context.UpdateRelationshipAsync(sl => sl.Media);
            });
        }

        public override async Task<StorageLocation> UpdateAsync(StorageLocation entity)
        {
            return await base.UpdateAsync(entity, async context =>
            {
                await context.UpdateRelationshipAsync(sl => sl.Archives);
                await context.UpdateRelationshipAsync(sl => sl.GameSaves);
                await context.UpdateRelationshipAsync(sl => sl.Media);
            });
        }

        public async Task<StorageLocation> DefaultAsync(StorageLocationType type)
        {
            return await FirstOrDefaultAsync(sl => sl.Default && sl.Type == type);
        }

        public async Task<StorageLocation> GetOrDefaultAsync(Guid? id, StorageLocationType type)
        {
            if (!id.HasValue)
                return await DefaultAsync(type);
            
            var storageLocation = await FirstOrDefaultAsync(sl => sl.Id == id && sl.Type == type);

            if (storageLocation != null)
                return storageLocation;
            
            return await DefaultAsync(type); 
        }
    }
}