add repo url to repo_info

Author: swilly22

code_graph/__init__.py

@@ -1,3 +1,4 @@
+from .info import *
 from .graph import *
 from .entities import *
 from .git_utils import *

code_graph/info.py

@@ -0,0 +1,76 @@
+import os
+import redis
+import logging
+from typing import Optional, Dict
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+
+def get_redis_connection() -> redis.Redis:
+    """
+    Establishes a connection to Redis using environment variables.
+
+    Returns:
+        redis.Redis: A Redis connection object.
+    """
+    try:
+        return redis.Redis(
+            host             = os.getenv('FALKORDB_HOST'),
+            port             = os.getenv('FALKORDB_PORT'),
+            username         = os.getenv('FALKORDB_USERNAME'),
+            password         = os.getenv('FALKORDB_PASSWORD'),
+            decode_responses = True  # To ensure string responses
+        )
+    except Exception as e:
+        logging.error(f"Error connecting to Redis: {e}")
+        raise
+
+
+def save_repo_info(repo_name: str, repo_url: str) -> None:
+    """
+    Saves repository information (URL) to Redis under a hash named {repo_name}_info.
+
+    Args:
+        repo_name (str): The name of the repository.
+        repo_url (str): The URL of the repository.
+    """
+
+    try:
+        r = get_redis_connection()
+        key = f"{{ {repo_name} }}_info"  # Safely format the key
+
+        # Save the repository URL
+        r.hset(key, 'repo_url', repo_url)
+        logging.info(f"Repository info saved for {repo_name}")
+
+    except Exception as e:
+        logging.error(f"Error saving repo info for '{repo_name}': {e}")
+        raise
+
+def get_repo_info(repo_name: str) -> Optional[Dict[str, str]]:
+    """
+    Retrieves repository information from Redis.
+
+    Args:
+        repo_name (str): The name of the repository.
+
+    Returns:
+        Optional[Dict[str, str]]: A dictionary of repository information, or None if not found.
+    """
+    try:
+        r = get_redis_connection()
+        key = f"{{{repo_name}}}_info"
+        
+        # Retrieve all information about the repository
+        repo_info = r.hgetall(key)
+        if not repo_info:
+            logging.warning(f"No repository info found for {repo_name}")
+            return None
+        
+        logging.info(f"Repository info retrieved for {repo_name}")
+        return repo_info
+
+    except Exception as e:
+        logging.error(f"Error retrieving repo info for '{repo_name}': {e}")
+        raise
+

main.py

@@ -1,5 +1,4 @@
 import os
-import redis
 import datetime
 from code_graph import *
 from typing import Optional
@@ -70,6 +69,7 @@ def get_neighbors():
     Endpoint to get neighbors of a specific node in the graph.
     Expects 'repo' and 'node_id' as query parameters.
     """
+
     repo = request.args.get('repo')
     node_id = request.args.get('node_id')
 
@@ -155,6 +155,8 @@ def process_repo():
         logger.error(f'An error occurred: {e}')
         return jsonify({'status': f'Failed to process repository: {git_url}'}), 400
 
+    save_repo_info(repo_name, repo_url)
+
     # Create a response
     response = {
         'status': 'success',
@@ -358,7 +360,7 @@ def repo_info():
     g = Graph(repo)
 
     # Retrieve statistics from the graph
-    stats = g.stats()
+    stats = g.stats() | get_repo_info(repo)
 
     # Create a response
     response = {