feat(server): refactor main entry point to use asyncio for better concurrency

feat(shell_executor): add get_allowed_commands method to retrieve allowed commands
fix(server): include allowed commands in tool description for better clarity
chore(pyproject): update entry point in scripts to reflect new main function location
chore(pyproject): add asyncio as a dependency for proper functionality

Author: tumf

mcp_shell_server/__init__.py

@@ -1,6 +1,15 @@
 """MCP Shell Server Package."""
 
-from .server import main
+import asyncio
 
+from . import server
+
+
+def main():
+    """Main entry point for the package."""
+    asyncio.run(server.main())
+
+
+# Optionally expose other important items at package level
+__all__ = ["main", "server"]
 __version__ = "0.1.0"
-__all__ = ["main"]

mcp_shell_server/server.py

@@ -24,11 +24,15 @@ class ExecuteToolHandler:
     def __init__(self):
         self.executor = ShellExecutor()
 
+    def get_allowed_commands(self) -> list[str]:
+        """Get the allowed commands"""
+        return self.executor.get_allowed_commands()
+
     def get_tool_description(self) -> Tool:
         """Get the tool description for the execute command"""
         return Tool(
             name=self.name,
-            description=self.description,
+            description=f"{self.description}\nAllowed commands: {', '.join(self.get_allowed_commands())}",
             inputSchema={
                 "type": "object",
                 "properties": {
@@ -101,6 +105,7 @@ async def call_tool(name: str, arguments: Any) -> Sequence[TextContent]:
 
 async def main() -> None:
     """Main entry point for the MCP shell server"""
+    logger.info("Starting MCP shell server")
     try:
         from mcp.server.stdio import stdio_server
 

mcp_shell_server/shell_executor.py

@@ -74,6 +74,10 @@ def _validate_command(self, command: List[str]) -> None:
                         f"Command not allowed after {cleaned_arg}: {next_cmd}"
                     )
 
+    def get_allowed_commands(self) -> list[str]:
+        """Get the allowed commands"""
+        return list(self._get_allowed_commands())
+
     async def execute(
         self, command: List[str], stdin: Optional[str] = None
     ) -> Dict[str, Any]:

pyproject.toml

@@ -7,14 +7,15 @@ authors = [
     { name = "tumf" }
 ]
 dependencies = [
+    "asyncio>=3.4.3",
     "mcp>=1.1.0",
 ]
 requires-python = ">=3.11"
 readme = "README.md"
 license = { text = "MIT" }
 
 [project.scripts]
-mcp-shell-server = "mcp_shell_server.server:main"
+mcp-shell-server = "mcp_shell_server:main"
 
 [project.optional-dependencies]
 test = [