using Furion; using Furion.DependencyInjection; using Furion.DynamicApiController; using Furion.FriendlyException; using GDZZ.Core.Entity; using Mapster; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.Logging; using Microsoft.Extensions.Options; using SqlSugar; using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using System.Threading.Tasks; using System.Web; using Yitter.IdGenerator; namespace GDZZ.Core.Service; /// /// 文档服务 /// [ApiDescriptionSettings(Name = "Document", Order = 150)] public class DocumentService : IDocumentService, IDynamicApiController, ITransient { private readonly SqlSugarRepository _rep; private readonly ISysDictDataService _sysDictDataService; private readonly UploadFileOptions _options; private readonly ILogger _logger; public DocumentService(ILogger logger, SqlSugarRepository rep, IOptions options, ISysDictDataService sysDictDataService) { this._logger = logger; _rep = rep; this._sysDictDataService = sysDictDataService; this._options = options.Value; } #region API /// /// 分页查询文档 /// /// /// [HttpGet("/Document/page")] public async Task Page([FromQuery] DocumentInput input) { //获取字典 var fileTypes = await _sysDictDataService.GetDictDataByCode("file_type"); List fileTypeList = null; #region 判断是否根据类型搜索 if (!string.IsNullOrEmpty(input.FileType)) { if (input.FileType == "文件夹") { input.DocumentType = DocumentType.Folder; } else { var data = fileTypes.Where(it => it.Value == input.FileType).FirstOrDefault(); if (data != null) { fileTypeList = new List(); data.Code.Split(",").ToList().ForEach(it => { fileTypeList.Add($".{it}"); }); } } } #endregion var entities = await _rep.AsQueryable() .WhereIF(!string.IsNullOrWhiteSpace(input.Name), u => u.Name.Contains(input.Name.Trim())) .WhereIF(fileTypeList != null, u => fileTypeList.Contains(u.FileSuffix)) .WhereIF(input.Label != null, u => u.Label == input.Label) .WhereIF(input.DocumentType != null, u => u.DocumentType == input.DocumentType) .WhereIF(input.CreateTimeStart != null && input.CreateTimeEnd != null, u => SqlFunc.Between(u.CreatedTime, input.CreateTimeStart, input.CreateTimeEnd)) .WhereIF(input.UpdateTimeStart != null && input.UpdateTimeEnd != null, u => SqlFunc.Between(u.UpdatedTime, input.UpdateTimeStart, input.UpdateTimeEnd)) .WhereIF(input.PId != null, u => u.PId == input.PId) .Where(u => u.IsDeleted == input.IsDelete && u.Visible == true) .Filter(null, true) .OrderBy(u => u.DocumentType) .Select() .Mapper(async it => { if (it.FileSizeKb != null) { it.FileSize = GetUnit(it.FileSizeKb.Value);//文件大小转成字符串 } }) .ToPagedListAsync(input.PageNo, input.PageSize); #region 获取文件类型 //遍历获取到的数据 foreach (var it in entities.Items) { if (string.IsNullOrWhiteSpace(it.FileSuffix)) { it.FileType = "文件夹"; } else { //获取后缀名 var fileSuffix = it.FileSuffix.Split(".")[1]; foreach (var filetype in fileTypes) { var types = filetype.Code.Split(',').ToArray();//获取所有后缀名 if (types.Contains(fileSuffix))//是否存在 { it.FileType = filetype.Value; break; } } if (string.IsNullOrEmpty(it.FileType)) { it.FileType = "文件"; } } } #endregion return entities.XnPagedResult(); } /// /// 文件夹树 /// /// [HttpGet("/Document/tree")] public async Task Tree() { var tree = await _rep .Where(it => it.DocumentType == DocumentType.Folder) .ToTreeAsync(it => it.Children, it => it.PId, 0); var result = tree.Adapt>(); return result; } /// /// 上传文件 /// /// /// [HttpPost("/Document/upload")] public async Task Upload([FromForm] DocumentUploadInput input) { var fileNames = input.Files.Select(it => it.FileName).ToList();//获取文件列表名称 var exist = await _rep.Where(it => fileNames.Contains(it.Name) && it.PId == input.PId).Select(it => it.Name).ToListAsync();//获取数据库中同名文件 if (exist.Count > 0) { throw Oops.Oh($"同级目录下已存在相同文件名称{string.Join(",", exist)}"); } else { foreach (var file in input.Files) { await UploadFile(file, input.PId.Value); } } } /// /// 上传文件夹 /// /// /// [HttpPost("/Document/uploadfolder")] public async Task UploadFolder([FromForm] DocumentUploadInput input) { var pid = input.PId.Value; long firtId = 0;//主文件夹的ID Dictionary> fileDic = new Dictionary>(); input.Files.ForEach(it => { fileDic.Add(it.FileName, it.FileName.Split('/').ToList());//将文件名和分割的数组加到字典 }); //循环从0开始,小于字典中value最多的数量-1,因为后面要加2去拿 for (int i = 0; i < fileDic.OrderByDescending(it => it.Value.Count).First().Value.Count - 1; i++) { //重新获取新的字典 var files = fileDic.Where(it => it.Value.Count == i + 2).ToDictionary(it => it.Key, it => it.Value); if (firtId != 0) pid = firtId;//如果主文件夹id不为0表示创建过第一层的文件夹,后面的文件夹在第一层文件夹下面创建 //循环创建文件夹 var folders = files.ElementAt(0).Value; folders.RemoveAt(folders.Count - 1);//删掉最后一个文件夹名字 bool isFirst = folders.Count == 1 ? true : false;//只有一个表示顶层文件夹 if (folders.Count > 1) { folders.RemoveAt(0);//删掉第一层 } foreach (var folder in folders) { Console.WriteLine($"新建文件夹:{folder}"); pid = await Add(new AddDocumentInput { PId = pid, Name = folder });//创建文件夹 } if (isFirst)//表示顶层 { firtId = pid; } foreach (var dic in files) { var info = fileDic.Where(it => it.Key == dic.Key).First();//文件信息 //var filename = dic.Value[i + 1]; var file = input.Files.Where(it => it.FileName == info.Key).First(); await UploadFile(file, pid); } } } /// /// 新建文件夹 /// /// /// [HttpPost("/Document/add")] public async Task Add(AddDocumentInput input) { var entity = input.Adapt(); entity.DocumentType = DocumentType.Folder; var exist = await _rep.AnyAsync(it => it.Name == input.Name && it.PId == input.PId); if (exist) { throw Oops.Oh("ErrorMessage.E2000"); } var result = await _rep.AsInsertable(entity).IgnoreColumns(true).ExecuteReturnEntityAsync(); return result.Id; } /// /// 删除文档 /// /// /// [HttpPost("/Document/delete")] public async Task Delete(DeleteDocumentInput input) { var entity = await _rep.FirstOrDefaultAsync(it => it.Id == input.Id); if (entity != null) { entity.IsDeleted = true; await _rep.UpdateAsync(entity); } else { throw Oops.Oh("ErrorMessage.E404"); } } /// /// 批量删除 /// /// /// [HttpPost("/Document/deletes")] public async Task Deletes(DeletesDocumentInput input) { #region 所有isdelete=true var documents = await _rep.ToListAsync(it => input.Ids.Contains(it.Id)); #endregion documents.ForEach(document => document.IsDeleted = true); await _rep.AsUpdateable(documents).UpdateColumns(it => new { it.IsDeleted, it.UpdatedTime, it.UpdatedUserId, it.UpdatedUserName }).ExecuteCommandAsync(); } /// /// 移动文档 /// /// /// [HttpPost("/Document/move")] public async Task Move(MoveDocumentInput input) { var tree = await GetParentList(input.PId.Value);//获取上级 var ids = tree.Select(it => it.Id).ToList();//获取id列表 if (ids.Count > 0)//去掉自己 { ids.Remove(input.PId.Value); } var same = ids.Intersect(input.Ids).ToArray();//检查目标ID的父ID是否有选中的文件 if (ids.Contains(input.PId.Value) || same.Length > 0)//判断父ID是否在列表 { throw Oops.Oh("ErrorMessage.E2001"); } //判断有没有相同名称的文件或者文件夹 var names = await _rep.Where(it => input.Ids.Contains(it.Id)).Select(it => it.Name).ToListAsync();//文档列表 var existName = await _rep.Where(it => it.PId == input.PId && names.Contains(it.Name)).AnyAsync();//文档列表 if (existName) { throw Oops.Oh("ErrorMessage.E2002"); } var pFolder = _rep.FirstOrDefault(it => it.Id == input.PId.Value);//获取父文件夹 _rep.CurrentBeginTran(); try { await _rep.UpdateAsync(it => input.Ids.Contains(it.Id), it => new Documentation { PId = input.PId.Value }); if (pFolder != null) { await _rep.AsUpdateable(pFolder).UpdateColumns(it => new { it.UpdatedTime, it.UpdatedUserId, it.UpdatedUserName }).ExecuteCommandAsync(); } _rep.CurrentCommitTran(); } catch (Exception ex) { _rep.CurrentRollbackTran(); _logger.LogError(ex, $"移动失败:{ex.Message}"); throw Oops.Oh("ErrorMessage.E500"); } } /// /// 更新文件夹 /// /// /// [HttpPost("/Document/edit")] public async Task Update(UpdateDocumentInput input) { var exist = await _rep.FirstOrDefaultAsync(it => it.Id == input.Id); if (exist != null) { if (!string.IsNullOrEmpty(exist.FileSuffix))//如果有后缀名加上后缀名 { input.Name += $".{exist.FileSuffix}"; } exist.Name = input.Name; exist.Remark = input.Remark; exist.Label = input.Label; //获取上级 var tree = await GetParentList(exist.Id); _rep.CurrentBeginTran(); try { if (tree.Count > 0) { //更新上级的修改时间 await _rep.AsUpdateable(tree).UpdateColumns(it => new { it.UpdatedTime, it.UpdatedUserId, it.UpdatedUserName }).ExecuteCommandAsync(); } await _rep.AsUpdateable(exist).IgnoreColumns(ignoreAllNullColumns: true).ExecuteCommandAsync(); _rep.CurrentCommitTran(); } catch (Exception ex) { _rep.CurrentRollbackTran(); _logger.LogError(ex, $"更新文件夹失败:{ex.Message}"); throw Oops.Oh("ErrorMessage.E500"); } } else { throw Oops.Oh(ErrorCode.D1002); } } /// /// 获取文档 /// /// /// [HttpGet("/Document/detail")] public async Task Get([FromQuery] QueryeDocumentInput input) { var file = await _rep.FirstOrDefaultAsync(u => u.Id == input.Id); if (file == null) throw Oops.Oh(ErrorCode.D1002); return file; } /// /// 下载文件 /// /// /// [HttpGet("/Document/download")] public async Task Download([FromQuery] QueryeDocumentInput input) { var file = await Get(input); var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.FilePath, file.FileObjectName); var fileName = HttpUtility.UrlEncode(file.Name, Encoding.GetEncoding("UTF-8")); var result = new FileStreamResult(new FileStream(filePath, FileMode.Open), "application/octet-stream") { FileDownloadName = fileName }; return result; } /// /// 预览文件 /// /// /// [HttpGet("/Document/preview")] public async Task Preview([FromQuery] QueryeDocumentInput input) { var file = await Get(input); return $"{file.FilePath}/{file.FileObjectName}"; } #endregion #region 方法 /// /// 创建Pids格式 /// 如果pid是0顶级节点,pids就是 [0]; /// 如果pid不是顶级节点,pids就是 pid文档的 pids + [pid] + , /// /// /// private async Task CreateNewPids(long pid) { if (pid == 0L) { return "0,"; } else { var pmenu = await _rep.FirstOrDefaultAsync(u => u.Id == pid); return pmenu.PIds + $"{pid},"; } } /// /// 获取文件大小 /// /// /// [NonAction] private string GetUnit(int fileSizeKb) { var b = fileSizeKb * 1024; const int MB = 1024 * 1024; const int KB = 1024; if (b / MB >= 1) { return Math.Round(b / (float)MB, 2) + "MB"; } if (b / KB >= 1) { return Math.Round(b / (float)KB, 2) + "KB"; } if (b == 0) { return "0B"; } return null; } /// /// 上传文件 /// /// /// /// [NonAction] private async Task UploadFile(IFormFile file, long pid) { var now = DateTime.Now.ToString("d"); var localPath = $"{_options.Document.path}/{now}"; var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, localPath); if (!Directory.Exists(filePath)) Directory.CreateDirectory(filePath); var fileSizeKb = (int)(file.Length / 1024.0); // 文件大小KB var originalFilename = Path.GetFileName(file.FileName); // 文件原始名称 var fileSuffix = Path.GetExtension(file.FileName).ToLower(); // 文件后缀 var objectName = $"{ YitIdHelper.NextId()}{fileSuffix}";//存储文件名 Documentation document = new Documentation { PId = pid, Name = originalFilename, FileSuffix = fileSuffix, FileSizeKb = fileSizeKb, DocumentType = DocumentType.File, FileObjectName = objectName, FilePath = localPath, }; var result = await _rep.InsertAsync(document);//添加到数据库 if (result > 0) { try { //上传文件 using (var stream = File.Create(Path.Combine(filePath, objectName))) { await file.CopyToAsync(stream); } } catch (Exception ex) { throw Oops.Oh($"保存文件{originalFilename}失败,请重新上传"); } } } /// /// 获取父级 /// /// /// private async Task> GetParentList(long id) { return await _rep.Where(it => it.DocumentType == DocumentType.Folder).ToParentListAsync(it => it.PId, id); } #endregion }