get_neighbors get list of ids

Author: Anchel123

api/graph.py

@@ -1,7 +1,7 @@
 import os
 import time
 from .entities import *
-from typing import Dict, List, Optional, Tuple
+from typing import Dict, Optional, List, Tuple
 from falkordb import FalkorDB, Path, Node, QueryResult
 
 # Configure the logger
@@ -194,12 +194,12 @@ def get_sub_graph(self, l: int) -> dict:
         return sub_graph
 
 
-    def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[str] = None) -> Dict[str, List[dict]]:
+    def get_neighbors(self, node_ids: List[int], rel: Optional[str] = None, lbl: Optional[str] = None) -> Dict[str, List[dict]]:
         """
-        Fetch the neighbors of a given node in the graph based on relationship type and/or label.
+        Fetch the neighbors of a given nodes in the graph based on relationship type and/or label.
 
         Args:
-            node_id (int): The ID of the source node.
+            node_ids (List[int]): The IDs of the source nodes.
             rel (str, optional): The type of relationship to filter by. Defaults to None.
             lbl (str, optional): The label of the destination node to filter by. Defaults to None.
 
@@ -208,8 +208,8 @@ def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[s
         """
 
         # Validate inputs
-        if not isinstance(node_id, int):
-            raise ValueError("node_id must be an integer")
+        if not all(isinstance(node_id, int) for node_id in node_ids):
+            raise ValueError("node_ids must be an integer list")
 
         # Build relationship and label query parts
         rel_query = f":{rel}" if rel else ""
@@ -218,7 +218,7 @@ def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[s
         # Parameterized Cypher query to find neighbors
         query = f"""
             MATCH (n)-[e{rel_query}]->(dest{lbl_query})
-            WHERE ID(n) = $node_id
+            WHERE ID(n) IN $node_ids
             RETURN e, dest
         """
 
@@ -227,7 +227,7 @@ def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[s
 
         try:
             # Execute the graph query with node_id parameter
-            result_set = self._query(query, {'node_id': node_id}).result_set
+            result_set = self._query(query, {'node_ids': node_ids}).result_set
 
             # Iterate over the result set and process nodes and edges
             for edge, destination_node in result_set:
@@ -237,7 +237,7 @@ def get_neighbors(self, node_id: int, rel: Optional[str] = None, lbl: Optional[s
             return neighbors
 
         except Exception as e:
-            logging.error(f"Error fetching neighbors for node {node_id}: {e}")
+            logging.error(f"Error fetching neighbors for node {node_ids}: {e}")
             return {'nodes': [], 'edges': []}
 
 

api/index.py

@@ -79,28 +79,31 @@ def graph_entities():
         return jsonify({"status": "Internal server error"}), 500
 
 
-@app.route('/get_neighbors', methods=['GET'])
+@app.route('/get_neighbors', methods=['POST'])
 @token_required  # Apply token authentication decorator
 def get_neighbors():
     """
-    Endpoint to get neighbors of a specific node in the graph.
-    Expects 'repo' and 'node_id' as query parameters.
+    Endpoint to get neighbors of a nodes list in the graph.
+    Expects 'repo' and 'node_ids' as body parameters.
 
     Returns:
         JSON response containing neighbors or error messages.
     """
 
+    # Get JSON data from the request
+    data = request.get_json()
+
     # Get query parameters
-    repo    = request.args.get('repo')
-    node_id = request.args.get('node_id')
+    repo    = data.get('repo')
+    node_ids = data.get('node_ids')
 
     # Validate 'repo' parameter
     if not repo:
         logging.error("Repository name is missing in the request.")
         return jsonify({"status": "Repository name is required."}), 400
 
     # Validate 'node_id' parameter
-    if not node_id:
+    if not node_ids:
         logging.error("Node ID is missing in the request.")
         return jsonify({"status": "Node ID is required."}), 400
 
@@ -111,19 +114,19 @@ def get_neighbors():
 
     # Try converting node_id to an integer
     try:
-        node_id = int(node_id)
+        node_ids = [int(node_id) for node_id in node_ids]
     except ValueError:
-        logging.error(f"Invalid node ID: {node_id}. It must be an integer.")
+        logging.error(f"Invalid node ID: {node_ids}. It must be an integer.")
         return jsonify({"status": "Invalid node ID. It must be an integer."}), 400
 
     # Initialize the graph with the provided repository
     g = Graph(repo)
 
     # Fetch the neighbors of the specified node
-    neighbors = g.get_neighbors(node_id)
+    neighbors = g.get_neighbors(node_ids)
 
     # Log and return the neighbors
-    logging.info(f"Successfully retrieved neighbors for node ID {node_id} in repo '{repo}'.")
+    logging.info(f"Successfully retrieved neighbors for node ID {node_ids} in repo '{repo}'.")
 
     response = {
         'status': 'success',
@@ -170,6 +173,41 @@ def auto_complete():
 
     return jsonify(response), 200
 
+@app.route('/process_repo', methods=['POST'])
+@token_required  # Apply token authentication decorator
+def process_repo():
+     """
+     Process a GitHub repository.
+
+     Expected JSON payload:
+     {
+         "repo_url": "string",
+         "ignore": ["string"]  # optional
+     }
+
+     Returns:
+         JSON response with processing status
+     """
+
+     data = request.get_json()
+     url = data.get('repo_url')
+     if url is None:
+         return jsonify({'status': f'Missing mandatory parameter "url"'}), 400
+     logger.debug(f'Received repo_url: {url}')
+
+     ignore = data.get('ignore', [])
+
+     proj = Project.from_git_repository(url)
+     proj.analyze_sources(ignore)
+    #  proj.process_git_history(ignore)
+
+     # Create a response
+     response = {
+         'status': 'success',
+     }
+
+     return jsonify(response), 200
+
 
 @app.route('/list_repos', methods=['GET'])
 @token_required  # Apply token authentication decorator
@@ -183,7 +221,7 @@ def list_repos():
 
     # Fetch list of repositories
     repos = get_repos()
-
+    print(repos)
     # Create a success response with the list of repositories
     response = {
         'status': 'success',