Iceshrimp.NET/Iceshrimp.Backend/Pages/Queue.cshtml.cs
2024-06-29 01:06:19 +02:00

73 lines
No EOL
2.3 KiB
C#

using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Middleware;
using Iceshrimp.Backend.Core.Services;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.RazorPages;
using Microsoft.EntityFrameworkCore;
namespace Iceshrimp.Backend.Pages;
public class QueueModel(DatabaseContext db, QueueService queueSvc) : PageModel
{
public int? DelayedCount;
public Job.JobStatus? Filter;
public List<Job> Jobs = [];
public int? NextPage;
public int? PrevPage;
public string? Queue;
public int? QueuedCount;
public int? RunningCount;
public int? TotalCount;
public async Task<IActionResult> OnGet(
[FromRoute] string? queue, [FromRoute(Name = "pagination")] int? page, [FromRoute] string? status
)
{
if (!Request.Cookies.TryGetValue("admin_session", out var cookie))
return Redirect("/login");
if (!await db.Sessions.AnyAsync(p => p.Token == cookie && p.Active && p.User.IsAdmin))
return Redirect("/login");
if (queue == null)
return Page();
if (!queueSvc.QueueNames.Contains(queue))
throw GracefulException.BadRequest($"Unknown queue: {queue}");
Queue = queue;
if (page is null or < 1)
page = 1;
var query = db.Jobs.Where(p => p.Queue == queue);
if (status is { Length: > 0 })
{
if (!Enum.TryParse<Job.JobStatus>(status, true, out var jobStatus))
throw GracefulException.BadRequest($"Unknown status: {status}");
query = query.Where(p => p.Status == jobStatus);
Filter = jobStatus;
}
Jobs = await query.OrderByDescending(p => p.Id)
.Skip((page.Value - 1) * 50)
.Take(50)
.ToListAsync();
if (Filter == null)
{
TotalCount = await db.Jobs.CountAsync(p => p.Queue == queue);
QueuedCount = await db.Jobs.CountAsync(p => p.Queue == queue && p.Status == Job.JobStatus.Queued);
RunningCount = await db.Jobs.CountAsync(p => p.Queue == queue && p.Status == Job.JobStatus.Running);
DelayedCount = await db.Jobs.CountAsync(p => p.Queue == queue && p.Status == Job.JobStatus.Delayed);
}
else
{
TotalCount = await query.CountAsync();
}
if (Jobs.Count >= 50)
NextPage = page + 1;
if (page is > 1)
PrevPage = page - 1;
return Page();
}
}