FalkorDB · swilly22 · Dec 4, 2024 · Dec 3, 2024 · Dec 3, 2024 · Dec 3, 2024
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,44 @@
+# Byte-compiled files
+*.pyc
+*.pyo
+*.pyd
+__pycache__/
+
+# Virtual environments
+venv/
+env/
+.virtualenv/
+
+# Distribution/build files
+build/
+dist/
+*.egg-info/
+.eggs/
+
+# IDE settings
+.vscode/
+.idea/
+*.swp
+
+# Logs and debugging
+*.log
+*.trace
+
+# OS-specific files
+.DS_Store
+Thumbs.db
+
+# Testing and coverage
+htmlcov/
+*.cover
+.coverage
+.cache/
+pytest_cache/
+
+# Jupyter Notebook checkpoints
+.ipynb_checkpoints/
+
+# Custom settings
 .env
+*.sqlite3
 .vercel
diff --git a/api/__pycache__/__init__.cpython-312.pyc b/api/__pycache__/__init__.cpython-312.pyc
diff --git a/api/__pycache__/app.cpython-312.pyc b/api/__pycache__/app.cpython-312.pyc
diff --git a/api/__pycache__/auto_complete.cpython-312.pyc b/api/__pycache__/auto_complete.cpython-312.pyc
diff --git a/api/__pycache__/graph.cpython-312.pyc b/api/__pycache__/graph.cpython-312.pyc
diff --git a/api/__pycache__/info.cpython-312.pyc b/api/__pycache__/info.cpython-312.pyc
diff --git a/api/__pycache__/llm.cpython-312.pyc b/api/__pycache__/llm.cpython-312.pyc
diff --git a/api/__pycache__/project.cpython-312.pyc b/api/__pycache__/project.cpython-312.pyc
diff --git a/api/analyzers/__pycache__/__init__.cpython-312.pyc b/api/analyzers/__pycache__/__init__.cpython-312.pyc
diff --git a/api/analyzers/__pycache__/analyzer.cpython-312.pyc b/api/analyzers/__pycache__/analyzer.cpython-312.pyc
diff --git a/api/analyzers/__pycache__/source_analyzer.cpython-312.pyc b/api/analyzers/__pycache__/source_analyzer.cpython-312.pyc
diff --git a/api/analyzers/__pycache__/utils.cpython-312.pyc b/api/analyzers/__pycache__/utils.cpython-312.pyc
diff --git a/api/analyzers/c/__pycache__/analyzer.cpython-312.pyc b/api/analyzers/c/__pycache__/analyzer.cpython-312.pyc
diff --git a/api/analyzers/python/__pycache__/analyzer.cpython-312.pyc b/api/analyzers/python/__pycache__/analyzer.cpython-312.pyc
diff --git a/api/analyzers/source_analyzer.py b/api/analyzers/source_analyzer.py
@@ -135,7 +135,7 @@ def analyze_sources(self, ignore: List[str]) -> None:
             # Second pass analysis of the source code
             self.second_pass(ignore, executor)
 
-    def analyze(self, path: str, g: Graph, ignore: Optional[List[str]] = []) -> None:
+    def analyze_local_folder(self, path: str, g: Graph, ignore: Optional[List[str]] = []) -> None:
         """
         Analyze path.
 
@@ -144,6 +144,8 @@ def analyze(self, path: str, g: Graph, ignore: Optional[List[str]] = []) -> None
             ignore (List(str)): List of paths to skip
         """
 
+        logging.info(f"Analyzing local folder {path}")
+
         # Save original working directory for later restore
         original_dir = Path.cwd()
 
@@ -179,4 +181,4 @@ def analyze_local_repository(self, path: str, ignore: Optional[List[str]] = [])
         self.graph.set_graph_commit(head.hexsha)
 
         return self.graph
-
+    
diff --git a/api/code_coverage/__pycache__/__init__.cpython-312.pyc b/api/code_coverage/__pycache__/__init__.cpython-312.pyc
diff --git a/api/code_coverage/lcov/__pycache__/__init__.cpython-312.pyc b/api/code_coverage/lcov/__pycache__/__init__.cpython-312.pyc
diff --git a/api/code_coverage/lcov/__pycache__/lcov.cpython-312.pyc b/api/code_coverage/lcov/__pycache__/lcov.cpython-312.pyc
diff --git a/api/entities/__pycache__/__init__.cpython-312.pyc b/api/entities/__pycache__/__init__.cpython-312.pyc
diff --git a/api/entities/__pycache__/argument.cpython-312.pyc b/api/entities/__pycache__/argument.cpython-312.pyc
diff --git a/api/entities/__pycache__/cls.cpython-312.pyc b/api/entities/__pycache__/cls.cpython-312.pyc
diff --git a/api/entities/__pycache__/entity_encoder.cpython-312.pyc b/api/entities/__pycache__/entity_encoder.cpython-312.pyc
diff --git a/api/entities/__pycache__/file.cpython-312.pyc b/api/entities/__pycache__/file.cpython-312.pyc
diff --git a/api/entities/__pycache__/function.cpython-312.pyc b/api/entities/__pycache__/function.cpython-312.pyc
diff --git a/api/entities/__pycache__/struct.cpython-312.pyc b/api/entities/__pycache__/struct.cpython-312.pyc
diff --git a/api/index.py b/api/index.py
@@ -1,6 +1,7 @@
 import os
 import datetime
 from api import *
+from pathlib import Path
 from typing import Optional
 from functools import wraps
 from falkordb import FalkorDB
@@ -309,3 +310,43 @@ def chat():
     response = { 'status': 'success', 'response': answer }
 
     return jsonify(response), 200
+
+@app.route('/analyze_folder', methods=['POST'])
+@token_required  # Apply token authentication decorator
+def analyze_folder():
+    """
+    Endpoint to analyze local source code
+    Expects 'path' and optionaly an ignore list.
+
+    Returns:
+        Status code
+    """
+
+    # Get JSON data from the request
+    data = request.get_json()
+
+    # Get query parameters
+    path      = data.get('path')
+    ignore    = data.get('ignore', [])
+
+    # Validate 'path' parameter
+    if not path:
+        logging.error("'path' is missing from the request.")
+        return jsonify({"status": "'path' is required."}), 400
+
+    proj_name = Path(path).name
+
-    proj_name = Path(path).name
+    proj_name = Path(path).name
+    if not proj_name or proj_name.startswith('.'):
+        logging.error(f"Invalid project name derived from path: {proj_name}")
+        return jsonify({"status": "Invalid path: cannot derive valid project name"}), 400
-    proj_name = Path(path).name
+    proj_name = Path(path).name
+    if not proj_name or proj_name.startswith('.'):
+        logging.error(f"Invalid project name derived from path: {proj_name}")
+        return jsonify({"status": "Invalid path: cannot derive valid project name"}), 400
+    # Initialize the graph with the provided project name
+    g = Graph(proj_name)
+
+    # Analyze source code within given folder
+    analyzer = SourceAnalyzer()
+    analyzer.analyze_local_folder(path, g, ignore)
+
+    # Return response
+    response = {
+        'status': 'success'
+    }
+
+    return jsonify(response), 200
-@app.route('/analyze_folder', methods=['POST'])
-@token_required  # Apply token authentication decorator
-def analyze_folder():
-    """
-    Endpoint to analyze local source code
-    Expects 'path' and optionaly an ignore list.
-
-    Returns:
-        Status code
-    """
-
-    # Get JSON data from the request
-    data = request.get_json()
-
-    # Get query parameters
-    path      = data.get('path')
-    ignore    = data.get('ignore', [])
-
-    # Validate 'path' parameter
-    if not path:
-        logging.error("'path' is missing from the request.")
-        return jsonify({"status": "'path' is required."}), 400
-
-    proj_name = Path(path).name
-
-    # Initialize the graph with the provided project name
-    g = Graph(proj_name)
-
-    # Analyze source code within given folder
-    analyzer = SourceAnalyzer()
-    analyzer.analyze_local_folder(path, g, ignore)
-
-    # Return response
-    response = {
-        'status': 'success'
-    }
-
-    return jsonify(response), 200
+@app.route('/analyze_folder', methods=['POST'])
+@token_required
+def analyze_folder():
+    """
+    Endpoint to analyze local source code
+    Expects 'path' and optionally an ignore list.
+
+    Returns:
+        JSON response with status and error message if applicable
+        Status codes:
+            200: Success
+            400: Invalid input
+            500: Internal server error
+    """
+    # Get JSON data from the request
+    data = request.get_json()
+
+    # Get query parameters
+    path      = data.get('path')
+    ignore    = data.get('ignore', [])
+
+    # Validate input parameters
+    if not path:
+        logging.error("'path' is missing from the request.")
+        return jsonify({"status": "'path' is required."}), 400
+
+    # Validate path exists and is a directory
+    if not os.path.isdir(path):
+        logging.error(f"Path '{path}' does not exist or is not a directory")
+        return jsonify({"status": "Invalid path: must be an existing directory"}), 400
+
+    # Validate ignore list contains valid paths
+    if not isinstance(ignore, list):
+        logging.error("'ignore' must be a list of paths")
+        return jsonify({"status": "'ignore' must be a list of paths"}), 400
+
+    proj_name = Path(path).name
+
+    try:
+        # Initialize the graph with the provided project name
+        g = Graph(proj_name)
+
+        # Analyze source code within given folder
+        analyzer = SourceAnalyzer()
+        analyzer.analyze_local_folder(path, g, ignore)
+
+        # Return response
+        response = {
+            'status': 'success',
+            'project': proj_name
+        }
+        return jsonify(response), 200
+
+    except Exception as e:
+        logging.error(f"Error analyzing folder {path}: {e}")
+        return jsonify({
+            'status': 'error',
+            'message': str(e)
+        }), 500
-@app.route('/analyze_folder', methods=['POST'])
-@token_required  # Apply token authentication decorator
-def analyze_folder():
-    """
-    Endpoint to analyze local source code
-    Expects 'path' and optionaly an ignore list.
-
-    Returns:
-        Status code
-    """
-
-    # Get JSON data from the request
-    data = request.get_json()
-
-    # Get query parameters
-    path      = data.get('path')
-    ignore    = data.get('ignore', [])
-
-    # Validate 'path' parameter
-    if not path:
-        logging.error("'path' is missing from the request.")
-        return jsonify({"status": "'path' is required."}), 400
-
-    proj_name = Path(path).name
-
-    # Initialize the graph with the provided project name
-    g = Graph(proj_name)
-
-    # Analyze source code within given folder
-    analyzer = SourceAnalyzer()
-    analyzer.analyze_local_folder(path, g, ignore)
-
-    # Return response
-    response = {
-        'status': 'success'
-    }
-
-    return jsonify(response), 200
+@app.route('/analyze_folder', methods=['POST'])
+@token_required
+def analyze_folder():
+    """
+    Endpoint to analyze local source code
+    Expects 'path' and optionally an ignore list.
+
+    Returns:
+        JSON response with status and error message if applicable
+        Status codes:
+            200: Success
+            400: Invalid input
+            500: Internal server error
+    """
+    # Get JSON data from the request
+    data = request.get_json()
+
+    # Get query parameters
+    path      = data.get('path')
+    ignore    = data.get('ignore', [])
+
+    # Validate input parameters
+    if not path:
+        logging.error("'path' is missing from the request.")
+        return jsonify({"status": "'path' is required."}), 400
+
+    # Validate path exists and is a directory
+    if not os.path.isdir(path):
+        logging.error(f"Path '{path}' does not exist or is not a directory")
+        return jsonify({"status": "Invalid path: must be an existing directory"}), 400
+
+    # Validate ignore list contains valid paths
+    if not isinstance(ignore, list):
+        logging.error("'ignore' must be a list of paths")
+        return jsonify({"status": "'ignore' must be a list of paths"}), 400
+
+    proj_name = Path(path).name
+
+    try:
+        # Initialize the graph with the provided project name
+        g = Graph(proj_name)
+
+        # Analyze source code within given folder
+        analyzer = SourceAnalyzer()
+        analyzer.analyze_local_folder(path, g, ignore)
+
+        # Return response
+        response = {
+            'status': 'success',
+            'project': proj_name
+        }
+        return jsonify(response), 200
+
+    except Exception as e:
+        logging.error(f"Error analyzing folder {path}: {e}")
+        return jsonify({
+            'status': 'error',
+            'message': str(e)
+        }), 500
+