#56 update Fortiguard URLCategory summary() and short reports

TheHive-Project · Jun 16, 2017 · 60b64e1 · 60b64e1
1 parent a05169e
commit 60b64e1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/analyzers/Fortiguard/urlcategory.py b/analyzers/Fortiguard/urlcategory.py
@@ -10,7 +10,16 @@
 class URLCategoryAnalyzer(Analyzer):
 
     def summary(self, raw):
-        return {'category': raw['category']}
+        taxonomy = {"level": "info", "namespace": "Fortiguard", "predicate": "URLCat", "value": 0}
+        taxonomies = []
+
+        if 'category' in raw:
+            r = raw.get('category')
+            taxonomy["value"] = r
+            taxonomies.append(taxonomy)
+
+        result = {"taxonomies": taxonomies}
+        return result
 
     def run(self):
         Analyzer.run(self)

diff --git a/thehive-templates/Fortiguard_URLCategory_1_1/short.html b/thehive-templates/Fortiguard_URLCategory_1_1/short.html
@@ -1,4 +1,3 @@
-<span class="label label-info">
-	Fortiguard:URLCat=
-	<span ng-if="content.category">{{content.category}}&nbsp;</span>
-</span>
+<span class="label" ng-repeat="t in content.taxonomies" ng-class="{'info': 'label-info', 'safe': 'label-success', 'suspicious': 'label-warning', 'malicious':'label-danger'}[t.level]">
+    {{t.namespace}}:{{t.predicate}}={{t.value}}
+</span>&nbsp;