Merge pull request #44 from DerekMelchin/bug-docker-mount

Mount local organization workspace

Author: DerekMelchin

src/main.py

@@ -15,6 +15,7 @@
 from tools.lean_versions import register_lean_version_tools
 from tools.ai import register_ai_tools
 from tools.mcp_server_version import register_mcp_server_version_tools
+from organization_workspace import OrganizationWorkspace
 
 transport = os.getenv('MCP_TRANSPORT', 'stdio')
 
@@ -45,5 +46,7 @@
     f(mcp)
 
 if __name__ == "__main__":
+    # Load the organization workspace.
+    OrganizationWorkspace.load()
     # Run the server.
     mcp.run(transport=transport)

src/organization_workspace.py

@@ -0,0 +1,41 @@
+import os
+import json
+
+
+class OrganizationWorkspace:
+
+    # Load mount destination and source from environment variables.
+    MOUNT_SOURCE = os.getenv('MOUNT_SOURCE_PATH')
+    MOUNT_DESTINATION = os.getenv('MOUNT_DST_PATH')
+    
+    available = False  # Indicate if local disk access is available
+    project_id_by_path = {}
+
+    @classmethod
+    def load(cls):
+        if not (cls.MOUNT_SOURCE and cls.MOUNT_DESTINATION):
+            return
+        if not os.path.exists(cls.MOUNT_DESTINATION):
+            return
+        for name in os.listdir(cls.MOUNT_DESTINATION):
+            if name in ['.QuantConnect', 'data', 'lean.json']:
+                continue
+            cls._process_directory(os.path.join(cls.MOUNT_DESTINATION, name))
+        cls.available = True
+
+    @classmethod
+    def _process_directory(cls, path):
+        # If the current directory contains a config.json file, then 
+        # it's a project, so save it's Id and path.
+        config_path = os.path.join(path, 'config.json')
+        if os.path.isfile(config_path):
+            with open(config_path, 'r') as f:
+                config_data = json.load(f)
+                if 'cloud-id' in config_data:
+                    cls.project_id_by_path[path] = config_data['cloud-id']
+        # Otherwise, it's a directory of projects, so recurse.
+        else:
+            for dir_name in os.listdir(path):
+                sub_path = os.path.join(path, dir_name)
+                if os.path.isdir(sub_path):
+                    cls._process_directory(sub_path)

tests/test_organization_workspace.py

@@ -0,0 +1,39 @@
+import pytest
+import time
+import docker
+
+from organization_workspace import OrganizationWorkspace
+from api_connection import USER_ID, API_TOKEN
+
+def test_organization_workspace_mount():
+    # Ensure the MOUNT_SOURCE environment variable is set.
+    assert OrganizationWorkspace.MOUNT_SOURCE, 'MOUNT_SOURCE env var is not set.'
+    # Create a Docker client.
+    client = docker.from_env()
+    # Start MCP Server inside a container.
+    container = client.containers.run(
+        image='quantconnect/mcp-server',
+        environment={
+            'QUANTCONNECT_USER_ID': USER_ID,
+            'QUANTCONNECT_API_TOKEN': API_TOKEN
+        },
+        platform='linux/amd64',
+        volumes={
+            OrganizationWorkspace.MOUNT_SOURCE: {
+                'bind': OrganizationWorkspace.MOUNT_DESTINATION, 'mode': 'ro'}
+            },
+        detach=True,  # Run in background
+        auto_remove=True  # Equivalent to --rm
+    )
+    # Wait for the container to start running.
+    time.sleep(5)
+    # Check if the expected mount exists.
+    assert any(
+        mount['Type'] == 'bind' and
+        mount['Source'] == OrganizationWorkspace.MOUNT_SOURCE and
+        mount['Destination'] == OrganizationWorkspace.MOUNT_DESTINATION and
+        mount['Mode'] == 'ro'
+        for mount in container.attrs['Mounts']
+    )
+    # Stop the container.
+    container.stop()