mcpower-proxy 0.0.64__py3-none-any.whl → 0.0.66__py3-none-any.whl

{mcpower_proxy-0.0.64.dist-info → mcpower_proxy-0.0.66.dist-info}/METADATA

@@ -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.dist-info → mcpower_proxy-0.0.66.dist-info}/RECORD

@@ -1,10 +1,10 @@
 main.py,sha256=4BnzO7q9Atpzgr-_NTc1loRnrRY0m5OxeG9biI-C0es,3707
-mcpower_proxy-0.0.64.dist-info/licenses/LICENSE,sha256=U6WUzdnBrbmVxBmY75ikW-KtinwYnowZ7yNb5hECrvY,11337
+mcpower_proxy-0.0.66.dist-info/licenses/LICENSE,sha256=U6WUzdnBrbmVxBmY75ikW-KtinwYnowZ7yNb5hECrvY,11337
 modules/__init__.py,sha256=mJglXQwSRhU-bBv4LXgfu7NfGN9K4BeQWMPApen5rAA,30
 modules/apis/__init__.py,sha256=Y5WZpKJzHpnRJebk0F80ZRTjR2PpA2LlYLgqI3XlmRo,15
 modules/apis/security_policy.py,sha256=AZDHTuOf99WhhzNw9AwC-0KACx1-ZjtQx-Ve3gAKYPM,15000
 modules/logs/__init__.py,sha256=dpboUQjuO02z8K-liCbm2DYkCa-CB_ZDV9WSSjNm7Fs,15
-modules/logs/audit_trail.py,sha256=PjVplAjCSUef9ps8P8xF3PSy6F4DLJicSQoTwpY4y6g,6096
+modules/logs/audit_trail.py,sha256=r8aIjaW-jBXXhSwdafzgOn0AvvdTZG8UPnkI0GmbJnA,6199
 modules/logs/logger.py,sha256=dfYRLnABZB07SBfoYV4DsD8-ZCpzEeoewFCBGHyqo9k,4171
 modules/redaction/__init__.py,sha256=e5NTmp-zonUdzzscih-w_WQ-X8Nvb8CE8b_d6SbrwWg,316
 modules/redaction/constants.py,sha256=xbDSX8n72FuJu6JJ_sbBE0f5OcWuwEwHxBZuK9Xz-TI,1213
@@ -28,16 +28,16 @@ modules/utils/__init__.py,sha256=Ptwu1epT_dW6EHjGkzGHAB-MbrrmYAlcPXGGcr4PvwE,20
 modules/utils/cli.py,sha256=qYgf7TsWKjwPsCItbDYzNZCih2vfGAbAl2MIem320_Y,1517
 modules/utils/config.py,sha256=YuGrIYfBsOYABWjFoZosObPz-R7Wdul16RnDed_glYI,6654
 modules/utils/copy.py,sha256=9OJIqWn8PxPZXr3DTt_01jp0YgmPimckab1969WFh0c,1075
-modules/utils/ids.py,sha256=i2MjU_sLFpowYb_5pYQKsbz_wJfjpDXkA9Svqy_wiJQ,4622
+modules/utils/ids.py,sha256=eYN31WDlfCyRSsdNu9F_hr_duuEpYEUJ5-ZMqo1fiAQ,4701
 modules/utils/json.py,sha256=8GA2akQsufXIn9HIP4SkFGFShzngexEBzejXi4B-Mfg,4031
 modules/utils/mcp_configs.py,sha256=DZaujZnF9LlPDJHzyepH7fWSt1GTr-FEmShPCqnZ5aI,1829
 wrapper/__init__.py,sha256=OJUsuWSoN1JqIHq4bSrzuL7ufcYJcwAmYCrJjLH44LM,22
-wrapper/__version__.py,sha256=2sNRZJT14qRkv4tRvpyg-2idt5inPq2tBnNecy0GWLI,82
-wrapper/middleware.py,sha256=HrxjC0puUxbh7lA0Yp36wJqtJwVsAVZleiqsOKgSt6w,34498
+wrapper/__version__.py,sha256=NMwcKl_v88ECQhojnCXMfvRbXACvhh32_M57YGOUiTE,82
+wrapper/middleware.py,sha256=g8jgIA-ciJMshSYflw9HSv_j6i-t1mQvkhc4pjww7Zc,35080
 wrapper/schema.py,sha256=O-CtKI9eJ4eEnqeUXPCrK7QJAFJrdp_cFbmMyg452Aw,7952
 wrapper/server.py,sha256=uVtxELALRrQNd-VrPWyLQPiEzxOpG-oCU7bItAeSjYU,2981
-mcpower_proxy-0.0.64.dist-info/METADATA,sha256=yxtClnw6IoEMoxWKA5kBFh6ZjGTk_-sHpDGlu8q_Hps,15667
-mcpower_proxy-0.0.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mcpower_proxy-0.0.64.dist-info/entry_points.txt,sha256=0smL8dxE7ERNz6XEggNaUC3QzKp8mD-v4q5nVEo0MXE,48
-mcpower_proxy-0.0.64.dist-info/top_level.txt,sha256=FLbRkTTggoMB-kq14IH4ZUbNGMGtbxtmiWw0QykRlkU,21
-mcpower_proxy-0.0.64.dist-info/RECORD,,
+mcpower_proxy-0.0.66.dist-info/METADATA,sha256=j3rSpJS3RFqhTP7QmONIhOVN1l7rIGvB8WXGlcoOIds,15667
+mcpower_proxy-0.0.66.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcpower_proxy-0.0.66.dist-info/entry_points.txt,sha256=0smL8dxE7ERNz6XEggNaUC3QzKp8mD-v4q5nVEo0MXE,48
+mcpower_proxy-0.0.66.dist-info/top_level.txt,sha256=FLbRkTTggoMB-kq14IH4ZUbNGMGtbxtmiWw0QykRlkU,21
+mcpower_proxy-0.0.66.dist-info/RECORD,,

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:

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})")

wrapper/__version__.py

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

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)