dotnet/taobao-sdk-netcore-auto_1719572210878-20240628-source/Util/AtsUtils.cs

using System.Collections.Generic;
using System.IO;
using System.IO.Compression;
using System.Net;
using System.Security.Cryptography;
using System.Text;
using System.Text.RegularExpressions;
using ICSharpCode.SharpZipLib.Zip;

namespace Top.Api.Util
{
    /// <summary>
    /// 异步API下载工具类。
    /// </summary>
    public abstract class AtsUtils
    {
        private const string CTYPE_OCTET = "application/octet-stream";
        private static Regex regex = new Regex("attachment;filename=\"([\\w\\-]+)\"", RegexOptions.Compiled);

        /// <summary>
        /// 通过HTTP GET方式下载文件到指定的目录。
        /// </summary>
        /// <param name="url">需要下载的URL</param>
        /// <param name="destDir">需要下载到的目录</param>
        /// <returns>下载后的文件</returns>
        public static string Download(string url, string destDir)
        {
            string file = null;

            try
            {
                WebUtils wu = new WebUtils();
                HttpWebRequest req = wu.GetWebRequest(url, "GET", null);
                HttpWebResponse rsp = (HttpWebResponse)req.GetResponse();
                if (CTYPE_OCTET.Equals(rsp.ContentType))
                {
                    file = Path.Combine(destDir, GetFileName(rsp.Headers["Content-Disposition"]));
                    using (System.IO.Stream rspStream = rsp.GetResponseStream())
                    {
                        int len = 0;
                        byte[] buf = new byte[8192];
                        using (FileStream fileStream = new FileStream(file, FileMode.OpenOrCreate))
                        {
                            while ((len = rspStream.Read(buf, 0, buf.Length)) > 0)
                            {
                                fileStream.Write(buf, 0, len);
                            }
                        }
                    }
                }
                else
                {
                    throw new TopException(wu.GetResponseAsString(rsp, Encoding.UTF8));
                }
            }
            catch (WebException e)
            {
                throw new TopException("isv.file-already-download", e.Message);
            }
            return file;
        }

        /// <summary>
        /// 解压gzip文件到指定的目录，目前只能解压gzip包里面只包含一个文件的压缩包。
        /// </summary>
        /// <param name="gzipFile">需要解压的gzip文件</param>
        /// <param name="destDir">需要解压到的目录（不能和压缩文件在同一个目录）</param>
        /// <returns>解压后的文件</returns>
        public static string Ungzip(string gzipFile, string destDir)
        {
            string destFile = Path.Combine(destDir, Path.GetFileName(gzipFile));
            using (System.IO.Stream output = File.Create(destFile))
            {
                using (System.IO.Stream input = new GZipStream(File.Open(gzipFile, FileMode.Open), CompressionMode.Decompress))
                {
                    int size = 0;
                    byte[] buf = new byte[8192];
                    while ((size = input.Read(buf, 0, buf.Length)) > 0)
                    {
                        output.Write(buf, 0, size);
                    }
                }
            }

            return destFile;
        }

        /// <summary>
        /// 解压zip文件到指定的目录。
        /// </summary>
        /// <param name="zipFile">需要解压的zip文件</param>
        /// <param name="destDir">需要解压到的目录</param>
        /// <returns>解压后的文件列表（不包含文件夹）</returns>
        public static List<string> Unzip(string zipFile, string destDir)
        {
            List<string> files = new List<string>();

            using (ZipInputStream s = new ZipInputStream(File.OpenRead(zipFile)))
            {
                ZipEntry theEntry;
                while ((theEntry = s.GetNextEntry()) != null)
                {
                    if (theEntry.IsDirectory)
                    {
                        Directory.CreateDirectory(Path.Combine(destDir, theEntry.Name));
                        continue;
                    }

                    string fileName = Path.Combine(destDir, theEntry.Name);
                    using (FileStream streamWriter = File.Create(fileName))
                    {
                        int size = 0;
                        byte[] buf = new byte[8192];
                        while ((size = s.Read(buf, 0, buf.Length)) > 0)
                        {
                            streamWriter.Write(buf, 0, size);
                        }
                    }
                    files.Add(fileName);
                }
            }

            return files;
        }

        /// <summary>
        /// 检查指定文件的md5sum和指定的检验码是否一致。
        /// </summary>
        /// <param name="fileName">需要检验的文件</param>
        /// <param name="checkCode">已知的md5sum检验码</param>
        /// <returns>true/false</returns>
        public static bool CheckMd5sum(string fileName, string checkCode)
        {
            using (FileStream stream = new FileStream(fileName, FileMode.Open))
            {
                MD5 md5 = new MD5CryptoServiceProvider();
                byte[] retVal = md5.ComputeHash(stream);

                StringBuilder sb = new StringBuilder();
                for (int i = 0; i < retVal.Length; i++)
                {
                    sb.Append(retVal[i].ToString("x2"));
                }

                return sb.ToString().Equals(checkCode);
            }
        }

        private static string GetFileName(string contentDisposition)
        {
            Match match = regex.Match(contentDisposition);
            if (match.Success)
            {
                return match.Groups[1].ToString();
            }
            else
            {
                throw new TopException("Invalid response header format!");
            }
        }
    }
}