From 6c98513153ce0cd365c1d35a54b8307f275cdb89 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Sat, 25 Aug 2018 21:48:20 -0500
Subject: [PATCH] Add referer to /data_control

---
 src/invidious/views/data_control.ecr         | 2 +-
 src/invidious/views/preferences.ecr          | 4 ++--
 src/invidious/views/subscription_manager.ecr | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/invidious/views/data_control.ecr b/src/invidious/views/data_control.ecr
index 87c7e64a..a710b8fd 100644
--- a/src/invidious/views/data_control.ecr
+++ b/src/invidious/views/data_control.ecr
@@ -3,7 +3,7 @@
 <% end %>
 
 <div class="h-box">
-    <form class="pure-form pure-form-aligned" enctype="multipart/form-data" action="/data_control" method="post">
+    <form class="pure-form pure-form-aligned" enctype="multipart/form-data" action="/data_control?referer=<%= referer %>" method="post">
         <fieldset>
             <legend>Import</legend>
 
diff --git a/src/invidious/views/preferences.ecr b/src/invidious/views/preferences.ecr
index ffd3e69f..5ee98141 100644
--- a/src/invidious/views/preferences.ecr
+++ b/src/invidious/views/preferences.ecr
@@ -137,11 +137,11 @@ function update_value(element) {
 
             <legend>Data preferences</legend>
             <div class="pure-control-group">
-                <a href="/clear_watch_history">Clear watch history</a>
+                <a href="/clear_watch_history?referer=<%= referer %>">Clear watch history</a>
             </div>
             
             <div class="pure-control-group">
-                <a href="/data_control">Import/Export data</a>
+                <a href="/data_control?referer=<%= referer %>">Import/Export data</a>
             </div>
 
             <div class="pure-control-group">
diff --git a/src/invidious/views/subscription_manager.ecr b/src/invidious/views/subscription_manager.ecr
index 22c30b9e..b942e686 100644
--- a/src/invidious/views/subscription_manager.ecr
+++ b/src/invidious/views/subscription_manager.ecr
@@ -8,7 +8,7 @@
     </div>
     <div class="pure-u-1-3" style="text-align:right;">
         <h3>
-            <a href="/data_control">Import/Export</a>
+            <a href="/data_control?referer=<%= referer %>">Import/Export</a>
         </h3>
     </div>
 </div>