From 8cfad64271cf72ed4572c4d3a2c5ff6885bc8b95 Mon Sep 17 00:00:00 2001
From: Cristian <cristian@swapps.com>
Date: Thu, 12 Nov 2020 11:09:34 -0500
Subject: [PATCH] feat: Add specific logic for archive_org icon

---
 archivebox/core/utils.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/archivebox/core/utils.py b/archivebox/core/utils.py
index 67b8004d..6266024b 100644
--- a/archivebox/core/utils.py
+++ b/archivebox/core/utils.py
@@ -23,7 +23,7 @@ def get_icons(snapshot: Snapshot) -> str:
         "mercury": "🅼",
         "warc": "📦"
     }
-    exclude = ["favicon"]
+    exclude = ["favicon", "archive_org"]
     # Missing specific entry for WARC
 
     for extractor, _ in EXTRACTORS:
@@ -40,6 +40,14 @@ def get_icons(snapshot: Snapshot) -> str:
                     output += output_template.format(exists[0], "",
                                                      True, "warc", icons.get("warc", "?"))
 
+            if extractor == "archive_org" and exists:
+                # The check for archive_org is different, so it has to be handled separately
+                target_path = Path(path) / "archive.org.txt"
+                exists = target_path.exists()
+                if exists:
+                    output += '<a href="{}" class="exists-{}" title="{}">{} </a>'.format(canon["archive_org_path"],
+                                                                                         True, "archive_org", icons.get("archive_org", "?"))
+
         except Exception as e:
             print(e)