Provider.cs

Go to the documentation of this file.

using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Api.Models;

namespace Tgstation.Server.Host.Components.Chat.Providers
{
    abstract class Provider : IProvider
    {
        readonly Queue<Message> messageQueue;

        TaskCompletionSource<object> nextMessage;

        protected Provider()
        {
            messageQueue = new Queue<Message>();
            nextMessage = new TaskCompletionSource<object>();
        }

        public abstract bool Connected { get; }

        public abstract string BotMention { get; }

        protected void EnqueueMessage(Message message)
        {
            lock (messageQueue)
            {
                messageQueue.Enqueue(message);
                nextMessage.TrySetResult(null);
            }
        }

        public abstract void Dispose();

        public abstract Task<bool> Connect(CancellationToken cancellationToken);

        public abstract Task Disconnect(CancellationToken cancellationToken);

        public abstract Task<IReadOnlyList<Channel>> MapChannels(IEnumerable<ChatChannel> channels, CancellationToken cancellationToken);

        public async Task<Message> NextMessage(CancellationToken cancellationToken)
        {
            var cancelTcs = new TaskCompletionSource<object>();
            using (cancellationToken.Register(() => cancelTcs.SetCanceled()))
                await Task.WhenAny(nextMessage.Task, cancelTcs.Task).ConfigureAwait(false);
            cancellationToken.ThrowIfCancellationRequested();
            lock (messageQueue)
            {
                var result = messageQueue.Dequeue();
                if (messageQueue.Count == 0)
                    nextMessage = new TaskCompletionSource<object>();
                return result;
            }
        }

        public abstract Task SendMessage(ulong channelId, string message, CancellationToken cancellationToken);
    }
}