namespace DTLib.Network;

//
// передача файлов по сети
//
public class FSP
{
    public int BufferSize { get; init; } = 512 * 1024;
    
    readonly byte[] buffer;
    
    Socket MainSocket { get; init; }
    public FSP(Socket _mainSocket)
    {
        buffer = new byte[BufferSize];
        MainSocket = _mainSocket;
    }

    public long BytesDownloaded { get; private set; }
    public long BytesUploaded { get; private set; }
    public long Filesize { get; private set; }

    // скачивает файл с помощью FSP протокола
    public void DownloadFile(IOPath filePath_server, IOPath filePath_client)
    {
        filePath_server.ThrowIfEscapes();
        filePath_client.ThrowIfEscapes();
        lock (MainSocket)
        {
            MainSocket.SendPackage("requesting file download");
            MainSocket.SendPackage(filePath_server.Str);
        }
        DownloadFile(filePath_client);
    }

    public void DownloadFile(IOPath filePath_client)
    {
        filePath_client.ThrowIfEscapes();
        using System.IO.Stream fileStream = File.OpenWrite(filePath_client);
        Download_SharedCode(fileStream);
    }

    public byte[] DownloadFileToMemory(IOPath filePath_server)
    {
        filePath_server.ThrowIfEscapes();
        lock (MainSocket)
        {
            MainSocket.SendPackage("requesting file download");
            MainSocket.SendPackage(filePath_server.Str);
        }
        return DownloadFileToMemory();
    }

    public byte[] DownloadFileToMemory()
    {
        using var fileStream = new System.IO.MemoryStream();
        Download_SharedCode(fileStream);
        return fileStream.ToArray();
    }

    private void Download_SharedCode(System.IO.Stream fileStream)
    {
        lock (MainSocket)
        {
            BytesDownloaded = 0;
            Filesize = BitConverter.ToInt64(MainSocket.GetPackage(), 0);
            if (Filesize < 0)
                throw new Exception("FileSize < 0");
            MainSocket.SendPackage("ready");
            int recievedCount;
            do
            {
                recievedCount = MainSocket.Receive(buffer);
                fileStream.Write(buffer, 0, recievedCount);
                BytesDownloaded += recievedCount;
            } while (recievedCount == buffer.Length);

            if (BytesDownloaded != Filesize)
                throw new Exception($"expected {Filesize} bytes, but downloaded {BytesDownloaded} bytes");
        }
    }

    // отдаёт файл с помощью FSP протокола
    public void UploadFile(IOPath filePath)
    {
        filePath.ThrowIfEscapes();
        BytesUploaded = 0;
        using System.IO.FileStream fileStream = File.OpenRead(filePath);
        Filesize = fileStream.Length;
        lock (MainSocket)
        {
            MainSocket.SendPackage(BitConverter.GetBytes(Filesize));
            MainSocket.GetAnswer("ready");
            int readCount;
            do
            {
                readCount = fileStream.Read(buffer, 0, buffer.Length);
                MainSocket.Send(buffer, 0, readCount, SocketFlags.None);
                BytesUploaded += readCount;
            } while (readCount == buffer.Length);
            
            if (BytesUploaded != Filesize)
                throw new Exception($"expected {Filesize} bytes, but uploaded {BytesDownloaded} bytes");
        }
    }
}