mcpower-proxy 0.0.64__tar.gz → 0.0.66__tar.gz

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcpower-proxy
-Version: 0.0.64
+Version: 0.0.66
 Summary: MCPower Security proxy
 Author-email: MCPower Security <support@mcpower.tech>
 License:                                  Apache License

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcpower-proxy"
-version = "0.0.64"
+version = "0.0.66"
 description = "MCPower Security proxy"
 readme = "README.md"
 requires-python = "~=3.11.0"

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/src/mcpower_proxy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcpower-proxy
-Version: 0.0.64
+Version: 0.0.66
 Summary: MCPower Security proxy
 Author-email: MCPower Security <support@mcpower.tech>
 License:                                  Apache License

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/src/modules/logs/audit_trail.py

@@ -130,16 +130,20 @@ class AuditTrailLogger:
         
         This is called by the middleware after workspace roots are available.
         All queued logs will be written with app_uid as the first key.
+        Supports updating app_uid when workspace context changes.
         
         Args:
             app_uid: The application UID from workspace root
         """
-        if self.app_uid is not None:
-            self.logger.warning(f"app_uid already set to {self.app_uid}, ignoring new value {app_uid}")
+        if self.app_uid == app_uid:
             return
         
+        if self.app_uid is not None:
+            self.logger.info(f"app_uid changed from {self.app_uid} to {app_uid}")
+        else:
+            self.logger.debug(f"app_uid set to: {app_uid}")
+        
         self.app_uid = app_uid
-        self.logger.debug(f"✅ app_uid set to: {app_uid}")
         
         # Flush all pending logs
         if self._pending_logs:

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/src/modules/utils/ids.py

@@ -2,6 +2,7 @@
 Utilities for generating event IDs, session IDs, app UIDs, and timing helpers
 """
 import os
+import sys
 import time
 import uuid
 from pathlib import Path
@@ -67,7 +68,8 @@ def _atomic_write_uuid(file_path: Path, new_uuid: str) -> bool:
         True if write succeeded, False if file exists
     """
     try:
-        fd = os.open(str(file_path), os.O_CREAT | os.O_EXCL | os.O_WRONLY, 0o600)
+        mode = 0o666 if sys.platform == 'win32' else 0o600
+        fd = os.open(str(file_path), os.O_CREAT | os.O_EXCL | os.O_WRONLY, mode)
         try:
             os.write(fd, new_uuid.encode('utf-8'))
         finally:
@@ -91,7 +93,7 @@ def _get_or_create_uuid(uid_path: Path, logger, id_type: str) -> str:
         UUID string
     """
     uid_path.parent.mkdir(parents=True, exist_ok=True)
-    
+
     max_attempts = 3
     for attempt in range(max_attempts):
         if uid_path.exists():
@@ -111,7 +113,7 @@ def _get_or_create_uuid(uid_path: Path, logger, id_type: str) -> str:
         new_uid = str(uuid.uuid4())
         
         if _atomic_write_uuid(uid_path, new_uid):
-            logger.info(f"Generated {id_type}: {new_uid}")
+            logger.info(f"Generated {id_type}: {new_uid} at {uid_path}")
             return new_uid
         
         logger.debug(f"{id_type.title()} file created by another process, reading (attempt {attempt + 1}/{max_attempts})")

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/src/wrapper/__version__.py

@@ -3,4 +3,4 @@
 Wrapper MCP Server Version
 """
 
-__version__ = "0.0.64"
+__version__ = "0.0.66"

{mcpower_proxy-0.0.64 → mcpower_proxy-0.0.66}/src/wrapper/middleware.py

@@ -2,6 +2,7 @@
 FastMCP middleware for security policy enforcement
 Implements pre/post interception for all MCP operations
 """
+import sys
 import time
 import urllib.parse
 from datetime import datetime, timezone
@@ -55,6 +56,7 @@ class SecurityMiddleware(Middleware):
     app_id: str = ""
     _TOOLS_INIT_DEBOUNCE_SECONDS = 60
     _last_tools_init_time: Optional[float] = None
+    _last_workspace_root: Optional[str] = None
 
     def __init__(self,
                  wrapped_server_configs: dict,
@@ -69,6 +71,7 @@ class SecurityMiddleware(Middleware):
         self.logger = logger
         self.audit_logger = audit_logger
         self.app_id = ""
+        self._last_workspace_root = None
 
         self.wrapped_server_name, self.wrapped_server_transport = (
             extract_wrapped_server_info(self.wrapper_server_name, self.logger, self.wrapped_server_configs)
@@ -85,14 +88,13 @@ class SecurityMiddleware(Middleware):
     async def on_message(self, context: MiddlewareContext, call_next: CallNext) -> Any:
         self.logger.info(f"on_message: {redact(safe_json_dumps(context))}")
 
-        # Ensure app_id is set before making API calls
-        if not self.app_id:
-            workspace_roots = await self._extract_workspace_roots(context)
-            if workspace_roots:
-                self.app_id = read_app_uid(logger=self.logger, project_folder_path=workspace_roots[0])
-            else:
-                # Fallback: read app_uid from ~/.mcpower when no workspace roots
-                self.app_id = read_app_uid(logger=self.logger, project_folder_path=str(Path.home() / ".mcpower"))
+        # Check workspace roots and re-initialize app_uid if workspace changed
+        workspace_roots = await self._extract_workspace_roots(context)
+        current_workspace_root = workspace_roots[0] if workspace_roots else str(Path.home() / ".mcpower")
+        if current_workspace_root != self._last_workspace_root:
+            self.logger.debug(f"Workspace root changed from {self._last_workspace_root} to {current_workspace_root}")
+            self._last_workspace_root = current_workspace_root
+            self.app_id = read_app_uid(logger=self.logger, project_folder_path=current_workspace_root)
             self.audit_logger.set_app_uid(self.app_id)
 
         operation_type = "message"
@@ -480,6 +482,12 @@ class SecurityMiddleware(Middleware):
                         file_path_prefix = 'file://'
                         if uri.startswith(file_path_prefix):
                             path = urllib.parse.unquote(uri[len(file_path_prefix):])
+                            
+                            # Windows fix: remove leading slash before drive letter
+                            # file:///C:/path becomes /C:/path, should be C:/path
+                            if sys.platform == 'win32' and len(path) >= 3 and path[0] == '/' and path[2] == ':':
+                                path = path[1:]
+                            
                             try:
                                 resolved_path = str(Path(path).resolve())
                                 workspace_roots.append(resolved_path)