基本通信测试通过。

2026-03-10 11:51:02 +08:00
commit b361eb3f88
15 changed files with 690 additions and 0 deletions
--- a/main.cpp
+++ b/main.cpp
@@ -0,0 +1,113 @@
+#include <CLI/CLI.hpp>
+#include <iostream>
+#include <string>
+#include <vector>
+#include <xlnt/xlnt.hpp>
+
+#include "bf.request.h"
+#include "bf.util.h"
+
+#ifdef _WIN32
+#include "Windows.h"
+#endif
+
+int main(int argc, char** argv)
+{
+#ifdef _WIN32
+    SetConsoleOutputCP(CP_UTF8);
+#endif
+    CLI::App app{"百度翻译xlsx批量工具。"};
+
+    std::string workbook;
+    std::string sheetName;
+    std::string from = "auto";
+    std::string to = "en";
+    bool testRequest = false;
+    std::vector<std::string> tasks;   // 存储多个字符串
+
+    app.add_option("-t,--task", tasks, "要处理的行列范围（起始行,起始列,结束行,结束列），如：1,1,10,10）")->multi_option_policy();
+    app.add_option("-s,--sheet", sheetName, "工作表名称");
+    app.add_option("-w,--workbook", workbook, "工作簿");
+    app.add_option("-f,--from", from, "源语言（默认auto）");
+    app.add_option("--to-language,--to", to, "目标语言（默认en）");
+    app.add_flag("--test-request", testRequest, "测试请求");
+
+    // 添加其他可选参数
+    std::string config = "baidu_fanyi.json";
+    app.add_option("-c,--config", config, "配置文件（默认baidu_fanyi.json）");
+
+    bool verbose = false;
+    app.add_flag("-v,--verbose", verbose, "详细输出");
+
+    // 解析命令行参数
+    CLI11_PARSE(app, argc, argv);
+
+    BF_Util::initLogger();
+    BF_Util::setLogLevel(spdlog::level::debug);
+
+    // 测试参数，这里直接返回。
+    // for (const auto& task : tasks) {
+    //     gLogger->info("任务：{}", task);
+    // }
+    // return 0;
+
+    BF_Interface* request = new BF_Request();
+    std::shared_ptr<void> del(nullptr, [request](void*) { delete request; });
+
+    std::shared_ptr<BF_Config> bc = std::make_shared<BF_Config>();
+    if (!bc->parseConfig(config)) {
+        gLogger->error("解析配置文件失败！");
+        return -1;
+    }
+    request->setConfig(bc);
+
+    if (testRequest) {
+        gLogger->info("测试请求开始...");
+        gLogger->info("测试内容：[你好，世界。] from auto to en.");
+        std::string readyData = request->doReady("你好，世界。", from, to);
+        std::string data = fmt::format("{}{}", bc->baseUrl, readyData);
+        std::string result = request->getResult(data);
+        gLogger->info("测试请求结束！");
+        gLogger->info("结果：[{}]", result);
+        return 0;
+    }
+
+    xlnt::workbook wb;
+    xlnt::worksheet ws;
+    try {
+        wb.load("");
+        ws = wb.sheet_by_title(sheetName);
+        std::vector<XlsxTask> tasks;
+        for (auto& task : tasks) {
+            for (int i = task.startRow; i <= task.endRow; i++) {
+                for (int j = task.startCol; j <= task.endCol; j++) {
+                    xlnt::cell cell = ws.cell(i, j);
+                    std::string text = cell.value<std::string>();
+                    if (text.empty()) {
+                        gLogger->error("数据为空，跳过:[{}][{}]", i, j);
+                        continue;
+                    }
+                    auto readyData = request->doReady(text, from, to);
+                    if (readyData.empty()) {
+                        gLogger->error("doReady数据为空，跳过:[{}][{}][{}]", i, j, text);
+                        continue;
+                    }
+                    auto data = fmt::format("{}{}", bc->baseUrl, readyData);
+                    std::string result = request->getResult(data);
+                    if (result.empty()) {
+                        gLogger->error("getResult数据为空，跳过:[{}][{}][{}]", i, j, text);
+                        continue;
+                    }
+                    gLogger->info("翻译成功:[{}][{}][{}][{}]", i, j, text, result);
+                    cell.value(result);
+                }
+            }
+        }
+    } catch (const std::exception& e) {
+        gLogger->error(e.what());
+    }
+
+    wb.save(fmt::format("{}_out.xlsx", workbook));
+    gLogger->info("结束！");
+    return 0;
+}