60 lines
No EOL
2.8 KiB
C#
60 lines
No EOL
2.8 KiB
C#
using Iceshrimp.Backend.Controllers.Mastodon.Renderers;
|
|
using Iceshrimp.Backend.Controllers.Mastodon.Schemas;
|
|
using Iceshrimp.Backend.Controllers.Mastodon.Schemas.Entities;
|
|
using Iceshrimp.Backend.Core.Database.Tables;
|
|
using Iceshrimp.Backend.Core.Middleware;
|
|
using Microsoft.EntityFrameworkCore;
|
|
|
|
namespace Iceshrimp.Backend.Core.Extensions;
|
|
|
|
public static class NoteQueryableExtensions {
|
|
public static IQueryable<Note> WithIncludes(this IQueryable<Note> query) {
|
|
return query.Include(p => p.User)
|
|
.Include(p => p.Renote)
|
|
.ThenInclude(p => p != null ? p.User : null)
|
|
.Include(p => p.Reply)
|
|
.ThenInclude(p => p != null ? p.User : null);
|
|
}
|
|
|
|
public static IQueryable<Note> Paginate(this IQueryable<Note> query, PaginationQuery p, int defaultLimit,
|
|
int maxLimit) {
|
|
if (p is { SinceId: not null, MinId: not null })
|
|
throw GracefulException.BadRequest("Can't use sinceId and minId params simultaneously");
|
|
|
|
query = p switch {
|
|
{ SinceId: not null, MaxId: not null } => query
|
|
.Where(note => note.Id.IsGreaterThan(p.SinceId) &&
|
|
note.Id.IsLessThan(p.MaxId))
|
|
.OrderByDescending(note => note.Id),
|
|
{ MinId: not null, MaxId: not null } => query
|
|
.Where(note => note.Id.IsGreaterThan(p.MinId) &&
|
|
note.Id.IsLessThan(p.MaxId))
|
|
.OrderBy(note => note.Id),
|
|
{ SinceId: not null } => query.Where(note => note.Id.IsGreaterThan(p.SinceId))
|
|
.OrderByDescending(note => note.Id),
|
|
{ MinId: not null } => query.Where(note => note.Id.IsGreaterThan(p.MinId)).OrderBy(note => note.Id),
|
|
{ MaxId: not null } => query.Where(note => note.Id.IsLessThan(p.MaxId)).OrderByDescending(note => note.Id),
|
|
_ => query
|
|
};
|
|
|
|
return query.Take(Math.Min(p.Limit ?? defaultLimit, maxLimit));
|
|
}
|
|
|
|
public static IQueryable<Note> HasVisibility(this IQueryable<Note> query, Note.NoteVisibility visibility) {
|
|
return query.Where(note => note.Visibility == visibility);
|
|
}
|
|
|
|
public static IQueryable<Note> FilterByFollowingAndOwn(this IQueryable<Note> query, User user) {
|
|
return query.Where(note => note.User == user || note.User.IsFollowedBy(user));
|
|
}
|
|
|
|
public static IQueryable<Note> OrderByIdDesc(this IQueryable<Note> query) {
|
|
return query.OrderByDescending(note => note.Id);
|
|
}
|
|
|
|
public static async Task<IEnumerable<Status>> RenderAllForMastodonAsync(
|
|
this IQueryable<Note> notes, NoteRenderer renderer) {
|
|
var list = await notes.ToListAsync();
|
|
return await list.Select(renderer.RenderAsync).AwaitAllAsync();
|
|
}
|
|
} |