Luski.Net/Luski.net/Structures/Main/MainSocketTextChannel.cs

using Luski.net.Enums;
using Luski.net.Interfaces;
using Luski.net.JsonTypes.BaseTypes;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text.Json.Serialization;
using System.Threading;
using System.Threading.Tasks;
using JacobTechEncryption;
using JacobTechEncryption.Enums;
using Luski.net.Enums.Main;

namespace Luski.net.Structures.Main;

public class MainSocketTextChannel : MainSocketChannel
{
    public async Task<MainSocketMessage> GetMessage(long ID, CancellationToken CancellationToken)
    {
        return await Server.GetMessage(ID, CancellationToken);
    }

    public async Task<Stream> GetPicture(CancellationToken CancellationToken)
    {
        if (Type == ChannelType.DM) return Members.First().GetAvatar(CancellationToken).Result;
        else
        {
            bool isc = System.IO.File.Exists(Server.Storage.GetStorageDirectory(StorageDirectory.ChannelIcons) + Id.ToString());
            if (!isc) await Server.GetFromServer($"SocketChannel/GetPicture/{Id}", Server.Storage.GetStorageDirectory(StorageDirectory.ChannelIcons) + Id.ToString(), CancellationToken);
            return Server.Storage.GetResourceStream(StorageDirectory.ChannelIcons, Id.ToString());
        }
    }

    public async Task<IReadOnlyList<MainSocketMessage>> GetMessages(long Message_Id, CancellationToken CancellationToken, int count = 50)
    {
        if (count > 200)
        {
            throw new Exception("You can not request more than 200 messages at a time");
        }
        else if (count < 1)
        {
            throw new Exception("You must request at least 1 message");
        }
        else
        {
            SocketBulkMessage data = await Server.GetFromServer("SocketBulkMessage",
                SocketBulkMessageContext.Default.SocketBulkMessage,
                CancellationToken,
                new KeyValuePair<string, string?>("channel_id", Id.ToString()),
                new KeyValuePair<string, string?>("messages", count.ToString()),
                new KeyValuePair<string, string?>("mostrecentid", Message_Id.ToString()));
            if (data.Error is null)
            {
                int num = Convert.ToInt32(Math.Ceiling((Environment.ProcessorCount * 5) * 2.0));
                if (num == 0) num = 1;
                
                string key = Server.EncryptionHandler.GetChannelKey(Id);
                if (data is null) throw new Exception("Invalid data from server");
                if (data.Messages is null) data.Messages = Array.Empty<MainSocketMessage>();
                Parallel.ForEach(data.Messages, new ParallelOptions()
                {
                    MaxDegreeOfParallelism = num
                }, i =>
                {
                    i.decrypt(key, CancellationToken);
                });
                key = null;
                return await Task.FromResult(data.Messages.ToList().AsReadOnly() as IReadOnlyList<MainSocketMessage>);
            }
            else
            {
                throw new Exception(data.ErrorMessage);
            }
        }
    }

    public async Task<IReadOnlyList<MainSocketMessage>> GetMessages(CancellationToken CancellationToken, int count = 50)
    {
        try
        {
            if (count > 200)
            {
                throw new Exception("You can not request more than 200 messages at a time");
            }
            else if (count < 1)
            {
                throw new Exception("You must request at least 1 message");
            }
            else
            {
                DateTime start = DateTime.Now;
                SocketBulkMessage data = await Server.GetFromServer("SocketBulkMessage",
                    SocketBulkMessageContext.Default.SocketBulkMessage,
                    CancellationToken,
                    new KeyValuePair<string, string?>("id", Id.ToString()),
                    new KeyValuePair<string, string?>("messages", count.ToString()));
                Console.WriteLine($"Messages downloaded in {(DateTime.Now - start).TotalSeconds}");
                start = DateTime.Now;
                if (data is not null && !data.Error.HasValue)
                {
                    int num = Convert.ToInt32(Math.Ceiling((Environment.ProcessorCount * 5) * 2.0));
                    if (num == 0) num = 1;
                    string key = Server.EncryptionHandler.GetChannelKey(Id);
                    if (data.Messages is null) data.Messages = Array.Empty<MainSocketMessage>();
                    Parallel.ForEach(data.Messages, new ParallelOptions()
                    {
                        MaxDegreeOfParallelism = num
                    }, i =>
                    {
                        i.decrypt(key, CancellationToken);
                    });
                    key = null;
                    Console.WriteLine($"Messages decrypted in {(DateTime.Now - start).TotalSeconds}");
                    return await Task.FromResult(data.Messages.ToList().AsReadOnly() as IReadOnlyList<MainSocketMessage>);
                }
                else
                {
                    throw data?.Error switch
                    {
                        ErrorCode.InvalidToken => new Exception("Your current token is no longer valid"),
                        ErrorCode.ServerError => new Exception($"Error from server: {data.ErrorMessage}"),
                        ErrorCode.InvalidHeader => new Exception(data.ErrorMessage),
                        ErrorCode.MissingHeader => new Exception("The header sent to the server was not found. This may be because you app is couropt or you are using the wron API version"),
                        ErrorCode.Forbidden => new Exception("You are not allowed to do this request"),
                        _ => new Exception(data?.Error.ToString()),
                    };
                }
            }
        }
        catch (Exception)
        {
            throw;
        }
    }

    public async Task<Task> SendMessage(string Message, CancellationToken CancellationToken, params File?[] Files)
    {
        string key = Server.EncryptionHandler.GetChannelKey(Id);
        JsonTypes.HTTP.Message m = new()
        {
            Context = Convert.ToBase64String(Encryption.RSA.Encrypt(Message, key, EncoderType.UTF8)),
            Channel = Id,
        };
        if (Files is not null && Files.Length > 0)
        {
            List<long> bb = new();
            for (int i = 0; i < Files.Length; i++)
            {
                File? ff = Files[i];
                if (ff is not null)
                {
                    bb.Add(await ff.Upload(key, CancellationToken));
                    Files[i] = null;
                }
            }
            m.Files = bb.ToArray();
        }
        IncomingHTTP data = await Server.SendServer("socketmessage", m, net.JsonTypes.HTTP.MessageContext.Default.Message, IncomingHTTPContext.Default.IncomingHTTP, CancellationToken);
        if (data.Error is not null && data.ErrorMessage != "Server responded with empty data") throw new Exception(data.ErrorMessage);
        return Task.CompletedTask;
    }
}

[JsonSerializable(typeof(MainSocketTextChannel))]
[JsonSourceGenerationOptions(
    GenerationMode = JsonSourceGenerationMode.Default,
    PropertyNamingPolicy = JsonKnownNamingPolicy.CamelCase,
    WriteIndented = false)]
internal partial class MainSocketTextChannelContext : JsonSerializerContext
{

}