Hotfix/discover

assemblyline/common/classification.py

@@ -100,9 +100,7 @@ def __init__(self, classification_definition: Dict):
                     self.levels_aliases[a.upper()] = short_name
                 self.params_map[short_name] = {k: v for k, v in x.items() if k not in banned_params_keys}
                 self.params_map[name] = self.params_map[short_name]
-                self.levels_styles_map[short_name] = x.get('css', {'banner': 'alert-default',
-                                                                   'label': 'label-default',
-                                                                   'text': 'text-muted'})
+                self.levels_styles_map[short_name] = x.get('css', {'color': 'default'})
                 self.levels_styles_map[name] = self.levels_styles_map[short_name]
                 self.description[short_name] = x.get('description', "N/A")
                 self.description[name] = self.description[short_name]

assemblyline/common/classification.yml

@@ -18,12 +18,8 @@ levels:
       - U
     # Stylesheet applied in the UI for the different levels
     css:
-      # Top banner stylesheet (alert-* because it's based of bootstrap alert dialogs)
-      banner: alert-default
-      # Label stylesheet (label-* because it's based of bootstrap label components)
-      label: label-default
-      # Text stylesheet (text-* because it's based of bootstrap text color styles)
-      text: text-muted
+      # Name of the color scheme used for display (default, primary, secondary, success, info, warning, error)
+      color: default
     # Description of the classification level
     description: Subject to standard copyright rules, TLP:WHITE information may be distributed without restriction.
     # Interger value of the Classification level (higher is more classified)
@@ -34,9 +30,7 @@ levels:
     short_name: TLP:W
   - aliases: []
     css:
-      banner: alert-success
-      label: label-success
-      text: text-success
+      color: success
     description:
       Recipients may share TLP:GREEN information with peers and partner organizations
       within their sector or community, but not via publicly accessible channels. Information
@@ -48,9 +42,7 @@ levels:
   - aliases:
       - RESTRICTED
     css:
-      banner: alert-warning2
-      label: label-warning
-      text: text-warning2
+      color: warning
     description:
       Recipients may only share TLP:AMBER information with members of their
       own organization and with clients or customers who need to know the information

assemblyline/odm/models/config.py

@@ -788,6 +788,8 @@ class UI(odm.Model):
     banner_level: str = odm.Enum(values=["info", "warning", "success", "error"])
     # Turn on debugging
     debug: bool = odm.Boolean()
+    # Discover URL
+    discover_url: str = odm.Optional(odm.Keyword())
     # Which encoding will be used
     download_encoding = odm.Enum(values=["raw", "cart"])
     # Assemblyline admins email address
@@ -832,6 +834,7 @@ class UI(odm.Model):
     "banner": None,
     "banner_level": 'info',
     "debug": False,
+    "discover_url": None,
     "download_encoding": "cart",
     "email": None,
     "enforce_quota": True,