From 1f09a6de97493eec8e99f1f83a45bf0b171b0a5b Mon Sep 17 00:00:00 2001
From: Steve Garon <steve.garon@cyber.gc.ca>
Date: Fri, 24 Sep 2021 16:10:35 +0000
Subject: [PATCH] Don't add you own site to the app list

---
 assemblyline_ui/helper/discover.py | 26 ++++++++++++++++----------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/assemblyline_ui/helper/discover.py b/assemblyline_ui/helper/discover.py
index a9a761bf..0737f343 100644
--- a/assemblyline_ui/helper/discover.py
+++ b/assemblyline_ui/helper/discover.py
@@ -14,16 +14,22 @@ def get_apps_list():
             if resp.ok:
                 data = resp.json()
                 for app in data['applications']['application']:
-                    apps['apps'].append(
-                        {
-                            "alt": app['instance'][0]['metadata']['alternateText'],
-                            "name": app['name'],
-                            "img_d": app['instance'][0]['metadata']['imageDark'],
-                            "img_l": app['instance'][0]['metadata']['imageLight'],
-                            "route": app['instance'][0]['hostName'],
-                            "classification": app['instance'][0]['metadata']['classification']
-                        }
-                    )
+                    try:
+                        url = app['instance'][0]['hostName']
+
+                        if config.ui.fqdn not in url:
+                            apps['apps'].append(
+                                {
+                                    "alt": app['instance'][0]['metadata']['alternateText'],
+                                    "name": app['name'],
+                                    "img_d": app['instance'][0]['metadata']['imageDark'],
+                                    "img_l": app['instance'][0]['metadata']['imageLight'],
+                                    "route": url,
+                                    "classification": app['instance'][0]['metadata']['classification']
+                                }
+                            )
+                    except Exception:
+                        logger.exception(f'Failed to parse get app: {str(app)}')
             else:
                 logger.warning(f'Invalid response from server for apps discovery: {config.ui.discover_url}')
         except Exception: