Rewrite template providing to allow custom ones

Signed-off-by: Eloi DEMOLIS <[email protected]>
sozu-proxy · Dec 17, 2024 · 663b67f · 663b67f
1 parent 4102871
commit 663b67f
Show file tree

Hide file tree

Showing 16 changed files with 371 additions and 479 deletions.
diff --git a/bin/src/ctl/request_builder.rs b/bin/src/ctl/request_builder.rs
@@ -252,6 +252,7 @@ impl CommandManager {
                     },
                     redirect: todo!(),
                     redirect_scheme: todo!(),
+                    redirect_template: todo!(),
                     rewrite_host: todo!(),
                     rewrite_path: todo!(),
                     rewrite_port: todo!(),
@@ -305,6 +306,7 @@ impl CommandManager {
                     },
                     redirect: todo!(),
                     redirect_scheme: todo!(),
+                    redirect_template: todo!(),
                     rewrite_host: todo!(),
                     rewrite_path: todo!(),
                     rewrite_port: todo!(),
@@ -351,8 +353,8 @@ impl CommandManager {
             } => {
                 let https_listener = ListenerBuilder::new_https(address.into())
                     .with_public_address(public_address)
-                    .with_answer_404_path(answer_404)
-                    .with_answer_503_path(answer_503)
+                    .with_answer("404", answer_404)
+                    .with_answer("503", answer_503)
                     .with_tls_versions(tls_versions)
                     .with_cipher_list(cipher_list)
                     .with_expect_proxy(expect_proxy)
@@ -394,8 +396,8 @@ impl CommandManager {
             } => {
                 let http_listener = ListenerBuilder::new_http(address.into())
                     .with_public_address(public_address)
-                    .with_answer_404_path(answer_404)
-                    .with_answer_503_path(answer_503)
+                    .with_answer("404", answer_404)
+                    .with_answer("503", answer_503)
                     .with_expect_proxy(expect_proxy)
                     .with_sticky_name(sticky_name)
                     .with_front_timeout(front_timeout)

diff --git a/command/assets/config.toml b/command/assets/config.toml
@@ -17,7 +17,7 @@ protocol = "http"
 [[listeners]]
 address = "0.0.0.0:443"
 protocol = "https"
-answer_404 = "./assets/custom_404.html"
+answers = { "404" = "./assets/custom_404.html" }
 tls_versions = ["TLS_V12"]
 
 [[listeners]]
@@ -28,7 +28,7 @@ expect_proxy = true
 [clusters]
 [clusters.MyCluster]
 protocol = "http"
-answer_503 = "./assets/custom_503.html"
+answers = { "503" = "./assets/custom_503.html" }
 #sticky_session = false
 #https_redirect = false
 frontends = [

diff --git a/command/src/command.proto b/command/src/command.proto
@@ -129,7 +129,7 @@ message HttpListenerConfig {
     required uint32 request_timeout = 10 [default = 10];
     // wether the listener is actively listening on its socket
     required bool active = 11 [default = false];
-    optional CustomHttpAnswers http_answers = 12;
+    map<string, string> answers = 13;
 }
 
 // details of an HTTPS listener
@@ -161,7 +161,7 @@ message HttpsListenerConfig {
     // The tickets allow the client to resume a session. This protects the client
     // agains session tracking. Defaults to 4.
     required uint64 send_tls13_tickets = 20;
-    optional CustomHttpAnswers http_answers = 21;
+    map<string, string> answers = 22;
 }
 
 // details of an TCP listener
@@ -179,31 +179,6 @@ message TcpListenerConfig {
     required bool active = 7 [default = false];
 }
 
-// custom HTTP answers, useful for 404, 503 pages
-message CustomHttpAnswers {
-    // MovedPermanently
-    optional string answer_301 = 1;
-    // BadRequest
-    optional string answer_400 = 2;
-    // Unauthorized
-    optional string answer_401 = 3;
-    // NotFound
-    optional string answer_404 = 4;
-    // RequestTimeout
-    optional string answer_408 = 5;
-    // PayloadTooLarge
-    optional string answer_413 = 6;
-    // BadGateway
-    optional string answer_502 = 7;
-    // ServiceUnavailable
-    optional string answer_503 = 8;
-    // GatewayTimeout
-    optional string answer_504 = 9;
-    // InsufficientStorage
-    optional string answer_507 = 10;
-
-}
-
 message ActivateListener {
     required SocketAddress address = 1;
     required ListenerType proxy = 2;
@@ -239,10 +214,9 @@ message ListenersList {
 
 enum RedirectPolicy {
     FORWARD = 0;
-    FORCE_HTTPS = 1;
-    TEMPORARY = 2;
-    PERMANENT = 3;
-    UNAUTHORIZED = 4;
+    TEMPORARY = 1;
+    PERMANENT = 2;
+    UNAUTHORIZED = 3;
 }
 
 enum RedirectScheme {
@@ -263,9 +237,10 @@ message RequestHttpFrontend {
     map<string, string> tags = 7;
     optional RedirectPolicy redirect = 8;
     optional RedirectScheme redirect_scheme = 9;
-    optional string rewrite_host = 10;
-    optional string rewrite_path = 11;
-    optional uint32 rewrite_port = 12;
+    optional string redirect_template = 10;
+    optional string rewrite_host = 11;
+    optional string rewrite_path = 12;
+    optional uint32 rewrite_port = 13;
 }
 
 message RequestTcpFrontend {
@@ -393,9 +368,9 @@ message Cluster {
     required bool https_redirect = 3;
     optional ProxyProtocolConfig proxy_protocol = 4;
     required LoadBalancingAlgorithms load_balancing = 5 [default = ROUND_ROBIN];
-    optional string answer_503 = 6;
     optional LoadMetric load_metric = 7;
     optional uint32 https_redirect_port = 8;
+    map<string, string> answers = 9;
 }
 
 enum LoadBalancingAlgorithms {