Telegrator/Telegrator/TelegratorClient.cs

using Telegram.Bot;
using Telegram.Bot.Polling;
using Telegrator.Configuration;
using Telegrator.Logging;
using Telegrator.MadiatorCore;
using Telegrator.Polling;
using Telegrator.Providers;

namespace Telegrator
{
    /// <summary>
    /// Main client class for the Telegrator library.
    /// Extends TelegramBotClient with reactive capabilities for handling updates.
    /// </summary>
    public class TelegratorClient : TelegramBotClient, ITelegratorBot, ICollectingProvider
    {
        /// <summary>
        /// The update router for handling incoming updates.
        /// </summary>
        private IUpdateRouter? updateRouter = null;

        /// <inheritdoc/>
        public TelegratorOptions Options { get; private set; }

        /// <inheritdoc/>
        public IHandlersCollection Handlers { get; private set; }

        /// <inheritdoc/>
        public ITelegramBotInfo BotInfo { get; private set; }

        /// <inheritdoc/>
        public IUpdateRouter UpdateRouter { get => updateRouter ?? throw new Exception(); }

        /// <summary>
        /// Initializes a new instance of the <see cref="TelegratorClient"/> class with a bot token.
        /// </summary>
        /// <param name="token">The bot token from BotFather.</param>
        /// <param name="httpClient">Optional HTTP client for making requests.</param>
        /// <param name="cancellationToken">The cancellation token.</param>
        public TelegratorClient(string token, HttpClient? httpClient = null, CancellationToken cancellationToken = default)
            : this(new TelegramBotClientOptions(token), null, httpClient, cancellationToken) { }

        /// <summary>
        /// Initializes a new instance of the <see cref="TelegratorClient"/> class with bot options.
        /// </summary>
        /// <param name="options">The Telegram bot client options.</param>
        /// <param name="httpClient">Optional HTTP client for making requests.</param>
        /// <param name="cancellationToken">The cancellation token.</param>
        public TelegratorClient(TelegramBotClientOptions options, HttpClient? httpClient = null, CancellationToken cancellationToken = default) 
            : this(options, null, httpClient, cancellationToken) { }

        /// <summary>
        /// Initializes a new instance of the <see cref="TelegratorClient"/> class with bot options and Telegrator options.
        /// </summary>
        /// <param name="options">The Telegram bot client options.</param>
        /// <param name="telegratorOptions">The Telegrator options.</param>
        /// <param name="httpClient">Optional HTTP client for making requests.</param>
        /// <param name="cancellationToken">The cancellation token.</param>
        public TelegratorClient(TelegramBotClientOptions options, TelegratorOptions? telegratorOptions, HttpClient? httpClient = null, CancellationToken cancellationToken = default) : base(options, httpClient, cancellationToken)
        {
            Options = telegratorOptions ?? new TelegratorOptions();
            Handlers = new HandlersCollection(default);
            BotInfo = new TelegramBotInfo(this.GetMe(cancellationToken).Result);
        }

        /// <summary>
        /// Starts receiving updates from Telegram.
        /// Initializes the update router and begins polling for updates.
        /// </summary>
        /// <param name="receiverOptions">Optional receiver options for configuring update polling.</param>
        /// <param name="cancellationToken">The cancellation token to stop receiving updates.</param>
        public void StartReceiving(ReceiverOptions? receiverOptions = null, CancellationToken cancellationToken = default)
        {
            if (Options.GlobalCancellationToken == CancellationToken.None)
                Options.GlobalCancellationToken = cancellationToken;

            HandlersProvider handlerProvider = new HandlersProvider(Handlers, Options);
            AwaitingProvider awaitingProvider = new AwaitingProvider(Options);

            updateRouter = new UpdateRouter(handlerProvider, awaitingProvider, Options, BotInfo);
            
            // Log startup
            Alligator.LogInformation($"Telegrator bot starting up - BotId: {BotInfo.User.Id}, Username: {BotInfo.User.Username}, MaxParallelHandlers: {Options.MaximumParallelWorkingHandlers ?? -1}");

            StartReceivingInternal(receiverOptions, cancellationToken);
        }

        /// <summary>
        /// Internal method that starts the update receiving process.
        /// Handles the reactive update receiver and error handling.
        /// </summary>
        /// <param name="receiverOptions">Optional receiver options for configuring update polling.</param>
        /// <param name="cancellationToken">The cancellation token to stop receiving updates.</param>
        private async void StartReceivingInternal(ReceiverOptions? receiverOptions, CancellationToken cancellationToken)
        {
            try
            {
                try
                {
                    await new ReactiveUpdateReceiver(this, receiverOptions)
                        .ReceiveAsync(UpdateRouter, cancellationToken)
                        .ConfigureAwait(false);
                }
                catch (Exception exception)
                {
                    await UpdateRouter
                        .HandleErrorAsync(this, exception, HandleErrorSource.FatalError, cancellationToken)
                        .ConfigureAwait(false);
                }
            }
            catch (OperationCanceledException)
            {
                // Cancelled
                Alligator.LogInformation("Telegrator bot stopped (cancelled)");
            }
        }


    }
}