EntrustSettle/EntrustSettle.Api/Controllers/AnnexController.cs

using EntrustSettle.Common;
using EntrustSettle.Controllers;
using EntrustSettle.IServices;
using EntrustSettle.Model;
using EntrustSettle.Model.Models;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using System.ComponentModel.DataAnnotations;
using System.Web;

namespace EntrustSettle.Api.Controllers
{
    /// <summary>
    /// 附件
    /// </summary>
    public class AnnexController : BaseApiController
    {
        private readonly IAnnexService annexService;
        private readonly IHttpContextAccessor httpContextAccessor;

        public AnnexController(IAnnexService annexService, IHttpContextAccessor httpContextAccessor)
        {
            this.annexService = annexService;
            this.httpContextAccessor = httpContextAccessor;
        }

        /// <summary>
        /// 附件上传接口
        /// </summary>
        /// <param name="file">附件文件</param>
        /// <param name="fileType">文件类型</param>
        [HttpPost]
        public async Task<MessageModel<long>> Upload([FromForm] IFormFile file,
                                                     [FromForm] FileTypeEnum fileType)
        {
            if (file == null)
            {
                return Failed<long>("附件不能为空");
            }
            //new FileTypeEnumValidator().ValidateAndThrow(fileType);

            // 文件目录
            var dir = Path.Combine(App.WebHostEnvironment.WebRootPath, "files");
            if (!Directory.Exists(dir))
            {
                Directory.CreateDirectory(dir);
            }
            // 文件名
            var newFileName = Guid.NewGuid().ToString("N") + Path.GetExtension(file.FileName);

            // 完整路径
            var fullPath = Path.Combine(dir, newFileName);

            // 保存
            using (var stream = new FileStream(fullPath, FileMode.Create))
            {
                await file.CopyToAsync(stream);
            }

            // 保存文件信息到数据库
            var model = new Annex()
            {
                Name = file.FileName,
                Path = Path.Combine(@"files", newFileName),
                Type = (int)fileType,
                Key = Guid.NewGuid().ToString("N")
            };
            var annexId = await annexService.Add(model);

            return Success(annexId);
        }

        /// <summary>
        /// 文件下载接口
        /// </summary>
        /// <param name="annexId">文件记录主键</param>
        [HttpGet]
        public async Task<IActionResult> DownloadFile([FromQuery] long annexId)
        {
            // 如果当前登陆人非本公司，判断要下载的文件所属订单是否属于登陆人所在公司
            //if (!App.User.CompanyName.Contains("东胜伟业") && !App.User.CompanyName.Contains("大简云"))
            //{
            //    var compId = await orderAnnexService.AsQueryable()
            //                                  .InnerJoin<Order>((an, o) => an.OrderId == o.Id)
            //                                  .Where(an => an.AnnexId == annexId)
            //                                  .Select((an, o) => o.CompanyId)
            //                                  .FirstAsync();

            //    if (compId != App.User.CompanyId)
            //    {
            //        throw new Exception("附件所属订单与登陆人不匹配");
            //    }
            //}
            var file = await annexService.QueryById(annexId);
            if (file == null)
            {
                throw new FileNotFoundException("文件记录不存在");
            }
            var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.Path);
            if (!System.IO.File.Exists(filePath))
            {
                throw new FileNotFoundException("文件不存在");
            }
            
            httpContextAccessor.HttpContext.Response.Headers.TryAdd("Access-Control-Expose-Headers", "Content-Disposition,File-Name");

            // 读取filePath文件的内容，并返回给客户端
            var fileName = HttpUtility.UrlEncode(file.Name, System.Text.Encoding.UTF8);
            httpContextAccessor.HttpContext.Response.Headers.TryAdd("File-Name", fileName);

            return PhysicalFile(filePath, "application/octet-stream", fileName);
        }

        /// <summary>
        /// 提供给外部的文件下载接口
        /// </summary>
        [HttpGet]
        [AllowAnonymous]
        public async Task<IActionResult> Download([Required][FromQuery] string key)
        {
            var file = await annexService.QueryFirst(x => x.Key == key);
            if (file == null)
            {
                throw new FileNotFoundException("文件记录不存在");
            }
            var filePath = Path.Combine(App.WebHostEnvironment.WebRootPath, file.Path);
            if (!System.IO.File.Exists(filePath))
            {
                throw new FileNotFoundException("文件不存在");
            }
            // 读取filePath文件的内容，并返回给客户端
            var p = HttpUtility.UrlEncode(file.Name, System.Text.Encoding.UTF8);
            var p2 = file.Name;
            HttpContext.Response.Headers.TryAdd("Access-Control-Expose-Headers", "Content-Disposition");
            return PhysicalFile(filePath, "application/octet-stream", p);
        }
    }
}