mcp-proxy-adapter 4.1.1__py3-none-any.whl → 6.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (253) hide show
  1. mcp_proxy_adapter/__main__.py +12 -0
  2. mcp_proxy_adapter/api/app.py +254 -33
  3. mcp_proxy_adapter/api/handlers.py +32 -6
  4. mcp_proxy_adapter/api/middleware/__init__.py +36 -30
  5. mcp_proxy_adapter/api/middleware/command_permission_middleware.py +148 -0
  6. mcp_proxy_adapter/api/middleware/error_handling.py +9 -0
  7. mcp_proxy_adapter/api/middleware/factory.py +243 -0
  8. mcp_proxy_adapter/api/middleware/logging.py +32 -6
  9. mcp_proxy_adapter/api/middleware/protocol_middleware.py +135 -0
  10. mcp_proxy_adapter/api/middleware/transport_middleware.py +122 -0
  11. mcp_proxy_adapter/api/middleware/unified_security.py +152 -0
  12. mcp_proxy_adapter/api/middleware/user_info_middleware.py +83 -0
  13. mcp_proxy_adapter/commands/__init__.py +19 -4
  14. mcp_proxy_adapter/commands/auth_validation_command.py +408 -0
  15. mcp_proxy_adapter/commands/base.py +66 -32
  16. mcp_proxy_adapter/commands/builtin_commands.py +95 -0
  17. mcp_proxy_adapter/commands/catalog_manager.py +838 -0
  18. mcp_proxy_adapter/commands/cert_monitor_command.py +620 -0
  19. mcp_proxy_adapter/commands/certificate_management_command.py +608 -0
  20. mcp_proxy_adapter/commands/command_registry.py +711 -354
  21. mcp_proxy_adapter/commands/dependency_manager.py +245 -0
  22. mcp_proxy_adapter/commands/echo_command.py +81 -0
  23. mcp_proxy_adapter/commands/health_command.py +7 -0
  24. mcp_proxy_adapter/commands/help_command.py +21 -14
  25. mcp_proxy_adapter/commands/hooks.py +200 -167
  26. mcp_proxy_adapter/commands/key_management_command.py +506 -0
  27. mcp_proxy_adapter/commands/load_command.py +176 -0
  28. mcp_proxy_adapter/commands/plugins_command.py +235 -0
  29. mcp_proxy_adapter/commands/protocol_management_command.py +232 -0
  30. mcp_proxy_adapter/commands/proxy_registration_command.py +409 -0
  31. mcp_proxy_adapter/commands/reload_command.py +48 -50
  32. mcp_proxy_adapter/commands/result.py +1 -0
  33. mcp_proxy_adapter/commands/role_test_command.py +141 -0
  34. mcp_proxy_adapter/commands/roles_management_command.py +697 -0
  35. mcp_proxy_adapter/commands/security_command.py +488 -0
  36. mcp_proxy_adapter/commands/ssl_setup_command.py +483 -0
  37. mcp_proxy_adapter/commands/token_management_command.py +529 -0
  38. mcp_proxy_adapter/commands/transport_management_command.py +144 -0
  39. mcp_proxy_adapter/commands/unload_command.py +158 -0
  40. mcp_proxy_adapter/config.py +159 -2
  41. mcp_proxy_adapter/core/app_factory.py +326 -0
  42. mcp_proxy_adapter/core/auth_validator.py +606 -0
  43. mcp_proxy_adapter/core/certificate_utils.py +827 -0
  44. mcp_proxy_adapter/core/client_security.py +384 -0
  45. mcp_proxy_adapter/core/config_converter.py +405 -0
  46. mcp_proxy_adapter/core/config_validator.py +218 -0
  47. mcp_proxy_adapter/core/logging.py +19 -3
  48. mcp_proxy_adapter/core/mtls_asgi.py +156 -0
  49. mcp_proxy_adapter/core/mtls_asgi_app.py +187 -0
  50. mcp_proxy_adapter/core/protocol_manager.py +235 -0
  51. mcp_proxy_adapter/core/proxy_client.py +602 -0
  52. mcp_proxy_adapter/core/proxy_registration.py +522 -0
  53. mcp_proxy_adapter/core/role_utils.py +426 -0
  54. mcp_proxy_adapter/core/security_adapter.py +370 -0
  55. mcp_proxy_adapter/core/security_factory.py +239 -0
  56. mcp_proxy_adapter/core/security_integration.py +277 -0
  57. mcp_proxy_adapter/core/server_adapter.py +345 -0
  58. mcp_proxy_adapter/core/server_engine.py +364 -0
  59. mcp_proxy_adapter/core/settings.py +1 -0
  60. mcp_proxy_adapter/core/ssl_utils.py +233 -0
  61. mcp_proxy_adapter/core/transport_manager.py +292 -0
  62. mcp_proxy_adapter/core/unified_config_adapter.py +579 -0
  63. mcp_proxy_adapter/custom_openapi.py +22 -11
  64. mcp_proxy_adapter/examples/README.md +230 -97
  65. mcp_proxy_adapter/examples/README_EN.md +258 -0
  66. mcp_proxy_adapter/examples/SECURITY_TESTING.md +455 -0
  67. mcp_proxy_adapter/examples/__pycache__/security_configurations.cpython-312.pyc +0 -0
  68. mcp_proxy_adapter/examples/__pycache__/security_test_client.cpython-312.pyc +0 -0
  69. mcp_proxy_adapter/examples/basic_framework/configs/http_auth.json +37 -0
  70. mcp_proxy_adapter/examples/basic_framework/configs/http_simple.json +23 -0
  71. mcp_proxy_adapter/examples/basic_framework/configs/https_auth.json +39 -0
  72. mcp_proxy_adapter/examples/basic_framework/configs/https_simple.json +25 -0
  73. mcp_proxy_adapter/examples/basic_framework/configs/mtls_no_roles.json +39 -0
  74. mcp_proxy_adapter/examples/basic_framework/configs/mtls_with_roles.json +45 -0
  75. mcp_proxy_adapter/examples/basic_framework/main.py +63 -0
  76. mcp_proxy_adapter/examples/basic_framework/roles.json +21 -0
  77. mcp_proxy_adapter/examples/cert_config.json +9 -0
  78. mcp_proxy_adapter/examples/certs/admin.crt +32 -0
  79. mcp_proxy_adapter/examples/certs/admin.key +52 -0
  80. mcp_proxy_adapter/examples/certs/admin_cert.pem +21 -0
  81. mcp_proxy_adapter/examples/certs/admin_key.pem +28 -0
  82. mcp_proxy_adapter/examples/certs/ca_cert.pem +23 -0
  83. mcp_proxy_adapter/examples/certs/ca_cert.srl +1 -0
  84. mcp_proxy_adapter/examples/certs/ca_key.pem +28 -0
  85. mcp_proxy_adapter/examples/certs/cert_config.json +9 -0
  86. mcp_proxy_adapter/examples/certs/client.crt +32 -0
  87. mcp_proxy_adapter/examples/certs/client.key +52 -0
  88. mcp_proxy_adapter/examples/certs/client_admin.crt +32 -0
  89. mcp_proxy_adapter/examples/certs/client_admin.key +52 -0
  90. mcp_proxy_adapter/examples/certs/client_user.crt +32 -0
  91. mcp_proxy_adapter/examples/certs/client_user.key +52 -0
  92. mcp_proxy_adapter/examples/certs/guest_cert.pem +21 -0
  93. mcp_proxy_adapter/examples/certs/guest_key.pem +28 -0
  94. mcp_proxy_adapter/examples/certs/mcp_proxy_adapter_ca_ca.crt +23 -0
  95. mcp_proxy_adapter/examples/certs/proxy_cert.pem +21 -0
  96. mcp_proxy_adapter/examples/certs/proxy_key.pem +28 -0
  97. mcp_proxy_adapter/examples/certs/readonly.crt +32 -0
  98. mcp_proxy_adapter/examples/certs/readonly.key +52 -0
  99. mcp_proxy_adapter/examples/certs/readonly_cert.pem +21 -0
  100. mcp_proxy_adapter/examples/certs/readonly_key.pem +28 -0
  101. mcp_proxy_adapter/examples/certs/server.crt +32 -0
  102. mcp_proxy_adapter/examples/certs/server.key +52 -0
  103. mcp_proxy_adapter/examples/certs/server_cert.pem +32 -0
  104. mcp_proxy_adapter/examples/certs/server_key.pem +52 -0
  105. mcp_proxy_adapter/examples/certs/test_ca_ca.crt +20 -0
  106. mcp_proxy_adapter/examples/certs/user.crt +32 -0
  107. mcp_proxy_adapter/examples/certs/user.key +52 -0
  108. mcp_proxy_adapter/examples/certs/user_cert.pem +21 -0
  109. mcp_proxy_adapter/examples/certs/user_key.pem +28 -0
  110. mcp_proxy_adapter/examples/client_configs/api_key_client.json +13 -0
  111. mcp_proxy_adapter/examples/client_configs/basic_auth_client.json +13 -0
  112. mcp_proxy_adapter/examples/client_configs/certificate_client.json +22 -0
  113. mcp_proxy_adapter/examples/client_configs/jwt_client.json +15 -0
  114. mcp_proxy_adapter/examples/client_configs/no_auth_client.json +9 -0
  115. mcp_proxy_adapter/examples/commands/__init__.py +1 -0
  116. mcp_proxy_adapter/examples/create_certificates_simple.py +307 -0
  117. mcp_proxy_adapter/examples/debug_request_state.py +144 -0
  118. mcp_proxy_adapter/examples/debug_role_chain.py +205 -0
  119. mcp_proxy_adapter/examples/demo_client.py +341 -0
  120. mcp_proxy_adapter/examples/full_application/commands/custom_echo_command.py +99 -0
  121. mcp_proxy_adapter/examples/full_application/commands/dynamic_calculator_command.py +106 -0
  122. mcp_proxy_adapter/examples/full_application/configs/http_auth.json +37 -0
  123. mcp_proxy_adapter/examples/full_application/configs/http_simple.json +23 -0
  124. mcp_proxy_adapter/examples/full_application/configs/https_auth.json +39 -0
  125. mcp_proxy_adapter/examples/full_application/configs/https_simple.json +25 -0
  126. mcp_proxy_adapter/examples/full_application/configs/mtls_no_roles.json +39 -0
  127. mcp_proxy_adapter/examples/full_application/configs/mtls_with_roles.json +45 -0
  128. mcp_proxy_adapter/examples/full_application/hooks/application_hooks.py +97 -0
  129. mcp_proxy_adapter/examples/full_application/hooks/builtin_command_hooks.py +95 -0
  130. mcp_proxy_adapter/examples/full_application/main.py +138 -0
  131. mcp_proxy_adapter/examples/full_application/roles.json +21 -0
  132. mcp_proxy_adapter/examples/generate_all_certificates.py +429 -0
  133. mcp_proxy_adapter/examples/generate_certificates.py +121 -0
  134. mcp_proxy_adapter/examples/keys/ca_key.pem +28 -0
  135. mcp_proxy_adapter/examples/keys/mcp_proxy_adapter_ca_ca.key +28 -0
  136. mcp_proxy_adapter/examples/keys/test_ca_ca.key +28 -0
  137. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log +220 -0
  138. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log.1 +1 -0
  139. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log.2 +1 -0
  140. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log.3 +1 -0
  141. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log.4 +1 -0
  142. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter.log.5 +1 -0
  143. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log +220 -0
  144. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log.1 +1 -0
  145. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log.2 +1 -0
  146. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log.3 +1 -0
  147. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log.4 +1 -0
  148. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_access.log.5 +1 -0
  149. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log +2 -0
  150. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log.1 +1 -0
  151. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log.2 +1 -0
  152. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log.3 +1 -0
  153. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log.4 +1 -0
  154. mcp_proxy_adapter/examples/logs/mcp_proxy_adapter_error.log.5 +1 -0
  155. mcp_proxy_adapter/examples/proxy_registration_example.py +401 -0
  156. mcp_proxy_adapter/examples/roles.json +38 -0
  157. mcp_proxy_adapter/examples/run_example.py +81 -0
  158. mcp_proxy_adapter/examples/run_security_tests.py +326 -0
  159. mcp_proxy_adapter/examples/run_security_tests_fixed.py +300 -0
  160. mcp_proxy_adapter/examples/security_test_client.py +743 -0
  161. mcp_proxy_adapter/examples/server_configs/config_basic_http.json +204 -0
  162. mcp_proxy_adapter/examples/server_configs/config_http_token.json +238 -0
  163. mcp_proxy_adapter/examples/server_configs/config_https.json +215 -0
  164. mcp_proxy_adapter/examples/server_configs/config_https_token.json +231 -0
  165. mcp_proxy_adapter/examples/server_configs/config_mtls.json +215 -0
  166. mcp_proxy_adapter/examples/server_configs/config_proxy_registration.json +250 -0
  167. mcp_proxy_adapter/examples/server_configs/config_simple.json +46 -0
  168. mcp_proxy_adapter/examples/server_configs/roles.json +38 -0
  169. mcp_proxy_adapter/examples/test_examples.py +344 -0
  170. mcp_proxy_adapter/examples/universal_client.py +628 -0
  171. mcp_proxy_adapter/main.py +186 -0
  172. mcp_proxy_adapter/utils/config_generator.py +639 -0
  173. mcp_proxy_adapter/version.py +2 -1
  174. mcp_proxy_adapter-6.1.0.dist-info/METADATA +205 -0
  175. mcp_proxy_adapter-6.1.0.dist-info/RECORD +193 -0
  176. mcp_proxy_adapter-6.1.0.dist-info/entry_points.txt +2 -0
  177. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.1.0.dist-info}/licenses/LICENSE +2 -2
  178. mcp_proxy_adapter/api/middleware/auth.py +0 -146
  179. mcp_proxy_adapter/api/middleware/rate_limit.py +0 -152
  180. mcp_proxy_adapter/commands/reload_settings_command.py +0 -125
  181. mcp_proxy_adapter/examples/__init__.py +0 -7
  182. mcp_proxy_adapter/examples/basic_server/README.md +0 -60
  183. mcp_proxy_adapter/examples/basic_server/__init__.py +0 -7
  184. mcp_proxy_adapter/examples/basic_server/basic_custom_settings.json +0 -39
  185. mcp_proxy_adapter/examples/basic_server/config.json +0 -35
  186. mcp_proxy_adapter/examples/basic_server/custom_settings_example.py +0 -238
  187. mcp_proxy_adapter/examples/basic_server/server.py +0 -103
  188. mcp_proxy_adapter/examples/custom_commands/README.md +0 -127
  189. mcp_proxy_adapter/examples/custom_commands/__init__.py +0 -27
  190. mcp_proxy_adapter/examples/custom_commands/advanced_hooks.py +0 -250
  191. mcp_proxy_adapter/examples/custom_commands/auto_commands/__init__.py +0 -6
  192. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_echo_command.py +0 -103
  193. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_info_command.py +0 -111
  194. mcp_proxy_adapter/examples/custom_commands/config.json +0 -35
  195. mcp_proxy_adapter/examples/custom_commands/custom_health_command.py +0 -169
  196. mcp_proxy_adapter/examples/custom_commands/custom_help_command.py +0 -215
  197. mcp_proxy_adapter/examples/custom_commands/custom_openapi_generator.py +0 -76
  198. mcp_proxy_adapter/examples/custom_commands/custom_settings.json +0 -96
  199. mcp_proxy_adapter/examples/custom_commands/custom_settings_manager.py +0 -241
  200. mcp_proxy_adapter/examples/custom_commands/data_transform_command.py +0 -135
  201. mcp_proxy_adapter/examples/custom_commands/echo_command.py +0 -122
  202. mcp_proxy_adapter/examples/custom_commands/hooks.py +0 -230
  203. mcp_proxy_adapter/examples/custom_commands/intercept_command.py +0 -123
  204. mcp_proxy_adapter/examples/custom_commands/manual_echo_command.py +0 -103
  205. mcp_proxy_adapter/examples/custom_commands/server.py +0 -228
  206. mcp_proxy_adapter/examples/custom_commands/test_hooks.py +0 -176
  207. mcp_proxy_adapter/examples/deployment/README.md +0 -49
  208. mcp_proxy_adapter/examples/deployment/__init__.py +0 -7
  209. mcp_proxy_adapter/examples/deployment/config.development.json +0 -8
  210. mcp_proxy_adapter/examples/deployment/config.json +0 -29
  211. mcp_proxy_adapter/examples/deployment/config.production.json +0 -12
  212. mcp_proxy_adapter/examples/deployment/config.staging.json +0 -11
  213. mcp_proxy_adapter/examples/deployment/docker-compose.yml +0 -31
  214. mcp_proxy_adapter/examples/deployment/run.sh +0 -43
  215. mcp_proxy_adapter/examples/deployment/run_docker.sh +0 -84
  216. mcp_proxy_adapter/schemas/base_schema.json +0 -114
  217. mcp_proxy_adapter/schemas/openapi_schema.json +0 -314
  218. mcp_proxy_adapter/tests/__init__.py +0 -0
  219. mcp_proxy_adapter/tests/api/__init__.py +0 -3
  220. mcp_proxy_adapter/tests/api/test_cmd_endpoint.py +0 -115
  221. mcp_proxy_adapter/tests/api/test_custom_openapi.py +0 -617
  222. mcp_proxy_adapter/tests/api/test_handlers.py +0 -522
  223. mcp_proxy_adapter/tests/api/test_middleware.py +0 -340
  224. mcp_proxy_adapter/tests/api/test_schemas.py +0 -546
  225. mcp_proxy_adapter/tests/api/test_tool_integration.py +0 -531
  226. mcp_proxy_adapter/tests/commands/__init__.py +0 -3
  227. mcp_proxy_adapter/tests/commands/test_config_command.py +0 -211
  228. mcp_proxy_adapter/tests/commands/test_echo_command.py +0 -127
  229. mcp_proxy_adapter/tests/commands/test_help_command.py +0 -136
  230. mcp_proxy_adapter/tests/conftest.py +0 -131
  231. mcp_proxy_adapter/tests/functional/__init__.py +0 -3
  232. mcp_proxy_adapter/tests/functional/test_api.py +0 -253
  233. mcp_proxy_adapter/tests/integration/__init__.py +0 -3
  234. mcp_proxy_adapter/tests/integration/test_cmd_integration.py +0 -129
  235. mcp_proxy_adapter/tests/integration/test_integration.py +0 -255
  236. mcp_proxy_adapter/tests/performance/__init__.py +0 -3
  237. mcp_proxy_adapter/tests/performance/test_performance.py +0 -189
  238. mcp_proxy_adapter/tests/stubs/__init__.py +0 -10
  239. mcp_proxy_adapter/tests/stubs/echo_command.py +0 -104
  240. mcp_proxy_adapter/tests/test_api_endpoints.py +0 -271
  241. mcp_proxy_adapter/tests/test_api_handlers.py +0 -289
  242. mcp_proxy_adapter/tests/test_base_command.py +0 -123
  243. mcp_proxy_adapter/tests/test_batch_requests.py +0 -117
  244. mcp_proxy_adapter/tests/test_command_registry.py +0 -281
  245. mcp_proxy_adapter/tests/test_config.py +0 -127
  246. mcp_proxy_adapter/tests/test_utils.py +0 -65
  247. mcp_proxy_adapter/tests/unit/__init__.py +0 -3
  248. mcp_proxy_adapter/tests/unit/test_base_command.py +0 -436
  249. mcp_proxy_adapter/tests/unit/test_config.py +0 -217
  250. mcp_proxy_adapter-4.1.1.dist-info/METADATA +0 -200
  251. mcp_proxy_adapter-4.1.1.dist-info/RECORD +0 -110
  252. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.1.0.dist-info}/WHEEL +0 -0
  253. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.1.0.dist-info}/top_level.txt +0 -0
@@ -1,111 +0,0 @@
1
- """
2
- Auto-registered Info Command
3
-
4
- This command will be automatically discovered and registered by the framework.
5
- """
6
-
7
- from typing import Dict, Any, Optional
8
- from mcp_proxy_adapter.commands.base import Command
9
- from mcp_proxy_adapter.commands.result import CommandResult
10
-
11
-
12
- class AutoInfoResult(CommandResult):
13
- """
14
- Result of the auto-registered info command execution.
15
- """
16
-
17
- def __init__(self, info: Dict[str, Any], auto_registered: bool = True):
18
- """
19
- Initialize auto info command result.
20
-
21
- Args:
22
- info: Information data
23
- auto_registered: Flag indicating this was auto-registered
24
- """
25
- self.info = info
26
- self.auto_registered = auto_registered
27
-
28
- def to_dict(self) -> Dict[str, Any]:
29
- """
30
- Convert result to dictionary.
31
-
32
- Returns:
33
- Dict[str, Any]: Result as dictionary
34
- """
35
- return {
36
- "info": self.info,
37
- "auto_registered": self.auto_registered,
38
- "command_type": "auto_info"
39
- }
40
-
41
- @classmethod
42
- def get_schema(cls) -> Dict[str, Any]:
43
- """
44
- Get JSON schema for the result.
45
-
46
- Returns:
47
- Dict[str, Any]: JSON schema
48
- """
49
- return {
50
- "type": "object",
51
- "properties": {
52
- "info": {"type": "object"},
53
- "auto_registered": {"type": "boolean"},
54
- "command_type": {"type": "string"}
55
- }
56
- }
57
-
58
-
59
- class AutoInfoCommand(Command):
60
- """
61
- Auto-registered info command.
62
- """
63
-
64
- name = "auto_info"
65
- result_class = AutoInfoResult
66
-
67
- async def execute(self, topic: Optional[str] = None, **kwargs) -> AutoInfoResult:
68
- """
69
- Execute auto-registered info command.
70
-
71
- Args:
72
- topic: Information topic
73
- **kwargs: Additional parameters
74
-
75
- Returns:
76
- AutoInfoResult: Auto info command result
77
- """
78
- if topic is None:
79
- topic = "general"
80
-
81
- info_data = {
82
- "topic": topic,
83
- "auto_registered": True,
84
- "discovery_method": "automatic",
85
- "registration_time": "startup",
86
- "command_type": "auto_info"
87
- }
88
-
89
- return AutoInfoResult(
90
- info=info_data,
91
- auto_registered=True
92
- )
93
-
94
- @classmethod
95
- def get_schema(cls) -> Dict[str, Any]:
96
- """
97
- Get JSON schema for command parameters.
98
-
99
- Returns:
100
- Dict[str, Any]: JSON schema
101
- """
102
- return {
103
- "type": "object",
104
- "properties": {
105
- "topic": {
106
- "type": "string",
107
- "description": "Information topic",
108
- "default": "general"
109
- }
110
- }
111
- }
@@ -1,35 +0,0 @@
1
- {
2
- "server": {
3
- "host": "0.0.0.0",
4
- "port": 8000,
5
- "debug": true,
6
- "log_level": "DEBUG"
7
- },
8
- "logging": {
9
- "level": "DEBUG",
10
- "log_dir": "./logs/custom_commands",
11
- "log_file": "custom_commands.log",
12
- "error_log_file": "custom_commands_error.log",
13
- "access_log_file": "custom_commands_access.log",
14
- "max_file_size": "5MB",
15
- "backup_count": 3,
16
- "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
17
- "date_format": "%Y-%m-%d %H:%M:%S",
18
- "console_output": true,
19
- "file_output": true
20
- },
21
- "commands": {
22
- "auto_discovery": true,
23
- "discovery_path": "mcp_proxy_adapter.examples.custom_commands.auto_commands",
24
- "custom_commands_path": null
25
- },
26
- "custom": {
27
- "server_name": "Custom Commands MCP Proxy Server",
28
- "description": "Example server with custom commands",
29
- "features": {
30
- "hooks_enabled": true,
31
- "custom_commands_enabled": true,
32
- "advanced_logging": true
33
- }
34
- }
35
- }
@@ -1,169 +0,0 @@
1
- """
2
- Custom Health Command Example
3
-
4
- A custom health command with enhanced system information.
5
- """
6
-
7
- import os
8
- import platform
9
- import sys
10
- import psutil
11
- from datetime import datetime
12
- from typing import Dict, Any
13
-
14
- from mcp_proxy_adapter.commands.base import Command
15
- from mcp_proxy_adapter.commands.result import SuccessResult
16
- from mcp_proxy_adapter.commands.command_registry import registry
17
-
18
-
19
- class CustomHealthResult(SuccessResult):
20
- """
21
- Result of the custom health command execution.
22
- """
23
-
24
- def __init__(self, status: str, version: str, uptime: float,
25
- components: Dict[str, Any], custom_metrics: Dict[str, Any]):
26
- """
27
- Initialize custom health command result.
28
-
29
- Args:
30
- status: Server status ("ok" or "error")
31
- version: Server version
32
- uptime: Server uptime in seconds
33
- components: Dictionary with components status
34
- custom_metrics: Additional custom metrics
35
- """
36
- super().__init__(
37
- data={
38
- "status": status,
39
- "version": version,
40
- "uptime": uptime,
41
- "components": components,
42
- "custom_metrics": custom_metrics,
43
- "custom_health": True
44
- }
45
- )
46
-
47
- @classmethod
48
- def get_schema(cls) -> Dict[str, Any]:
49
- """
50
- Get JSON schema for result validation.
51
-
52
- Returns:
53
- Dict[str, Any]: JSON schema
54
- """
55
- return {
56
- "type": "object",
57
- "properties": {
58
- "data": {
59
- "type": "object",
60
- "properties": {
61
- "status": {"type": "string"},
62
- "version": {"type": "string"},
63
- "uptime": {"type": "number"},
64
- "components": {"type": "object"},
65
- "custom_metrics": {"type": "object"},
66
- "custom_health": {"type": "boolean"}
67
- },
68
- "required": ["status", "version", "uptime", "components", "custom_metrics", "custom_health"]
69
- }
70
- },
71
- "required": ["data"]
72
- }
73
-
74
-
75
- class CustomHealthCommand(Command):
76
- """
77
- Custom health command with enhanced system information.
78
- """
79
-
80
- name = "health"
81
- result_class = CustomHealthResult
82
-
83
- async def execute(self, **kwargs) -> CustomHealthResult:
84
- """
85
- Execute custom health command.
86
-
87
- Returns:
88
- CustomHealthResult: Custom health command result
89
- """
90
- # Get version from package
91
- try:
92
- from mcp_proxy_adapter.version import __version__ as version
93
- except ImportError:
94
- version = "unknown"
95
-
96
- # Get process start time
97
- process = psutil.Process(os.getpid())
98
- start_time = datetime.fromtimestamp(process.create_time())
99
- uptime_seconds = (datetime.now() - start_time).total_seconds()
100
-
101
- # Get system information
102
- system_info = {
103
- "platform": platform.platform(),
104
- "python_version": sys.version,
105
- "cpu_count": psutil.cpu_count(),
106
- "memory_total": psutil.virtual_memory().total,
107
- "memory_available": psutil.virtual_memory().available
108
- }
109
-
110
- # Get process information
111
- process_info = {
112
- "pid": os.getpid(),
113
- "memory_usage": process.memory_info().rss,
114
- "cpu_percent": process.cpu_percent(),
115
- "create_time": process.create_time()
116
- }
117
-
118
- # Get commands information
119
- from mcp_proxy_adapter.commands.command_registry import registry
120
- commands_info = {
121
- "total_commands": len(registry.get_all_commands()),
122
- "custom_commands": ["echo", "help", "health"],
123
- "registry_status": "active"
124
- }
125
-
126
- # Custom metrics
127
- custom_metrics = {
128
- "custom_health_check": True,
129
- "enhanced_monitoring": True,
130
- "system_load": psutil.getloadavg() if hasattr(psutil, 'getloadavg') else None,
131
- "disk_usage": {
132
- "total": psutil.disk_usage('/').total,
133
- "free": psutil.disk_usage('/').free,
134
- "used": psutil.disk_usage('/').used
135
- },
136
- "network_interfaces": len(psutil.net_if_addrs()),
137
- "custom_features": ["enhanced_metrics", "system_monitoring", "custom_health"],
138
- "hook_enhanced": kwargs.get("hook_enhanced", False),
139
- "health_check_id": kwargs.get("health_check_id"),
140
- "global_hook_processed": kwargs.get("global_hook_processed", False)
141
- }
142
-
143
- components = {
144
- "system": system_info,
145
- "process": process_info,
146
- "commands": commands_info
147
- }
148
-
149
- return CustomHealthResult(
150
- status="ok",
151
- version=version,
152
- uptime=uptime_seconds,
153
- components=components,
154
- custom_metrics=custom_metrics
155
- )
156
-
157
- @classmethod
158
- def get_schema(cls) -> Dict[str, Any]:
159
- """
160
- Get JSON schema for command parameters.
161
-
162
- Returns:
163
- Dict[str, Any]: JSON schema
164
- """
165
- return {
166
- "type": "object",
167
- "properties": {},
168
- "description": "Get enhanced system health information"
169
- }
@@ -1,215 +0,0 @@
1
- """
2
- Custom Help Command Example
3
-
4
- A custom help command that provides enhanced help information.
5
- """
6
-
7
- from typing import Dict, Any, Optional
8
- from mcp_proxy_adapter.commands.base import Command
9
- from mcp_proxy_adapter.commands.result import CommandResult
10
- from mcp_proxy_adapter.commands.command_registry import registry
11
- from mcp_proxy_adapter.core.errors import NotFoundError
12
-
13
-
14
- class CustomHelpResult(CommandResult):
15
- """
16
- Result of the custom help command execution.
17
- """
18
-
19
- def __init__(self, commands_info: Optional[Dict[str, Any]] = None,
20
- command_info: Optional[Dict[str, Any]] = None,
21
- custom_info: Optional[Dict[str, Any]] = None):
22
- """
23
- Initialize custom help command result.
24
-
25
- Args:
26
- commands_info: Information about all commands
27
- command_info: Information about a specific command
28
- custom_info: Additional custom information
29
- """
30
- self.commands_info = commands_info
31
- self.command_info = command_info
32
- self.custom_info = custom_info or {}
33
-
34
- def to_dict(self) -> Dict[str, Any]:
35
- """
36
- Convert result to dictionary.
37
-
38
- Returns:
39
- Dict[str, Any]: Result as dictionary
40
- """
41
- if self.command_info is not None:
42
- return {
43
- "cmdname": self.command_info.get("name", "unknown"),
44
- "info": {
45
- "description": self.command_info.get("description", ""),
46
- "summary": self.command_info.get("summary", ""),
47
- "params": self.command_info.get("params", {}),
48
- "examples": self.command_info.get("examples", []),
49
- "custom_help": True
50
- }
51
- }
52
-
53
- if self.commands_info is None:
54
- return {
55
- "tool_info": {
56
- "name": "Custom MCP-Proxy API Service",
57
- "description": "Enhanced JSON-RPC API with custom commands",
58
- "version": "2.0.0",
59
- "custom_help": True
60
- },
61
- "help_usage": {
62
- "description": "Get enhanced information about commands",
63
- "examples": [
64
- {"command": "help", "description": "List of all available commands"},
65
- {"command": "help", "params": {"cmdname": "command_name"}, "description": "Get detailed information about a specific command"}
66
- ]
67
- },
68
- "commands": {},
69
- "total": 0,
70
- "custom_features": self.custom_info
71
- }
72
-
73
- result = self.commands_info.copy()
74
- commands = result.get("commands", {})
75
- result["total"] = len(commands)
76
- result["custom_help"] = True
77
- result["custom_features"] = self.custom_info
78
-
79
- return result
80
-
81
- @classmethod
82
- def get_schema(cls) -> Dict[str, Any]:
83
- """
84
- Get JSON schema for the result.
85
-
86
- Returns:
87
- Dict[str, Any]: JSON schema
88
- """
89
- return {
90
- "type": "object",
91
- "properties": {
92
- "cmdname": {
93
- "type": "string",
94
- "description": "Name of the command"
95
- },
96
- "info": {
97
- "type": "object",
98
- "properties": {
99
- "description": {"type": "string"},
100
- "summary": {"type": "string"},
101
- "params": {"type": "object"},
102
- "examples": {"type": "array"},
103
- "custom_help": {"type": "boolean"}
104
- }
105
- },
106
- "tool_info": {
107
- "type": "object",
108
- "properties": {
109
- "name": {"type": "string"},
110
- "description": {"type": "string"},
111
- "version": {"type": "string"},
112
- "custom_help": {"type": "boolean"}
113
- }
114
- },
115
- "help_usage": {
116
- "type": "object",
117
- "properties": {
118
- "description": {"type": "string"},
119
- "examples": {"type": "array"}
120
- }
121
- },
122
- "commands": {"type": "object"},
123
- "total": {"type": "integer"},
124
- "custom_features": {"type": "object"}
125
- }
126
- }
127
-
128
-
129
- class CustomHelpCommand(Command):
130
- """
131
- Custom help command with enhanced functionality.
132
- """
133
-
134
- name = "help"
135
- result_class = CustomHelpResult
136
-
137
- async def execute(self, cmdname: Optional[str] = None, **kwargs) -> CustomHelpResult:
138
- """
139
- Execute custom help command.
140
-
141
- Args:
142
- cmdname: Name of specific command to get help for
143
- **kwargs: Additional parameters
144
-
145
- Returns:
146
- CustomHelpResult: Custom help command result
147
- """
148
- try:
149
- # Check if hook processed this request
150
- request_id = kwargs.get("request_id")
151
- hook_processed = kwargs.get("hook_processed", False)
152
-
153
- if cmdname is not None:
154
- # Get specific command info
155
- command_info = registry.get_command_info(cmdname)
156
- if command_info is None:
157
- raise NotFoundError(f"Command '{cmdname}' not found")
158
-
159
- custom_info = {
160
- "enhanced": True,
161
- "command_specific": True,
162
- "request_id": request_id,
163
- "hook_processed": hook_processed
164
- }
165
-
166
- return CustomHelpResult(
167
- command_info=command_info,
168
- custom_info=custom_info
169
- )
170
- else:
171
- # Get all commands info
172
- commands_info = registry.get_all_commands_info()
173
-
174
- custom_info = {
175
- "enhanced": True,
176
- "total_commands": len(commands_info.get("commands", {})),
177
- "custom_commands": ["echo", "help", "health"],
178
- "request_id": request_id,
179
- "hook_processed": hook_processed
180
- }
181
-
182
- return CustomHelpResult(
183
- commands_info=commands_info,
184
- custom_info=custom_info
185
- )
186
-
187
- except Exception as e:
188
- # Return error result
189
- return CustomHelpResult(
190
- custom_info={
191
- "error": str(e),
192
- "enhanced": True,
193
- "error_handling": True,
194
- "request_id": request_id,
195
- "hook_processed": hook_processed
196
- }
197
- )
198
-
199
- @classmethod
200
- def get_schema(cls) -> Dict[str, Any]:
201
- """
202
- Get JSON schema for command parameters.
203
-
204
- Returns:
205
- Dict[str, Any]: JSON schema
206
- """
207
- return {
208
- "type": "object",
209
- "properties": {
210
- "cmdname": {
211
- "type": "string",
212
- "description": "Name of specific command to get help for"
213
- }
214
- }
215
- }
@@ -1,76 +0,0 @@
1
- """
2
- Custom OpenAPI generator example for the extended server.
3
-
4
- This demonstrates how to create a custom OpenAPI schema generator
5
- that can be registered with the framework.
6
- """
7
-
8
- from typing import Dict, Any
9
- from fastapi import FastAPI
10
-
11
- from mcp_proxy_adapter.custom_openapi import register_openapi_generator, CustomOpenAPIGenerator
12
- from mcp_proxy_adapter.core.logging import logger
13
-
14
-
15
- def custom_openapi_generator(app: FastAPI) -> Dict[str, Any]:
16
- """
17
- Custom OpenAPI generator for the extended server example.
18
-
19
- This generator extends the default generator with additional
20
- information specific to the extended server.
21
-
22
- Args:
23
- app: FastAPI application instance.
24
-
25
- Returns:
26
- Custom OpenAPI schema.
27
- """
28
- # Use the default generator as base
29
- default_generator = CustomOpenAPIGenerator()
30
- schema = default_generator.generate(
31
- title=getattr(app, 'title', None),
32
- description=getattr(app, 'description', None),
33
- version=getattr(app, 'version', None)
34
- )
35
-
36
- # Add custom information to the schema
37
- schema["info"]["description"] += "\n\n## Extended Server Features:\n"
38
- schema["info"]["description"] += "- Custom commands with hooks\n"
39
- schema["info"]["description"] += "- Data transformation hooks\n"
40
- schema["info"]["description"] += "- Command interception hooks\n"
41
- schema["info"]["description"] += "- Auto-registration and manual registration examples\n"
42
-
43
- # Add custom tags
44
- if "tags" not in schema:
45
- schema["tags"] = []
46
-
47
- schema["tags"].extend([
48
- {
49
- "name": "custom-commands",
50
- "description": "Custom commands with advanced features"
51
- },
52
- {
53
- "name": "hooks",
54
- "description": "Command hooks for data transformation and interception"
55
- },
56
- {
57
- "name": "registration",
58
- "description": "Command registration examples"
59
- }
60
- ])
61
-
62
- # Add custom server information
63
- if "servers" not in schema:
64
- schema["servers"] = []
65
-
66
- schema["servers"].append({
67
- "url": "http://localhost:8000",
68
- "description": "Extended server with custom features"
69
- })
70
-
71
- logger.info("Generated custom OpenAPI schema for extended server")
72
- return schema
73
-
74
-
75
- # Register the custom generator
76
- register_openapi_generator("extended_server", custom_openapi_generator)