From 973d0fda6f4a69fed465da3d668536c7f7fc2c5b Mon Sep 17 00:00:00 2001
From: Sheppy <sheppy@atlantishq.de>
Date: Sat, 24 Dec 2022 01:52:29 +0100
Subject: [PATCH] feat: proxy timeout option

---
 templates/nginx_stream_block.conf.j2 | 2 +-
 vm.py                                | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/templates/nginx_stream_block.conf.j2 b/templates/nginx_stream_block.conf.j2
index c00ae70..f4b3be4 100644
--- a/templates/nginx_stream_block.conf.j2
+++ b/templates/nginx_stream_block.conf.j2
@@ -3,7 +3,7 @@ server {
     # {{ comment }}
     listen {{ portstring }} {% if udp %} udp {% endif %}{% if ssl %} ssl {% endif %};
 
-    proxy_timeout 10s;
+    proxy_timeout {{ proxy_timeout }};
     proxy_responses 1;
     proxy_pass {{ targetip }}:$server_port;
 
diff --git a/vm.py b/vm.py
index 4669809..2a70142 100644
--- a/vm.py
+++ b/vm.py
@@ -43,11 +43,12 @@ class VM:
             transparent = portStruct.get("transparent")
             proto = portStruct.get("proto") or "tcp"
             isUDP = proto == "udp"
+            proxy_timeout = portStruct.get("proxy_timeout") or "10s"
 
             compositeName = "-".join((self.hostname, name, portstring, proto))
 
             component = template.render(targetip=self.ip, udp=isUDP, portstring=portstring,
-                                            transparent=transparent)
+                                        transparent=transparent, proxy_timeout=proxy_timeout)
             components.append(component)
 
         return components