feat: Add graph extraction service for SigmaJS

Author: GermanMT

app/controllers/__init__.py

@@ -1,4 +1,4 @@
-from .vul_dbs.update_nvd_controller import nvd_updater
+from .vul_dbs.nvd_controller import nvd_updater
 
 __all__ = [
     'nvd_updater'

app/controllers/file_operation_controller.py

@@ -7,8 +7,7 @@
     MaximizeImpact,
     FilterConfigs
 )
-
-from app.controllers.graph_to_smt import GraphToSMT
+from flamapy.metamodels.smt_metamodel.transformations import GraphToSMT
 from app.services import (
     read_data_for_smt_transform,
     read_releases_by_counts,
@@ -93,14 +92,14 @@ async def minimize_impact(requirement_file_id: str, agregator: str, file_name: s
     - **agregator**: agregator function to build the smt model ('mean' or 'weighted_mean')
     - **file_name**: name of requirement file belonging to graph
     - **limit**: the number of configurations to return
+    - **max_level**: the depth of the graph to be analysed
     '''
     package_manager = await get_manager(file_name)
-    graph_data = await read_data_for_smt_transform(requirement_file_id, package_manager, max_level)
+    graph_data = await read_data_for_smt_transform(requirement_file_id, package_manager, max_level*2)
     smt_transform = GraphToSMT(graph_data, file_name, package_manager, agregator)
     smt_transform.transform()
-    smt_model = smt_transform.destination_model
     operation = MinimizeImpact(limit)
-    operation.execute(smt_model)
+    operation.execute(smt_transform.destination_model)
     result = await read_releases_by_counts(operation.get_result(), package_manager)
     return JSONResponse(status_code=status.HTTP_200_OK, content=json_encoder({'result': result}))
 
@@ -129,7 +128,6 @@ async def maximize_impact(requirement_file_id: str, agregator: str, file_name: s
     result = await read_releases_by_counts(operation.get_result(), package_manager)
     return JSONResponse(status_code=status.HTTP_200_OK, content=json_encoder({'result': result}))
 
-
 @router.post(
     '/operation/file/filter_configs/{requirement_file_id}',
     summary='Filter configurations of a graph',

app/controllers/graph_controller.py

@@ -6,7 +6,7 @@
 from fastapi.responses import JSONResponse
 from app.services import (
     create_repository,
-    read_graph_by_repository_id,
+    read_graphs_by_owner_name_for_sigma,
     read_repositories,
     read_repositories_moment,
     read_requirement_files_by_repository,
@@ -27,19 +27,22 @@
 router = APIRouter()
 
 @router.get(
-    '/graph/{repository_id}',
-    summary='Get a graph by a repository id',
+    '/graph/{owner}/{name}',
+    summary='Get each package manager graph by the owner and name of a repository',
     response_description='Return a graph'
 )
-async def get_graph(repository_id: str) -> JSONResponse:
+async def get_graphs(owner: str, name: str) -> JSONResponse:
     '''
-    Return a graph by a given repository id. If attribute is_complete is True
-    the graph is wholly built:
+    Get each package manager graph by the owner and name of a repository:
 
-    - **repository_id**: the id of a repository
+    - **owner**: the owner of a repository
+    - **name**: the name of a repository
     '''
-    graph = await read_graph_by_repository_id(repository_id, '_')
-    return JSONResponse(status_code=status.HTTP_200_OK, content=graph)
+    # TODO: Hacer un servicio que devuelva todos los grafos por owner y name para mostrarlos en el FrontEnd
+    # TODO: Añadir a Neo4J owner y name como un índice
+    # TODO: Cambiar en los servicios la palabra relationships por edges
+    graphs = await read_graphs_by_owner_name_for_sigma(owner, name)
+    return JSONResponse(status_code=status.HTTP_200_OK, content=json_encoder(graphs))
 
 
 @router.post(
@@ -69,6 +72,9 @@ async def init_graph(repository: RepositoryModel) -> JSONResponse:
                     await update_repository_is_complete(repository_id, False, package_manager)
                     await replace_repository(raw_requirement_files, repository_id, package_manager)
                 await update_repository_is_complete(repository_id, True, package_manager)
+        else:
+            # TODO: Devolver el grafo actual de la misma forma que en get_graphs
+            pass
     return JSONResponse(status_code=status.HTTP_200_OK, content=json_encoder({'message': 'initializing'}))
 
 
@@ -114,7 +120,7 @@ async def select_manager(package_manager: str, name: str, file: dict[str, Any],
     match package_manager:
         case 'PIP':
             await pip_extract_graph(name, file, repository_id)
-        # case 'NPM':
-        #     await npm_extract_graph(name, file, repository_json, repository_id)
-        # case 'MVN':
-        #     await mvn_extract_graph(name, file, repository_json, repository_id)
+        case 'NPM':
+            await npm_extract_graph(name, file, repository_id)
+        case 'MVN':
+            await mvn_extract_graph(name, file, repository_id)

app/controllers/graph_to_smt.py

@@ -4,8 +4,8 @@
     get_all_versions,
     requires_packages
 )
-from app.services import read_cpe_matches_by_package_name
 from app.services import (
+    read_cpe_matches_by_package_name,
     read_package_by_name,
     create_requirement_file,
     relate_package,

app/controllers/managers/mvn_generate_controller.py

@@ -4,8 +4,8 @@
     get_all_versions,
     requires_packages
 )
-from app.services import read_cpe_matches_by_package_name
 from app.services import (
+    read_cpe_matches_by_package_name,
     read_package_by_name,
     create_requirement_file,
     relate_package,