Program.cs

using ArtemisC2Server.Commands;
using ArtemisC2Server.Commands.Handlers;
using ArtemisC2Server.Components;
using ArtemisC2Server.Data;
using ArtemisC2Server.Data.Entities;
using ArtemisC2Server.Dns;
using ArtemisC2Server.Results.Handlers;
using ArtemisC2Server.Services;
using ArtemisC2Server.Services.Interfaces;
using ArtemisC2Server.Shared;
using ArtemisC2Server.Signals;
using ArtemisC2Server.Signals.Handlers;
using Microsoft.EntityFrameworkCore;
using MudBlazor.Services;

var builder = WebApplication.CreateBuilder(args);

// Add services to the container.

// Add services to the container.
builder.Services.AddRazorComponents()
    .AddInteractiveServerComponents();

builder.Services.AddDbContext<ArtemisC2Server.Data.AppDbContext>(options =>
    options.UseSqlite(builder.Configuration.GetConnectionString("DefaultConnection")));

builder.Services.AddHostedService<ArtemisC2Server.Dns.DnsService>();

//Services
builder.Services.AddScoped<IAgentService, AgentService>();
builder.Services.AddScoped<IJobService, JobService>();

//Dispatchers
builder.Services.AddScoped<ResultDispatcher>();
builder.Services.AddScoped<SignalDispatcher>();

//Result handlers
builder.Services.AddScoped<CompleteHandshakeHandler>();
builder.Services.AddScoped<WhoamiHandler>();
builder.Services.AddScoped<SetBeaconIntervalHandler>();
builder.Services.AddScoped<ShellHandler>();
builder.Services.AddScoped<DownloadHandler>();

//Signal handlers
builder.Services.AddScoped<BeaconHandler>();

//Entity change notifier
builder.Services.AddSingleton<EntityChangeNotifier<Agent>>();
builder.Services.AddSingleton<EntityChangeNotifier<Job>>();

//Chunked transfers
builder.Services.AddSingleton<ChunkedTrasfers>();

//Shared
builder.Services.AddScoped<TimeProvider, BrowserTimeProvider>();

builder.Services.AddMudServices();
var app = builder.Build();

// Configure the HTTP request pipeline.
if (!app.Environment.IsDevelopment())
{
    app.UseExceptionHandler("/Error", createScopeForErrors: true);
    // The default HSTS value is 30 days. You may want to change this for production scenarios, see https://aka.ms/aspnetcore-hsts.
    app.UseHsts();
}

app.UseHttpsRedirection();

app.UseStaticFiles();
app.UseAntiforgery();

app.MapRazorComponents<App>()
    .AddInteractiveServerRenderMode();

app.Run();