using System.Buffers.Binary;
using System.Net;
using Common;
using Microsoft.AspNetCore.Cors;
using Microsoft.AspNetCore.Mvc;
using Newtonsoft.Json;
using WebProtocol;

namespace server.Controllers;

/// <summary>
/// UDP API 
/// </summary>
[ApiController]
[Route("api/[controller]")]
public class UDPController : ControllerBase
{
    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();

    private const string LOCALHOST = "127.0.0.1";

    /// <summary>
    /// 页面
    /// </summary>
    [HttpGet]
    public string Index()
    {
        return "This is UDP Controller";
    }

    /// <summary>
    /// 发送字符串
    /// </summary>
    /// <param name="address">IPV4 或者 IPV6 地址</param>
    /// <param name="port">设备端口号</param>
    /// <param name="text">发送的文本</param>
    /// <response code="200">发送成功</response>
    /// <response code="500">发送失败</response>
    [HttpPost("SendString")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> SendString(string address = LOCALHOST, int port = 1234, string text = "Hello Server!")
    {
        var endPoint = new IPEndPoint(IPAddress.Parse(address), port);
        var ret = await UDPClientPool.SendStringAsync(endPoint, [text]);

        if (ret) { return TypedResults.Ok(); }
        else { return TypedResults.InternalServerError(); }
    }

    /// <summary>
    /// 发送二进制数据
    /// </summary>
    /// <param name="address" example="127.0.0.1">IPV4 或者 IPV6 地址</param>
    /// <param name="port" example="1234">设备端口号</param>
    /// <param name="bytes" example="FFFFAAAA">16进制文本</param>
    [HttpPost("SendBytes")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> SendBytes(string address, int port, string bytes)
    {
        var endPoint = new IPEndPoint(IPAddress.Parse(address), port);
        var ret = await UDPClientPool.SendBytesAsync(endPoint, Number.StringToBytes(bytes));

        if (ret) { return TypedResults.Ok(); }
        else { return TypedResults.InternalServerError(); }
    }


    /// <summary>
    /// 发送地址包
    /// </summary>
    /// <param name="address">IP地址</param>
    /// <param name="port">UDP 端口号</param>
    /// <param name="opts">地址包选项</param>
    [HttpPost("SendAddrPackage")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> SendAddrPackage(
        string address,
        int port,
        [FromBody] SendAddrPackOptions opts)
    {
        var endPoint = new IPEndPoint(IPAddress.Parse(address), port);
        var ret = await UDPClientPool.SendAddrPackAsync(endPoint, new WebProtocol.SendAddrPackage(opts));

        if (ret) { return TypedResults.Ok(); }
        else { return TypedResults.InternalServerError(); }
    }

    /// <summary>
    /// 发送数据包
    /// </summary>
    /// <param name="address">IP地址</param>
    /// <param name="port">UDP 端口号</param>
    /// <param name="data">16进制数据</param>
    [HttpPost("SendDataPackage")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> SendDataPackage(string address, int port, string data)
    {
        var endPoint = new IPEndPoint(IPAddress.Parse(address), port);
        var ret = await UDPClientPool.SendDataPackAsync(endPoint,
            new WebProtocol.SendDataPackage(Number.StringToBytes(data)));

        if (ret) { return TypedResults.Ok(); }
        else { return TypedResults.InternalServerError(); }
    }

    /// <summary>
    /// 获取指定IP地址接受的数据列表
    /// </summary>
    /// <param name="address">IP地址</param>
    [HttpGet("GetRecvDataArray")]
    [ProducesResponseType(typeof(List<UDPData>), StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> GetRecvDataArray(string address)
    {
        var ret = await MsgBus.UDPServer.GetDataArrayAsync(address);

        if (ret.HasValue)
        {
            var dataJson = JsonConvert.SerializeObject(ret.Value);
            logger.Debug($"Get Receive Successfully: {dataJson}");

            return TypedResults.Ok(ret.Value);
        }
        else
        {
            logger.Debug("Get Receive Failed");
            return TypedResults.InternalServerError();
        }
    }

}

/// <summary>
/// Jtag API
/// </summary>
[ApiController]
[Route("api/[controller]")]
public class JtagController : ControllerBase
{
    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();

    /// <summary>
    /// 页面
    /// </summary>
    [HttpGet]
    public string Index()
    {
        return "This is Jtag Controller";
    }

    /// <summary>
    /// 执行一个Jtag命令
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="port"> 设备端口 </param>
    /// <param name="hexDevAddr"> 16进制设备目的地址(Jtag) </param>
    /// <param name="hexCmd"> 16进制命令 </param>
    [HttpPost("RunCommand")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> RunCommand(string address, int port, string hexDevAddr, string hexCmd)
    {
        var jtagCtrl = new JtagClient.Jtag(address, port);
        var ret = await jtagCtrl.WriteFIFO(Convert.ToUInt32(hexDevAddr, 16), Convert.ToUInt32(hexCmd, 16));

        if (ret.IsSuccessful) { return TypedResults.Ok(ret.Value); }
        else { return TypedResults.InternalServerError(ret.Error); }
    }


    /// <summary>
    /// 获取Jtag ID Code
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="port"> 设备端口 </param>
    [HttpGet("GetDeviceIDCode")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> GetDeviceIDCode(string address, int port)
    {
        var jtagCtrl = new JtagClient.Jtag(address, port);
        var ret = await jtagCtrl.ReadIDCode();

        if (ret.IsSuccessful)
        {
            logger.Info($"Get device {address} ID code: 0x{ret.Value:X4}");
            return TypedResults.Ok(ret.Value);
        }
        else
        {
            logger.Error(ret.Error);
            return TypedResults.InternalServerError(ret.Error);
        }
    }

    /// <summary>
    /// 获取状态寄存器
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="port"> 设备端口 </param>
    [HttpGet("ReadStatusReg")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> ReadStatusReg(string address, int port)
    {
        var jtagCtrl = new JtagClient.Jtag(address, port);
        var ret = await jtagCtrl.ReadStatusReg();

        if (ret.IsSuccessful)
        {
            var binaryValue = Common.String.Reverse(Convert.ToString(ret.Value, 2).PadLeft(32, '0'));
            var decodeValue = new JtagClient.JtagStatusReg(ret.Value);
            logger.Info($"Read device {address} Status Register: \n\t 0b{binaryValue} \n\t {decodeValue}");
            return TypedResults.Ok(new
            {
                original = ret.Value,
                binaryValue,
                decodeValue,
            });
        }
        else
        {
            logger.Error(ret.Error);
            return TypedResults.InternalServerError(ret.Error);
        }
    }

    /// <summary>
    /// 上传比特流文件
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="file">比特流文件</param>
    [HttpPost("UploadBitstream")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    public async ValueTask<IResult> UploadBitstream(string address, IFormFile file)
    {
        if (file == null || file.Length == 0)
            return TypedResults.BadRequest("未选择文件");

        // 生成安全的文件名（避免路径遍历攻击）
        var fileName = Path.GetRandomFileName();
        var uploadsFolder = Path.Combine(Environment.CurrentDirectory, $"bitstream/{address}");

        // 如果存在文件，则删除原文件再上传
        if (Directory.Exists(uploadsFolder))
        {
            Directory.Delete(uploadsFolder, true);
        }
        Directory.CreateDirectory(uploadsFolder);

        var filePath = Path.Combine(uploadsFolder, fileName);

        using (var stream = new FileStream(filePath, FileMode.Create))
        {
            await file.CopyToAsync(stream);
        }

        logger.Info($"Device {address} Upload Bitstream Successfully");
        return TypedResults.Ok("Bitstream Upload Successfully");
    }

    /// <summary>
    /// 通过Jtag下载比特流文件
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="port"> 设备端口 </param>
    [HttpPost("DownloadBitstream")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> DownloadBitstream(string address, int port)
    {
        // 检查文件
        var fileDir = Path.Combine(Environment.CurrentDirectory, $"bitstream/{address}");
        if (!Directory.Exists(fileDir))
            return TypedResults.BadRequest("Empty bitstream, Please upload it first");

        try
        {
            // 读取文件
            var filePath = Directory.GetFiles(fileDir)[0];

            using (var fileStream = System.IO.File.Open(filePath, System.IO.FileMode.Open))
            {
                if (fileStream is null || fileStream.Length <= 0)
                    return TypedResults.BadRequest("Wrong bitstream, Please upload it again");

                // 定义缓冲区大小: 32KB
                byte[] buffer = new byte[32 * 1024];
                byte[] revBuffer = new byte[32 * 1024];
                long totalBytesRead = 0;

                // 使用异步流读取文件
                using (var memoryStream = new MemoryStream())
                {
                    int bytesRead;
                    while ((bytesRead = await fileStream.ReadAsync(buffer, 0, buffer.Length)) > 0)
                    {
                        // 反转 32bits
                        var retBuffer = Common.Number.ReverseBytes(buffer, 4);
                        if (!retBuffer.IsSuccessful)
                            return TypedResults.InternalServerError(retBuffer.Error);
                        revBuffer = retBuffer.Value;
                        for (int i = 0; i < buffer.Length; i++)
                        {
                            revBuffer[i] = BinaryPrimitives.ReverseEndianness(revBuffer[i]);
                        }

                        await memoryStream.WriteAsync(revBuffer, 0, bytesRead);
                        totalBytesRead += bytesRead;
                    }

                    // 将所有数据转换为字节数组（注意：如果文件非常大，可能不适合完全加载到内存）
                    var fileBytes = memoryStream.ToArray();

                    // 下载比特流
                    var jtagCtrl = new JtagClient.Jtag(address, port);
                    var ret = await jtagCtrl.DownloadBitstream(fileBytes);

                    if (ret.IsSuccessful)
                    {
                        logger.Info($"Device {address} dowload bitstream successfully");
                        return TypedResults.Ok(ret.Value);
                    }
                    else
                    {
                        logger.Error(ret.Error);
                        return TypedResults.InternalServerError(ret.Error);
                    }
                }

            }

        }
        catch (Exception error)
        {
            return TypedResults.InternalServerError(error);
        }
        finally
        {

        }
    }
}

/// <summary>
/// 远程更新
/// </summary>
[ApiController]
[Route("api/[controller]")]
public class RemoteUpdater : ControllerBase
{
    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();

    /// <summary>
    /// 上传远程更新比特流文件
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="file">比特流文件</param>
    [HttpPost("UploadBitstream")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    public async ValueTask<IResult> UploadBitstream(string address, IFormFile file)
    {
        if (file == null || file.Length == 0)
            return TypedResults.BadRequest("未选择文件");

        // 生成安全的文件名（避免路径遍历攻击）
        var fileName = Path.GetRandomFileName();
        var uploadsFolder = Path.Combine(Environment.CurrentDirectory, $"bitstream/RemoteUpdate/{address}");

        // 如果存在文件，则删除原文件再上传
        if (Directory.Exists(uploadsFolder))
        {
            Directory.Delete(uploadsFolder, true);
        }
        Directory.CreateDirectory(uploadsFolder);

        var filePath = Path.Combine(uploadsFolder, fileName);

        using (var stream = new FileStream(filePath, FileMode.Create))
        {
            await file.CopyToAsync(stream);
        }

        logger.Info($"Device {address} Upload Bitstream Successfully");
        return TypedResults.Ok("Bitstream Upload Successfully");
    }

    /// <summary>
    /// 远程更新比特流文件
    /// </summary>
    /// <param name="address"> 设备地址 </param>
    /// <param name="port"> 设备端口 </param>
    /// <param name="bitstreamNum"> 比特流位号 </param>
    [HttpPost("DownloadBitstream")]
    [ProducesResponseType(StatusCodes.Status200OK)]
    [ProducesResponseType(StatusCodes.Status400BadRequest)]
    [ProducesResponseType(StatusCodes.Status500InternalServerError)]
    public async ValueTask<IResult> UpdateBitstream(string address, int port, int bitstreamNum)
    {
        // 检查文件
        var fileDir = Path.Combine(Environment.CurrentDirectory, $"bitstream/RemoteUpdate/{address}");
        if (!Directory.Exists(fileDir))
            return TypedResults.BadRequest("Empty bitstream, Please upload it first");

        try
        {
            // 读取文件
            var filePath = Directory.GetFiles(fileDir)[0];

            using (var fileStream = System.IO.File.Open(filePath, System.IO.FileMode.Open))
            {
                if (fileStream is null || fileStream.Length <= 0)
                    return TypedResults.BadRequest("Wrong bitstream, Please upload it again");

                // 定义缓冲区大小: 32KB
                byte[] buffer = new byte[32 * 1024];
                byte[] revBuffer = new byte[32 * 1024];
                long totalBytesRead = 0;

                // 使用异步流读取文件
                using (var memoryStream = new MemoryStream())
                {
                    int bytesRead;
                    while ((bytesRead = await fileStream.ReadAsync(buffer, 0, buffer.Length)) > 0)
                    {
                        // 反转 32bits
                        var retBuffer = Common.Number.ReverseBytes(buffer, 4);
                        if (!retBuffer.IsSuccessful)
                            return TypedResults.InternalServerError(retBuffer.Error);
                        revBuffer = retBuffer.Value;
                        for (int i = 0; i < buffer.Length; i++)
                        {
                            revBuffer[i] = BinaryPrimitives.ReverseEndianness(revBuffer[i]);
                        }

                        await memoryStream.WriteAsync(revBuffer, 0, bytesRead);
                        totalBytesRead += bytesRead;
                    }

                    // 将所有数据转换为字节数组（注意：如果文件非常大，可能不适合完全加载到内存）
                    var restStreamLen = memoryStream.Length % (4 * 1024);
                    if (restStreamLen != 0)
                    {
                        var appendLen = ((int)(4 * 1024 - restStreamLen));
                        var bytesAppend = new byte[appendLen];
                        Array.Fill<byte>(bytesAppend, 0xFF);
                        await memoryStream.WriteAsync(bytesAppend, 0, appendLen);
                    }
                    var fileBytes = memoryStream.ToArray();

                    // 下载比特流
                    var remoteUpdater = new RemoteUpdate.RemoteUpdateClient(address, port);
                    var ret = await remoteUpdater.UpdateBitstream(bitstreamNum, fileBytes);

                    if (ret.IsSuccessful)
                    {
                        logger.Info($"Device {address} Update bitstream successfully");
                        return TypedResults.Ok(ret.Value);
                    }
                    else
                    {
                        logger.Error(ret.Error);
                        return TypedResults.InternalServerError(ret.Error);
                    }
                }

            }

        }
        catch (Exception error)
        {
            return TypedResults.InternalServerError(error);
        }
    }
}


/// <summary>
/// 数据控制器
/// </summary>
[ApiController]
[Route("api/[controller]")]
public class Data : ControllerBase
{
    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();

    /// <summary>
    /// 创建数据库表
    /// </summary>
    /// <returns>插入的记录数</returns>
    [EnableCors("Development")]
    [HttpPost("CreateTable")]
    public IResult CreateTables()
    {
        using var db = new Database.AppDataConnection();
        db.CreateAllTables();
        return TypedResults.Ok();
    }

    /// <summary>
    /// 删除数据库表
    /// </summary>
    /// <returns>插入的记录数</returns>
    [EnableCors("Development")]
    [HttpDelete("DropTables")]
    public IResult DropTables()
    {
        using var db = new Database.AppDataConnection();
        db.DropAllTables();
        return TypedResults.Ok();
    }

    /// <summary>
    /// 获取所有用户
    /// </summary>
    /// <returns>用户列表</returns>
    [HttpGet("AllUsers")]
    public IResult AllUsers()
    {
        using var db = new Database.AppDataConnection();
        var ret = db.User.ToList();
        return TypedResults.Ok(ret);
    }

    /// <summary>
    /// 注册新用户
    /// </summary>
    /// <param name="name">用户名</param>
    /// <returns>操作结果</returns>
    [HttpPost("SignUpUser")]
    public IResult SignUpUser(string name)
    {
        if (name.Length > 255)
            return TypedResults.BadRequest("Name Couln't over 255 characters");

        using var db = new Database.AppDataConnection();
        var ret = db.AddUser(name);
        return TypedResults.Ok(ret);
    }
}

/// <summary>
/// 日志控制器
/// </summary>
[ApiController]
[Route("api/[controller]")]
public class Log : ControllerBase
{
    private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();

    /// <summary>
    /// 日志文件路径
    /// </summary>
    private readonly string _logFilePath = Directory.GetFiles(Directory.GetCurrentDirectory())[0];

}