Minecraft-Console-Client/MinecraftClient/Protocol/Session/SessionToken.cs

using System;
using System.IO;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using MessagePack;
using MinecraftClient.Scripting;
using static MinecraftClient.Settings.MainConfigHelper.MainConfig.GeneralConfig;

namespace MinecraftClient.Protocol.Session
{
    [Serializable]
    [MessagePackObject]
    public class SessionToken
    {
        private static readonly Regex JwtRegex = new("^[A-Za-z0-9-_]+\\.[A-Za-z0-9-_]+\\.[A-Za-z0-9-_]+$");

        [Key(0)]
        public string ID { get; set; }
        [Key(1)]
        public string PlayerName { get; set; }
        [Key(2)]
        public string PlayerID { get; set; }
        [Key(3)]
        public string ClientID { get; set; }
        [Key(4)]
        public string RefreshToken { get; set; }
        [Key(5)]
        public string ServerIDhash { get; set; }
        [Key(6)]
        public byte[]? ServerPublicKey { get; set; }
        
        [IgnoreMember]
        public Task<bool>? SessionPreCheckTask = null;

        public SessionToken()
        {
            ID = String.Empty;
            PlayerName = String.Empty;
            PlayerID = String.Empty;
            ClientID = String.Empty;
            RefreshToken = String.Empty;
            ServerIDhash = String.Empty;
            ServerPublicKey = null;
        }

        public bool SessionPreCheck(LoginType type)
        {
            if (ID == string.Empty || PlayerID == String.Empty || ServerPublicKey is null)
                return false;
            Crypto.CryptoHandler.ClientAESPrivateKey ??= Crypto.CryptoHandler.GenerateAESPrivateKey();
            string serverHash = Crypto.CryptoHandler.GetServerHash(ServerIDhash, ServerPublicKey, Crypto.CryptoHandler.ClientAESPrivateKey);
            if (ProtocolHandler.SessionCheck(PlayerID, ID, serverHash, type))
                return true;
            return false;
        }

        public override string ToString()
        {
            return String.Join(",", ID, PlayerName, PlayerID, ClientID, RefreshToken, ServerIDhash,
                (ServerPublicKey is null) ? String.Empty : Convert.ToBase64String(ServerPublicKey));
        }

        public static SessionToken FromString(string tokenString)
        {
            string[] fields = tokenString.Split(',');
            if (fields.Length < 4)
                throw new InvalidDataException("Invalid string format");

            SessionToken session = new()
            {
                ID = fields[0],
                PlayerName = fields[1],
                PlayerID = fields[2],
                ClientID = fields[3]
            };
            // Backward compatible with old session file without refresh token field
            if (fields.Length > 4)
                session.RefreshToken = fields[4];
            else
                session.RefreshToken = String.Empty;
            if (fields.Length > 5)
                session.ServerIDhash = fields[5];
            else
                session.ServerIDhash = String.Empty;
            if (fields.Length > 6)
            {
                try
                {
                    session.ServerPublicKey = Convert.FromBase64String(fields[6]);
                }
                catch
                {
                    session.ServerPublicKey = null;
                }
            }
            else
                session.ServerPublicKey = null;
            if (!JwtRegex.IsMatch(session.ID))
                throw new InvalidDataException("Invalid session ID");
            if (!ChatBot.IsValidName(session.PlayerName))
                throw new InvalidDataException("Invalid player name");
            if (!Guid.TryParseExact(session.PlayerID, "N", out _))
                throw new InvalidDataException("Invalid player ID");
            if (!Guid.TryParseExact(session.ClientID, "N", out _))
                throw new InvalidDataException("Invalid client ID");
            // No validation on refresh token because it is custom format token (not Jwt)

            return session;
        }
    }
}