DiscordChatExporter/DiscordChatExporter.Core/Exporting/ChannelExporter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using DiscordChatExporter.Core.Discord;
using DiscordChatExporter.Core.Discord.Data;
using DiscordChatExporter.Core.Discord.Data.Common;
using DiscordChatExporter.Core.Exceptions;
using DiscordChatExporter.Core.Utils.Extensions;
using Gress;

namespace DiscordChatExporter.Core.Exporting;

public class ChannelExporter
{
    private readonly DiscordClient _discord;

    public ChannelExporter(DiscordClient discord) => _discord = discord;

    public async ValueTask ExportChannelAsync(
        ExportRequest request,
        IProgress<Percentage>? progress = null,
        CancellationToken cancellationToken = default)
    {
        // Build context
        var contextMembers = new HashSet<Member>(IdBasedEqualityComparer.Instance);
        var contextChannels = await _discord.GetGuildChannelsAsync(request.Guild.Id, cancellationToken);
        var contextRoles = await _discord.GetGuildRolesAsync(request.Guild.Id, cancellationToken);

        var context = new ExportContext(
            request,
            contextMembers,
            contextChannels,
            contextRoles
        );

        // Export messages
        await using var messageExporter = new MessageExporter(context);

        var exportedAnything = false;
        var encounteredUsers = new HashSet<User>(IdBasedEqualityComparer.Instance);

        await foreach (var message in _discord.GetMessagesAsync(
                           request.Channel.Id,
                           request.After,
                           request.Before,
                           progress,
                           cancellationToken))
        {
            cancellationToken.ThrowIfCancellationRequested();

            // Skips any messages that fail to pass the supplied filter
            if (!request.MessageFilter.IsMatch(message))
                continue;

            // Resolve members for referenced users
            foreach (var referencedUser in message.MentionedUsers.Prepend(message.Author))
            {
                if (!encounteredUsers.Add(referencedUser))
                    continue;

                var member = await _discord.GetGuildMemberAsync(
                    request.Guild.Id,
                    referencedUser,
                    cancellationToken
                );

                contextMembers.Add(member);
            }

            // Export message
            await messageExporter.ExportMessageAsync(message, cancellationToken);
            exportedAnything = true;
        }

        // Throw if no messages were exported
        if (!exportedAnything)
            throw DiscordChatExporterException.ChannelIsEmpty();
    }
}