From 171cd4ecd22cf884750809bee702e5a707fca1b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Tue, 3 Nov 2020 19:26:24 +0100
Subject: [PATCH] chg: Pass max depth as parameter in the capture page

Related to #117
---
 website/web/__init__.py            |  3 ++-
 website/web/templates/capture.html | 14 +++++---------
 2 files changed, 7 insertions(+), 10 deletions(-)

diff --git a/website/web/__init__.py b/website/web/__init__.py
index f0e398b..cff62a4 100644
--- a/website/web/__init__.py
+++ b/website/web/__init__.py
@@ -46,6 +46,7 @@ lookyloo: Lookyloo = Lookyloo()
 user = get_config('generic', 'cache_clean_user')
 time_delta_on_index = get_config('generic', 'time_delta_on_index')
 blur_screenshot = get_config('generic', 'enable_default_blur_screenshot')
+max_depth = get_config('generic', 'max_depth')
 
 logging.basicConfig(level=get_config('generic', 'loglevel'))
 
@@ -478,7 +479,7 @@ def capture_web():
     if not user_agents:
         user_agents = get_user_agents()
     user_agents.pop('by_frequency')
-    return render_template('capture.html', user_agents=user_agents)
+    return render_template('capture.html', user_agents=user_agents, max_depth=max_depth)
 
 
 @app.route('/cookies/<string:cookie_name>', methods=['GET'])
diff --git a/website/web/templates/capture.html b/website/web/templates/capture.html
index de8d380..2029c1f 100644
--- a/website/web/templates/capture.html
+++ b/website/web/templates/capture.html
@@ -26,22 +26,18 @@
         </div>
       </div>
 
+      {% if max_depth >= 2 %}
       <div class="form-group row">
         <label for="depth" class="col-sm-2 col-form-label">Link Depth:</label>
         <div class="col-sm-1">
           <select class="form-control" name="depth" id=depth>
-            <option value="1">1</option>
-            <option value="2">2</option>
-            <option value="3">3</option>
-            <option value="4">4</option>
-            <option value="5">5</option>
-            <option value="6">6</option>
-            <option value="7">7</option>
-            <option value="8">8</option>
-            <option value="9">9</option>
+          {% for depth in range(max_depth) %}
+          <option value="{{ depth + 1 }}">{{ depth + 1 }}</option>
+          {% endfor %}
           </select>
         </div>
       </div>
+      {%endif%}
 
       <div class="form-group row">
         <label for="referer" class="col-sm-2 col-form-label">Referer (Optional):</label>