115 lines
3.2 KiB
C++
115 lines
3.2 KiB
C++
#include "cadsdk.h"
|
|
#include "mcp_tool.h"
|
|
#include "mcp_utils.h"
|
|
|
|
namespace
|
|
{
|
|
bool ReplaceTemplateArgs(const std::string &_tpl, const nlohmann::json &args, std::wstring &outScript)
|
|
{
|
|
if (_tpl.empty()) return false;
|
|
|
|
std::string tpl(_tpl);
|
|
for (auto it = args.begin(); it != args.end(); ++it)
|
|
{
|
|
std::string placeholder = "{" + it.key() + "}";
|
|
std::string valueStr;
|
|
|
|
if (it.value().is_string())
|
|
{
|
|
valueStr = it.value().get<std::string>();
|
|
}
|
|
else if (it.value().is_boolean())
|
|
{
|
|
valueStr = it.value().get<bool>() ? "T" : "nil"; // 适配 AutoLISP 的布尔逻辑
|
|
}
|
|
else
|
|
{
|
|
valueStr = it.value().dump(); // 数字类型直接转字符串
|
|
}
|
|
|
|
size_t pos = 0;
|
|
while ((pos = tpl.find(placeholder, pos)) != std::string::npos)
|
|
{
|
|
tpl.replace(pos, placeholder.length(), valueStr);
|
|
pos += valueStr.length();
|
|
}
|
|
}
|
|
outScript = mcp_utils::utf8ToWString(tpl);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
#pragma region LispInline
|
|
nlohmann::json mcp_Tool_LispInline::Execute(const nlohmann::json &args)
|
|
{
|
|
std::string tpl = m_config["backend"].value("script_template", "");
|
|
std::wstring finalScript;
|
|
|
|
if (!ReplaceTemplateArgs(tpl, args, finalScript))
|
|
return mcp_Tool_Utility::make_error(-32603, "Script template replacement failed.");
|
|
|
|
finalScript += _T(" "); // 尾部加空格，等同于在 CAD 命令行敲回车
|
|
|
|
// 异步推送到 AutoCAD 命令行执行
|
|
acDocManager->sendStringToExecute(acDocManager->curDocument(), finalScript.c_str());
|
|
|
|
return mcp_Tool_Utility::make_text_result("LISP inline command issued for execution (asynchronous)");
|
|
}
|
|
#pragma endregion LispInline
|
|
|
|
#pragma region LispFile
|
|
nlohmann::json mcp_Tool_LispFile::Execute(const nlohmann::json &args)
|
|
{
|
|
fs::path filePath
|
|
= mcp_utils::relative2absolute(m_config["backend"].value("file_name", "")
|
|
, fs::path(m_lispDir));
|
|
|
|
if (!fs::exists(filePath))
|
|
{
|
|
return mcp_Tool_Utility::make_error(-32603, "LISP file does not exist: " + filePath.string());
|
|
}
|
|
|
|
std::string callTpl = m_config["backend"].value("call_template", "");
|
|
|
|
// 组装前置加载命令，末尾的 nil 是为了防止文件不存在时 LISP 引擎报错弹窗
|
|
std::wstring loadCmd = std::wstring(_T("(load \"")) + filePath.generic_wstring() + _T("\" nil) ");
|
|
std::wstring executeCmd;
|
|
if (!ReplaceTemplateArgs(callTpl, args, executeCmd))
|
|
{
|
|
return mcp_Tool_Utility::make_error(-32603, "Script template replacement failed.");
|
|
}
|
|
executeCmd += _T(" ");
|
|
|
|
std::wstring finalScript = loadCmd + executeCmd;
|
|
|
|
acDocManager->sendStringToExecute(acDocManager->curDocument(), finalScript.c_str());
|
|
|
|
return mcp_Tool_Utility::make_text_result("LISP external file loaded and issued for execution (asynchronous)");
|
|
}
|
|
#pragma endregion LispFile
|
|
|
|
#pragma region DllProxy
|
|
// RAII：基座销毁工具时，自动通知 DLL 释放内存，并卸载模块
|
|
mcp_Tool_DllProxy::~mcp_Tool_DllProxy()
|
|
{
|
|
if (m_pRealTool && m_destroyFunc)
|
|
{
|
|
m_destroyFunc(m_pRealTool);
|
|
}
|
|
if (m_hModule)
|
|
{
|
|
FreeLibrary(m_hModule);
|
|
}
|
|
}
|
|
|
|
nlohmann::json mcp_Tool_DllProxy::Execute(const nlohmann::json &args)
|
|
{
|
|
if (m_pRealTool)
|
|
{
|
|
return m_pRealTool->Execute(args); // 完美转发
|
|
}
|
|
return mcp_Tool_Utility::make_error(-32603, "Underlying DLL tool instance is invalid");
|
|
}
|
|
#pragma endregion DllProxy
|
|
|