From f73342cb6b0592d3a6310f9a12fd40b4bf218e5c Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 18 Dec 2022 20:07:29 +0800
Subject: Develop secret api. v42

---
 .../CrupestApi.Commons/HttpContextExtensions.cs    | 110 +++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 docker/crupest-api/CrupestApi/CrupestApi.Commons/HttpContextExtensions.cs

(limited to 'docker/crupest-api/CrupestApi/CrupestApi.Commons/HttpContextExtensions.cs')

diff --git a/docker/crupest-api/CrupestApi/CrupestApi.Commons/HttpContextExtensions.cs b/docker/crupest-api/CrupestApi/CrupestApi.Commons/HttpContextExtensions.cs
new file mode 100644
index 0000000..1d7d858
--- /dev/null
+++ b/docker/crupest-api/CrupestApi/CrupestApi.Commons/HttpContextExtensions.cs
@@ -0,0 +1,110 @@
+using System.Text.Json;
+using Microsoft.Extensions.Options;
+
+namespace CrupestApi.Commons;
+
+public delegate void HttpResponseAction(HttpResponse response);
+
+public class MessageBody
+{
+    public MessageBody(string message)
+    {
+        Message = message;
+    }
+
+    public string Message { get; set; }
+}
+
+public static class CrupestApiJsonExtensions
+{
+    public static IServiceCollection AddJsonOptions(this IServiceCollection services)
+    {
+        services.AddOptions<JsonSerializerOptions>();
+        services.Configure<JsonSerializerOptions>(config =>
+        {
+            config.AllowTrailingCommas = true;
+            config.PropertyNameCaseInsensitive = true;
+            config.PropertyNamingPolicy = JsonNamingPolicy.CamelCase;
+        });
+
+        return services;
+    }
+
+    public static async Task<JsonDocument> ReadJsonAsync(this HttpRequest request)
+    {
+        using var stream = request.Body;
+        return await JsonDocument.ParseAsync(stream);
+    }
+
+    public static async Task WriteJsonAsync<T>(this HttpResponse response, T bodyObject, int statusCode = 200, HttpResponseAction? beforeWriteBody = null, CancellationToken cancellationToken = default)
+    {
+        var jsonOptions = response.HttpContext.RequestServices.GetRequiredService<IOptionsSnapshot<JsonSerializerOptions>>();
+        byte[] json = JsonSerializer.SerializeToUtf8Bytes<T>(bodyObject, jsonOptions.Value);
+
+        var byteCount = json.Length;
+
+        response.StatusCode = statusCode;
+        response.Headers.ContentType = "application/json; charset=utf-8";
+        response.Headers.ContentLength = byteCount;
+
+        if (beforeWriteBody is not null)
+        {
+            beforeWriteBody(response);
+        }
+
+        await response.Body.WriteAsync(json, cancellationToken);
+    }
+
+    public static async Task WriteMessageAsync(this HttpResponse response, string message, int statusCode = 400, HttpResponseAction? beforeWriteBody = null, CancellationToken cancellationToken = default)
+    {
+        await response.WriteJsonAsync(new MessageBody(message), statusCode: statusCode, beforeWriteBody, cancellationToken);
+    }
+
+    public static Task ResponseJsonAsync<T>(this HttpContext context, T bodyObject, int statusCode = 200, HttpResponseAction? beforeWriteBody = null, CancellationToken cancellationToken = default)
+    {
+        return context.Response.WriteJsonAsync<T>(bodyObject, statusCode, beforeWriteBody, cancellationToken);
+    }
+
+    public static Task ResponseMessageAsync(this HttpContext context, string message, int statusCode = 400, HttpResponseAction? beforeWriteBody = null, CancellationToken cancellationToken = default)
+    {
+        return context.Response.WriteMessageAsync(message, statusCode, beforeWriteBody, cancellationToken);
+    }
+
+    public static string? GetToken(this HttpRequest request)
+    {
+        var token = request.Headers["Authorization"].ToString();
+        if (token.StartsWith("Bearer "))
+        {
+            token = token.Substring("Bearer ".Length);
+            return token;
+        }
+
+        if (request.Query.TryGetValue("token", out var tokenValues))
+        {
+            return tokenValues.Last();
+        }
+
+        return null;
+    }
+
+    public static bool RequirePermission(this HttpContext context, string? permission)
+    {
+        if (permission is null) return true;
+
+        var token = context.Request.GetToken();
+        if (token is null)
+        {
+            context.ResponseMessageAsync("Unauthorized", 401);
+            return false;
+        }
+
+        var secretService = context.RequestServices.GetRequiredService<ISecretService>();
+        var permissions = secretService.GetPermissions(token);
+        if (!permissions.Contains(permission))
+        {
+            context.ResponseMessageAsync("Forbidden", 403);
+            return false;
+        }
+        return true;
+    }
+}
-- 
cgit v1.2.3