using AutoMapper; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.StaticFiles; using MTWorkHR.Application.Models; using MTWorkHR.Application.Services.Interfaces; using MTWorkHR.Core.Global; using System.Net.Http.Headers; namespace MTWorkHR.Application.Services { public class FileService : IFileService { private readonly AppSettingsConfiguration settings; public FileService(AppSettingsConfiguration settings) { this.settings = settings; } public string UploadFile(IFormFile file) { return UploadFiles(new List { file }).First(); } public List UploadFiles(List files) { if (!AttachmentsMust(files)) throw new AppException(ExceptionEnum.InvalidFileType); string pathToSave = GetTempAttachmentPath(); if (!Directory.Exists(pathToSave)) Directory.CreateDirectory(pathToSave); var fileNames = new List(); foreach (var formFile in files) { var fname = ContentDispositionHeaderValue.Parse(formFile.ContentDisposition).FileName.Trim('"'); var fnameSplit = fname.Split("."); var fNewName = Guid.NewGuid().ToString() + "." + fnameSplit[fnameSplit.Length - 1]; var fullPath = Path.Combine(pathToSave, fNewName); fileNames.Add(fNewName); if (formFile.Length > 0) { using (var stream = new FileStream(fullPath, FileMode.Create)) { formFile.CopyToAsync(stream); } } else throw new AppException(ExceptionEnum.FileLengthNotCorrect); } return fileNames; } public async Task> GetFileDownloadInfo(string fileUrl) { var filePath = Path.Combine(GetActualAttachmentPath(), fileUrl); if (!System.IO.File.Exists(filePath)) throw new AppException(ExceptionEnum.RecordNotExist); var memory = new MemoryStream(); await using (var stream = new FileStream(filePath, FileMode.Open)) { await stream.CopyToAsync(memory); } memory.Position = 0; return new Tuple(memory, GetContentType(filePath), filePath); } private string GetContentType(string path) { var provider = new FileExtensionContentTypeProvider(); string contentType; if (!provider.TryGetContentType(path, out contentType)) { contentType = "application/octet-stream"; } return contentType; } public async Task CopyFileToActualFolder(List attachments) { foreach (var attachment in attachments) { var files = attachments.Select(a => a.FileData); foreach (var formFile in files) { if (formFile.Length > 0) { var filePath = Path.GetTempFileName(); using (var stream = System.IO.File.Create(filePath)) { await formFile.CopyToAsync(stream); } } } if (!Directory.Exists(GetActualAttachmentPath())) Directory.CreateDirectory(GetActualAttachmentPath()); var tempFilePath = Path.Combine(GetTempAttachmentPath(), attachment.FileName); var destinationFilePath = Path.Combine(GetActualAttachmentPath(), attachment.FileName); if (File.Exists(tempFilePath)) { if (!File.Exists(destinationFilePath)) File.Copy(tempFilePath, destinationFilePath); } else return false; } return true; } public bool CopyFileToActualFolder(string FileName) { if (!Directory.Exists(GetActualAttachmentPath())) Directory.CreateDirectory(GetActualAttachmentPath()); var tempFilePath = Path.Combine(GetTempAttachmentPath(), FileName); var destinationFilePath = Path.Combine(GetActualAttachmentPath(), FileName); if (File.Exists(tempFilePath)) { if (!File.Exists(destinationFilePath)) File.Copy(tempFilePath, destinationFilePath); return true; } return false; } public bool DeleteFileFromTempFolder(string FileName) { try { string pathToRemove = GetTempAttachmentPath(); if (Directory.Exists(pathToRemove)) { var tempFilePath = Path.Combine(pathToRemove, FileName); if (File.Exists(tempFilePath)) { File.Delete(tempFilePath); } } } catch (Exception ex) { } return true; } public string GetTempAttachmentPath() { var pathTemp = settings.AttachmentSettings.TempAttachment; return pathTemp; } public string GetActualAttachmentPath() { var pathActual = settings.AttachmentSettings.ActualAttachment ; return pathActual; } private bool AttachmentsMust(List files) { // extensions whitelist string[] allowedExtensions = new string[] { ".jpeg" , ".jpg" , ".png" , ".pdf" , ".doc" , ".docx" , ".csv" , ".xls" , ".xlsx" , ".txt" , ".pptx" , ".svg" , ".webp" }; // MIME types whitelist string[] allowedMIMETypes = new string[] { "image/jpeg" , "image/png" , "application/pdf" , "application/msword" , "application/vnd.openxmlformats-officedocument.wordprocessingml.document" , "text/csv" , "application/vnd.ms-excel" , "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" , "text/plain" , "application/vnd.openxmlformats-officedocument.presentationml.presentation" , "image/svg+xml" , "image/webp" }; foreach (IFormFile file in files) { if (!allowedExtensions.Contains(Path.GetExtension(file.FileName).ToLower())) return false; if (!allowedMIMETypes.Contains(file.ContentType.ToLower())) return false; } return true; } } }