From 52a3f77ced68a582f3555ebfaeb95fa453fea905 Mon Sep 17 00:00:00 2001
From: Marcus Bakker <marcus@mbsecure.nl>
Date: Fri, 30 Oct 2020 21:19:31 +0100
Subject: [PATCH] Added support for Navigator v4.0

---
 generic.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/generic.py b/generic.py
index c8a2b37..6afe8ba 100644
--- a/generic.py
+++ b/generic.py
@@ -219,15 +219,11 @@ def _get_base_template(name, description, stage, platform, sorting):
     """
     layer = dict()
     layer['name'] = name
-    layer['version'] = '3.0'
-    layer['domain'] = 'mitre-enterprise'
+    layer['versions'] = {'navigator': '4.0', 'layer': '4.0'}
+    layer['domain'] = 'enterprise-attack'
     layer['description'] = description
 
-    if stage == 'attack':
-        layer['filters'] = {'stages': ['act'], 'platforms': platform}
-    else:
-        layer['filters'] = {'stages': ['prepare'], 'platforms': platform}
-
+    layer['filters'] = {'platforms': platform}
     layer['sorting'] = sorting
     layer['layout'] = {"layout": "flat", "showName": True, "showID": False}
     layer['hideDisable'] = False