1 files changed, 62 insertions, 0 deletions
diff --git a/ReferenceClientProxyImplementation/Services/ClientStoreService.cs b/ReferenceClientProxyImplementation/Services/ClientStoreService.cs
new file mode 100644
index 0000000..6bd7418
--- /dev/null
+++ b/ReferenceClientProxyImplementation/Services/ClientStoreService.cs
@@ -0,0 +1,62 @@
+using ArcaneLibs.Extensions.Streams;
+using ReferenceClientProxyImplementation.Configuration;
+using ReferenceClientProxyImplementation.Patches.Implementations;
+
+namespace ReferenceClientProxyImplementation.Services;
+
+public class ClientStoreService(ProxyConfiguration config, PatchSet patches) {
+    private static readonly HttpClient HttpClient = new();
+    
+    public async Task<Stream> GetPatchedClientAsset(string relativePath) {
+        if (relativePath.StartsWith("/")) {
+            relativePath = relativePath[1..];
+        }
+        
+        var path = Path.Combine(config.TestClient.RevisionPath, "patched", relativePath);
+
+        if (File.Exists(path))
+            return File.OpenRead(path);
+        
+        var srcAsset = (await GetOrDownloadRawAsset(relativePath)).ReadToEnd().ToArray();
+        var result = await patches.ApplyPatches(relativePath, srcAsset);
+        Directory.CreateDirectory(Path.GetDirectoryName(path)!);
+        if (!result.SequenceEqual(srcAsset)) {
+            await File.WriteAllBytesAsync(path, result);
+            return File.OpenRead(path);
+        }
+        
+        Console.WriteLine($"No patches applied for {relativePath}, returning original asset.");
+        return new MemoryStream(srcAsset);
+    }
+
+    public async Task<Stream> GetOrDownloadRawAsset(string relativePath) {
+        relativePath = relativePath.TrimStart('/');
+        var assetPath = Path.Combine(config.TestClient.RevisionPath, "src", relativePath);
+        if (File.Exists(assetPath)) {
+            Console.WriteLine($"Asset {relativePath} already exists at {assetPath}, returning existing file.");
+            return File.OpenRead(assetPath);
+        }
+        
+        var url = $"{config.TestClient.RevisionBaseUrl}/{relativePath}";
+        var response = await HttpClient.GetAsync(url, HttpCompletionOption.ResponseHeadersRead);
+        if (!response.IsSuccessStatusCode) {
+            Console.WriteLine($"Failed to download asset {relativePath} from {url}, status code: {response.StatusCode}");
+            throw new FileNotFoundException($"Asset not found: {relativePath}");
+        }
+        var contentStream = await response.Content.ReadAsStreamAsync();
+        Directory.CreateDirectory(Path.GetDirectoryName(assetPath)!);
+        await using var fileStream = File.Create(assetPath);
+        await contentStream.CopyToAsync(fileStream);
+        fileStream.Close();
+        contentStream.Close();
+        Console.WriteLine($"Downloaded asset {relativePath} to {assetPath}");
+
+        return File.OpenRead(assetPath);
+    }
+
+    public bool HasRawAsset(string relativePath) {
+        relativePath = relativePath.TrimStart('/');
+        var assetPath = Path.Combine(config.TestClient.RevisionPath, "src", relativePath);
+        return File.Exists(assetPath);
+    }
+}
\ No newline at end of file