From d4bc6b545c6ff38173af57b5dfe8354fcd4a85a2 Mon Sep 17 00:00:00 2001
From: johnd0e <1838643+johnd0e@users.noreply.github.com>
Date: Sun, 20 Oct 2024 18:01:56 +0200
Subject: [PATCH] socket.headers.canonic: use dynamic canonization instead of
 limited set of predefined values

---
 src/headers.lua | 101 ++++--------------------------------------------
 1 file changed, 7 insertions(+), 94 deletions(-)

diff --git a/src/headers.lua b/src/headers.lua
index 1eb8223..5fb3af5 100644
--- a/src/headers.lua
+++ b/src/headers.lua
@@ -7,98 +7,11 @@ local socket = require("socket")
 socket.headers = {}
 local _M = socket.headers
 
-_M.canonic = {
-    ["accept"] = "Accept",
-    ["accept-charset"] = "Accept-Charset",
-    ["accept-encoding"] = "Accept-Encoding",
-    ["accept-language"] = "Accept-Language",
-    ["accept-ranges"] = "Accept-Ranges",
-    ["action"] = "Action",
-    ["alternate-recipient"] = "Alternate-Recipient",
-    ["age"] = "Age",
-    ["allow"] = "Allow",
-    ["arrival-date"] = "Arrival-Date",
-    ["authorization"] = "Authorization",
-    ["bcc"] = "Bcc",
-    ["cache-control"] = "Cache-Control",
-    ["cc"] = "Cc",
-    ["comments"] = "Comments",
-    ["connection"] = "Connection",
-    ["content-description"] = "Content-Description",
-    ["content-disposition"] = "Content-Disposition",
-    ["content-encoding"] = "Content-Encoding",
-    ["content-id"] = "Content-ID",
-    ["content-language"] = "Content-Language",
-    ["content-length"] = "Content-Length",
-    ["content-location"] = "Content-Location",
-    ["content-md5"] = "Content-MD5",
-    ["content-range"] = "Content-Range",
-    ["content-transfer-encoding"] = "Content-Transfer-Encoding",
-    ["content-type"] = "Content-Type",
-    ["cookie"] = "Cookie",
-    ["date"] = "Date",
-    ["diagnostic-code"] = "Diagnostic-Code",
-    ["dsn-gateway"] = "DSN-Gateway",
-    ["etag"] = "ETag",
-    ["expect"] = "Expect",
-    ["expires"] = "Expires",
-    ["final-log-id"] = "Final-Log-ID",
-    ["final-recipient"] = "Final-Recipient",
-    ["from"] = "From",
-    ["host"] = "Host",
-    ["if-match"] = "If-Match",
-    ["if-modified-since"] = "If-Modified-Since",
-    ["if-none-match"] = "If-None-Match",
-    ["if-range"] = "If-Range",
-    ["if-unmodified-since"] = "If-Unmodified-Since",
-    ["in-reply-to"] = "In-Reply-To",
-    ["keywords"] = "Keywords",
-    ["last-attempt-date"] = "Last-Attempt-Date",
-    ["last-modified"] = "Last-Modified",
-    ["location"] = "Location",
-    ["max-forwards"] = "Max-Forwards",
-    ["message-id"] = "Message-ID",
-    ["mime-version"] = "MIME-Version",
-    ["original-envelope-id"] = "Original-Envelope-ID",
-    ["original-recipient"] = "Original-Recipient",
-    ["pragma"] = "Pragma",
-    ["proxy-authenticate"] = "Proxy-Authenticate",
-    ["proxy-authorization"] = "Proxy-Authorization",
-    ["range"] = "Range",
-    ["received"] = "Received",
-    ["received-from-mta"] = "Received-From-MTA",
-    ["references"] = "References",
-    ["referer"] = "Referer",
-    ["remote-mta"] = "Remote-MTA",
-    ["reply-to"] = "Reply-To",
-    ["reporting-mta"] = "Reporting-MTA",
-    ["resent-bcc"] = "Resent-Bcc",
-    ["resent-cc"] = "Resent-Cc",
-    ["resent-date"] = "Resent-Date",
-    ["resent-from"] = "Resent-From",
-    ["resent-message-id"] = "Resent-Message-ID",
-    ["resent-reply-to"] = "Resent-Reply-To",
-    ["resent-sender"] = "Resent-Sender",
-    ["resent-to"] = "Resent-To",
-    ["retry-after"] = "Retry-After",
-    ["return-path"] = "Return-Path",
-    ["sender"] = "Sender",
-    ["server"] = "Server",
-    ["smtp-remote-recipient"] = "SMTP-Remote-Recipient",
-    ["status"] = "Status",
-    ["subject"] = "Subject",
-    ["te"] = "TE",
-    ["to"] = "To",
-    ["trailer"] = "Trailer",
-    ["transfer-encoding"] = "Transfer-Encoding",
-    ["upgrade"] = "Upgrade",
-    ["user-agent"] = "User-Agent",
-    ["vary"] = "Vary",
-    ["via"] = "Via",
-    ["warning"] = "Warning",
-    ["will-retry-until"] = "Will-Retry-Until",
-    ["www-authenticate"] = "WWW-Authenticate",
-    ["x-mailer"] = "X-Mailer",
-}
+_M.canonic = setmetatable({},{
+  __index=function (t,k)
+    t[k] = string.gsub(k, "%f[%w]%l", string.upper)
+    return t[k]
+  end;
+})
 
-return _M
\ No newline at end of file
+return _M