using System;
using System.Collections.Generic;
using System.Net.Sockets;
using System.Threading;

namespace DTLib.Network
{
    //
    // отправка/получение пакетов
    //
    public static class Package
    {
        // принимает пакет
        public static byte[] GetPackage(this Socket socket)
        {
            int packageSize = 0;
            byte[] data = new byte[2];
            // цикл выполняется пока не пройдёт 2000 мс
            for(ushort s = 0; s<400; s+=1)
            {
                if(packageSize==0&&socket.Available>=2)
                {
                    socket.Receive(data, data.Length, 0);
                    packageSize=data.ToInt();
                }
                if(packageSize!=0&&socket.Available>=packageSize)
                {
                    data=new byte[packageSize];
                    socket.Receive(data, data.Length, 0);
                    return data;
                }
                else
                    Thread.Sleep(5);
            }
            throw new Exception($"GetPackage() error: timeout. socket.Available={socket.Available}\n");
        }

        // отправляет пакет
        public static void SendPackage(this Socket socket, byte[] data)
        {
            if(data.Length>65536)
                throw new Exception($"SendPackage() error: package is too big ({data.Length} bytes)");
            if(data.Length==0)
                throw new Exception($"SendPackage() error: package has zero size");
            var list = new List<byte>();
            byte[] packageSize = data.Length.ToBytes();
            if(packageSize.Length==1)
                list.Add(0);
            list.AddRange(packageSize);
            list.AddRange(data);
            socket.Send(list.ToArray());
        }
        public static void SendPackage(this Socket socket, string data) => SendPackage(socket, data.ToBytes());

        // получает пакет и выбрасывает исключение, если пакет не соответствует образцу
        public static void GetAnswer(this Socket socket, string answer)
        {
            string rec = socket.GetPackage().BytesToString();
            if(rec!=answer)
                throw new Exception($"GetAnswer() error: invalid answer: <{rec}>");
        }

        public static byte[] RequestPackage(this Socket socket, byte[] request)
        {
            socket.SendPackage(request);
            return socket.GetPackage();
        }
        public static byte[] RequestPackage(this Socket socket, string request) => socket.RequestPackage(request.ToBytes());
    }
}