forked from kimhyungsik/ax_hub_mcp_tool
33 lines
1.6 KiB
Python
33 lines
1.6 KiB
Python
import os
|
|
|
|
# Fix ToolAuthorizationService
|
|
auth_path = r"C:\eGovFrameDev-4.3.1-64bit\workspace-egov\axhub-backend-main\axhub-gateway\src\main\java\io\shinhanlife\axhub\biz\mcp\gateway\security\ToolAuthorizationService.java"
|
|
with open(auth_path, "r", encoding="utf-8") as f:
|
|
auth_content = f.read()
|
|
|
|
auth_content = auth_content.replace("metadata.name()", "metadata.getName()")
|
|
auth_content = auth_content.replace("metadata.operationType()", "metadata.getOperationType()")
|
|
auth_content = auth_content.replace("metadata.requiresApproval()", "(metadata.getRequiresApproval() != null && metadata.getRequiresApproval())")
|
|
|
|
with open(auth_path, "w", encoding="utf-8") as f:
|
|
f.write(auth_content)
|
|
|
|
# Fix RedisToolTraceService
|
|
redis_path = r"C:\eGovFrameDev-4.3.1-64bit\workspace-egov\axhub-backend-main\axhub-gateway\src\main\java\io\shinhanlife\axhub\biz\mcp\gateway\redis\RedisToolTraceService.java"
|
|
with open(redis_path, "r", encoding="utf-8") as f:
|
|
redis_content = f.read()
|
|
|
|
redis_content = redis_content.replace("metadata.name()", "metadata.getName()")
|
|
redis_content = redis_content.replace("metadata.operationType()", "metadata.getOperationType()")
|
|
|
|
arg_names_old = 'trace.put("argumentNames", arguments.propertyNames());'
|
|
arg_names_new = ''' java.util.List<String> argNames = new java.util.ArrayList<>();
|
|
arguments.fieldNames().forEachRemaining(argNames::add);
|
|
trace.put("argumentNames", argNames);'''
|
|
redis_content = redis_content.replace(arg_names_old, arg_names_new)
|
|
|
|
with open(redis_path, "w", encoding="utf-8") as f:
|
|
f.write(redis_content)
|
|
|
|
print("Files fixed!")
|