using OnceMi.AspNetCore.OSS;
namespace Admin.NET.Core.Service;
///
/// 系统文件服务
///
[ApiDescriptionSettings(Name = "文件管理", 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(_OSSProviderOptions.Provider.ToString());
}
///
/// 获取文件分页列表
///
///
///
[HttpGet("/sysFile/page")]
public async Task> QueryFilePageList([FromQuery] PageFileInput input)
{
var files = 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, SqlSugar.OrderByType.Desc)
.ToPagedListAsync(input.Page, input.PageSize);
return files;
}
///
/// 获取文件列表
///
///
[HttpGet("/sysFile/list")]
public async Task> GetFileList()
{
return await _sysFileRep.GetListAsync();
}
///
/// 上传文件
///
///
///
[HttpPost("/sysFile/upload")]
[AllowAnonymous]
public async Task UploadFile([Required] IFormFile file)
{
var sysFile = await HandleUploadFile(file);
return new FileOutput
{
Id = sysFile.Id,
Url = _commonService.GetFileUrl(sysFile),
SizeKb = sysFile.SizeKb,
Suffix = sysFile.Suffix,
FilePath = sysFile.FilePath,
};
}
///
/// 下载文件(文件流)
///
///
///
[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, "application/octet-stream") { FileDownloadName = fileName };
}
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 };
}
}
///
/// 删除文件
///
///
///
[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)
{
if (file == null) throw Oops.Oh(ErrorCodeEnum.D8000);
string 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(); // 后缀
// 先存库获取Id
var newFile = await _sysFileRep.AsInsertable(new SysFile
{
BucketName = _OSSProviderOptions.IsEnable ? _OSSProviderOptions.Provider.ToString() : "Local",
FileName = Path.GetFileNameWithoutExtension(file.FileName),
Suffix = suffix,
SizeKb = sizeKb.ToString(),
FilePath = path
}).ExecuteReturnEntityAsync();
var finalName = newFile.Id + suffix; // 文件最终名称
if (_OSSProviderOptions.IsEnable)
{
var filePath = string.Concat(path, "/", finalName);
await _OSSService.PutObjectAsync(newFile.BucketName.ToString(), filePath, file.OpenReadStream());
}
else
{
var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, path);
if (!Directory.Exists(filePath))
Directory.CreateDirectory(filePath);
using var stream = File.Create(Path.Combine(filePath, finalName));
await file.CopyToAsync(stream);
}
return newFile;
}
}