Refactor and optimize route handling, clean up code structure, and apply minor improvements

This commit is contained in:
2025-08-20 13:52:52 +03:30
parent 64e025d9f7
commit 462f10ffb2
4 changed files with 60 additions and 63 deletions

View File

@@ -5,6 +5,6 @@ namespace VirtualDDNSRouter.Server.Context;
[YamlStaticContext]
[YamlSerializable(typeof(Rule))]
public partial class YamlStaticContext : YamlDotNet.Serialization.StaticContext
public partial class YamlStaticContext : StaticContext
{
}

View File

@@ -1,5 +1,4 @@
using System.Net;
using YamlDotNet.Serialization;
namespace VirtualDDNSRouter.Server.Models;
@@ -9,16 +8,19 @@ public record Rule
public string apiKey { get; set; } = string.Empty;
public string path { get; set; } = string.Empty;
public Rule() { } // Needed for AOT static deserializer
public Rule()
{
} // Needed for AOT static deserializer
}
public record Route
{
public string path { get; set; } = string.Empty;
public IPAddress ipAddress { get; set; } = IPAddress.None;
public UInt16 port { get; set; } = 80;
public ushort port { get; set; } = 80;
public Route() { }
public Route()
{
}
}

View File

@@ -19,14 +19,12 @@ app.UseForwardedHeaders(new ForwardedHeadersOptions
ForwardedHeaders.XForwardedProto
});
if (app.Environment.IsDevelopment())
{
app.MapOpenApi();
}
if (app.Environment.IsDevelopment()) app.MapOpenApi();
List<Route> routes = new List<Route>();
var routes = new List<Route>();
app.MapGet("/setip/{path}/{port}/{apiKey}", async (IYamlParser yamlParser ,HttpContext context, string path, UInt16 port, string apiKey) =>
app.MapGet("/setip/{path}/{port}/{apiKey}",
async (IYamlParser yamlParser, HttpContext context, string path, ushort port, string apiKey) =>
{
var rules = await yamlParser.GetRules();
app.Logger.LogInformation($"New setip request for {path} with port {port}.");
@@ -36,27 +34,24 @@ app.MapGet("/setip/{path}/{port}/{apiKey}", async (IYamlParser yamlParser ,HttpC
app.Logger.LogInformation($"Invalid rule for {path} with port {port}.");
return Results.StatusCode(StatusCodes.Status403Forbidden);
}
var clientIp = context.Connection.RemoteIpAddress;
if (clientIp is null)
{
app.Logger.LogInformation($"Could not get the client ip address for {path} with port {port}.");
return Results.BadRequest("Could not get the client ip address");
}
var existingRoute = routes.FirstOrDefault(r => r.path == path);
if (existingRoute is not null)
{
existingRoute.ipAddress = clientIp;
existingRoute.port = port;
}
else
{
if (existingRoute is not null) routes.Remove(existingRoute);
routes.Add(new Route
{
ipAddress = clientIp,
path = path,
port = port
});
}
return Results.Ok($"goto/{path}");
});