feat(mcp): add basic logger (#8740)

mcp_server/.gitignore

@@ -14,7 +14,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
 lib64/
 parts/
 sdist/

mcp_server/prowler_mcp_server/lib/__init__.py

@@ -0,0 +1,3 @@
+from prowler_mcp_server.lib.logger import logger
+
+__all__ = ["logger"]

mcp_server/prowler_mcp_server/lib/logger.py

@@ -0,0 +1,4 @@
+from fastmcp.utilities.logging import get_logger
+
+# Create and export logger
+logger = get_logger("prowler-mcp-server")

mcp_server/prowler_mcp_server/main.py

@@ -1,6 +1,8 @@
 import asyncio
 import sys
-from prowler_mcp_server.server import setup_main_server, prowler_mcp_server
+
+from prowler_mcp_server.lib.logger import logger
+from prowler_mcp_server.server import prowler_mcp_server, setup_main_server
 
 
 def main():
@@ -9,10 +11,10 @@ def main():
         asyncio.run(setup_main_server())
         prowler_mcp_server.run()
     except KeyboardInterrupt:
-        print("\nShutting down Prowler MCP server...")
+        logger.info("Shutting down Prowler MCP server...")
         sys.exit(0)
     except Exception as e:
-        print(f"Error: {e}", file=sys.stderr)
+        logger.error(f"Error starting server: {e}")
         sys.exit(1)
 
 

mcp_server/prowler_mcp_server/server.py

@@ -1,5 +1,5 @@
 from fastmcp import FastMCP
-
+from prowler_mcp_server.lib.logger import logger
 
 # Initialize main Prowler MCP server
 prowler_mcp_server = FastMCP("prowler-mcp-server")
@@ -10,9 +10,10 @@ async def setup_main_server():
 
     # Import Prowler Hub tools with prowler_hub_ prefix
     try:
+        logger.info("Importing Prowler Hub server...")
         from prowler_mcp_server.prowler_hub.server import hub_mcp_server
 
         await prowler_mcp_server.import_server(hub_mcp_server, prefix="prowler_hub")
-    except Exception:
-        # TODO: Add error logging
-        pass
+        logger.info("Successfully imported Prowler Hub server")
+    except Exception as e:
+        logger.error(f"Failed to import Prowler Hub server: {e}")