Program.cs

// Copyright © 2026 Xpl0itR
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

using System.Net;
using System.Threading.RateLimiting;
using Microsoft.AspNetCore.Connections;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Proto.Promises;
using Xport.AspNetCore.Adapters.Middleware.Logging;
using Xport.AspNetCore.Adapters.Transport.Sockets;
using Xport.Hosting;
using Xport.Middleware;
using Xport.Middleware.ConnectionLimiter;
using XportExample;

HostApplicationBuilder builder = Host.CreateApplicationBuilder(args);

#if DEBUG
builder.Logging.SetMinimumLevel(LogLevel.Debug);
#endif

builder.Services.AddSocketTransport()
                .AddHostedTransportServer(builder.Configuration.GetSection("Xport"))
                .AddConnectionServicesMiddleware()
                .AddLoggingTransportMiddleware()
                .AddConnectionLimiterMiddleware()
                .AddConnectionLimiterPolicy(
                    PartitionedRateLimiter.Create<BaseConnectionContext, string>(static _ =>
                        RateLimitPartition.GetConcurrencyLimiter("global", static _ =>
                            new ConcurrencyLimiterOptions { PermitLimit = 10000 })));

using IHost host = builder.Build();

ILogger logger = host.Services.GetRequiredService<ILogger<Program>>();
Promise.Config.UncaughtRejectionHandler = ex => logger.LogError(ex, "An unhandled exception has occurred in an unawaited promise.");

host.ConfigureTransportServer(static server =>
{
    server.ListenIp(IPAddress.Loopback, 1234, static options => // programmatically configure an endpoint
        options.UseSocketTransport()
               .UseSingleplexedMiddleware<ConnectionLimiterMiddleware>()
#if DEBUG
               .UseSingleplexedMiddleware<LoggingTransportMiddleware>()
#endif
               .UseSingleplexedMiddleware<ConnectionServicesMiddleware>()
               .UseSingleplexedConnectionHandler<SessionManager>());

    server.ConfigureEndpoint("EchoServer", static options => // configure an endpoint from configuration (see appsettings.json)
        options.UseSingleplexedMiddleware<ConnectionLimiterMiddleware>()
#if DEBUG
               .UseSingleplexedMiddleware<LoggingTransportMiddleware>()
#endif
               .Run(static async conn =>
               {
                   await conn.Transport.Output.WriteAsync("Welcome to the echo test server!\n"u8.ToArray(), conn.ConnectionClosed);
                   await conn.Transport.Input.CopyToAsync(conn.Transport.Output, conn.ConnectionClosed);
               }));
});

await host.RunAsync();