mcp-proxy-adapter 4.1.1__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 (200) hide show
  1. mcp_proxy_adapter/__main__.py +32 -0
  2. mcp_proxy_adapter/api/app.py +290 -33
  3. mcp_proxy_adapter/api/handlers.py +32 -6
  4. mcp_proxy_adapter/api/middleware/__init__.py +38 -32
  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 +201 -0
  10. mcp_proxy_adapter/api/middleware/transport_middleware.py +122 -0
  11. mcp_proxy_adapter/api/middleware/unified_security.py +197 -0
  12. mcp_proxy_adapter/api/middleware/user_info_middleware.py +158 -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 +8 -1
  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 +366 -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 +394 -14
  41. mcp_proxy_adapter/core/app_factory.py +410 -0
  42. mcp_proxy_adapter/core/app_runner.py +272 -0
  43. mcp_proxy_adapter/core/auth_validator.py +606 -0
  44. mcp_proxy_adapter/core/certificate_utils.py +1045 -0
  45. mcp_proxy_adapter/core/client.py +574 -0
  46. mcp_proxy_adapter/core/client_manager.py +284 -0
  47. mcp_proxy_adapter/core/client_security.py +384 -0
  48. mcp_proxy_adapter/core/config_converter.py +405 -0
  49. mcp_proxy_adapter/core/config_validator.py +218 -0
  50. mcp_proxy_adapter/core/logging.py +19 -3
  51. mcp_proxy_adapter/core/mtls_asgi.py +156 -0
  52. mcp_proxy_adapter/core/mtls_asgi_app.py +187 -0
  53. mcp_proxy_adapter/core/protocol_manager.py +385 -0
  54. mcp_proxy_adapter/core/proxy_client.py +602 -0
  55. mcp_proxy_adapter/core/proxy_registration.py +522 -0
  56. mcp_proxy_adapter/core/role_utils.py +426 -0
  57. mcp_proxy_adapter/core/security_adapter.py +370 -0
  58. mcp_proxy_adapter/core/security_factory.py +239 -0
  59. mcp_proxy_adapter/core/security_integration.py +286 -0
  60. mcp_proxy_adapter/core/server_adapter.py +282 -0
  61. mcp_proxy_adapter/core/server_engine.py +270 -0
  62. mcp_proxy_adapter/core/settings.py +1 -0
  63. mcp_proxy_adapter/core/ssl_utils.py +234 -0
  64. mcp_proxy_adapter/core/transport_manager.py +292 -0
  65. mcp_proxy_adapter/core/unified_config_adapter.py +579 -0
  66. mcp_proxy_adapter/custom_openapi.py +22 -11
  67. mcp_proxy_adapter/examples/__init__.py +13 -4
  68. mcp_proxy_adapter/examples/basic_framework/__init__.py +9 -0
  69. mcp_proxy_adapter/examples/basic_framework/commands/__init__.py +4 -0
  70. mcp_proxy_adapter/examples/basic_framework/hooks/__init__.py +4 -0
  71. mcp_proxy_adapter/examples/basic_framework/main.py +44 -0
  72. mcp_proxy_adapter/examples/commands/__init__.py +5 -0
  73. mcp_proxy_adapter/examples/create_certificates_simple.py +550 -0
  74. mcp_proxy_adapter/examples/debug_request_state.py +112 -0
  75. mcp_proxy_adapter/examples/debug_role_chain.py +158 -0
  76. mcp_proxy_adapter/examples/demo_client.py +275 -0
  77. mcp_proxy_adapter/examples/examples/basic_framework/__init__.py +9 -0
  78. mcp_proxy_adapter/examples/examples/basic_framework/commands/__init__.py +4 -0
  79. mcp_proxy_adapter/examples/examples/basic_framework/hooks/__init__.py +4 -0
  80. mcp_proxy_adapter/examples/examples/basic_framework/main.py +44 -0
  81. mcp_proxy_adapter/examples/examples/full_application/__init__.py +12 -0
  82. mcp_proxy_adapter/examples/examples/full_application/commands/__init__.py +7 -0
  83. mcp_proxy_adapter/examples/examples/full_application/commands/custom_echo_command.py +80 -0
  84. mcp_proxy_adapter/examples/examples/full_application/commands/dynamic_calculator_command.py +90 -0
  85. mcp_proxy_adapter/examples/examples/full_application/hooks/__init__.py +7 -0
  86. mcp_proxy_adapter/examples/examples/full_application/hooks/application_hooks.py +75 -0
  87. mcp_proxy_adapter/examples/examples/full_application/hooks/builtin_command_hooks.py +71 -0
  88. mcp_proxy_adapter/examples/examples/full_application/main.py +173 -0
  89. mcp_proxy_adapter/examples/examples/full_application/proxy_endpoints.py +154 -0
  90. mcp_proxy_adapter/examples/full_application/__init__.py +12 -0
  91. mcp_proxy_adapter/examples/full_application/commands/__init__.py +7 -0
  92. mcp_proxy_adapter/examples/full_application/commands/custom_echo_command.py +80 -0
  93. mcp_proxy_adapter/examples/full_application/commands/dynamic_calculator_command.py +90 -0
  94. mcp_proxy_adapter/examples/full_application/hooks/__init__.py +7 -0
  95. mcp_proxy_adapter/examples/full_application/hooks/application_hooks.py +75 -0
  96. mcp_proxy_adapter/examples/full_application/hooks/builtin_command_hooks.py +71 -0
  97. mcp_proxy_adapter/examples/full_application/main.py +173 -0
  98. mcp_proxy_adapter/examples/full_application/proxy_endpoints.py +154 -0
  99. mcp_proxy_adapter/examples/generate_all_certificates.py +362 -0
  100. mcp_proxy_adapter/examples/generate_certificates.py +177 -0
  101. mcp_proxy_adapter/examples/generate_certificates_and_tokens.py +369 -0
  102. mcp_proxy_adapter/examples/generate_test_configs.py +331 -0
  103. mcp_proxy_adapter/examples/proxy_registration_example.py +334 -0
  104. mcp_proxy_adapter/examples/run_example.py +59 -0
  105. mcp_proxy_adapter/examples/run_full_test_suite.py +318 -0
  106. mcp_proxy_adapter/examples/run_proxy_server.py +146 -0
  107. mcp_proxy_adapter/examples/run_security_tests.py +544 -0
  108. mcp_proxy_adapter/examples/run_security_tests_fixed.py +247 -0
  109. mcp_proxy_adapter/examples/scripts/config_generator.py +740 -0
  110. mcp_proxy_adapter/examples/scripts/create_certificates_simple.py +560 -0
  111. mcp_proxy_adapter/examples/scripts/generate_certificates_and_tokens.py +369 -0
  112. mcp_proxy_adapter/examples/security_test_client.py +782 -0
  113. mcp_proxy_adapter/examples/setup_test_environment.py +328 -0
  114. mcp_proxy_adapter/examples/test_config.py +148 -0
  115. mcp_proxy_adapter/examples/test_config_generator.py +86 -0
  116. mcp_proxy_adapter/examples/test_examples.py +281 -0
  117. mcp_proxy_adapter/examples/universal_client.py +620 -0
  118. mcp_proxy_adapter/main.py +93 -0
  119. mcp_proxy_adapter/utils/config_generator.py +1008 -0
  120. mcp_proxy_adapter/version.py +5 -2
  121. mcp_proxy_adapter-6.0.1.dist-info/METADATA +679 -0
  122. mcp_proxy_adapter-6.0.1.dist-info/RECORD +140 -0
  123. mcp_proxy_adapter-6.0.1.dist-info/entry_points.txt +2 -0
  124. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/licenses/LICENSE +2 -2
  125. mcp_proxy_adapter/api/middleware/auth.py +0 -146
  126. mcp_proxy_adapter/api/middleware/rate_limit.py +0 -152
  127. mcp_proxy_adapter/commands/reload_settings_command.py +0 -125
  128. mcp_proxy_adapter/examples/README.md +0 -124
  129. mcp_proxy_adapter/examples/basic_server/README.md +0 -60
  130. mcp_proxy_adapter/examples/basic_server/__init__.py +0 -7
  131. mcp_proxy_adapter/examples/basic_server/basic_custom_settings.json +0 -39
  132. mcp_proxy_adapter/examples/basic_server/config.json +0 -35
  133. mcp_proxy_adapter/examples/basic_server/custom_settings_example.py +0 -238
  134. mcp_proxy_adapter/examples/basic_server/server.py +0 -103
  135. mcp_proxy_adapter/examples/custom_commands/README.md +0 -127
  136. mcp_proxy_adapter/examples/custom_commands/__init__.py +0 -27
  137. mcp_proxy_adapter/examples/custom_commands/advanced_hooks.py +0 -250
  138. mcp_proxy_adapter/examples/custom_commands/auto_commands/__init__.py +0 -6
  139. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_echo_command.py +0 -103
  140. mcp_proxy_adapter/examples/custom_commands/auto_commands/auto_info_command.py +0 -111
  141. mcp_proxy_adapter/examples/custom_commands/config.json +0 -35
  142. mcp_proxy_adapter/examples/custom_commands/custom_health_command.py +0 -169
  143. mcp_proxy_adapter/examples/custom_commands/custom_help_command.py +0 -215
  144. mcp_proxy_adapter/examples/custom_commands/custom_openapi_generator.py +0 -76
  145. mcp_proxy_adapter/examples/custom_commands/custom_settings.json +0 -96
  146. mcp_proxy_adapter/examples/custom_commands/custom_settings_manager.py +0 -241
  147. mcp_proxy_adapter/examples/custom_commands/data_transform_command.py +0 -135
  148. mcp_proxy_adapter/examples/custom_commands/echo_command.py +0 -122
  149. mcp_proxy_adapter/examples/custom_commands/hooks.py +0 -230
  150. mcp_proxy_adapter/examples/custom_commands/intercept_command.py +0 -123
  151. mcp_proxy_adapter/examples/custom_commands/manual_echo_command.py +0 -103
  152. mcp_proxy_adapter/examples/custom_commands/server.py +0 -228
  153. mcp_proxy_adapter/examples/custom_commands/test_hooks.py +0 -176
  154. mcp_proxy_adapter/examples/deployment/README.md +0 -49
  155. mcp_proxy_adapter/examples/deployment/__init__.py +0 -7
  156. mcp_proxy_adapter/examples/deployment/config.development.json +0 -8
  157. mcp_proxy_adapter/examples/deployment/config.json +0 -29
  158. mcp_proxy_adapter/examples/deployment/config.production.json +0 -12
  159. mcp_proxy_adapter/examples/deployment/config.staging.json +0 -11
  160. mcp_proxy_adapter/examples/deployment/docker-compose.yml +0 -31
  161. mcp_proxy_adapter/examples/deployment/run.sh +0 -43
  162. mcp_proxy_adapter/examples/deployment/run_docker.sh +0 -84
  163. mcp_proxy_adapter/schemas/base_schema.json +0 -114
  164. mcp_proxy_adapter/schemas/openapi_schema.json +0 -314
  165. mcp_proxy_adapter/tests/__init__.py +0 -0
  166. mcp_proxy_adapter/tests/api/__init__.py +0 -3
  167. mcp_proxy_adapter/tests/api/test_cmd_endpoint.py +0 -115
  168. mcp_proxy_adapter/tests/api/test_custom_openapi.py +0 -617
  169. mcp_proxy_adapter/tests/api/test_handlers.py +0 -522
  170. mcp_proxy_adapter/tests/api/test_middleware.py +0 -340
  171. mcp_proxy_adapter/tests/api/test_schemas.py +0 -546
  172. mcp_proxy_adapter/tests/api/test_tool_integration.py +0 -531
  173. mcp_proxy_adapter/tests/commands/__init__.py +0 -3
  174. mcp_proxy_adapter/tests/commands/test_config_command.py +0 -211
  175. mcp_proxy_adapter/tests/commands/test_echo_command.py +0 -127
  176. mcp_proxy_adapter/tests/commands/test_help_command.py +0 -136
  177. mcp_proxy_adapter/tests/conftest.py +0 -131
  178. mcp_proxy_adapter/tests/functional/__init__.py +0 -3
  179. mcp_proxy_adapter/tests/functional/test_api.py +0 -253
  180. mcp_proxy_adapter/tests/integration/__init__.py +0 -3
  181. mcp_proxy_adapter/tests/integration/test_cmd_integration.py +0 -129
  182. mcp_proxy_adapter/tests/integration/test_integration.py +0 -255
  183. mcp_proxy_adapter/tests/performance/__init__.py +0 -3
  184. mcp_proxy_adapter/tests/performance/test_performance.py +0 -189
  185. mcp_proxy_adapter/tests/stubs/__init__.py +0 -10
  186. mcp_proxy_adapter/tests/stubs/echo_command.py +0 -104
  187. mcp_proxy_adapter/tests/test_api_endpoints.py +0 -271
  188. mcp_proxy_adapter/tests/test_api_handlers.py +0 -289
  189. mcp_proxy_adapter/tests/test_base_command.py +0 -123
  190. mcp_proxy_adapter/tests/test_batch_requests.py +0 -117
  191. mcp_proxy_adapter/tests/test_command_registry.py +0 -281
  192. mcp_proxy_adapter/tests/test_config.py +0 -127
  193. mcp_proxy_adapter/tests/test_utils.py +0 -65
  194. mcp_proxy_adapter/tests/unit/__init__.py +0 -3
  195. mcp_proxy_adapter/tests/unit/test_base_command.py +0 -436
  196. mcp_proxy_adapter/tests/unit/test_config.py +0 -217
  197. mcp_proxy_adapter-4.1.1.dist-info/METADATA +0 -200
  198. mcp_proxy_adapter-4.1.1.dist-info/RECORD +0 -110
  199. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/WHEEL +0 -0
  200. {mcp_proxy_adapter-4.1.1.dist-info → mcp_proxy_adapter-6.0.1.dist-info}/top_level.txt +0 -0
@@ -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
- }