hn_zw_back_main/Service/System/SysFileServices.cs

using System.ComponentModel;
using System.ComponentModel.DataAnnotations;
using System.Text;
using System.Text.RegularExpressions;
using System.Web;
using Aliyun.OSS.Util;
using Common.Enum;
using Common.Util;
using Entity.Base;
using Entity.DbModel.System;
using Entity.Dto.Req;
using Furion.VirtualFileServer;
using HybirdFrameworkCore.Autofac.Attribute;
using HybirdFrameworkCore.Configuration;
using HybirdFrameworkCore.Entity;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using OnceMi.AspNetCore.OSS;
using Repository.System;
using Service.Mgr;
using SqlSugar;
using Yitter.IdGenerator;

namespace Service.System
{
    [Scope("SingleInstance")]
    public class SysFileServices : BaseServices<SysFile>
    {
        private readonly SysFileRepository _sysFileRep;
        private readonly SysUserRepository _sysUserRep;

        private readonly SysFileMgr _sysFileMgr;
        private readonly string _imageType = ".jpg.png.bmp.gif.tif";


        public SysFileServices(
            SysUserRepository sysUserRepository,
            SysFileRepository sysFileRep,
            SysFileMgr sysFileMgr)
        {
            _sysUserRep = sysUserRepository;
            _sysFileRep = sysFileRep;
            _sysFileMgr = sysFileMgr;
        }



        /// <summary>
        /// 上传头像 🔖
        /// </summary>
        /// <param name="file"></param>
        /// <returns></returns>
        [DisplayName("上传头像")]
        public async Task<SysFile> UploadAvatar([Required] IFormFile file, string path)
        {
            var sysFile = await _sysFileMgr.HandleUploadFile(file, path, _imageType);
            var user = _sysUserRep.QueryByClause(u => u.Id == UserManager.UserId);
            await _sysUserRep.UpdateAsync(u => new SysUser() { Avatar = sysFile.Url }, u => u.Id == user.Id);
            return sysFile;
        }

        /// <summary>
        /// 获取文件分页列表 🔖
        /// </summary>
        /// <param name="input"></param>
        /// <returns></returns>
        [DisplayName("获取文件分页列表")]
        public async Task<PageResult<SysFile>> Page(PageFileReq input)
        {
            RefAsync<int> total = 0;
            var items = await _sysFileRep.QueryPageAsync(
                               entity => true, false, entity => true,
               !string.IsNullOrWhiteSpace(input.FileName), u => u.FileName.Contains(input.FileName.Trim()),
               !string.IsNullOrWhiteSpace(input.StartTime.ToString()) && !string.IsNullOrWhiteSpace(input.EndTime.ToString()),
                            u => u.CreateTime >= input.StartTime && u.CreateTime <= input.EndTime,
                u => u.CreateTime, input.PageNum, input.PageSize, total);
            return new PageResult<SysFile>()
            {
                PageNum = input.PageNum,
                PageSize = input.PageSize,
                ToTal = total,
                Rows = items,
            };
        }

        /// <summary>
        /// 上传文件 🔖
        /// </summary>
        /// <param name="file"></param>
        /// <param name="path"></param>
        /// <returns></returns>
        [DisplayName("上传文件")]
        public async Task<SysFile> UploadFile([Required] IFormFile file, [FromQuery] string? path)
        {
            return await _sysFileMgr.HandleUploadFile(file, path);
        }

        /// <summary>
        /// 上传文件Base64
        /// </summary>
        /// <param name="strBase64"></param>
        /// <param name="fileName"></param>
        /// <param name="contentType"></param>
        /// <param name="path"></param>
        /// <returns></returns>
        private async Task<SysFile> UploadFileFromBase64(string strBase64, string fileName, string contentType, string? path)
        {
            byte[] fileData = Convert.FromBase64String(strBase64);
            var ms = new MemoryStream();
            ms.Write(fileData);
            ms.Seek(0, SeekOrigin.Begin);
            if (string.IsNullOrEmpty(fileName))
                fileName = $"{YitIdHelper.NextId()}.jpg";
            if (string.IsNullOrEmpty(contentType))
                contentType = "image/jpg";
            IFormFile formFile = new FormFile(ms, 0, fileData.Length, "file", fileName)
            {
                Headers = new HeaderDictionary(),
                ContentType = contentType
            };
            return await UploadFile(formFile, path);
        }

        /// <summary>
        /// 上传文件Base64 🔖
        /// </summary>
        /// <param name="input"></param>
        /// <returns></returns>
        [DisplayName("上传文件Base64")]
        [HttpPost]
        public async Task<SysFile> UploadFileFromBase64(UploadFileFromBase64Req input)
        {
            return await UploadFileFromBase64(input.FileDataBase64, input.FileName, input.ContentType, input.Path);
        }

        /// <summary>
        /// 上传多文件 🔖
        /// </summary>
        /// <param name="files"></param>
        /// <returns></returns>
        [DisplayName("上传多文件")]
        public async Task<List<SysFile>> UploadFiles([Required] List<IFormFile> files)
        {
            var filelist = new List<SysFile>();
            foreach (var file in files)
            {
                filelist.Add(await UploadFile(file, ""));
            }
            return filelist;
        }

        ///// <summary>
        ///// 根据文件Id或Url下载 🔖
        ///// </summary>
        ///// <param name="input"></param>
        ///// <returns></returns>
        //[DisplayName("根据文件Id或Url下载")]
        //public async BusinessTask<IActionResult> DownloadFile(FileReq input)
        //{
        //    var file = input.Id > 0 ? await GetFile(input) : await _sysFileRep.QueryByClauseAsync(u => u.Url == input.Url);
        //    var fileName = HttpUtility.UrlEncode(file.FileName, Encoding.GetEncoding("UTF-8"));
        //    if (_OSSProviderOptions.IsEnable)
        //    {
        //        var filePath = string.Concat(file.FilePath, "/", file.Id.ToString() + file.Suffix);
        //        var stream = await (await _OSSService.PresignedGetObjectAsync(file.BucketName.ToString(), filePath, 5)).GetAsStreamAsync();
        //        return new FileStreamResult(stream.Stream, "application/octet-stream") { FileDownloadName = fileName + file.Suffix };
        //    }
        //    else
        //    {
        //        var filePath = Path.Combine(file.FilePath, file.Id.ToString() + file.Suffix);
        //        var path = Path.Combine(App.WebHostEnvironment.WebRootPath, filePath);
        //        return new FileStreamResult(new FileStream(path, FileMode.Open), "application/octet-stream") { FileDownloadName = fileName + file.Suffix };
        //    }
        //}


        /// <summary>
        /// 更新文件 🔖
        /// </summary>
        /// <param name="input"></param>
        /// <returns></returns>
        [ApiDescriptionSettings(Name = "Update"), HttpPost]
        [DisplayName("更新文件")]
        public async Task<string> UpdateFile(FileReq input)
        {
            string result = "";
            var isExist = await _sysFileRep.QueryByClauseAsync(u => u.Id == input.Id);
            if (isExist == null)
                result = "文件不存在";
            bool updateResult = await _sysFileRep.UpdateAsync(u => new SysFile() { FileName = input.FileName }, u => u.Id == input.Id);
            if (updateResult)
            {
                result = "更新文件成功";
            }
            return result;
        }

        /// <summary>
        /// 获取文件
        /// </summary>
        /// <param name="input"></param>
        /// <returns></returns>
        private async Task<SysFile> GetFile([FromQuery] FileReq input)
        {
            var file = await _sysFileRep.QueryByClauseAsync(u => u.Id == input.Id);
            if (file == null)
            {
                //文件不存在
                return new SysFile();
            }
            return file;
        }
    }
}