122 lines
2.7 KiB
C#
122 lines
2.7 KiB
C#
using FileStorageService.www.Atttributes;
|
|
using FileStorageService.www.Models;
|
|
using FileStorageService.www.Repositories;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
namespace FileStorageService.www.Controllers;
|
|
|
|
public class FilesController(
|
|
FileRepository fileRepository) : Controller
|
|
{
|
|
private const long MaxFileSize = 7L * 1024L * 1024L * 1024L;
|
|
|
|
// GET
|
|
public async Task<IActionResult> Index(Guid? id)
|
|
{
|
|
if (id is not null)
|
|
{
|
|
return await ViewSingleFile((Guid)id);
|
|
}
|
|
|
|
var fileHandles = await fileRepository.GetAllFilesAsync();
|
|
|
|
var handles = fileHandles.Select(e => new FileHandleModel
|
|
{
|
|
FileName = e.Name,
|
|
BlockCount = e.FileBlockCount,
|
|
Id = e.Id,
|
|
})
|
|
.ToList();
|
|
|
|
var model = new FileHandleListModel
|
|
{
|
|
FileHandles = handles
|
|
};
|
|
|
|
return View(model);
|
|
}
|
|
|
|
private async Task<IActionResult> ViewSingleFile(Guid id)
|
|
{
|
|
var fileHandle = await fileRepository.GetFileAsync(id);
|
|
|
|
var model = new FileHandleModel
|
|
{
|
|
FileName = fileHandle.Name,
|
|
BlockCount = fileHandle.FileBlockCount,
|
|
Id = fileHandle.Id,
|
|
};
|
|
|
|
return View("SingleFile", model);
|
|
}
|
|
|
|
public IActionResult New()
|
|
{
|
|
return View(new NewFileModel
|
|
{
|
|
Name = "",
|
|
FileContents = null!
|
|
});
|
|
}
|
|
|
|
[HttpPost]
|
|
[DisableFormValueModelBinding]
|
|
[RequestSizeLimit(MaxFileSize)]
|
|
[RequestFormLimits(MultipartBodyLengthLimit = MaxFileSize)]
|
|
public async Task<IActionResult> New(NewFileModel model)
|
|
{
|
|
if (!ModelState.IsValid)
|
|
{
|
|
return BadRequest(ModelState.Values.SelectMany(v => v.Errors));
|
|
}
|
|
|
|
var name = model.Name;
|
|
var stream = model.FileContents.OpenReadStream();
|
|
|
|
var fileHandleId = await fileRepository.TryNewFileAsync(name, stream);
|
|
|
|
return fileHandleId != null ? RedirectToAction("Index", "Files", new { id = fileHandleId }) : RedirectToAction("NotEnoughSpace");
|
|
}
|
|
|
|
public IActionResult NotEnoughSpace()
|
|
{
|
|
return View();
|
|
}
|
|
|
|
public async Task<IActionResult> ConfirmDelete(Guid id)
|
|
{
|
|
|
|
var handle = await fileRepository.GetFileHandle(id);
|
|
|
|
var model = new FileHandleDeleteModel()
|
|
{
|
|
Id = handle.Id,
|
|
FileName = handle.Name,
|
|
BlockCount = handle.FileBlockCount,
|
|
};
|
|
|
|
return View(model);
|
|
}
|
|
|
|
[HttpPost]
|
|
public async Task<IActionResult> Delete(FileHandleDeleteModel model)
|
|
{
|
|
if (!ModelState.IsValid)
|
|
return RedirectToAction("ConfirmDelete", new {id = model.Id});
|
|
|
|
if (model.ConfirmFileName != model.FileName)
|
|
return RedirectToAction("ConfirmDelete", new {id = model.Id});
|
|
|
|
await fileRepository.DeleteFileHandle(model.Id);
|
|
|
|
return RedirectToAction("Index");
|
|
}
|
|
|
|
public async Task<IActionResult> Download(Guid id)
|
|
{
|
|
var filename = await fileRepository.GetFileName(id);
|
|
var stream = fileRepository.GetFileStream(id);
|
|
|
|
return File(stream, "application/octet-stream", filename);
|
|
}
|
|
} |