using System.IO.Compression; using LANCommander.Server.Data; using LANCommander.Server.Data.Models; using LANCommander.Server.Services.Extensions; using AutoMapper; using LANCommander.SDK.Enums; using LANCommander.SDK.Services; using Microsoft.AspNetCore.Http; using Microsoft.Extensions.Logging; using Microsoft.EntityFrameworkCore; using ZiggyCreatures.Caching.Fusion; namespace LANCommander.Server.Services { public sealed class RedistributableService( ILogger logger, ArchiveService archiveService, StorageLocationService storageLocationService, SettingsProvider settingsProvider, ScriptClient scriptClient, IFusionCache cache, IMapper mapper, IHttpContextAccessor httpContextAccessor, IDbContextFactory contextFactory) : BaseDatabaseService(logger, settingsProvider, cache, mapper, httpContextAccessor, contextFactory) { public override async Task AddAsync(Redistributable entity) { await cache.ExpireGameCacheAsync(); return await base.AddAsync(entity, async context => { await context.UpdateRelationshipAsync(r => r.Archives); await context.UpdateRelationshipAsync(r => r.Games); await context.UpdateRelationshipAsync(r => r.Pages); await context.UpdateRelationshipAsync(r => r.Scripts); }); } public override async Task UpdateAsync(Redistributable entity) { if (entity.Games != null && entity.Games.Any()) { foreach (var game in entity.Games) { await cache.ExpireGameCacheAsync(game?.Id); } } return await base.UpdateAsync(entity, async context => { await context.UpdateRelationshipAsync(r => r.Archives); await context.UpdateRelationshipAsync(r => r.Games); await context.UpdateRelationshipAsync(r => r.Pages); await context.UpdateRelationshipAsync(r => r.Scripts); }); } public async Task GetManifestAsync(Guid manifestId) { var redistributable = await AsNoTracking() .AsSplitQuery() .Query(q => { return q .Include(r => r.Archives) .Include(r => r.Scripts); }) .GetAsync(manifestId); return mapper.Map(redistributable); } public async Task GetLatestArchiveAsync(Guid id) { var redistributable = await AsNoTracking() .AsSplitQuery() .Include(r => r.Archives) .GetAsync(id); var latestArchive = redistributable.Archives.OrderByDescending(a => a.CreatedOn).FirstOrDefault(); return latestArchive; } public async Task GetVersionAsync(Guid id) { var latestArchive = await GetLatestArchiveAsync(id); return latestArchive?.Version ?? String.Empty; } public async Task> GetUpdatesAsync(Guid redistributableId, string version) { var redistributable = await AsNoTracking() .AsSplitQuery() .Include(r => r.Archives) .GetAsync(redistributableId); if (redistributable?.Archives == null || !redistributable.Archives.Any()) return []; var orderedArchives = redistributable.Archives.OrderBy(a => a.CreatedOn).ToList(); if (string.IsNullOrWhiteSpace(version)) return [orderedArchives.Last()]; var installedArchive = orderedArchives.FirstOrDefault(a => a.Version == version); if (installedArchive == null) return [orderedArchives.Last()]; var newerArchives = orderedArchives .Where(a => a.CreatedOn > installedArchive.CreatedOn) .ToList(); return newerArchives; } public async Task PackageAsync(Guid id) { var redistributable = await AsNoTracking() .AsSplitQuery() .Include(r => r.Archives) .Include(r => r.Scripts) .Include(r => r.Games) .GetAsync(id); logger?.LogInformation("Packaging redistributable {RedistributableName}", redistributable.Name); var latestArchive = redistributable.Archives?.OrderByDescending(r => r.CreatedOn).FirstOrDefault(); var storageLocation = await storageLocationService.GetOrDefaultAsync(latestArchive?.StorageLocationId, StorageLocationType.Archive); string latestArchivePath = null; if (latestArchive != null) latestArchivePath = await archiveService.GetArchiveFileLocationAsync(latestArchive); if (redistributable.Scripts?.Any(s => s.Type == ScriptType.Package) ?? false) { foreach (var script in redistributable.Scripts.Where(s => s.Type == ScriptType.Package)) { var package = await scriptClient.Redistributable_RunPackageScriptAsync(mapper.Map(script), mapper.Map(redistributable), latestArchivePath); if (package == null) { var message = $"Could not package redistributable {redistributable.Name}, the package script did not return a result"; logger?.LogError(message); throw new Exception(message); } if (String.IsNullOrWhiteSpace(package.Path) || !Directory.Exists(package.Path)) { var message = $"Could not package redistributable {redistributable.Name}, the path {package.Path} could not be found"; logger?.LogError(message); throw new Exception(message); } var archive = new Archive { Version = package.Version, RedistributableId = redistributable.Id, ObjectKey = Guid.NewGuid().ToString(), LastVersion = latestArchive, StorageLocationId = storageLocation.Id, }; archive = await archiveService.AddAsync(archive); var destination = await archiveService.GetArchiveFileLocationAsync(archive); ZipFile.CreateFromDirectory(package.Path, destination); await archiveService.RecalculateFileSizeArchiveAsync(archive); logger?.LogInformation("Successfully packaged {RedistributableName} and created new archive with version number {RedistributableVersion}", redistributable.Name, archive.Version); } } else { var message = $"Could not package redistributable {redistributable.Name}, no packaging scripts are defined"; logger?.LogWarning(message); throw new Exception(message); } } } }