From 5687c87a65c7e614fa51e2356e6685d96593d1cf Mon Sep 17 00:00:00 2001
From: Ben Busby <ben.busby7@gmail.com>
Date: Tue, 7 Apr 2020 17:04:03 +0000
Subject: [PATCH] Adding optional nojs links to results page, changed nojs to a
 user setting

---
 app/routes.py | 22 ++++++++++++++--------
 run.sh        |  7 +------
 2 files changed, 15 insertions(+), 14 deletions(-)

diff --git a/app/routes.py b/app/routes.py
index ed80157..94ff473 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -23,10 +23,6 @@ DESKTOP_UA = mozilla + '/5.0 (Windows NT 6.1; Win64; x64; rv:59.0) Gecko/2010010
 # Base search url
 SEARCH_URL = 'https://www.google.com/search?gbv=1&q='
 
-# Optional nojs tag - opens links in a contained window with all js removed
-# (can be useful for achieving nojs on mobile)
-nojs = int(os.environ.get('NOJS'))
-
 user_config = json.load(open(STATIC_FOLDER + '/config.json'))
 
 
@@ -36,12 +32,15 @@ def get_ua(user_agent):
 
 def send_request(curl_url, ua):
     request_header = []
+    response_header = []
 
     b_obj = BytesIO()
     crl = pycurl.Curl()
     crl.setopt(crl.URL, curl_url)
     crl.setopt(crl.USERAGENT, ua)
     crl.setopt(crl.WRITEDATA, b_obj)
+    crl.setopt(crl.HEADERFUNCTION, response_header.append)
+    crl.setopt(pycurl.FOLLOWLOCATION, 1)
     crl.perform()
     crl.close()
 
@@ -127,13 +126,20 @@ def search():
         if '/advanced_search' in href:
             a.decompose()
             continue
+
         if 'url?q=' in href:
+            # Strip unneeded arguments
             href = urlparse.urlparse(href)
             href = parse_qs(href.query)['q'][0]
-        if nojs:
-            a['href'] = '/window?location=' + href
-        # else: # Automatically go to reader mode in ff? Not sure if possible
-        #    a['href'] = 'about:reader?url=' + href
+
+            # Add no-js option
+            if user_config['nojs']:
+                nojs_link = soup.new_tag('a')
+                nojs_link['href'] = '/window?location=' + href
+                nojs_link['style'] = 'display:block;width:100%;'
+                nojs_link.string = 'NoJS Link: ' + nojs_link['href']
+                a.append(BeautifulSoup('<br><hr><br>', 'html.parser'))
+                a.append(nojs_link)
 
     # Ensure no extra scripts passed through
     try:
diff --git a/run.sh b/run.sh
index 098a6dd..25460ba 100755
--- a/run.sh
+++ b/run.sh
@@ -2,11 +2,6 @@
 
 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 
-# Set NOJS mode to false if unavailable
-if [[ -z "$NOJS" ]]; then
-    export NOJS=0
-fi
-
 # Create config json if it doesn't exist
 if [[ ! -f $SCRIPT_DIR/app/static/config.json ]]; then
     echo "{}" > $SCRIPT_DIR/app/static/config.json
@@ -16,4 +11,4 @@ pkill flask
 
 # TODO: Set up the following for running over https
 #--cert=./app/cert.pem --key=./app/key.pem
-$SCRIPT_DIR/venv/bin/flask run
+$SCRIPT_DIR/venv/bin/flask run