TEAMModelHTEX/HTEX.Complex/Service/AzureStorage/AzureStorageBlobExtensions.cs

using Azure.Storage;
using Azure.Storage.Blobs;
using Azure.Storage.Blobs.Models;
using Azure.Storage.Sas;
using Microsoft.Extensions.DependencyInjection.Extensions;
using System.Configuration;

namespace HTEX.Complex.Service
{
    public static class AzureStorageBlobExtensions
    {
        public static IServiceCollection AddAzureStorage(this IServiceCollection services, string connectionString, string name = "Default")
        {
            if (services == null) throw new ArgumentNullException(nameof(services));
            if (connectionString == null) throw new ArgumentNullException(nameof(connectionString));

            services.TryAddSingleton<AzureStorageFactory>();
            services.Configure<AzureStorageFactoryOptions>(name, o => { o.Name = name; o.StorageAccountConnectionString = connectionString; });

            return services;
        }
        /// <summary>
        /// 系统管理员 资源，题目关联，htex关联，学习活动学生上传文件关联，基本信息关联，教室平面图关联，评测冷数据关联
        /// "system": [ "res", "item", "htex", "task", "info", "room", "exam" ],
        /// 资源，题目关联，htex关联，学习活动学生上传文件关联，基本信息关联，教室平面图关联，评测冷数据关联
        /// "school": [ "res", "item", "htex", "task", "info", "room", "exam" ],
        /// 资源，题目关联，htex关联，学习活动关联，教师基本信息关联
        /// "teacher": [ "res", "item", "htex", "task", "info" ],
        /// 答案及学习活动上传的文件，学生基本信息关联
        ///"student": [ "stu/{studentId}/ans", "stu/{studentId}/task" ]
        /// </summary>
        /// <param name="name">容器名称</param>
        /// <param name="stream">文件内容的流</param>
        /// <param name="folder">业务文件夹</param>
        /// <param name="fileName">文件名</param>
        /// <param name="contentTypeDefault">是否存放文件后缀对应的contentType</param>
        /// <returns></returns>
        public static async Task<(string blobName ,string url )> UploadFileByContainer(this BlobContainerClient blobContainer, Stream stream, string root, string blobpath, bool contentTypeDefault = true)
        {
            //BlobContainerClient blobContainer = azureStorage.GetBlobContainerClient(name.ToLower().Replace("#", "")); //blobClient.GetContainerReference(groupName); 
            Uri url = blobContainer.Uri;
            var blockBlob = blobContainer.GetBlobClient($"{root}/{blobpath}");
            string content_type = "application/octet-stream";
            if (!contentTypeDefault)
            {
                string fileext = blobpath.Substring(blobpath.LastIndexOf(".") > 0 ? blobpath.LastIndexOf(".") : 0);
                ContentTypeDict.dict.TryGetValue(fileext, out string contenttype);
                if (!string.IsNullOrEmpty(contenttype))
                {
                    content_type = contenttype;
                }
            }
            await blockBlob.UploadAsync(stream, true);
            blockBlob.SetHttpHeaders(new BlobHttpHeaders { ContentType = content_type });
          
            return (blockBlob.Name,blockBlob.Uri.ToString());
        }

        public static async Task<string> UploadFileByContainer(this BlobContainerClient blobContainer, string json, string root, string blobpath, bool contentTypeDefault = true)
        {
            // string groupName =folder;
            //BlobContainerClient blobContainer = azureStorage.GetBlobContainerClient(name.ToLower().Replace("#", "")); //blobClient.GetContainerReference(groupName);            
            var blockBlob = blobContainer.GetBlobClient($"{root}/{blobpath}");
            string content_type = "application/octet-stream";
            if (!contentTypeDefault)
            {
                string fileext = blobpath.Substring(blobpath.LastIndexOf(".") > 0 ? blobpath.LastIndexOf(".") : 0);
                ContentTypeDict.dict.TryGetValue(fileext, out string contenttype);
                if (!string.IsNullOrEmpty(contenttype))
                {
                    content_type = contenttype;
                }
            }
            byte[] bytes = System.Text.Encoding.Default.GetBytes(json);
            Stream streamBlob = new MemoryStream(bytes);
            await blockBlob.UploadAsync(streamBlob, true);
            blockBlob.SetHttpHeaders(new BlobHttpHeaders { ContentType = content_type });
            return blockBlob.Uri.ToString();
        }


        /// <summary>
        /// 取得Blob SAS (有效期預設一天)
        /// </summary>
        /// <param name="containerName">容器名稱</param>
        /// <param name="blobName"></param>
        /// <param name="blobSasPermissions"></param>
        /// <param name="name"></param>
        /// <returns></returns>
        public static BlobAuth GetBlobSasUriRead(this BlobContainerClient blobContainer,IConfiguration _configuration , string containerName, string blobName, string name = "Default")
        {
            try
            {
                var ConnectionString =   name.Equals("Default") ? _configuration.GetSection("Azure:Storage:ConnectionString").Value : _configuration.GetSection("Azure:Storage:ConnectionString-Test").Value;
                var keys = ParseConnectionString(ConnectionString);
                var accountname = keys["AccountName"];
                var accountkey = keys["AccountKey"];
                var endpoint = keys["EndpointSuffix"];
                DateTimeOffset dateTime = DateTimeOffset.UtcNow.Add(new TimeSpan(365 * 99, 0, 15, 0));
                long time = dateTime.ToUnixTimeMilliseconds();
                var blobSasBuilder = new BlobSasBuilder
                {
                    StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
                    ExpiresOn = dateTime,
                    BlobContainerName = containerName.ToLower(),
                    BlobName = blobName
                };

                blobSasBuilder.SetPermissions(BlobSasPermissions.Read);
                var sskc = new StorageSharedKeyCredential(accountname, accountkey);
                BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
                UriBuilder fullUri = new UriBuilder()
                {
                    Scheme = "https",
                    Host = $"{accountname}.blob.{endpoint}",
                    Path = $"{containerName.ToLower()}/{blobName}",
                    Query = sasQueryParameters.ToString()
                };
                return new BlobAuth { url = fullUri.Uri.ToString(), sas = sasQueryParameters.ToString(), timeout = time };
                // return fullUri.Uri.ToString();
            }
            catch
            {
                return null;
            }
        }
        public static async Task<List<string>> List(this BlobContainerClient client, string prefix = null)
        {
            try
            {
                List<string> items = new List<string>();
                await foreach (BlobItem item in client.GetBlobsAsync(BlobTraits.None, BlobStates.None, prefix))
                {
                    items.Add(item.Name);
                }
                return items;
            }
            catch
            {
                return null;
            }
        }

        public static Dictionary<string, string> ParseConnectionString(string connectionString)
        {
            var d = new Dictionary<string, string>();
            foreach (var item in connectionString.Split(';', StringSplitOptions.RemoveEmptyEntries))
            {
                var a = item.IndexOf('=');
                d.Add(item.Substring(0, a), item.Substring(a + 1));
            }
            return d;
        }
        public class BlobAuth
        {
            public string url { get; set; }
            public string sas { get; set; }
            public long timeout { get; set; }
            public string name { get; set; }
        }
    }
}