85 lines
No EOL
2.8 KiB
C#
85 lines
No EOL
2.8 KiB
C#
using System.Net.Mime;
|
|
using Iceshrimp.Backend.Controllers.Mastodon.Attributes;
|
|
using Iceshrimp.Backend.Controllers.Mastodon.Schemas;
|
|
using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
|
|
using Iceshrimp.Backend.Core.Database;
|
|
using Iceshrimp.Backend.Core.Database.Tables;
|
|
using Iceshrimp.Backend.Core.Extensions;
|
|
using Iceshrimp.Backend.Core.Middleware;
|
|
using Iceshrimp.Backend.Core.Services;
|
|
using Microsoft.AspNetCore.Cors;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.RateLimiting;
|
|
using Microsoft.EntityFrameworkCore;
|
|
|
|
namespace Iceshrimp.Backend.Controllers.Mastodon;
|
|
|
|
[MastodonApiController]
|
|
[Authenticate]
|
|
[Authorize("write:media")]
|
|
[EnableCors("mastodon")]
|
|
[EnableRateLimiting("sliding")]
|
|
[Produces(MediaTypeNames.Application.Json)]
|
|
[ProducesResponseType(StatusCodes.Status200OK, Type = typeof(AttachmentEntity))]
|
|
public class MediaController(DriveService driveSvc, DatabaseContext db) : ControllerBase
|
|
{
|
|
[HttpPost("/api/v1/media")]
|
|
[HttpPost("/api/v2/media")]
|
|
public async Task<IActionResult> UploadAttachment(MediaSchemas.UploadMediaRequest request)
|
|
{
|
|
var user = HttpContext.GetUserOrFail();
|
|
var rq = new DriveFileCreationRequest
|
|
{
|
|
Filename = request.File.FileName,
|
|
IsSensitive = false,
|
|
Comment = request.Description,
|
|
MimeType = request.File.ContentType
|
|
};
|
|
var file = await driveSvc.StoreFile(request.File.OpenReadStream(), user, rq);
|
|
var res = RenderAttachment(file);
|
|
|
|
return Ok(res);
|
|
}
|
|
|
|
[HttpPut("/api/v1/media/{id}")]
|
|
[ProducesResponseType(StatusCodes.Status404NotFound, Type = typeof(MastodonErrorResponse))]
|
|
public async Task<IActionResult> UpdateAttachment(string id, [FromHybrid] MediaSchemas.UpdateMediaRequest request)
|
|
{
|
|
var user = HttpContext.GetUserOrFail();
|
|
var file = await db.DriveFiles.FirstOrDefaultAsync(p => p.Id == id && p.User == user) ??
|
|
throw GracefulException.RecordNotFound();
|
|
file.Comment = request.Description;
|
|
await db.SaveChangesAsync();
|
|
|
|
var res = RenderAttachment(file);
|
|
return Ok(res);
|
|
}
|
|
|
|
[HttpGet("/api/v1/media/{id}")]
|
|
[ProducesResponseType(StatusCodes.Status404NotFound, Type = typeof(MastodonErrorResponse))]
|
|
public async Task<IActionResult> GetAttachment(string id)
|
|
{
|
|
var user = HttpContext.GetUserOrFail();
|
|
var file = await db.DriveFiles.FirstOrDefaultAsync(p => p.Id == id && p.User == user) ??
|
|
throw GracefulException.RecordNotFound();
|
|
|
|
var res = RenderAttachment(file);
|
|
return Ok(res);
|
|
}
|
|
|
|
private static AttachmentEntity RenderAttachment(DriveFile file)
|
|
{
|
|
return new AttachmentEntity
|
|
{
|
|
Id = file.Id,
|
|
Type = AttachmentEntity.GetType(file.Type),
|
|
Url = file.PublicUrl,
|
|
Blurhash = file.Blurhash,
|
|
Description = file.Comment,
|
|
PreviewUrl = file.PublicThumbnailUrl,
|
|
RemoteUrl = file.Uri,
|
|
Sensitive = file.IsSensitive,
|
|
//Metadata = TODO,
|
|
};
|
|
}
|
|
} |