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

[backend/masto-client] Add EnumerableExtensions

Browse source
This commit is contained in:
Laura Hausmann 2024-02-02 21:04:41 +01:00
parent 519b5280ac
commit b015c464dd
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
2 changed files with 32 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
Iceshrimp.Backend/Controllers/Mastodon/MastodonTimelineController.cs
View file

@ -1,8 +1,10 @@
using Iceshrimp.Backend.Controllers.Attributes;
using Iceshrimp.Backend.Controllers.Mastodon.Renderers;
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.Helpers;
using Iceshrimp.Backend.Core.Middleware;
using Microsoft.AspNetCore.Mvc;
@ -26,14 +28,13 @@ public class MastodonTimelineController(DatabaseContext db, NoteRenderer noteRen
[ProducesResponseType(StatusCodes.Status200OK, Type = typeof(IEnumerable<Status>))]
public async Task<IActionResult> GetHomeTimeline() {
var user = HttpContext.GetOauthUser() ?? throw new GracefulException("Failed to get user from HttpContext");
var notes = await db.Notes
.WithIncludes()
.IsFollowedBy(user)
.OrderByIdDesc()
.Take(40)
.ToListAsync();
var res = await Task.WhenAll(notes.Select(async p => await noteRenderer.RenderAsync(p)));
var notes = db.Notes
.WithIncludes()
.IsFollowedBy(user)
.OrderByIdDesc()
.Take(40);
var res = await notes.RenderAllForMastodonAsync(noteRenderer);
return Ok(res);
}
@ -42,14 +43,13 @@ public class MastodonTimelineController(DatabaseContext db, NoteRenderer noteRen
[Produces("application/json")]
[ProducesResponseType(StatusCodes.Status200OK, Type = typeof(IEnumerable<Status>))]
public async Task<IActionResult> GetPublicTimeline() {
var notes = await db.Notes
.WithIncludes()
.HasVisibility(Note.NoteVisibility.Public)
.OrderByIdDesc()
.Take(40)
.ToListAsync();
var res = await Task.WhenAll(notes.Select(async p => await noteRenderer.RenderAsync(p)));
var notes = db.Notes
.WithIncludes()
.HasVisibility(Note.NoteVisibility.Public)
.OrderByIdDesc()
.Take(40);
var res = await notes.RenderAllForMastodonAsync(noteRenderer);
return Ok(res);
}
}

16
Iceshrimp.Backend/Core/Extensions/EnumerableExtensions.cs Normal file
View file

@ -0,0 +1,16 @@
using Iceshrimp.Backend.Controllers.Mastodon.Renderers;
using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
using Iceshrimp.Backend.Core.Database.Tables;
namespace Iceshrimp.Backend.Core.Extensions;
public static class EnumerableExtensions {
public static async Task<IEnumerable<T>> AwaitAllAsync<T>(this IEnumerable<Task<T>> tasks) {
return await Task.WhenAll(tasks);
}
public static async Task<IEnumerable<Status>> RenderAllForMastodonAsync(
this IEnumerable<Note> notes, NoteRenderer renderer) {
return await notes.Select(async p => await renderer.RenderAsync(p)).AwaitAllAsync();
}
}