mcp-proxy-adapter 6.0.0__py3-none-any.whl → 6.0.1__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 (212) hide show
  1. mcp_proxy_adapter/__main__.py +27 -7
  2. mcp_proxy_adapter/api/app.py +209 -79
  3. mcp_proxy_adapter/api/handlers.py +16 -5
  4. mcp_proxy_adapter/api/middleware/__init__.py +14 -9
  5. mcp_proxy_adapter/api/middleware/command_permission_middleware.py +148 -0
  6. mcp_proxy_adapter/api/middleware/factory.py +36 -12
  7. mcp_proxy_adapter/api/middleware/protocol_middleware.py +84 -18
  8. mcp_proxy_adapter/api/middleware/unified_security.py +197 -0
  9. mcp_proxy_adapter/api/middleware/user_info_middleware.py +158 -0
  10. mcp_proxy_adapter/commands/__init__.py +7 -1
  11. mcp_proxy_adapter/commands/base.py +7 -4
  12. mcp_proxy_adapter/commands/builtin_commands.py +8 -2
  13. mcp_proxy_adapter/commands/command_registry.py +8 -0
  14. mcp_proxy_adapter/commands/echo_command.py +81 -0
  15. mcp_proxy_adapter/commands/health_command.py +1 -1
  16. mcp_proxy_adapter/commands/help_command.py +21 -14
  17. mcp_proxy_adapter/commands/proxy_registration_command.py +326 -185
  18. mcp_proxy_adapter/commands/role_test_command.py +141 -0
  19. mcp_proxy_adapter/commands/security_command.py +488 -0
  20. mcp_proxy_adapter/commands/ssl_setup_command.py +234 -351
  21. mcp_proxy_adapter/commands/token_management_command.py +1 -1
  22. mcp_proxy_adapter/config.py +323 -40
  23. mcp_proxy_adapter/core/app_factory.py +410 -0
  24. mcp_proxy_adapter/core/app_runner.py +272 -0
  25. mcp_proxy_adapter/core/certificate_utils.py +291 -73
  26. mcp_proxy_adapter/core/client.py +574 -0
  27. mcp_proxy_adapter/core/client_manager.py +284 -0
  28. mcp_proxy_adapter/core/client_security.py +384 -0
  29. mcp_proxy_adapter/core/logging.py +8 -3
  30. mcp_proxy_adapter/core/mtls_asgi.py +156 -0
  31. mcp_proxy_adapter/core/mtls_asgi_app.py +187 -0
  32. mcp_proxy_adapter/core/protocol_manager.py +169 -10
  33. mcp_proxy_adapter/core/proxy_client.py +602 -0
  34. mcp_proxy_adapter/core/proxy_registration.py +299 -47
  35. mcp_proxy_adapter/core/security_adapter.py +12 -15
  36. mcp_proxy_adapter/core/security_integration.py +286 -0
  37. mcp_proxy_adapter/core/server_adapter.py +282 -0
  38. mcp_proxy_adapter/core/server_engine.py +270 -0
  39. mcp_proxy_adapter/core/ssl_utils.py +13 -12
  40. mcp_proxy_adapter/core/transport_manager.py +5 -5
  41. mcp_proxy_adapter/core/unified_config_adapter.py +579 -0
  42. mcp_proxy_adapter/examples/__init__.py +13 -4
  43. mcp_proxy_adapter/examples/basic_framework/__init__.py +9 -0
  44. mcp_proxy_adapter/examples/basic_framework/commands/__init__.py +4 -0
  45. mcp_proxy_adapter/examples/basic_framework/hooks/__init__.py +4 -0
  46. mcp_proxy_adapter/examples/basic_framework/main.py +44 -0
  47. mcp_proxy_adapter/examples/commands/__init__.py +5 -0
  48. mcp_proxy_adapter/examples/create_certificates_simple.py +550 -0
  49. mcp_proxy_adapter/examples/debug_request_state.py +112 -0
  50. mcp_proxy_adapter/examples/debug_role_chain.py +158 -0
  51. mcp_proxy_adapter/examples/demo_client.py +275 -0
  52. mcp_proxy_adapter/examples/examples/basic_framework/__init__.py +9 -0
  53. mcp_proxy_adapter/examples/examples/basic_framework/commands/__init__.py +4 -0
  54. mcp_proxy_adapter/examples/examples/basic_framework/hooks/__init__.py +4 -0
  55. mcp_proxy_adapter/examples/examples/basic_framework/main.py +44 -0
  56. mcp_proxy_adapter/examples/examples/full_application/__init__.py +12 -0
  57. mcp_proxy_adapter/examples/examples/full_application/commands/__init__.py +7 -0
  58. mcp_proxy_adapter/examples/examples/full_application/commands/custom_echo_command.py +80 -0
  59. mcp_proxy_adapter/examples/examples/full_application/commands/dynamic_calculator_command.py +90 -0
  60. mcp_proxy_adapter/examples/examples/full_application/hooks/__init__.py +7 -0
  61. mcp_proxy_adapter/examples/examples/full_application/hooks/application_hooks.py +75 -0
  62. mcp_proxy_adapter/examples/examples/full_application/hooks/builtin_command_hooks.py +71 -0
  63. mcp_proxy_adapter/examples/examples/full_application/main.py +173 -0
  64. mcp_proxy_adapter/examples/examples/full_application/proxy_endpoints.py +154 -0
  65. mcp_proxy_adapter/examples/full_application/__init__.py +12 -0
  66. mcp_proxy_adapter/examples/full_application/commands/__init__.py +7 -0
  67. mcp_proxy_adapter/examples/full_application/commands/custom_echo_command.py +80 -0
  68. mcp_proxy_adapter/examples/full_application/commands/dynamic_calculator_command.py +90 -0
  69. mcp_proxy_adapter/examples/full_application/hooks/__init__.py +7 -0
  70. mcp_proxy_adapter/examples/full_application/hooks/application_hooks.py +75 -0
  71. mcp_proxy_adapter/examples/full_application/hooks/builtin_command_hooks.py +71 -0
  72. mcp_proxy_adapter/examples/full_application/main.py +173 -0
  73. mcp_proxy_adapter/examples/full_application/proxy_endpoints.py +154 -0
  74. mcp_proxy_adapter/examples/generate_all_certificates.py +362 -0
  75. mcp_proxy_adapter/examples/generate_certificates.py +177 -0
  76. mcp_proxy_adapter/examples/generate_certificates_and_tokens.py +369 -0
  77. mcp_proxy_adapter/examples/generate_test_configs.py +331 -0
  78. mcp_proxy_adapter/examples/proxy_registration_example.py +334 -0
  79. mcp_proxy_adapter/examples/run_example.py +59 -0
  80. mcp_proxy_adapter/examples/run_full_test_suite.py +318 -0
  81. mcp_proxy_adapter/examples/run_proxy_server.py +146 -0
  82. mcp_proxy_adapter/examples/run_security_tests.py +544 -0
  83. mcp_proxy_adapter/examples/run_security_tests_fixed.py +247 -0
  84. mcp_proxy_adapter/examples/scripts/config_generator.py +740 -0
  85. mcp_proxy_adapter/examples/scripts/create_certificates_simple.py +560 -0
  86. mcp_proxy_adapter/examples/scripts/generate_certificates_and_tokens.py +369 -0
  87. mcp_proxy_adapter/examples/security_test_client.py +782 -0
  88. mcp_proxy_adapter/examples/setup_test_environment.py +328 -0
  89. mcp_proxy_adapter/examples/test_config.py +148 -0
  90. mcp_proxy_adapter/examples/test_config_generator.py +86 -0
  91. mcp_proxy_adapter/examples/test_examples.py +281 -0
  92. mcp_proxy_adapter/examples/universal_client.py +620 -0
  93. mcp_proxy_adapter/main.py +66 -148
  94. mcp_proxy_adapter/utils/config_generator.py +1008 -0
  95. mcp_proxy_adapter/version.py +5 -2
  96. mcp_proxy_adapter-6.0.1.dist-info/METADATA +679 -0
  97. mcp_proxy_adapter-6.0.1.dist-info/RECORD +140 -0
  98. mcp_proxy_adapter-6.0.1.dist-info/entry_points.txt +2 -0
  99. {mcp_proxy_adapter-6.0.0.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/licenses/LICENSE +2 -2
  100. mcp_proxy_adapter/api/middleware/auth.py +0 -146
  101. mcp_proxy_adapter/api/middleware/auth_adapter.py +0 -235
  102. mcp_proxy_adapter/api/middleware/mtls_adapter.py +0 -305
  103. mcp_proxy_adapter/api/middleware/mtls_middleware.py +0 -296
  104. mcp_proxy_adapter/api/middleware/rate_limit.py +0 -152
  105. mcp_proxy_adapter/api/middleware/rate_limit_adapter.py +0 -241
  106. mcp_proxy_adapter/api/middleware/roles_adapter.py +0 -365
  107. mcp_proxy_adapter/api/middleware/roles_middleware.py +0 -381
  108. mcp_proxy_adapter/api/middleware/security.py +0 -376
  109. mcp_proxy_adapter/api/middleware/token_auth_middleware.py +0 -261
  110. mcp_proxy_adapter/examples/README.md +0 -124
  111. mcp_proxy_adapter/examples/basic_server/README.md +0 -60
  112. mcp_proxy_adapter/examples/basic_server/__init__.py +0 -7
  113. mcp_proxy_adapter/examples/basic_server/basic_custom_settings.json +0 -39
  114. mcp_proxy_adapter/examples/basic_server/config.json +0 -70
  115. mcp_proxy_adapter/examples/basic_server/config_all_protocols.json +0 -54
  116. mcp_proxy_adapter/examples/basic_server/config_http.json +0 -70
  117. mcp_proxy_adapter/examples/basic_server/config_http_only.json +0 -52
  118. mcp_proxy_adapter/examples/basic_server/config_https.json +0 -58
  119. mcp_proxy_adapter/examples/basic_server/config_mtls.json +0 -58
  120. mcp_proxy_adapter/examples/basic_server/config_ssl.json +0 -46
  121. mcp_proxy_adapter/examples/basic_server/custom_settings_example.py +0 -238
  122. mcp_proxy_adapter/examples/basic_server/server.py +0 -114
  123. mcp_proxy_adapter/examples/custom_commands/README.md +0 -127
  124. mcp_proxy_adapter/examples/custom_commands/__init__.py +0 -27
  125. mcp_proxy_adapter/examples/custom_commands/advanced_hooks.py +0 -566
  126. mcp_proxy_adapter/examples/custom_commands/auto_commands/__init__.py +0 -6
  127. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_echo_command.py +0 -103
  128. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_info_command.py +0 -111
  129. mcp_proxy_adapter/examples/custom_commands/auto_commands/test_command.py +0 -105
  130. mcp_proxy_adapter/examples/custom_commands/catalog/commands/test_command.py +0 -129
  131. mcp_proxy_adapter/examples/custom_commands/config.json +0 -118
  132. mcp_proxy_adapter/examples/custom_commands/config_all_protocols.json +0 -46
  133. mcp_proxy_adapter/examples/custom_commands/config_https_only.json +0 -46
  134. mcp_proxy_adapter/examples/custom_commands/config_https_transport.json +0 -33
  135. mcp_proxy_adapter/examples/custom_commands/config_mtls_only.json +0 -46
  136. mcp_proxy_adapter/examples/custom_commands/config_mtls_transport.json +0 -33
  137. mcp_proxy_adapter/examples/custom_commands/config_single_transport.json +0 -33
  138. mcp_proxy_adapter/examples/custom_commands/custom_health_command.py +0 -169
  139. mcp_proxy_adapter/examples/custom_commands/custom_help_command.py +0 -215
  140. mcp_proxy_adapter/examples/custom_commands/custom_openapi_generator.py +0 -76
  141. mcp_proxy_adapter/examples/custom_commands/custom_settings.json +0 -96
  142. mcp_proxy_adapter/examples/custom_commands/custom_settings_manager.py +0 -241
  143. mcp_proxy_adapter/examples/custom_commands/data_transform_command.py +0 -135
  144. mcp_proxy_adapter/examples/custom_commands/echo_command.py +0 -122
  145. mcp_proxy_adapter/examples/custom_commands/full_help_response.json +0 -1
  146. mcp_proxy_adapter/examples/custom_commands/generated_openapi.json +0 -629
  147. mcp_proxy_adapter/examples/custom_commands/get_openapi.py +0 -103
  148. mcp_proxy_adapter/examples/custom_commands/hooks.py +0 -230
  149. mcp_proxy_adapter/examples/custom_commands/intercept_command.py +0 -123
  150. mcp_proxy_adapter/examples/custom_commands/loadable_commands/test_ignored.py +0 -129
  151. mcp_proxy_adapter/examples/custom_commands/manual_echo_command.py +0 -103
  152. mcp_proxy_adapter/examples/custom_commands/proxy_connection_manager.py +0 -278
  153. mcp_proxy_adapter/examples/custom_commands/server.py +0 -252
  154. mcp_proxy_adapter/examples/custom_commands/simple_openapi_server.py +0 -75
  155. mcp_proxy_adapter/examples/custom_commands/start_server_with_proxy_manager.py +0 -299
  156. mcp_proxy_adapter/examples/custom_commands/start_server_with_registration.py +0 -278
  157. mcp_proxy_adapter/examples/custom_commands/test_hooks.py +0 -176
  158. mcp_proxy_adapter/examples/custom_commands/test_openapi.py +0 -27
  159. mcp_proxy_adapter/examples/custom_commands/test_registry.py +0 -23
  160. mcp_proxy_adapter/examples/custom_commands/test_simple.py +0 -19
  161. mcp_proxy_adapter/examples/custom_project_example/README.md +0 -103
  162. mcp_proxy_adapter/examples/custom_project_example/README_EN.md +0 -103
  163. mcp_proxy_adapter/examples/deployment/README.md +0 -49
  164. mcp_proxy_adapter/examples/deployment/__init__.py +0 -7
  165. mcp_proxy_adapter/examples/deployment/config.development.json +0 -8
  166. mcp_proxy_adapter/examples/deployment/config.json +0 -29
  167. mcp_proxy_adapter/examples/deployment/config.production.json +0 -12
  168. mcp_proxy_adapter/examples/deployment/config.staging.json +0 -11
  169. mcp_proxy_adapter/examples/deployment/docker-compose.yml +0 -31
  170. mcp_proxy_adapter/examples/deployment/run.sh +0 -43
  171. mcp_proxy_adapter/examples/deployment/run_docker.sh +0 -84
  172. mcp_proxy_adapter/examples/simple_custom_commands/README.md +0 -149
  173. mcp_proxy_adapter/examples/simple_custom_commands/README_EN.md +0 -149
  174. mcp_proxy_adapter/schemas/base_schema.json +0 -114
  175. mcp_proxy_adapter/schemas/openapi_schema.json +0 -314
  176. mcp_proxy_adapter/schemas/roles_schema.json +0 -162
  177. mcp_proxy_adapter/tests/__init__.py +0 -0
  178. mcp_proxy_adapter/tests/api/__init__.py +0 -3
  179. mcp_proxy_adapter/tests/api/test_cmd_endpoint.py +0 -115
  180. mcp_proxy_adapter/tests/api/test_custom_openapi.py +0 -617
  181. mcp_proxy_adapter/tests/api/test_handlers.py +0 -522
  182. mcp_proxy_adapter/tests/api/test_middleware.py +0 -340
  183. mcp_proxy_adapter/tests/api/test_schemas.py +0 -546
  184. mcp_proxy_adapter/tests/api/test_tool_integration.py +0 -531
  185. mcp_proxy_adapter/tests/commands/__init__.py +0 -3
  186. mcp_proxy_adapter/tests/commands/test_config_command.py +0 -211
  187. mcp_proxy_adapter/tests/commands/test_echo_command.py +0 -127
  188. mcp_proxy_adapter/tests/commands/test_help_command.py +0 -136
  189. mcp_proxy_adapter/tests/conftest.py +0 -131
  190. mcp_proxy_adapter/tests/functional/__init__.py +0 -3
  191. mcp_proxy_adapter/tests/functional/test_api.py +0 -253
  192. mcp_proxy_adapter/tests/integration/__init__.py +0 -3
  193. mcp_proxy_adapter/tests/integration/test_cmd_integration.py +0 -129
  194. mcp_proxy_adapter/tests/integration/test_integration.py +0 -255
  195. mcp_proxy_adapter/tests/performance/__init__.py +0 -3
  196. mcp_proxy_adapter/tests/performance/test_performance.py +0 -189
  197. mcp_proxy_adapter/tests/stubs/__init__.py +0 -10
  198. mcp_proxy_adapter/tests/stubs/echo_command.py +0 -104
  199. mcp_proxy_adapter/tests/test_api_endpoints.py +0 -271
  200. mcp_proxy_adapter/tests/test_api_handlers.py +0 -289
  201. mcp_proxy_adapter/tests/test_base_command.py +0 -123
  202. mcp_proxy_adapter/tests/test_batch_requests.py +0 -117
  203. mcp_proxy_adapter/tests/test_command_registry.py +0 -281
  204. mcp_proxy_adapter/tests/test_config.py +0 -127
  205. mcp_proxy_adapter/tests/test_utils.py +0 -65
  206. mcp_proxy_adapter/tests/unit/__init__.py +0 -3
  207. mcp_proxy_adapter/tests/unit/test_base_command.py +0 -436
  208. mcp_proxy_adapter/tests/unit/test_config.py +0 -270
  209. mcp_proxy_adapter-6.0.0.dist-info/METADATA +0 -201
  210. mcp_proxy_adapter-6.0.0.dist-info/RECORD +0 -179
  211. {mcp_proxy_adapter-6.0.0.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/WHEEL +0 -0
  212. {mcp_proxy_adapter-6.0.0.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/top_level.txt +0 -0
@@ -1,33 +0,0 @@
1
- {
2
- "server": {
3
- "host": "0.0.0.0",
4
- "port": 8000,
5
- "debug": true,
6
- "log_level": "INFO"
7
- },
8
- "logging": {
9
- "level": "INFO",
10
- "log_dir": "./logs",
11
- "console_output": true
12
- },
13
- "commands": {
14
- "auto_discovery": true,
15
- "discovery_path": "mcp_proxy_adapter.commands"
16
- },
17
- "transport": {
18
- "type": "http",
19
- "port": null,
20
- "ssl": {
21
- "enabled": false,
22
- "cert_file": null,
23
- "key_file": null,
24
- "ca_cert": null,
25
- "verify_client": false,
26
- "client_cert_required": false
27
- }
28
- },
29
- "custom": {
30
- "description": "Single transport server example",
31
- "server_name": "Single Transport MCP Proxy Adapter Server"
32
- }
33
- }
@@ -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)
@@ -1,96 +0,0 @@
1
- {
2
- "application": {
3
- "name": "Extended MCP Proxy Server with Custom Settings",
4
- "version": "2.1.0",
5
- "environment": "development",
6
- "description": "Advanced server with custom settings management"
7
- },
8
- "features": {
9
- "advanced_hooks": true,
10
- "custom_commands": true,
11
- "data_transformation": true,
12
- "command_interception": true,
13
- "performance_monitoring": true,
14
- "custom_settings_manager": true
15
- },
16
- "security": {
17
- "enable_authentication": false,
18
- "max_request_size": "15MB",
19
- "rate_limiting": {
20
- "enabled": true,
21
- "requests_per_minute": 200,
22
- "burst_limit": 50
23
- },
24
- "cors": {
25
- "enabled": true,
26
- "allowed_origins": ["*"],
27
- "allowed_methods": ["GET", "POST", "PUT", "DELETE"]
28
- }
29
- },
30
- "monitoring": {
31
- "enable_metrics": true,
32
- "metrics_interval": 30,
33
- "health_check_interval": 15,
34
- "log_level": "DEBUG",
35
- "performance_tracking": {
36
- "enabled": true,
37
- "track_response_times": true,
38
- "track_memory_usage": true
39
- }
40
- },
41
- "custom_commands": {
42
- "auto_echo": {
43
- "enabled": true,
44
- "max_length": 2000,
45
- "enable_formatting": true
46
- },
47
- "data_transform": {
48
- "enabled": true,
49
- "transform_types": ["uppercase", "lowercase", "reverse", "capitalize", "title"],
50
- "enable_custom_transforms": true
51
- },
52
- "intercept": {
53
- "enabled": true,
54
- "bypass_conditions": ["input_value == 0", "input_value < 0"],
55
- "intercept_all_commands": false
56
- },
57
- "manual_echo": {
58
- "enabled": true,
59
- "description": "Manually registered echo command with custom settings"
60
- }
61
- },
62
- "hooks": {
63
- "data_transform": {
64
- "enabled": true,
65
- "transform_types": ["uppercase", "lowercase", "reverse"],
66
- "apply_to_all_commands": false
67
- },
68
- "intercept": {
69
- "enabled": true,
70
- "bypass_conditions": ["input_value == 0"],
71
- "log_interceptions": true
72
- }
73
- },
74
- "database": {
75
- "enabled": false,
76
- "type": "sqlite",
77
- "connection_string": "sqlite:///custom_server.db",
78
- "pool_size": 10,
79
- "max_overflow": 20
80
- },
81
- "cache": {
82
- "enabled": true,
83
- "type": "memory",
84
- "ttl": 300,
85
- "max_size": 1000
86
- },
87
- "api": {
88
- "version": "v1",
89
- "prefix": "/api/v1",
90
- "documentation": {
91
- "enabled": true,
92
- "title": "Extended MCP Proxy API",
93
- "description": "Advanced API with custom settings support"
94
- }
95
- }
96
- }