Improved structure of generated feature models

Author: SvenPeldszus

Resources/taxonomy.feature_model

@@ -1,5 +1,5 @@
 Security_Features
-    Access_Control
+	Access_Control
 		Authentication
 			Credentials
 			One_Time_Password
@@ -38,7 +38,7 @@ Security_Features
 			Message_Authentication
 			Digital_Watermarking
 		Steganography
-    Security_Monitoring
+	Security_Monitoring
 		Logging
 		Automated_Response
 		History_Maintenance

SecurityKeywordsBasedSearchTool/SecFeatFinder/Feature.py

@@ -0,0 +1,19 @@
+class Feature:
+    def __init__(self, name, parent):
+        self.name = name
+        self.parent = parent
+        self.sub_features = []
+        if parent is not None:
+            parent.sub_features.append(self)
+
+    # Searches in the feature tree for a feature with the given name
+    def dfs(self, name):
+        stack = [self]
+        while len(stack) > 0:
+            feature = stack.pop()
+            if feature.name == name:
+                return feature
+            stack.extend(feature.sub_features)
+
+    def __str__(self):
+        return "Feature: " + self.name

SecurityKeywordsBasedSearchTool/SecFeatFinder/FeatureModel.py

@@ -0,0 +1,76 @@
+import os
+
+from SecurityKeywordsBasedSearchTool.SecFeatFinder.Feature import Feature
+
+
+def add_to_fm(fm, taxonomy, feature_name, tag):
+    taxo_feature = taxonomy.dfs(feature_name)
+
+    feature = taxo_feature
+    parents = [feature]
+    while feature.parent is not None:
+        feature = feature.parent
+        parents.append(feature)
+
+    if parents.pop().name == fm.name:
+        for feature in reversed(parents):
+            exists = False
+            for proposal in fm.sub_features:
+                if proposal.name == feature.name:
+                    fm = proposal
+                    exists = True
+                    break
+            if not exists:
+                fm = Feature(feature.name, fm)
+        return Feature(tag, fm)
+
+
+def create_feature_model_file(repo_dir, custom_features, library_features, fm):
+    # Save the feature model structure
+    feature_model_path = os.path.join(repo_dir, ".feature-model")
+    with open(feature_model_path, "w") as file:
+        indentation = 0
+        save_feature_model(fm, file, indentation)
+    print(f".feature-model file created at: {feature_model_path}")
+
+
+def read_feature_model(file):
+    with open(file, "r") as f:
+        depths = {-1: 0}
+        stack = []
+        for line in f.readlines():
+            indentation = 0
+            for c in line:
+                if c == '\t':
+                    indentation += 4
+                elif c == ' ':
+                    indentation += 1
+                else:
+                    break
+            name = line.strip()
+
+            stack_depth = len(stack) - 1
+            if indentation in depths.keys():
+                new_feature_depth = depths[indentation]
+            else:
+                new_feature_depth = len(stack)
+                depths[indentation] = new_feature_depth
+
+            while new_feature_depth <= stack_depth:
+                stack.pop()
+                stack_depth -= 1
+
+            if len(stack) > 0:
+                parent = stack[len(stack) - 1]
+            else:
+                parent = None
+
+            feature = Feature(name, parent)
+            stack.append(feature)
+        return stack[0]
+
+
+def save_feature_model(fm, file, indentation):
+    file.write("\t" * indentation + fm.name + "\n")
+    for child in fm.sub_features:
+        save_feature_model(child, file, indentation + 1)

SecurityKeywordsBasedSearchTool/SecFeatFinder/GitClient.py

@@ -0,0 +1,22 @@
+import os
+import subprocess
+
+
+def clone_repository(repo_link):
+    """Clone the repository into a 'repos' directory with the project name."""
+    # Extract project name from the repo URL
+    project_name = os.path.basename(repo_link).replace(".git", "")
+    repos_dir = os.path.join(os.getcwd(), "repos")
+    project_dir = os.path.join(repos_dir, project_name)
+    # Create the 'repos' directory if it doesn't exist
+    os.makedirs(repos_dir, exist_ok=True)
+    if os.path.exists(project_dir):
+        print(f"Repository '{project_name}' already exists in 'repos'.")
+    else:
+        try:
+            subprocess.run(["git", "clone", repo_link, project_dir],
+                           check=True)
+            print(f"Cloned '{project_name}' into 'repos'.")
+        except subprocess.CalledProcessError as e:
+            raise RuntimeError(f"Failed to clone repository: {e}")
+    return project_dir, project_name