Moonbase/jumpstart-games-reduxed
1
0
Fork 0
forked from SoDOff-Project/sodoff
Code Issues 1 Pull Requests Projects 1 Activity

add "auto encrypt" function for *Main.xml

Browse Source
This commit is contained in:
Robert Paciorek 2024-02-04 12:32:52 +00:00
parent bd6df27c92
commit e1a8f47a27
2 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Configuration/AssetServerConfig.cs
View File

@ -5,6 +5,8 @@ public class AssetServerConfig {
public string URLPrefix { get; set; } = string.Empty;
public AssetServerMode Mode { get; set; }
public string ProviderURL { get; set; } = string.Empty;
public string AutoEncryptRegexp { get; set; } = string.Empty;
public string AutoEncryptKey { get; set; } = string.Empty;
public bool SubstituteMissingLocalAssets { get; set; } = false;
public bool UseCache { get; set; } = true;
}

20
src/Middleware/AssetMiddleware.cs
View File

@ -1,18 +1,25 @@
using Microsoft.Extensions.Options;
using System.Net;
using System.IO;
using System.Text.RegularExpressions;
using sodoff.Configuration;
using sodoff.Util;
namespace sodoff.Middleware;
public class AssetMiddleware
{
private readonly RequestDelegate _next;
private readonly IOptions<AssetServerConfig> config;
private readonly Regex autoEncryptRegexp;
public AssetMiddleware(RequestDelegate next, IOptions<AssetServerConfig> config)
{
_next = next;
this.config = config;
if (config.Value.AutoEncryptRegexp != "")
autoEncryptRegexp = new Regex(config.Value.AutoEncryptRegexp);
else
autoEncryptRegexp = null;
}
public async Task Invoke(HttpContext context)
@ -33,7 +40,6 @@ public class AssetMiddleware
}
string assetPath = path.Remove(0, config.Value.URLPrefix.Length + 1);
string localPath = GetLocalPath("assets/" + assetPath);
if (localPath == string.Empty && config.Value.Mode == AssetServerMode.Partial && config.Value.UseCache)
@ -47,7 +53,17 @@ public class AssetMiddleware
}
else {
context.Response.Headers["Content-Type"] = "application/octet-stream";
await context.Response.SendFileAsync(Path.GetFullPath(localPath));
bool needEncrypt = autoEncryptRegexp != null && autoEncryptRegexp.IsMatch(localPath);
if (needEncrypt) {
await context.Response.WriteAsync(
TripleDES.EncryptASCII(
System.IO.File.ReadAllText(localPath),
config.Value.AutoEncryptKey
)
);
} else {
await context.Response.SendFileAsync(Path.GetFullPath(localPath));
}
}
}