BookingHeChuan/Myshipping.Core/Service/File/SysFileService.cs

using Furion;
using Furion.DependencyInjection;
using Furion.DynamicApiController;
using Furion.FriendlyException;
using Myshipping.Core.Entity;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using SqlSugar;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Threading.Tasks;
using System.Web;
using Yitter.IdGenerator;

namespace Myshipping.Core.Service;

/// <summary>
/// 文件服务
/// </summary>
[ApiDescriptionSettings(Name = "File", Order = 100)]
public class SysFileService : ISysFileService, IDynamicApiController, ITransient
{
    private readonly SqlSugarRepository<SysFile> _sysFileInfoRep;  // 文件信息表仓储 

    private readonly UploadFileOptions _options;

    public SysFileService(SqlSugarRepository<SysFile> sysFileInfoRep, IOptions<UploadFileOptions> options)
    {
        _sysFileInfoRep = sysFileInfoRep;
        _options = options.Value;
    }

    /// <summary>
    /// 分页获取文件列表
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysFileInfo/page")]
    public async Task<dynamic> QueryFileInfoPageList([FromQuery] FileInput input)
    {
        var fileBucket = !string.IsNullOrEmpty(input.FileBucket?.Trim());
        var fileOriginName = !string.IsNullOrEmpty(input.FileOriginName?.Trim());

        var files = await _sysFileInfoRep.AsQueryable()
            .WhereIF(input.FileLocation > 0, u => u.FileLocation == input.FileLocation)
            .WhereIF(!string.IsNullOrWhiteSpace(input.FileBucket), u => u.FileBucket.Contains(input.FileBucket.Trim()))
            .WhereIF(!string.IsNullOrWhiteSpace(input.FileOriginName), u => u.FileOriginName.Contains(input.FileOriginName.Trim()))
            .Select<FileOutput>()
            .ToPagedListAsync(input.PageNo, input.PageSize);
        return files.XnPagedResult();
    }

    /// <summary>
    /// 获取文件列表
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysFileInfo/list")]
    public async Task<List<SysFile>> GetFileInfoList([FromQuery] FileOutput input)
    {
        return await _sysFileInfoRep.ToListAsync();
    }

    /// <summary>
    /// 删除文件
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/delete")]
    public async Task DeleteFileInfo(DeleteFileInfoInput input)
    {
        var file = await _sysFileInfoRep.FirstOrDefaultAsync(u => u.Id == input.Id);
        if (file != null)
        {
            await _sysFileInfoRep.DeleteAsync(file);

            var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.FileBucket, file.FileObjectName);
            if (File.Exists(filePath))
                File.Delete(filePath);
        }
    }

    /// <summary>
    /// 获取文件详情
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysFileInfo/detail")]
    public async Task<SysFile> GetFileInfo([FromQuery] QueryFileInoInput input)
    {
        var file = await _sysFileInfoRep.FirstOrDefaultAsync(u => u.Id == input.Id);
        if (file == null)
            throw Oops.Oh(ErrorCode.D8000);
        return file;
    }

    /// <summary>
    /// 预览文件
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysFileInfo/preview")]
    public async Task<IActionResult> PreviewFileInfo([FromQuery] QueryFileInoInput input)
    {
        return await DownloadFileInfo(input);
    }

    /// <summary>
    /// 上传文件
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/upload")]
    public async Task UploadFileDefault(IFormFile file)
    {
        // 可以读取系统配置来决定将文件存储到什么地方
        await UploadFile(file, _options.Default.path, FileLocation.LOCAL);
    }

    /// <summary>
    /// 下载文件
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysFileInfo/download")]
    public async Task<IActionResult> DownloadFileInfo([FromQuery] QueryFileInoInput input)
    {
        var file = await GetFileInfo(input);
        var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.FilePath, file.FileObjectName);
        var fileName = HttpUtility.UrlEncode(file.FileOriginName, Encoding.GetEncoding("UTF-8"));
        var result =  new FileStreamResult(new FileStream(filePath, FileMode.Open), "application/octet-stream") { FileDownloadName = fileName };
        return result;
    }

    /// <summary>
    /// 上传头像
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/uploadAvatar")]
    public async Task<long> UploadFileAvatar(IFormFile file)
    {
        return await UploadFile(file, _options.Avatar.path);
    }

    /// <summary>
    /// 上传文档
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/uploadDocument")]
    public async Task UploadFileDocument(IFormFile file)
    {
        await UploadFile(file, _options.Document.path);
    }
    /// <summary>
    /// 上传富文本图片
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/uploadEditor")]
    public async Task<string> UploadFileEditor(IFormFile file)
    {
        var id = await UploadFile(file, _options.Editor.path, FileLocation.LOCAL);
        var fileInfo = _sysFileInfoRep.Single(id);
        return Path.Combine(fileInfo.FilePath, fileInfo.FileObjectName);

    }
    /// <summary>
    /// 上传商店图片
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    [HttpPost("/sysFileInfo/uploadShop")]
    public async Task UploadFileShop(IFormFile file)
    {
        await UploadFile(file, _options.Shop.path);
    }

    /// <summary>
    /// 上传文件
    /// </summary>
    /// <param name="file"></param>
    /// <param name="pathType"></param>
    /// <returns></returns>
    private static async Task<long> UploadFile(IFormFile file, string pathType)
    {
        var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, pathType);
        if (!Directory.Exists(filePath))
            Directory.CreateDirectory(filePath);

        var fileSizeKb = (long)(file.Length / 1024.0); // 文件大小KB
        var originalFilename = file.FileName; // 文件原始名称
        var fileSuffix = Path.GetExtension(file.FileName).ToLower(); // 文件后缀  

        // 先存库获取Id
        var id = YitIdHelper.NextId();
        var newFile = new SysFile
        {
            Id=id,
            FileLocation = (int)FileLocation.LOCAL,
            FileBucket = FileLocation.LOCAL.ToString(),
            FileObjectName = $"{id}{fileSuffix}",
            FileOriginName = originalFilename,
            FileSuffix = fileSuffix.TrimStart('.'),
            FileSizeKb = fileSizeKb.ToString(),
            FilePath = pathType
        };
        newFile = await App.GetService<ISqlSugarClient>().Insertable(newFile).ExecuteReturnEntityAsync();    
        using (var stream = File.Create(Path.Combine(filePath, newFile.FileObjectName)))
        {
            await file.CopyToAsync(stream);
        }
        return newFile.Id; // 返回文件唯一标识
    }

    /// <summary>
    /// 上传文件
    /// </summary>
    /// <param name="file">文件</param>
    /// <param name="pathType">存储路径</param>
    /// <param name="fileLocation">文件存储位置</param>
    /// <returns></returns>
    private static async Task<long> UploadFile(IFormFile file, string pathType, FileLocation fileLocation)
    {
        var fileSizeKb = (long)(file.Length / 1024.0); // 文件大小KB
        var originalFilename = file.FileName; // 文件原始名称
        var fileSuffix = Path.GetExtension(file.FileName).ToLower(); // 文件后缀

        // 先存库获取Id
        var id = YitIdHelper.NextId();
        var newFile =  new SysFile
        {
            Id = id,
            FileLocation = (int)FileLocation.LOCAL,
            FileBucket = FileLocation.LOCAL.ToString(),
            FileObjectName = $"{YitIdHelper.NextId()}{fileSuffix}",
            FileOriginName = originalFilename,
            FileSuffix = fileSuffix.TrimStart('.'),
            FileSizeKb = fileSizeKb.ToString(),
            FilePath = pathType
        };
        newFile = await App.GetService<ISqlSugarClient>().Insertable(newFile).ExecuteReturnEntityAsync();

        var finalName = newFile.FileObjectName; // 生成文件的最终名称   
        if (fileLocation == FileLocation.LOCAL) // 本地存储
        {
            var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, pathType);
            if (!Directory.Exists(filePath))
                Directory.CreateDirectory(filePath);
            using (var stream = File.Create(Path.Combine(filePath, finalName)))
            {
                await file.CopyToAsync(stream);
            }
        }
        else if (fileLocation == FileLocation.ALIYUN) // 阿里云OSS
        {
            var filePath = pathType + finalName;
            OSSClientUtil.DeletefileCode(filePath);

            var stream = file.OpenReadStream();
            OSSClientUtil.PushMedia(stream, filePath);
        }
        newFile.FileObjectName = finalName;
        return newFile.Id; // 返回文件唯一标识
    }
}