ddirty830/OF-DL
2
0
Fork 0
forked from sim0n00ps/OF-DL
Code

Improve CLI output to show counts of media

Browse Source
This commit is contained in:
whimsical-c4lic0 2026-02-12 10:48:44 -06:00
parent fdb6d55454
commit f16ad7f138
1 changed files with 42 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
OF DL.Core/Services/DownloadOrchestrationService.cs
View File

@ -433,7 +433,7 @@ public class DownloadOrchestrationService(
: purchasedTabCollection.PaidPosts.PaidPosts.Count;
DownloadResult postResult = await eventHandler.WithProgressAsync(
$"Downloading {purchasedTabCollection.PaidPosts.PaidPosts.Count} Paid Posts",
$"Downloading {purchasedTabCollection.PaidPosts.PaidPosts.Count} Media from {purchasedTabCollection.PaidPosts.PaidPostObjects.Count} Paid Posts",
totalSize, config.ShowScrapeSize,
async reporter => await downloadService.DownloadPaidPostsPurchasedTab(
purchasedTabCollection.Username, path, users,
@ -461,7 +461,7 @@ public class DownloadOrchestrationService(
: purchasedTabCollection.PaidMessages.PaidMessages.Count;
DownloadResult msgResult = await eventHandler.WithProgressAsync(
$"Downloading {purchasedTabCollection.PaidMessages.PaidMessages.Count} Paid Messages",
$"Downloading {purchasedTabCollection.PaidMessages.PaidMessages.Count} Media from {purchasedTabCollection.PaidMessages.PaidMessageObjects.Count} Paid Messages",
totalSize, config.ShowScrapeSize,
async reporter => await downloadService.DownloadPaidMessagesPurchasedTab(
purchasedTabCollection.Username, path, users,
@ -502,48 +502,77 @@ public class DownloadOrchestrationService(
PurchasedEntities.SinglePaidMessageCollection singlePaidMessageCollection =
await apiService.GetPaidMessage($"/messages/{messageId}", path);
if (singlePaidMessageCollection.SingleMessages.Count == 0)
if (singlePaidMessageCollection.SingleMessages.Count == 0 &&
singlePaidMessageCollection.PreviewSingleMessages.Count == 0)
{
eventHandler.OnNoContentFound("Paid Messages");
return;
}
Config config = configService.CurrentConfig;
int messageCount = singlePaidMessageCollection.SingleMessageObjects.Count;
string messageLabel = messageCount == 1 ? "Paid Message" : "Paid Messages";
int previewCount = singlePaidMessageCollection.PreviewSingleMessages.Count;
int paidCount = singlePaidMessageCollection.SingleMessages.Count;
int totalCount = previewCount + paidCount;
// Handle preview messages
if (singlePaidMessageCollection.PreviewSingleMessages.Count > 0)
// Handle mixed paid + preview message media.
if (previewCount > 0 && paidCount > 0)
{
eventHandler.OnContentFound("Paid Messages",
totalCount,
singlePaidMessageCollection.SingleMessageObjects.Count);
List<string> allMessageUrls = [];
allMessageUrls.AddRange(singlePaidMessageCollection.PreviewSingleMessages.Values);
allMessageUrls.AddRange(singlePaidMessageCollection.SingleMessages.Values);
long totalSize = config.ShowScrapeSize
? await downloadService.CalculateTotalFileSize(allMessageUrls)
: totalCount;
DownloadResult result = await eventHandler.WithProgressAsync(
$"Downloading {totalCount} Media from {messageCount} {messageLabel} ({paidCount} Paid + {previewCount} Preview)",
totalSize, config.ShowScrapeSize,
async reporter => await downloadService.DownloadSinglePaidMessage(username, path, users,
clientIdBlobMissing, devicePrivateKeyMissing, singlePaidMessageCollection, reporter));
eventHandler.OnDownloadComplete("Paid Messages", result);
}
// Handle preview-only message media.
else if (previewCount > 0)
{
eventHandler.OnContentFound("Preview Paid Messages",
singlePaidMessageCollection.PreviewSingleMessages.Count,
previewCount,
singlePaidMessageCollection.SingleMessageObjects.Count);
long previewSize = config.ShowScrapeSize
? await downloadService.CalculateTotalFileSize(
singlePaidMessageCollection.PreviewSingleMessages.Values.ToList())
: singlePaidMessageCollection.PreviewSingleMessages.Count;
: previewCount;
DownloadResult previewResult = await eventHandler.WithProgressAsync(
$"Downloading {singlePaidMessageCollection.PreviewSingleMessages.Count} Preview Paid Messages",
$"Downloading {previewCount} Preview Media from {messageCount} {messageLabel}",
previewSize, config.ShowScrapeSize,
async reporter => await downloadService.DownloadSinglePaidMessage(username, path, users,
clientIdBlobMissing, devicePrivateKeyMissing, singlePaidMessageCollection, reporter));
eventHandler.OnDownloadComplete("Paid Messages", previewResult);
}
else if (singlePaidMessageCollection.SingleMessages.Count > 0)
else if (paidCount > 0)
{
// Only actual paid messages, no preview
eventHandler.OnContentFound("Paid Messages",
singlePaidMessageCollection.SingleMessages.Count,
paidCount,
singlePaidMessageCollection.SingleMessageObjects.Count);
long totalSize = config.ShowScrapeSize
? await downloadService.CalculateTotalFileSize(
singlePaidMessageCollection.SingleMessages.Values.ToList())
: singlePaidMessageCollection.SingleMessages.Count;
: paidCount;
DownloadResult result = await eventHandler.WithProgressAsync(
$"Downloading {singlePaidMessageCollection.SingleMessages.Count} Paid Messages",
$"Downloading {paidCount} Paid Media from {messageCount} {messageLabel}",
totalSize, config.ShowScrapeSize,
async reporter => await downloadService.DownloadSinglePaidMessage(username, path, users,
clientIdBlobMissing, devicePrivateKeyMissing, singlePaidMessageCollection, reporter));
@ -607,7 +636,7 @@ public class DownloadOrchestrationService(
: mediaCount;
DownloadResult result = await eventHandler.WithProgressAsync(
$"Downloading {mediaCount} {contentType}", totalSize, config.ShowScrapeSize,
$"Downloading {mediaCount} Media from {objectCount} {contentType}", totalSize, config.ShowScrapeSize,
async reporter => await downloadData(data, reporter));
eventHandler.OnDownloadComplete(contentType, result);