From 76dd6337a2a6b9979630680a9db745ddf2e92d7e Mon Sep 17 00:00:00 2001
From: Donovan Daniels <hewwo@yiff.rocks>
Date: Thu, 2 May 2024 23:09:44 -0500
Subject: [PATCH] log forwarded host

---
 config/environments/production.rb | 2 +-
 config/initializers/log_tags.rb   | 8 ++++++++
 docker-compose.prod.yml           | 2 --
 docker-compose.yml                | 2 --
 4 files changed, 9 insertions(+), 5 deletions(-)
 create mode 100644 config/initializers/log_tags.rb

diff --git a/config/environments/production.rb b/config/environments/production.rb
index 301ba84..41d88a0 100644
--- a/config/environments/production.rb
+++ b/config/environments/production.rb
@@ -59,7 +59,7 @@ Rails.application.configure do
                                        .then { |logger| ActiveSupport::TaggedLogging.new(logger) }
 
   # Prepend all log lines with the following tags.
-  config.log_tags = %i[request_id host]
+  config.log_tags = %i[request_id forwarded_host]
 
   # Info include generic and useful information about system operation, but avoids logging too much
   # information to avoid inadvertent exposure of personally identifiable information (PII). If you
diff --git a/config/initializers/log_tags.rb b/config/initializers/log_tags.rb
new file mode 100644
index 0000000..f308e93
--- /dev/null
+++ b/config/initializers/log_tags.rb
@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+module RequestExtension
+  def forwarded_host
+    headers["X-Forwarded-Host"] || host
+  end
+end
+
+ActiveSupport.on_load(:action_dispatch_request) { include RequestExtension }
diff --git a/docker-compose.prod.yml b/docker-compose.prod.yml
index 6e1be8b..700e647 100644
--- a/docker-compose.prod.yml
+++ b/docker-compose.prod.yml
@@ -1,5 +1,3 @@
-version: "3"
-
 x-environment: &common-env
   WEBSITES_READONLY: "${READONLY:-0}"
 
diff --git a/docker-compose.yml b/docker-compose.yml
index 7122964..80cb6d0 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,5 +1,3 @@
-version: "3"
-
 x-environment: &common-env
   WEBSITES_READONLY: "${READONLY:-0}"