From bed6b4e39e47bc6215207d8d4c2cb74af3750ee4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Wed, 27 May 2020 15:39:06 +0200
Subject: [PATCH] chg: Add initial URL in email notification

---
 config/email.tmpl    |  2 ++
 lookyloo/lookyloo.py | 10 ++++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/config/email.tmpl b/config/email.tmpl
index 8f1afdf..ecf83ce 100644
--- a/config/email.tmpl
+++ b/config/email.tmpl
@@ -3,6 +3,8 @@ Dear {recipient},
 Please have a look at this capture on lookyloo:
   * https://{domain}/tree/{uuid}
 
+Initial URL: {initial_url}
+
 {redirects}
 
 {comment}
diff --git a/lookyloo/lookyloo.py b/lookyloo/lookyloo.py
index 3933e81..d8c1200 100644
--- a/lookyloo/lookyloo.py
+++ b/lookyloo/lookyloo.py
@@ -257,12 +257,17 @@ class Lookyloo():
             return
 
         redirects = ''
+        initial_url = ''
         capture_dir = self.lookup_capture_dir(capture_uuid)
         if capture_dir:
             cache = self.capture_cache(capture_dir)
             if cache:
-                redirects = "Redirects:\n"
-                redirects += '\n'.join(cache['redirects'])
+                initial_url = cache['url']
+                if 'redirects' in cache and cache['redirects']:
+                    redirects = "Redirects:\n"
+                    redirects += '\n'.join(cache['redirects'])
+                else:
+                    redirects = "No redirects."
 
         email_config = self.get_config('email')
         msg = EmailMessage()
@@ -276,6 +281,7 @@ class Lookyloo():
             recipient=msg['To'].addresses[0].display_name,
             domain=email_config['domain'],
             uuid=capture_uuid,
+            initial_url=initial_url,
             redirects=redirects,
             comment=comment,
             sender=msg['From'].addresses[0].display_name,