mcp-proxy-adapter 6.3.4__py3-none-any.whl → 6.3.5__py3-none-any.whl

mcp_proxy_adapter/examples/generate_test_configs.py

@@ -30,15 +30,12 @@ def generate_http_simple_config(
                 "endpoints": {
                     "jsonrpc": "/api/jsonrpc",
                     "rest": "/cmd",
-                    "health": "/health"
-                }
+                    "health": "/health",
+                },
             },
-            "heartbeat": {"enabled": True, "interval": 30}
+            "heartbeat": {"enabled": True, "interval": 30},
         },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["http"]
-        }
+        "protocols": {"enabled": True, "allowed_protocols": ["http"]},
     }
 
 
@@ -46,7 +43,7 @@ def generate_http_token_config(
     port: int = 20001,
     certs_dir: str = "certs",
     keys_dir: str = "keys",
-    roles_file: str = "configs/roles.json"
+    roles_file: str = "configs/roles.json",
 ) -> Dict[str, Any]:
     """Generate HTTP configuration with token authorization."""
     return {
@@ -63,10 +60,10 @@ def generate_http_token_config(
                     "user-token-456": "user",
                     "readonly-token-123": "readonly",
                     "guest-token-123": "guest",
-                    "proxy-token-123": "proxy"
-                }
+                    "proxy-token-123": "proxy",
+                },
             },
-            "permissions": {"enabled": True, "roles_file": roles_file}
+            "permissions": {"enabled": True, "roles_file": roles_file},
         },
         "registration": {
             "enabled": True,
@@ -75,12 +72,9 @@ def generate_http_token_config(
             "capabilities": ["http", "token_auth"],
             "retry_count": 3,
             "retry_delay": 5,
-            "heartbeat": {"enabled": True, "interval": 30}
+            "heartbeat": {"enabled": True, "interval": 30},
         },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["http"]
-        }
+        "protocols": {"enabled": True, "allowed_protocols": ["http"]},
     }
 
 
@@ -93,7 +87,7 @@ def generate_https_simple_config(
         "ssl": {
             "enabled": True,
             "cert_file": f"{certs_dir}/localhost_server.crt",
-            "key_file": f"{keys_dir}/localhost_server.key"
+            "key_file": f"{keys_dir}/localhost_server.key",
         },
         "security": {"enabled": False},
         "registration": {
@@ -103,12 +97,9 @@ def generate_https_simple_config(
             "capabilities": ["https"],
             "retry_count": 3,
             "retry_delay": 5,
-            "heartbeat": {"enabled": True, "interval": 30}
+            "heartbeat": {"enabled": True, "interval": 30},
         },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["http", "https"]
-        }
+        "protocols": {"enabled": True, "allowed_protocols": ["http", "https"]},
     }
 
 
@@ -121,7 +112,7 @@ def generate_https_token_config(
         "ssl": {
             "enabled": True,
             "cert_file": f"{certs_dir}/localhost_server.crt",
-            "key_file": f"{keys_dir}/localhost_server.key"
+            "key_file": f"{keys_dir}/localhost_server.key",
         },
         "security": {
             "enabled": True,
@@ -133,10 +124,10 @@ def generate_https_token_config(
                     "user-token-456": "user",
                     "readonly-token-123": "readonly",
                     "guest-token-123": "guest",
-                    "proxy-token-123": "proxy"
-                }
+                    "proxy-token-123": "proxy",
+                },
             },
-            "permissions": {"enabled": True, "roles_file": "./configs/roles.json"}
+            "permissions": {"enabled": True, "roles_file": "./configs/roles.json"},
         },
         "registration": {
             "enabled": True,
@@ -145,12 +136,9 @@ def generate_https_token_config(
             "capabilities": ["https", "token_auth"],
             "retry_count": 3,
             "retry_delay": 5,
-            "heartbeat": {"enabled": True, "interval": 30}
+            "heartbeat": {"enabled": True, "interval": 30},
         },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["http", "https"]
-        }
+        "protocols": {"enabled": True, "allowed_protocols": ["http", "https"]},
     }
 
 
@@ -166,20 +154,15 @@ def generate_mtls_no_roles_config(
             "key_file": f"{keys_dir}/localhost_server.key",
             "ca_cert": f"{certs_dir}/mcp_proxy_adapter_ca_ca.crt",
             "verify_client": True,
-            "client_cert_required": True
+            "client_cert_required": True,
         },
         "security": {
             "enabled": True,
             "auth": {"enabled": True, "methods": ["certificate"]},
-            "permissions": {"enabled": False}
+            "permissions": {"enabled": False},
         },
-        "registration": {
-            "enabled": False
-        },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["https", "mtls"]
-        }
+        "registration": {"enabled": False},
+        "protocols": {"enabled": True, "allowed_protocols": ["https", "mtls"]},
     }
 
 
@@ -187,7 +170,7 @@ def generate_mtls_with_roles_config(
     port: int = 20005,
     certs_dir: str = "certs",
     keys_dir: str = "keys",
-    roles_file: str = "configs/roles.json"
+    roles_file: str = "configs/roles.json",
 ) -> Dict[str, Any]:
     """Generate mTLS configuration with roles."""
     return {
@@ -197,7 +180,7 @@ def generate_mtls_with_roles_config(
             "cert_file": f"{certs_dir}/localhost_server.crt",
             "key_file": f"{keys_dir}/localhost_server.key",
             "ca_cert": f"{certs_dir}/mcp_proxy_adapter_ca_ca.crt",
-            "verify_client": True
+            "verify_client": True,
         },
         "registration": {
             "enabled": True,
@@ -210,20 +193,17 @@ def generate_mtls_with_roles_config(
                 "endpoints": {
                     "jsonrpc": "/api/jsonrpc",
                     "rest": "/cmd",
-                    "health": "/health"
-                }
+                    "health": "/health",
+                },
             },
-            "heartbeat": {"enabled": True, "interval": 30}
+            "heartbeat": {"enabled": True, "interval": 30},
         },
         "security": {
             "enabled": True,
             "auth": {"enabled": True, "methods": ["certificate"]},
-            "permissions": {"enabled": True, "roles_file": roles_file}
+            "permissions": {"enabled": True, "roles_file": roles_file},
         },
-        "protocols": {
-            "enabled": True,
-            "allowed_protocols": ["https", "mtls"]
-        }
+        "protocols": {"enabled": True, "allowed_protocols": ["https", "mtls"]},
     }
 
 
@@ -241,9 +221,9 @@ def generate_roles_config() -> Dict[str, Any]:
                 "register",
                 "unregister",
                 "heartbeat",
-                "discover"
+                "discover",
             ],
-            "tokens": ["test-token-123"]
+            "tokens": ["test-token-123"],
         },
         "user": {
             "description": "User role with limited access",
@@ -253,36 +233,25 @@ def generate_roles_config() -> Dict[str, Any]:
                 "register",
                 "unregister",
                 "heartbeat",
-                "discover"
+                "discover",
             ],
-            "tokens": ["user-token-456"]
+            "tokens": ["user-token-456"],
         },
         "readonly": {
             "description": "Read-only role",
-            "permissions": [
-                "read",
-                "discover"
-            ],
-            "tokens": ["readonly-token-123"]
+            "permissions": ["read", "discover"],
+            "tokens": ["readonly-token-123"],
         },
         "guest": {
             "description": "Guest role with read-only access",
-            "permissions": [
-                "read",
-                "discover"
-            ],
-            "tokens": ["guest-token-123"]
+            "permissions": ["read", "discover"],
+            "tokens": ["guest-token-123"],
         },
         "proxy": {
             "description": "Proxy role for registration",
-            "permissions": [
-                "register",
-                "unregister",
-                "heartbeat",
-                "discover"
-            ],
-            "tokens": ["proxy-token-123"]
-        }
+            "permissions": ["register", "unregister", "heartbeat", "discover"],
+            "tokens": ["proxy-token-123"],
+        },
     }
 
 
@@ -290,7 +259,7 @@ def generate_all_configs(
     output_dir: str,
     certs_dir: str = "certs",
     keys_dir: str = "keys",
-    roles_file: str = "configs/roles.json"
+    roles_file: str = "configs/roles.json",
 ) -> None:
     """Generate all 6 configuration types and save them to files."""
     # Ensure output directory exists first
@@ -298,17 +267,21 @@ def generate_all_configs(
 
     configs = {
         "http_simple": generate_http_simple_config(20000, certs_dir, keys_dir),
-        "http_token": generate_http_token_config(20001, certs_dir, keys_dir, roles_file),
+        "http_token": generate_http_token_config(
+            20001, certs_dir, keys_dir, roles_file
+        ),
         "https_simple": generate_https_simple_config(20002, certs_dir, keys_dir),
         "https_token": generate_https_token_config(20003, certs_dir, keys_dir),
         "mtls_no_roles": generate_mtls_no_roles_config(20004, certs_dir, keys_dir),
-        "mtls_with_roles": generate_mtls_with_roles_config(20005, certs_dir, keys_dir, roles_file)
+        "mtls_with_roles": generate_mtls_with_roles_config(
+            20005, certs_dir, keys_dir, roles_file
+        ),
     }
 
     # Generate each configuration
     for name, config in configs.items():
         filename = os.path.join(output_dir, f"{name}.json")
-        with open(filename, 'w', encoding='utf-8') as f:
+        with open(filename, "w", encoding="utf-8") as f:
             json.dump(config, f, indent=2, ensure_ascii=False)
         print(f"Generated: {filename}")
 
@@ -324,13 +297,13 @@ def generate_all_configs(
         root_roles_filename = os.path.join(current_dir, "roles.json")
 
         # Create roles.json in the current working directory
-        with open(root_roles_filename, 'w', encoding='utf-8') as f:
+        with open(root_roles_filename, "w", encoding="utf-8") as f:
             json.dump(roles_config, f, indent=2, ensure_ascii=False)
         print(f"Generated: {root_roles_filename}")
 
         # Also create a copy in the output directory for reference
         backup_roles_filename = os.path.join(output_dir, "roles_backup.json")
-        with open(backup_roles_filename, 'w', encoding='utf-8') as f:
+        with open(backup_roles_filename, "w", encoding="utf-8") as f:
             json.dump(roles_config, f, indent=2, ensure_ascii=False)
         print(f"Generated backup: {backup_roles_filename}")
 
@@ -341,10 +314,12 @@ def generate_all_configs(
 
     # Also create roles.json in configs directory for reference
     roles_filename = os.path.join(output_dir, "roles.json")
-    with open(roles_filename, 'w', encoding='utf-8') as f:
+    with open(roles_filename, "w", encoding="utf-8") as f:
         json.dump(roles_config, f, indent=2, ensure_ascii=False)
     print(f"Generated: {roles_filename}")
-    print(f"\nGenerated {len(configs)} configuration files and roles.json in {output_dir}")
+    print(
+        f"\nGenerated {len(configs)} configuration files and roles.json in {output_dir}"
+    )
 
     print("\n" + "=" * 60)
     print("✅ CONFIGURATION GENERATION COMPLETED SUCCESSFULLY")
@@ -353,9 +328,13 @@ def generate_all_configs(
     print("1. Run security tests:")
     print("   python -m mcp_proxy_adapter.examples.run_security_tests")
     print("\n2. Start basic framework example:")
-    print("   python -m mcp_proxy_adapter.examples.basic_framework.main --config configs/https_simple.json")
+    print(
+        "   python -m mcp_proxy_adapter.examples.basic_framework.main --config configs/https_simple.json"
+    )
     print("\n3. Start full application example:")
-    print("   python -m mcp_proxy_adapter.examples.full_application.main --config configs/mtls_with_roles.json")
+    print(
+        "   python -m mcp_proxy_adapter.examples.full_application.main --config configs/mtls_with_roles.json"
+    )
     print("=" * 60)
 
 
@@ -367,27 +346,25 @@ def main() -> int:
     parser.add_argument(
         "--output-dir",
         default="configs",
-        help="Output directory for configuration files (default: configs)"
+        help="Output directory for configuration files (default: configs)",
     )
     parser.add_argument(
-        "--certs-dir",
-        default="certs",
-        help="Certificates directory (default: certs)"
+        "--certs-dir", default="certs", help="Certificates directory (default: certs)"
     )
     parser.add_argument(
-        "--keys-dir",
-        default="keys",
-        help="Keys directory (default: keys)"
+        "--keys-dir", default="keys", help="Keys directory (default: keys)"
     )
     parser.add_argument(
         "--roles-file",
         default="configs/roles.json",
-        help="Roles file path (default: configs/roles.json)"
+        help="Roles file path (default: configs/roles.json)",
     )
     args = parser.parse_args()
 
     try:
-        generate_all_configs(args.output_dir, args.certs_dir, args.keys_dir, args.roles_file)
+        generate_all_configs(
+            args.output_dir, args.certs_dir, args.keys_dir, args.roles_file
+        )
         print("Configuration generation completed successfully!")
     except Exception as e:
         print(f"\n❌ CONFIGURATION GENERATION FAILED: {e}")