Taiko_Stuff/TaikoLocalServer
1
0
Fork 0
mirror of synced 2025-02-07 07:01:16 +01:00
Code Releases Activity

Fixed unable to purchase ura fumen from don medal shop

Browse Source
This commit is contained in:
S-Sebb 2024-11-08 23:58:58 +08:00
parent 463f832b31
commit 064188171c
1 changed files with 55 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
TaikoLocalServer/Handlers/PurchaseSongCommand.cs
View File

@ -3,13 +3,66 @@ using Throw;
namespace TaikoLocalServer.Handlers;
public record PurchaseSongCommand(uint Baid, uint SongNo, uint TokenId, uint Price) : IRequest<CommonSongPurchaseResponse>;
public record PurchaseSongCommand(uint Baid, uint SongNo, uint Type, uint TokenId, uint Price) : IRequest<CommonSongPurchaseResponse>;
public record PurchaseSongCommandCN(uint Baid, uint SongNo, uint TokenId, uint Price) : IRequest<CommonSongPurchaseResponse>;
public class PurchaseSongCommandHandler(TaikoDbContext context, ILogger<PurchaseSongCommandHandler> logger)
: IRequestHandler<PurchaseSongCommand, CommonSongPurchaseResponse>
{
public async Task<CommonSongPurchaseResponse> Handle(PurchaseSongCommand request, CancellationToken cancellationToken)
{
var user = await context.UserData
.Include(u => u.Tokens)
.FirstOrDefaultAsync(u => u.Baid == request.Baid, cancellationToken);
user.ThrowIfNull($"User with baid {request.Baid} does not exist!");
var token = user.Tokens.FirstOrDefault(t => t.Id == request.TokenId);
if (token is not null && token.Count >= request.Price)
{
token.Count -= (int)request.Price;
}
else
{
logger.LogError("User with baid {Baid} does not have enough tokens to purchase song with id {SongNo}!", request.Baid, request.SongNo);
return new CommonSongPurchaseResponse { Result = 0 };
}
if (request.Type == 1)
{
if (user.UnlockedUraSongIdList.Contains(request.SongNo))
{
logger.LogWarning("User with baid {Baid} already has song with id {SongNo} unlocked!", request.Baid, request.SongNo);
return new CommonSongPurchaseResponse { Result = 0 };
}
user.UnlockedUraSongIdList.Add(request.SongNo);
}
else
{
if (user.UnlockedSongIdList.Contains(request.SongNo))
{
logger.LogWarning("User with baid {Baid} already has song with id {SongNo} unlocked!", request.Baid, request.SongNo);
return new CommonSongPurchaseResponse { Result = 0 };
}
user.UnlockedSongIdList.Add(request.SongNo);
}
context.UserData.Update(user);
await context.SaveChangesAsync(cancellationToken);
return new CommonSongPurchaseResponse { Result = 1, TokenCount = token.Count };
}
}
public class PurchaseSongCommandHandlerCN(TaikoDbContext context, ILogger<PurchaseSongCommandHandlerCN> logger)
: IRequestHandler<PurchaseSongCommandCN, CommonSongPurchaseResponse>
{
public async Task<CommonSongPurchaseResponse> Handle(PurchaseSongCommandCN request, CancellationToken cancellationToken)
{
var user = await context.UserData
.Include(u => u.Tokens)
@ -33,6 +86,7 @@ public class PurchaseSongCommandHandler(TaikoDbContext context, ILogger<Purchase
}
user.UnlockedSongIdList.Add(request.SongNo);
context.UserData.Update(user);
await context.SaveChangesAsync(cancellationToken);
return new CommonSongPurchaseResponse { Result = 1, TokenCount = token.Count };