ModernUO/Projects/Server/Serialization/BufferReader.cs

/*************************************************************************
 * ModernUO                                                              *
 * Copyright 2019-2020 - ModernUO Development Team                       *
 * Email: hi@modernuo.com                                                *
 * File: BufferReader.cs                                                 *
 *                                                                       *
 * This program is free software: you can redistribute it and/or modify  *
 * it under the terms of the GNU General Public License as published by  *
 * the Free Software Foundation, either version 3 of the License, or     *
 * (at your option) any later version.                                   *
 *                                                                       *
 * You should have received a copy of the GNU General Public License     *
 * along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 *************************************************************************/

using System;
using System.Buffers.Binary;
using System.Collections.Generic;
using System.Net;
using System.Text;
using Server.Guilds;

namespace Server
{
    public class BufferReader : IGenericReader
    {
        private readonly Encoding _encoding;
        private ArraySegment<byte> _segment;
        public int Position { get; private set; }

        public BufferReader(ArraySegment<byte> segment)
        {
            _segment = segment;
            _encoding = Utility.UTF8;
        }

        public string ReadString()
        {
            if (!ReadBool())
            {
                return null;
            }

            var length = ReadEncodedInt();
            var s = length == 0 ? "" : _encoding.GetString(_segment.AsSpan(Position, length));
            Position += length;
            return s;
        }

        public DateTime ReadDateTime() => new DateTime(ReadLong());

        public DateTimeOffset ReadDateTimeOffset() => new DateTimeOffset(ReadLong(), ReadTimeSpan());

        public TimeSpan ReadTimeSpan() => new TimeSpan(ReadLong());

        public DateTime ReadDeltaTime()
        {
            var ticks = ReadLong();
            var now = DateTime.UtcNow.Ticks;

            if (ticks > 0 && ticks + now < 0)
            {
                return DateTime.MaxValue;
            }

            if (ticks < 0 && ticks + now < 0)
            {
                return DateTime.MinValue;
            }

            try
            {
                return new DateTime(now + ticks);
            }
            catch
            {
                return ticks > 0 ? DateTime.MaxValue : DateTime.MinValue;
            }
        }

        public decimal ReadDecimal() => new decimal(new[] { ReadInt(), ReadInt(), ReadInt(), ReadInt() });

        public long ReadLong()
        {
            var v = BinaryPrimitives.ReadInt64LittleEndian(_segment.AsSpan(Position, 8));
            Position += 8;
            return v;
        }

        public ulong ReadULong()
        {
            var v = BinaryPrimitives.ReadUInt64LittleEndian(_segment.AsSpan(Position, 8));
            Position += 8;
            return v;
        }

        public int ReadInt()
        {
            var v = BinaryPrimitives.ReadInt32LittleEndian(_segment.AsSpan(Position, 4));
            Position += 4;
            return v;
        }

        public uint ReadUInt()
        {
            var v = BinaryPrimitives.ReadUInt32LittleEndian(_segment.AsSpan(Position, 4));
            Position += 4;
            return v;
        }

        public short ReadShort()
        {
            var v = BinaryPrimitives.ReadInt16LittleEndian(_segment.AsSpan(Position, 2));
            Position += 2;
            return v;
        }

        public ushort ReadUShort()
        {
            var v = BinaryPrimitives.ReadUInt16LittleEndian(_segment.AsSpan(Position, 2));
            Position += 2;
            return v;
        }

        public double ReadDouble()
        {
            var v = BinaryPrimitives.ReadDoubleLittleEndian(_segment.AsSpan(Position, 8));
            Position += 8;
            return v;
        }

        public float ReadFloat()
        {
            var v = BinaryPrimitives.ReadSingleLittleEndian(_segment.AsSpan(Position, 4));
            Position += 4;
            return v;
        }

        public byte ReadByte() => _segment[Position++];

        public sbyte ReadSByte() => (sbyte)_segment[Position++];

        public bool ReadBool() => _segment[Position++] != 0;

        public int ReadEncodedInt()
        {
            int v = 0, shift = 0;
            byte b;

            do
            {
                b = ReadByte();
                v |= (b & 0x7F) << shift;
                shift += 7;
            } while (b >= 0x80);

            return v;
        }

        public IPAddress ReadIPAddress() => new IPAddress(ReadLong());

        public Point3D ReadPoint3D() => new Point3D(ReadInt(), ReadInt(), ReadInt());

        public Point2D ReadPoint2D() => new Point2D(ReadInt(), ReadInt());

        public Rectangle2D ReadRect2D() => new Rectangle2D(ReadPoint2D(), ReadPoint2D());

        public Rectangle3D ReadRect3D() => new Rectangle3D(ReadPoint3D(), ReadPoint3D());

        public Map ReadMap() => Map.Maps[ReadByte()];

        public IEntity ReadEntity()
        {
            Serial serial = ReadUInt();
            return World.FindEntity(serial) ?? new Entity(serial, new Point3D(0, 0, 0), Map.Internal);
        }

        public Item ReadItem() => World.FindItem(ReadUInt());

        public Mobile ReadMobile() => World.FindMobile(ReadUInt());

        public BaseGuild ReadGuild() => World.FindGuild(ReadUInt());

        public T ReadItem<T>() where T : Item => ReadItem() as T;

        public T ReadMobile<T>() where T : Mobile => ReadMobile() as T;

        public T ReadGuild<T>() where T : BaseGuild => ReadGuild() as T;

        public List<Item> ReadStrongItemList() => ReadStrongItemList<Item>();

        public List<T> ReadStrongItemList<T>() where T : Item
        {
            var count = ReadInt();

            var list = new List<T>(count);

            for (var i = 0; i < count; ++i)
            {
                if (ReadItem() is T item)
                {
                    list.Add(item);
                }
            }

            return list;
        }

        public HashSet<Item> ReadItemSet() => ReadItemSet<Item>();

        public HashSet<T> ReadItemSet<T>() where T : Item
        {
            var count = ReadInt();

            var set = new HashSet<T>();

            for (var i = 0; i < count; ++i)
            {
                if (ReadItem() is T item)
                {
                    set.Add(item);
                }
            }

            return set;
        }

        public List<Mobile> ReadStrongMobileList() => ReadStrongMobileList<Mobile>();

        public List<T> ReadStrongMobileList<T>() where T : Mobile
        {
            var count = ReadInt();

            var list = new List<T>(count);

            for (var i = 0; i < count; ++i)
            {
                if (ReadMobile() is T m)
                {
                    list.Add(m);
                }
            }

            return list;
        }

        public HashSet<Mobile> ReadMobileSet() => ReadMobileSet<Mobile>();

        public HashSet<T> ReadMobileSet<T>() where T : Mobile
        {
            var count = ReadInt();
            var set = new HashSet<T>();

            for (var i = 0; i < count; ++i)
            {
                if (ReadMobile() is T item)
                {
                    set.Add(item);
                }
            }

            return set;
        }

        public List<BaseGuild> ReadStrongGuildList() => ReadStrongGuildList<BaseGuild>();

        public List<T> ReadStrongGuildList<T>() where T : BaseGuild
        {
            var count = ReadInt();
            var list = new List<T>(count);

            if (count > 0)
            {
                for (var i = 0; i < count; ++i)
                {
                    if (ReadGuild() is T g)
                    {
                        list.Add(g);
                    }
                }
            }

            return list;
        }

        public HashSet<BaseGuild> ReadGuildSet() => ReadGuildSet<BaseGuild>();

        public HashSet<T> ReadGuildSet<T>() where T : BaseGuild
        {
            var count = ReadInt();

            var set = new HashSet<T>();

            if (count > 0)
            {
                for (var i = 0; i < count; ++i)
                {
                    if (ReadGuild() is T item)
                    {
                        set.Add(item);
                    }
                }
            }

            return set;
        }

        public Race ReadRace() => Race.Races[ReadByte()];

        public int Read(Span<byte> buffer)
        {
            var length = buffer.Length;
            if (length > _segment.Count - Position)
            {
                throw new OutOfMemoryException();
            }

            _segment.AsSpan(Position, length).CopyTo(buffer);
            Position += length;
            return length;
        }
    }
}