notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/api] Allow optionally storing a drive file in a specified folder

Browse source
This commit is contained in:
pancakes 2024-12-16 14:41:06 +10:00 • committed by Laura Hausmann
parent 673e5c88bc
commit 98686e334e
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
2 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
Iceshrimp.Backend/Controllers/Web/DriveController.cs
View file

@ -118,7 +118,7 @@ public class DriveController(
[Produces(MediaTypeNames.Application.Json)]
[ProducesResults(HttpStatusCode.OK)]
[MaxRequestSizeIsMaxUploadSize]
public async Task<DriveFileResponse> UploadFile(IFormFile file)
public async Task<DriveFileResponse> UploadFile(IFormFile file, [FromQuery] string? folderId)
{
var user = HttpContext.GetUserOrFail();
var request = new DriveFileCreationRequest
@ -127,7 +127,7 @@ public class DriveController(
MimeType = file.ContentType,
IsSensitive = false
};
var res = await driveSvc.StoreFileAsync(file.OpenReadStream(), user, request);
var res = await driveSvc.StoreFileAsync(file.OpenReadStream(), user, request, folderId: folderId);
return await GetFileById(res.Id);
}

13
Iceshrimp.Backend/Core/Services/DriveService.cs
View file

@ -165,7 +165,7 @@ public class DriveService(
}
public async Task<DriveFile> StoreFileAsync(
Stream input, User user, DriveFileCreationRequest request, bool skipImageProcessing = false
Stream input, User user, DriveFileCreationRequest request, bool skipImageProcessing = false, string? folderId = null
)
{
if (user.IsLocalUser && input.Length > storageConfig.Value.MaxUploadSizeBytes)
@ -324,6 +324,14 @@ public class DriveService(
request.Filename += $".{fmt.Extension}";
}
// If the requested folder doesn't exist then store it in the root folder
if (folderId != null)
{
var folder = await db.DriveFolders
.FirstOrDefaultAsync(p => p.Id == folderId && p.UserId == user.Id);
if (folder == null) folderId = null;
}
file = new DriveFile
{
Id = IdHelpers.GenerateSnowflakeId(),
@ -351,7 +359,8 @@ public class DriveService(
ThumbnailMimeType = thumbnail?.format.Format.MimeType,
PublicUrl = @public?.url,
PublicAccessKey = @public?.accessKey,
PublicMimeType = @public?.format.Format.MimeType
PublicMimeType = @public?.format.Format.MimeType,
FolderId = folderId
};
await db.AddAsync(file);