using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Rhea.Common;
using Microsoft.AspNetCore.Cors;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Newtonsoft.Json;
using System.IO;
using ICSharpCode.SharpZipLib.Zip;
using Tiger.IBusiness;

namespace Tiger.Api.Controllers.Base
{
    /// <summary>
    /// Api自动更新服务
    /// </summary>
    [Route("api/[controller]/[action]")]
    [EnableCors("Any")]
    [ApiController]
    public partial class UpgradeController : ControllerBase
    {
        /// <summary>
        /// 获取文件列表
        /// </summary>
        /// <param name="path"></param>
        /// <returns></returns>
        [HttpGet]
        public async Task<IActionResult> GetFileListAsync(string? path)
        {
            var filePath = UpgradeFileSystem.UpgradeRootPath + path;
            var allFiles = new List<UpgradeFileSystem>();
            await Task.Run(() =>
            {
                if (Directory.Exists(filePath))
                {
                    allFiles = new UpgradeFileSystem(new DirectoryInfo(filePath)).ChildNodes;
                }
                else if (System.IO.File.Exists(filePath))
                {
                    allFiles.Add(new UpgradeFileSystem(new FileInfo(filePath)));
                }
            });
            return Ok(allFiles);
        }

        /// <summary>
        /// 获取文件
        /// </summary>
        /// <param name="path"></param>
        /// <returns></returns>
        [HttpGet]
        public async Task<IActionResult> GetFileAsync(string? path)
        {
            var filePath = UpgradeFileSystem.UpgradeRootPath + path;
            if (System.IO.File.Exists(filePath))
            {
                return await GetFileStreamResultAsync(new FileInfo(filePath));
                //return await SegmentDownloadFileAsync(new FileInfo(filePath));
            }
            return Ok(null);
        }

        /// <summary>
        /// 获取..\AutoUpgrade.exe
        /// </summary>
        /// <param name="path"></param>
        /// <returns></returns>
        [HttpGet]
        public async Task<IActionResult> GetAutoUpgradeProgramAsync()
        {
            var filePath = UpgradeFileSystem.UpgradeRootPath + "AutoUpgrade.exe";
            if (System.IO.File.Exists(filePath))
            {
                return await GetFileStreamResultAsync(new FileInfo(filePath));
            }
            return Ok($"请先把AutoUpgrade.exe放到更新目录的[{UpgradeFileSystem.UpgradeRootPath}]根目录下");
        }

        /// <summary>
        /// 获取..\AutoUpgrade.exe.config
        /// </summary>
        /// <param name="path"></param>
        /// <returns></returns>
        [HttpGet]
        public async Task<IActionResult> GetAutoUpgradeConfigAsync()
        {
            var filePath = UpgradeFileSystem.UpgradeRootPath + "AutoUpgrade.exe.config";
            if (System.IO.File.Exists(filePath))
            {
                return await GetFileStreamResultAsync(new FileInfo(filePath));
            }
            return Ok($"请先把AutoUpgrade.exe.config放到更新目录的[{UpgradeFileSystem.UpgradeRootPath}]根目录下");
        }

        /// <summary>
        /// 上传文件
        /// </summary>
        /// <param name="fileName"></param>
        /// <param name="filePath"></param>
        /// <returns></returns>
        [HttpPost]
        public JsonResult UploadFile([FromForm] string fileName, [FromForm] string filePath)
        {
            var files = Request.Form.Files;
            var msgResult = new List<string>();
            if (files.Count > 0)
            {
                foreach (var formFile in files)
                {
                    var fullPath = $"{filePath}\\{fileName}";
                    //save
                    using var stream = new FileStream(fullPath, FileMode.Create);
                    formFile.CopyTo(stream);
                    stream.Close();
                    stream.Dispose();
                    //unzip
                    var unZipDir = fullPath.Replace(Path.GetFileName(fullPath), "").Trim('\\');
                    //System.IO.Compression.ZipFile.ExtractToDirectory(fullPath, unZipDir);
                    var isOK = UnZip(fullPath, unZipDir);
                    //delete compress package
                    System.IO.File.Delete(fullPath);
                    msgResult.Add($"【{fileName}】上传成功，{(isOK ? "解压成功" : "解压失败")}");
                }
            }
            return new JsonResult(msgResult);
        }

        #region Function
        /// <summary>
        /// 获取到文件流结果
        /// </summary>
        /// <param name="fileInfo">FileInfo文件</param>
        /// <returns>返回FileStreamResult</returns>
        private async Task<IActionResult> GetFileStreamResultAsync(FileInfo fileInfo)
        {
            if (fileInfo == null) return NotFound();

            var memoryStream = new MemoryStream();
            using (var stream = new FileStream(fileInfo.FullName, FileMode.Open))
            {
                await stream.CopyToAsync(memoryStream);
            }
            memoryStream.Seek(0, SeekOrigin.Begin);

            //文件名必须编码，否则会有特殊字符(如中文)无法在此下载。
            string encodeFilename = System.Net.WebUtility.UrlEncode(fileInfo.Name);
            Response.Headers.Add("Content-Disposition", "attachment; filename=" + encodeFilename);
            return new FileStreamResult(memoryStream, "application/octet-stream");//文件流方式，指定文件流对应的ContenType。
        }

        /// <summary>
        /// 获取到文件流结果
        /// </summary>
        /// <param name="fileInfo">FileInfo文件</param>
        /// <returns>返回FileStreamResult</returns>
        private IActionResult GetFileStreamResult(FileInfo fileInfo)
        {
            if (fileInfo == null) return NotFound();

            var memoryStream = new MemoryStream();
            using (var stream = new FileStream(fileInfo.FullName, FileMode.Open))
            {
                stream.CopyTo(memoryStream);
            }
            memoryStream.Seek(0, SeekOrigin.Begin);

            //文件名必须编码，否则会有特殊字符(如中文)无法在此下载。
            string encodeFilename = System.Net.WebUtility.UrlEncode(fileInfo.Name);
            Response.Headers.Add("Content-Disposition", "attachment; filename=" + encodeFilename);
            return new FileStreamResult(memoryStream, "application/octet-stream");//文件流方式，指定文件流对应的ContenType。
        }

        /// <summary>
        /// 异步分段下载数据
        /// </summary>
        /// <param name="fileInfo">FileInfo文件</param>
        /// <returns>返回响应结果</returns>
        private async Task<IActionResult> SegmentDownloadFileAsync(FileInfo fileInfo)
        {
            if (fileInfo == null) return NotFound();

            int index = 0;

            using (FileStream fs = new FileStream(fileInfo.FullName, FileMode.Open))
            {
                if (fs.Length <= 0)
                {
                    return Ok(new { code = -1, msg = "文件尚未处理" });
                }
                int shardSize = 1 * 1024 * 1024;//一次1M
                int count = (int)(fs.Length / shardSize);

                if ((fs.Length % shardSize) > 0)
                {
                    count += 1;
                }
                if (index > count - 1)
                {
                    return Ok(new { code = -1, msg = "无效的下标" });
                }

                fs.Seek(index * shardSize, SeekOrigin.Begin);

                if (index == count - 1)
                {
                    //最后一片 = 总长 - (每次片段大小 * 已下载片段个数)
                    shardSize = (int)(fs.Length - (shardSize * index));
                }
                byte[] datas = new byte[shardSize];
                await fs.ReadAsync(datas.AsMemory(0, datas.Length));

                return File(datas, "application/octet-stream");
            }
        }

        /// <summary>
        /// 分段下载数据
        /// </summary>
        /// <param name="fileInfo">FileInfo文件</param>
        /// <returns>返回响应结果</returns>
        private IActionResult SegmentDownloadFile(FileInfo fileInfo)
        {
            if (fileInfo == null) return NotFound();

            int index = 0;

            using (FileStream fs = new FileStream(fileInfo.FullName, FileMode.Open))
            {
                if (fs.Length <= 0)
                {
                    return Ok(new { code = -1, msg = "文件尚未处理" });
                }
                int shardSize = 1 * 1024 * 1024;//一次1M
                int count = (int)(fs.Length / shardSize);

                if ((fs.Length % shardSize) > 0)
                {
                    count += 1;
                }
                if (index > count - 1)
                {
                    return Ok(new { code = -1, msg = "无效的下标" });
                }

                fs.Seek(index * shardSize, SeekOrigin.Begin);

                if (index == count - 1)
                {
                    //最后一片 = 总长 - (每次片段大小 * 已下载片段个数)
                    shardSize = (int)(fs.Length - (shardSize * index));
                }
                byte[] datas = new byte[shardSize];
                fs.Read(datas, 0, datas.Length);

                return File(datas, "application/octet-stream");
            }
        }
        /// <summary>  
        /// 功能：解压zip格式的文件。  
        /// </summary>  
        /// <param name="zipFilePath">压缩文件路径</param>  
        /// <param name="unZipDir">解压文件存放路径,为空时默认与压缩文件同一级目录下，跟压缩文件同名的文件夹</param>  
        /// <returns>解压是否成功</returns>  
        private bool UnZip(string zipFilePath, string unZipDir)
        {
            try
            {
                if (zipFilePath == string.Empty)
                {
                    throw new Exception("压缩文件不能为空！");
                }
                if (!System.IO.File.Exists(zipFilePath))
                {
                    throw new FileNotFoundException("压缩文件不存在！");
                }
                //解压文件夹为空时默认与压缩文件同一级目录下，跟压缩文件同名的文件夹  
                if (unZipDir == string.Empty)
                    unZipDir = zipFilePath.Replace(Path.GetFileName(zipFilePath), Path.GetFileNameWithoutExtension(zipFilePath));
                if (!unZipDir.EndsWith("/"))
                    unZipDir += "/";
                if (!Directory.Exists(unZipDir))
                    Directory.CreateDirectory(unZipDir);
                using (var s = new ZipInputStream(System.IO.File.OpenRead(zipFilePath)))
                {

                    ZipEntry theEntry;
                    while ((theEntry = s.GetNextEntry()) != null)
                    {
                        string directoryName = Path.GetDirectoryName(theEntry.Name);
                        string fileName = Path.GetFileName(theEntry.Name);
                        if (!string.IsNullOrEmpty(directoryName))
                        {
                            Directory.CreateDirectory(unZipDir + directoryName);
                        }
                        if (directoryName != null && !directoryName.EndsWith("/"))
                        {
                        }
                        if (fileName != string.Empty)
                        {
                            using FileStream streamWriter = System.IO.File.Create(unZipDir + theEntry.Name);

                            int size;
                            byte[] data = new byte[2048];
                            while (true)
                            {
                                size = s.Read(data, 0, data.Length);
                                if (size > 0)
                                {
                                    streamWriter.Write(data, 0, size);
                                }
                                else
                                {
                                    break;
                                }
                            }
                        }
                    }
                }
                return true;
            }
            catch (Exception ex)
            {
                //_logger.LogError(ex, $"解压zip格式的文件失败：{zipFilePath}");
                return false;
            }
        }


        #endregion

    }//endClass

}