Pack opening
This commit is contained in:
90
SVSim.Database/Repositories/Pack/PackRepository.cs
Normal file
90
SVSim.Database/Repositories/Pack/PackRepository.cs
Normal file
@@ -0,0 +1,90 @@
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using SVSim.Database.Models;
|
||||
|
||||
namespace SVSim.Database.Repositories.Pack;
|
||||
|
||||
public class PackRepository : IPackRepository
|
||||
{
|
||||
private readonly SVSimDbContext _db;
|
||||
public PackRepository(SVSimDbContext db) { _db = db; }
|
||||
|
||||
public async Task<List<PackConfigEntry>> GetActivePacks(DateTime now) =>
|
||||
await _db.Packs
|
||||
.Include(p => p.ChildGachas)
|
||||
.Include(p => p.Banners)
|
||||
.Where(p => p.CommenceDate <= now && p.CompleteDate >= now)
|
||||
.ToListAsync();
|
||||
|
||||
public async Task<PackConfigEntry?> GetPack(int parentGachaId) =>
|
||||
await _db.Packs
|
||||
.Include(p => p.ChildGachas)
|
||||
.Include(p => p.Banners)
|
||||
.FirstOrDefaultAsync(p => p.Id == parentGachaId);
|
||||
|
||||
public async Task<Dictionary<int, ViewerPackOpenCount>> GetOpenCountsForViewer(long viewerId)
|
||||
{
|
||||
var viewer = await _db.Viewers
|
||||
.Include(v => v.PackOpenCounts)
|
||||
.FirstOrDefaultAsync(v => v.Id == viewerId);
|
||||
return viewer?.PackOpenCounts.ToDictionary(p => p.PackId) ?? new();
|
||||
}
|
||||
|
||||
public async Task IncrementOpenCount(long viewerId, int parentGachaId, int by)
|
||||
{
|
||||
var viewer = await _db.Viewers
|
||||
.Include(v => v.PackOpenCounts)
|
||||
.FirstAsync(v => v.Id == viewerId);
|
||||
var row = viewer.PackOpenCounts.FirstOrDefault(p => p.PackId == parentGachaId);
|
||||
if (row is null)
|
||||
{
|
||||
viewer.PackOpenCounts.Add(new ViewerPackOpenCount { PackId = parentGachaId, OpenCount = by });
|
||||
}
|
||||
else
|
||||
{
|
||||
row.OpenCount += by;
|
||||
}
|
||||
await _db.SaveChangesAsync();
|
||||
}
|
||||
|
||||
public async Task MarkDailyFreeUsed(long viewerId, int parentGachaId, DateTime when)
|
||||
{
|
||||
var viewer = await _db.Viewers
|
||||
.Include(v => v.PackOpenCounts)
|
||||
.FirstAsync(v => v.Id == viewerId);
|
||||
var row = viewer.PackOpenCounts.FirstOrDefault(p => p.PackId == parentGachaId);
|
||||
if (row is null)
|
||||
{
|
||||
viewer.PackOpenCounts.Add(new ViewerPackOpenCount { PackId = parentGachaId, OpenCount = 0, LastDailyFreeAt = when });
|
||||
}
|
||||
else
|
||||
{
|
||||
row.LastDailyFreeAt = when;
|
||||
}
|
||||
await _db.SaveChangesAsync();
|
||||
}
|
||||
|
||||
public async Task GrantCardsToViewer(long viewerId, IEnumerable<long> cardIds)
|
||||
{
|
||||
var viewer = await _db.Viewers
|
||||
.Include(v => v.Cards).ThenInclude(c => c.Card)
|
||||
.FirstAsync(v => v.Id == viewerId);
|
||||
|
||||
var byId = viewer.Cards.ToDictionary(c => c.Card.Id);
|
||||
foreach (var grp in cardIds.GroupBy(id => id))
|
||||
{
|
||||
if (byId.TryGetValue(grp.Key, out var existing))
|
||||
{
|
||||
existing.Count += grp.Count();
|
||||
}
|
||||
else
|
||||
{
|
||||
// Look up the card by id and attach it so we don't insert a phantom Card row.
|
||||
var card = await _db.Cards.FirstAsync(c => c.Id == grp.Key);
|
||||
var owned = new OwnedCardEntry { Card = card, Count = grp.Count(), IsProtected = false };
|
||||
viewer.Cards.Add(owned);
|
||||
byId[grp.Key] = owned;
|
||||
}
|
||||
}
|
||||
await _db.SaveChangesAsync();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user