MARKUS

using System;

using System.Collections.Generic;

using System.IO;

using System.Linq;

using System.Threading.Tasks;

using FileDownloadSite.Models.Home;

using Microsoft.AspNetCore.Mvc;

using Microsoft.Extensions.FileProviders;

using MimeKit;

namespace FileDownloadSite.Controllers

{

    public class HomeController : Controller

    {

        private readonly IFileProvider fileProvider;

        public HomeController(IFileProvider fileProvider)

        {

            this.fileProvider = fileProvider;

        }

        public IActionResult Index()

        {

            return View();

        }

        /// <summary>

        /// 파일 폴더의 파일 리스트

        /// </summary>

        /// <param name="FileName"></param>

        /// <returns></returns>

        public IActionResult GetFileList()

        {

            var model = new FilesViewModel();

            foreach (var item in this.fileProvider.GetDirectoryContents(""))

            {

                model.Files.Add(

                    new FileDetails { Name = item.Name, Path = item.PhysicalPath });

            }

            return Ok(model.Files);

        }

        /// <summary>

        // 지정된 파일을 가져온다.

        /// </summary>

        /// <param name="FileName"></param>

        /// <returns></returns>

        public async Task<IActionResult> GetFile(string FileName)

        {

            var file = this.fileProvider.GetFileInfo(FileName);

            if (file.Exists)

            {

                var memory = new MemoryStream();

                using (var stream = file.CreateReadStream())

                {

                    await stream.CopyToAsync(memory);

                }

                memory.Position = 0;

                return File(memory, MimeKit.MimeTypes.GetMimeType(FileName), Path.GetFileName(FileName));

            }

            else

            {

                return new NotFoundResult();

            }

        }

    }

}