using Masuit.Tools.Files.FileDetector; using OnceMi.AspNetCore.OSS; namespace Admin.NET.Core.Service; /// /// 系统文件服务 /// [ApiDescriptionSettings(Order = 194)] public class SysFileService : IDynamicApiController, ITransient { private readonly SqlSugarRepository _sysFileRep; private readonly OSSProviderOptions _OSSProviderOptions; private readonly UploadOptions _uploadOptions; private readonly ICommonService _commonService; private readonly IOSSService _OSSService; public SysFileService(SqlSugarRepository sysFileRep, IOptions oSSProviderOptions, IOptions uploadOptions, ICommonService commonService, IOSSServiceFactory ossServiceFactory) { _sysFileRep = sysFileRep; _OSSProviderOptions = oSSProviderOptions.Value; _uploadOptions = uploadOptions.Value; _commonService = commonService; if (_OSSProviderOptions.IsEnable) _OSSService = ossServiceFactory.Create(Enum.GetName(_OSSProviderOptions.Provider)); } /// /// 获取文件分页列表 /// /// /// [HttpGet("/sysFile/page")] public async Task> GetFilePage([FromQuery] PageFileInput input) { return await _sysFileRep.AsQueryable() .WhereIF(!string.IsNullOrWhiteSpace(input.FileName), u => u.FileName.Contains(input.FileName.Trim())) .WhereIF(!string.IsNullOrWhiteSpace(input.StartTime.ToString()) && !string.IsNullOrWhiteSpace(input.EndTime.ToString()), u => u.CreateTime >= input.StartTime && u.CreateTime <= input.EndTime) .OrderBy(u => u.CreateTime, OrderByType.Desc) .ToPagedListAsync(input.Page, input.PageSize); } /// /// 上传文件 /// /// /// /// [HttpPost("/sysFile/upload")] public async Task UploadFile([Required] IFormFile file, string path = "") { var sysFile = await HandleUploadFile(file, path); return new FileOutput { Id = sysFile.Id, Url = sysFile.Url, // string.IsNullOrWhiteSpace(sysFile.Url) ? _commonService.GetFileUrl(sysFile) : sysFile.Url, SizeKb = sysFile.SizeKb, Suffix = sysFile.Suffix, FilePath = sysFile.FilePath, }; } /// /// 上传多文件 /// /// /// [HttpPost("/sysFile/uploads")] public async Task> UploadFiles([Required] List files) { var filelist = new List(); foreach (var file in files) { filelist.Add(await UploadFile(file)); } return filelist; } /// /// 下载文件(文件流) /// /// /// [HttpPost("/sysFile/download")] public async Task DownloadFile(FileInput input) { var file = await GetFile(input); var fileName = HttpUtility.UrlEncode(file.FileName, Encoding.GetEncoding("UTF-8")); if (_OSSProviderOptions.IsEnable) { var filePath = string.Concat(file.FilePath, "/", input.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, input.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 }; } } /// /// 删除文件 /// /// /// [HttpPost("/sysFile/delete")] public async Task DeleteFile(DeleteFileInput input) { var file = await _sysFileRep.GetFirstAsync(u => u.Id == input.Id); if (file != null) { await _sysFileRep.DeleteAsync(file); if (_OSSProviderOptions.IsEnable) { await _OSSService.RemoveObjectAsync(file.BucketName.ToString(), string.Concat(file.FilePath, "/", input.Id.ToString())); } else { var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.FilePath, input.Id.ToString() + file.Suffix); if (File.Exists(filePath)) File.Delete(filePath); } } } /// /// 获取文件 /// /// /// private async Task GetFile([FromQuery] FileInput input) { var file = await _sysFileRep.GetFirstAsync(u => u.Id == input.Id); return file ?? throw Oops.Oh(ErrorCodeEnum.D8000); } /// /// 上传文件 /// /// 文件 /// 路径 /// private async Task HandleUploadFile(IFormFile file, string savePath = "") { if (file == null) throw Oops.Oh(ErrorCodeEnum.D8000); var path = savePath; if (string.IsNullOrWhiteSpace(savePath)) { path = _uploadOptions.Path; var reg = new Regex(@"(\{.+?})"); var match = reg.Matches(path); match.ToList().ForEach(a => { var str = DateTime.Now.ToString(a.ToString().Substring(1, a.Length - 2)); // 每天一个目录 path = path.Replace(a.ToString(), str); }); } if (!_uploadOptions.ContentType.Contains(file.ContentType)) throw Oops.Oh(ErrorCodeEnum.D8001); var sizeKb = (long)(file.Length / 1024.0); // 大小KB if (sizeKb > _uploadOptions.MaxSize) throw Oops.Oh(ErrorCodeEnum.D8002); var suffix = Path.GetExtension(file.FileName).ToLower(); // 后缀 var newFile = new SysFile { Id = Yitter.IdGenerator.YitIdHelper.NextId(), // BucketName = _OSSProviderOptions.IsEnable ? _OSSProviderOptions.Provider.ToString() : "Local", // 阿里云对bucket名称有要求,1.只能包括小写字母,数字,短横线(-)2.必须以小写字母或者数字开头 3.长度必须在3-63字节之间 // 无法使用Provider BucketName = _OSSProviderOptions.IsEnable ? _OSSProviderOptions.Bucket : "Local", FileName = Path.GetFileNameWithoutExtension(file.FileName), Suffix = suffix, SizeKb = sizeKb.ToString(), FilePath = path, }; var finalName = newFile.Id + suffix; // 文件最终名称 if (_OSSProviderOptions.IsEnable) { newFile.Provider = Enum.GetName(_OSSProviderOptions.Provider); var filePath = string.Concat(path, "/", finalName); await _OSSService.PutObjectAsync(newFile.BucketName, filePath, file.OpenReadStream()); // http://<你的bucket名字>.oss.aliyuncs.com/<你的object名字> // 生成外链地址 方便前端预览 switch (_OSSProviderOptions.Provider) { case OSSProvider.Aliyun: newFile.Url = $"{(_OSSProviderOptions.IsEnableHttps ? "https" : "http")}://{newFile.BucketName}.{_OSSProviderOptions.Endpoint}/{filePath}"; break; case OSSProvider.Minio: // 获取Minio文件的下载或者预览地址 newFile.Url = await GetMinioPreviewFileUrl(newFile.BucketName, filePath); ; break; } } else { newFile.Provider = ""; // 本地存储 Provider 显示为空 var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, path); if (!Directory.Exists(filePath)) Directory.CreateDirectory(filePath); var realFile = Path.Combine(filePath, finalName); IDetector detector; using (var stream = File.Create(realFile)) { await file.CopyToAsync(stream); detector = stream.DetectFiletype(); } var realExt = detector.Extension; // 真实扩展名 // 二次校验扩展名 if (!string.Equals(realExt, suffix.Replace(".", ""), StringComparison.OrdinalIgnoreCase)) { var delFilePath = Path.Combine(App.WebHostEnvironment.WebRootPath, realFile); if (File.Exists(delFilePath)) File.Delete(delFilePath); throw Oops.Oh(ErrorCodeEnum.D8001); } // 生成外链 newFile.Url = _commonService.GetFileUrl(newFile); } await _sysFileRep.AsInsertable(newFile).ExecuteCommandAsync(); return newFile; } /// /// 获取Minio文件的下载或者预览地址 /// /// 桶名 /// 文件名 /// private async Task GetMinioPreviewFileUrl(string bucketName, string fileName) { return await _OSSService.PresignedGetObjectAsync(bucketName, fileName, 7); } /// /// 上传头像 /// /// /// [HttpPost("/sysFile/uploadAvatar")] public async Task UploadAvatar([Required] IFormFile file) { var res = await UploadFile(file, "Avatar"); var userId = App.GetService().UserId; await _sysFileRep.ChangeRepository>() .UpdateAsync(u => new SysUser() { Avatar = res.Url }, u => u.Id == userId); return res; } /// /// 上传电子签名 /// /// /// [HttpPost("/sysFile/uploadSignature")] public async Task UploadSignature([Required] IFormFile file) { var res = await UploadFile(file, "Signature"); var userId = App.GetService().UserId; await _sysFileRep.ChangeRepository>() .UpdateAsync(u => new SysUser() { Signature = res.Url }, u => u.Id == userId); return res; } }