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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcpower-proxy
-Version: 0.0.66
+Version: 0.0.68
 Summary: MCPower Security proxy
 Author-email: MCPower Security <support@mcpower.tech>
 License:                                  Apache License
@@ -209,7 +209,7 @@ Keywords: mcp,security,proxy,monitoring,audit,redaction,policy-enforcement
 Requires-Python: ~=3.11.0
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: fastmcp>=2.10.5
+Requires-Dist: fastmcp==2.13.0.1
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: mcp>=1.0.0
 Requires-Dist: watchdog>=3.0.0

{mcpower_proxy-0.0.66.dist-info → mcpower_proxy-0.0.68.dist-info}/RECORD

@@ -1,8 +1,8 @@
 main.py,sha256=4BnzO7q9Atpzgr-_NTc1loRnrRY0m5OxeG9biI-C0es,3707
-mcpower_proxy-0.0.66.dist-info/licenses/LICENSE,sha256=U6WUzdnBrbmVxBmY75ikW-KtinwYnowZ7yNb5hECrvY,11337
+mcpower_proxy-0.0.68.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/apis/security_policy.py,sha256=sgfkdE1X3P3t4d4WMt1caKXDhpjfBst-HVM0u3qcT9I,15100
 modules/logs/__init__.py,sha256=dpboUQjuO02z8K-liCbm2DYkCa-CB_ZDV9WSSjNm7Fs,15
 modules/logs/audit_trail.py,sha256=r8aIjaW-jBXXhSwdafzgOn0AvvdTZG8UPnkI0GmbJnA,6199
 modules/logs/logger.py,sha256=dfYRLnABZB07SBfoYV4DsD8-ZCpzEeoewFCBGHyqo9k,4171
@@ -25,19 +25,19 @@ modules/ui/xdialog/windows_structs.py,sha256=xzG44OGT5hBFnimJgOLXZBhmpQ_9CFxjtz-
 modules/ui/xdialog/yad_dialogs.py,sha256=EiajZVJg-xDwYymz1fyQwLtT5DzbJR3e8plMEnOgcpo,6933
 modules/ui/xdialog/zenity_dialogs.py,sha256=wE71I_Ovf0sjhxHVNocbrhhDd8Y8X8loLETp8TMGMPQ,4512
 modules/utils/__init__.py,sha256=Ptwu1epT_dW6EHjGkzGHAB-MbrrmYAlcPXGGcr4PvwE,20
-modules/utils/cli.py,sha256=qYgf7TsWKjwPsCItbDYzNZCih2vfGAbAl2MIem320_Y,1517
+modules/utils/cli.py,sha256=I8_Fz97J09gHL2PlAcgJzHmeNX9aF8XTzJOBjz3wBbk,1414
 modules/utils/config.py,sha256=YuGrIYfBsOYABWjFoZosObPz-R7Wdul16RnDed_glYI,6654
 modules/utils/copy.py,sha256=9OJIqWn8PxPZXr3DTt_01jp0YgmPimckab1969WFh0c,1075
 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=NMwcKl_v88ECQhojnCXMfvRbXACvhh32_M57YGOUiTE,82
-wrapper/middleware.py,sha256=g8jgIA-ciJMshSYflw9HSv_j6i-t1mQvkhc4pjww7Zc,35080
+wrapper/__version__.py,sha256=T9sGujOerGwd7GKxEy9ZnGh5ysLmP0HvBoFYoVUtHQ4,82
+wrapper/middleware.py,sha256=yvtYAssvcdbEPvLbc9FFhmZS0uToGcNyiXJK_cIKYC0,35364
 wrapper/schema.py,sha256=O-CtKI9eJ4eEnqeUXPCrK7QJAFJrdp_cFbmMyg452Aw,7952
 wrapper/server.py,sha256=uVtxELALRrQNd-VrPWyLQPiEzxOpG-oCU7bItAeSjYU,2981
-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,,
+mcpower_proxy-0.0.68.dist-info/METADATA,sha256=Wd48d6uOzsE9KwOaJPPO_dW0TYONi2T8hzo6p2-F7VM,15669
+mcpower_proxy-0.0.68.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcpower_proxy-0.0.68.dist-info/entry_points.txt,sha256=0smL8dxE7ERNz6XEggNaUC3QzKp8mD-v4q5nVEo0MXE,48
+mcpower_proxy-0.0.68.dist-info/top_level.txt,sha256=FLbRkTTggoMB-kq14IH4ZUbNGMGtbxtmiWw0QykRlkU,21
+mcpower_proxy-0.0.68.dist-info/RECORD,,

modules/apis/security_policy.py

@@ -1,9 +1,9 @@
 """Security Policy API Client"""
 
 import json
+import time
 import uuid
 from typing import Dict, Any, Optional, List
-import time
 
 import httpx
 
@@ -13,6 +13,7 @@ from modules.logs.logger import MCPLogger
 from modules.redaction import redact
 from modules.utils.config import get_api_url, get_user_id
 from modules.utils.json import safe_json_dumps, to_dict
+from wrapper.__version__ import __version__
 
 
 class SecurityAPIError(Exception):
@@ -166,6 +167,7 @@ class SecurityPolicyClient:
 
             headers = {
                 "Content-Type": "application/json",
+                "User-Agent": f"MCPower-{__version__}",
                 "X-User-UID": self.user_id,
                 "X-App-UID": self.app_id
             }

modules/utils/cli.py

@@ -7,7 +7,7 @@ import argparse
 def parse_args():
     """Parse command line arguments"""
     parser = argparse.ArgumentParser(
-        description="MCPower - Transparent 1:1 MCP Wrapper with security enforcement",
+        description="Transparent MCP wrapper with security middleware for real-time policy enforcement and monitoring.",
         formatter_class=argparse.RawDescriptionHelpFormatter,
         epilog="""
 Examples:
@@ -24,10 +24,8 @@ Examples:
   %(prog)s --wrapped-config '{"command": "node", "args": ["server.js"]}' --name MyWrapper
 
 Reference Links:
-  • FastMCP Proxy: https://gofastmcp.com/servers/proxy
-  • FastMCP Middleware: https://gofastmcp.com/servers/middleware  
+  • MCPower Proxy: https://github.com/ai-mcpower/mcpower-proxy
   • MCP Official: https://modelcontextprotocol.io
-  • Claude MCP Config: https://docs.anthropic.com/en/docs/claude-code/mcp
         """
     )
 

wrapper/__version__.py

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

wrapper/middleware.py

@@ -12,6 +12,10 @@ from typing import Any, Dict, List, Optional
 from fastmcp.exceptions import FastMCPError
 from fastmcp.server.middleware.middleware import Middleware, MiddlewareContext, CallNext
 from fastmcp.server.proxy import ProxyClient
+
+from mcpower_shared.mcp_types import (create_policy_request, create_policy_response, AgentContext, EnvironmentContext,
+                                      InitRequest,
+                                      ServerRef, ToolRef, UserConfirmation)
 from modules.apis.security_policy import SecurityPolicyClient
 from modules.logs.audit_trail import AuditTrailLogger
 from modules.logs.logger import MCPLogger
@@ -24,10 +28,6 @@ from modules.utils.json import safe_json_dumps, to_dict
 from modules.utils.mcp_configs import extract_wrapped_server_info
 from wrapper.schema import merge_input_schema_with_existing
 
-from mcpower_shared.mcp_types import (create_policy_request, create_policy_response, AgentContext, EnvironmentContext,
-                                      InitRequest,
-                                      ServerRef, ToolRef, UserConfirmation)
-
 
 class MockContext:
     """Mock context for internal operations"""
@@ -88,14 +88,18 @@ 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))}")
 
-        # 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)
+        # Skip workspace check for `initialize` calls to avoid premature app_uid changes.
+        # The `initialize` request doesn't contain workspace data, so checking it would
+        # cause unnecessary audit log flushes before the actual workspace init arrives.
+        if context.method != "initialize":
+            # 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"
         call_next_callback = call_next
@@ -116,7 +120,7 @@ class SecurityMiddleware(Middleware):
             case "tools/list":
                 # Special handling for tools/list - call /init instead of normal inspection
                 return await self._handle_tools_list(context, call_next)
-            case "resources/list" | "resources/templates/list" | "prompts/list":
+            case "initialize" | "resources/list" | "resources/templates/list" | "prompts/list":
                 return await call_next_callback(context)
 
         return await self._handle_operation(
@@ -184,12 +188,12 @@ class SecurityMiddleware(Middleware):
         # FIXME: log_message should be redacted before logging, 
         self.logger.info(f"secure_log_handler: {str(log_message)[:100]}...")
         # FIXME: log_message should be reviewed with policy before forwarding
-        
+
         # Handle case where log_message.data is a string instead of dict
         # The default_log_handler expects data to be a dict with 'msg' and 'extra' keys
         if hasattr(log_message, 'data') and isinstance(log_message.data, str):
             log_message = safe_copy(log_message, {'data': {'msg': log_message.data, 'extra': None}})
-        
+
         return await ProxyClient.default_log_handler(log_message)
 
     async def _handle_operation(self, context: MiddlewareContext, call_next, error_class, operation_type: str):
@@ -482,12 +486,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)