74 lines
3.2 KiB
C#
74 lines
3.2 KiB
C#
using JSMR.Application.Common.Caching;
|
|
using JSMR.Application.Integrations.Chobit.Models;
|
|
using JSMR.Application.Integrations.Chobit.Ports;
|
|
using JSMR.Application.Integrations.DLSite.Models;
|
|
using JSMR.Application.Integrations.DLSite.Models.ReleasedWorks;
|
|
using JSMR.Application.Integrations.DLSite.Ports;
|
|
using JSMR.Application.Scanning.Contracts;
|
|
using JSMR.Application.Scanning.Ports;
|
|
|
|
namespace JSMR.Application.Scanning;
|
|
|
|
public sealed class ScanVoiceWorksHandler(
|
|
IVoiceWorkScannerRepository scannerRepository,
|
|
IVoiceWorkUpdaterRepository updaterRepository,
|
|
IDLSiteClient dlsiteClient,
|
|
IChobitClient chobitClient,
|
|
ISpamCircleCache spamCircleCache,
|
|
IReleasedWorksProvider releasedWorksProvider,
|
|
IVoiceWorkSearchUpdater searchUpdater)
|
|
{
|
|
public async Task<ScanVoiceWorksResponse> HandleAsync(ScanVoiceWorksRequest request, CancellationToken cancellationToken)
|
|
{
|
|
IVoiceWorksScanner? scanner = scannerRepository.GetScanner(request.Locale);
|
|
IVoiceWorkUpdater? updater = updaterRepository.GetUpdater(request.Locale);
|
|
|
|
if (scanner is null)
|
|
throw new InvalidOperationException($"No scanner registered for locale {request.Locale}.");
|
|
|
|
if (updater is null)
|
|
throw new InvalidOperationException($"No updater registered for locale {request.Locale}.");
|
|
|
|
VoiceWorkScanOptions options = new(
|
|
PageNumber: request.PageNumber,
|
|
PageSize: request.PageSize,
|
|
ExcludedMakerIds: await spamCircleCache.GetAsync(cancellationToken),
|
|
ExcludePartiallyAIGeneratedWorks: true,
|
|
ExcludeAIGeneratedWorks: true
|
|
);
|
|
|
|
VoiceWorkScanResult scanResult = await scanner.ScanPageAsync(options, cancellationToken);
|
|
|
|
if (scanResult.EndOfResults)
|
|
{
|
|
return new ScanVoiceWorksResponse(
|
|
Results: [],
|
|
EndOfResults: true
|
|
);
|
|
}
|
|
|
|
string[] productIds = [.. scanResult.Works.Where(x => !string.IsNullOrWhiteSpace(x.ProductId)).Select(x => x.ProductId!)];
|
|
VoiceWorkDetailCollection voiceWorkDetails = await dlsiteClient.GetVoiceWorkDetailsAsync(productIds, cancellationToken);
|
|
ChobitResultCollection chobitResults = await chobitClient.GetSampleInfoAsync(productIds, cancellationToken);
|
|
ReleasedWorksCollection releasedWorkCollection = await releasedWorksProvider.GetReleasedWorksAsync(scanResult, cancellationToken);
|
|
|
|
VoiceWorkIngest[] ingests = [.. scanResult.Works.Select(work =>
|
|
{
|
|
voiceWorkDetails.TryGetValue(work.ProductId!, out VoiceWorkDetails? value);
|
|
chobitResults.TryGetValue(work.ProductId, out ChobitResult? chobit);
|
|
releasedWorkCollection.TryGetValue(work.ProductId, out ReleasedWork? releasedWork);
|
|
|
|
return VoiceWorkIngest.From(work, value, chobit);
|
|
})];
|
|
|
|
VoiceWorkUpsertResult[] upsertResults = await updater.UpsertAsync(ingests, cancellationToken);
|
|
int[] voiceWorkIds = [.. upsertResults.Where(x => x.VoiceWorkId.HasValue).Select(x => x.VoiceWorkId!.Value)];
|
|
|
|
await searchUpdater.UpdateAsync(voiceWorkIds, cancellationToken);
|
|
|
|
return new ScanVoiceWorksResponse(
|
|
Results: upsertResults,
|
|
EndOfResults: false
|
|
);
|
|
}
|
|
} |