add possibility to upload and download from specific folder

This commit is contained in:
Alex Bezdieniezhnykh
2024-12-24 12:48:02 +02:00
parent 81d2409009
commit 0945635a1c
2 changed files with 21 additions and 14 deletions
+6 -6
View File
@@ -135,15 +135,15 @@ app.MapGet("/users",
.RequireAuthorization(apiAdminPolicy) .RequireAuthorization(apiAdminPolicy)
.WithOpenApi(op => new(op){ Summary = "List users by criteria"}); .WithOpenApi(op => new(op){ Summary = "List users by criteria"});
app.MapPost("/resources", app.MapPost("/resources/{dataFolder?}",
async (IFormFile data, IResourcesService resourceService, CancellationToken cancellationToken) async ([FromRoute]string? dataFolder, IFormFile data, IResourcesService resourceService, CancellationToken cancellationToken)
=> await resourceService.SaveResource(data, cancellationToken)) => await resourceService.SaveResource(dataFolder, data, cancellationToken))
.Accepts<IFormFile>("multipart/form-data") .Accepts<IFormFile>("multipart/form-data")
.RequireAuthorization(apiUploaderPolicy) .RequireAuthorization(apiUploaderPolicy)
.DisableAntiforgery(); .DisableAntiforgery();
app.MapPost("/resources/get", //Need to have POST method for secure password app.MapPost("/resources/get/{dataFolder?}", //Need to have POST method for secure password
async ([FromBody]GetResourceRequest request, IAuthService authService, IUserService userService, IResourcesService resourcesService, CancellationToken cancellationToken) => async ([FromBody]GetResourceRequest request, [FromRoute]string? dataFolder, IAuthService authService, IUserService userService, IResourcesService resourcesService, CancellationToken cancellationToken) =>
{ {
var user = authService.CurrentUser; var user = authService.CurrentUser;
if (user == null) if (user == null)
@@ -152,7 +152,7 @@ app.MapPost("/resources/get", //Need to have POST method for secure password
await userService.CheckHardware(user, request); await userService.CheckHardware(user, request);
var key = Security.MakeEncryptionKey(user.Email, request.Password, request.Hardware.Hash); var key = Security.MakeEncryptionKey(user.Email, request.Password, request.Hardware.Hash);
var stream = await resourcesService.GetEncryptedResource(request.FileName, key, cancellationToken); var stream = await resourcesService.GetEncryptedResource(dataFolder, request.FileName, key, cancellationToken);
return Results.File(stream, "application/octet-stream", request.FileName); return Results.File(stream, "application/octet-stream", request.FileName);
}).RequireAuthorization() }).RequireAuthorization()
+15 -8
View File
@@ -11,15 +11,17 @@ namespace Azaion.Services;
public interface IResourcesService public interface IResourcesService
{ {
Task<Stream> GetEncryptedResource(string fileName, string key, CancellationToken cancellationToken = default); Task<Stream> GetEncryptedResource(string? dataFolder, string fileName, string key, CancellationToken cancellationToken = default);
Task SaveResource(IFormFile data, CancellationToken cancellationToken = default); Task SaveResource(string? dataFolder, IFormFile data, CancellationToken cancellationToken = default);
} }
public class ResourcesService(IOptions<ResourcesConfig> resourcesConfig, ILogger<ResourcesService> logger) : IResourcesService public class ResourcesService(IOptions<ResourcesConfig> resourcesConfig, ILogger<ResourcesService> logger) : IResourcesService
{ {
public async Task<Stream> GetEncryptedResource(string fileName, string key, CancellationToken cancellationToken = default) public async Task<Stream> GetEncryptedResource(string? dataFolder, string fileName, string key, CancellationToken cancellationToken = default)
{ {
var resourcePath = Path.Combine(resourcesConfig.Value.ResourcesFolder, fileName); var resourcePath = string.IsNullOrWhiteSpace(dataFolder)
? Path.Combine(resourcesConfig.Value.ResourcesFolder, fileName)
: Path.Combine(resourcesConfig.Value.ResourcesFolder, dataFolder, fileName);
var fileStream = new FileStream(resourcePath, FileMode.Open, FileAccess.Read); var fileStream = new FileStream(resourcePath, FileMode.Open, FileAccess.Read);
var ms = new MemoryStream(); var ms = new MemoryStream();
@@ -28,14 +30,19 @@ public class ResourcesService(IOptions<ResourcesConfig> resourcesConfig, ILogger
return ms; return ms;
} }
public async Task SaveResource(IFormFile data, CancellationToken cancellationToken = default) public async Task SaveResource(string? dataFolder, IFormFile data, CancellationToken cancellationToken = default)
{ {
if (data == null) if (data == null)
throw new BusinessException(ExceptionEnum.NoFileProvided); throw new BusinessException(ExceptionEnum.NoFileProvided);
if (!Directory.Exists(resourcesConfig.Value.ResourcesFolder))
Directory.CreateDirectory(resourcesConfig.Value.ResourcesFolder);
var resourcePath = Path.Combine(resourcesConfig.Value.ResourcesFolder, data.FileName); var resourceFolder = string.IsNullOrWhiteSpace(dataFolder)
? resourcesConfig.Value.ResourcesFolder
: Path.Combine(resourcesConfig.Value.ResourcesFolder, dataFolder);
if (!Directory.Exists(resourceFolder))
Directory.CreateDirectory(resourceFolder);
var resourcePath = Path.Combine(resourceFolder, data.FileName);
await using var fileStream = new FileStream(resourcePath, FileMode.OpenOrCreate, FileAccess.ReadWrite); await using var fileStream = new FileStream(resourcePath, FileMode.OpenOrCreate, FileAccess.ReadWrite);
await data.CopyToAsync(fileStream, cancellationToken); await data.CopyToAsync(fileStream, cancellationToken);
logger.LogInformation($"Resource {data.FileName} Saved Successfully"); logger.LogInformation($"Resource {data.FileName} Saved Successfully");