@rongchenglu/atc 1.0.6

package/index.js

@@ -0,0 +1,264 @@
+#!/usr/bin/env node --no-warnings
+
+import yaml from "js-yaml";
+import axios from "axios";
+import fs from "fs";
+import path from "path";
+import readline from "readline";
+import boxen from "boxen";
+import chalk from "chalk";
+
+function loadConfig() {
+  const configFile = path.join(process.cwd(), "atc-config.yaml");
+
+  if (!fs.existsSync(configFile)) {
+    console.log("错误：找不到atc-config.yaml文件");
+    process.exit(1);
+  }
+
+  try {
+    const fileContents = fs.readFileSync(configFile, "utf8");
+    const config = yaml.load(fileContents);
+
+    const requiredKeys = ["base_url", "ds_id", "apikey"];
+    for (const key of requiredKeys) {
+      if (!(key in config)) {
+        console.log(`错误：atc-config.yaml中缺少必需的配置项：${key}`);
+        process.exit(1);
+      }
+    }
+
+    // 处理可选的布尔配置项，设置默认值
+    if (config.omit_single_db !== undefined) {
+      config.omit_single_db = Boolean(config.omit_single_db);
+    }
+
+    if (config.camel_alias !== undefined) {
+      config.camel_alias = Boolean(config.camel_alias);
+    }
+
+    return config;
+  } catch (error) {
+    console.log(`错误：无法解析atc-config.yaml文件：${error.message}`);
+    process.exit(1);
+  }
+}
+
+async function fetchBackendVersion(baseUrl) {
+  try {
+    const response = await axios.get(`${baseUrl}/v1/version`, {
+      timeout: 5000,
+    });
+    return {
+      connected: true,
+      version: response.data.version || response.data,
+    };
+  } catch (error) {
+    return {
+      connected: false,
+      error: error.message,
+    };
+  }
+}
+
+function displayStartupScreen(backendInfo) {
+  const statusIcon = backendInfo.connected ? chalk.green("●") : chalk.red("●");
+
+  const caption = backendInfo.connected
+    ? `${statusIcon} Backend Version: ${chalk.cyan(
+        typeof backendInfo.version === "string"
+          ? backendInfo.version
+          : JSON.stringify(backendInfo.version)
+      )}`
+    : `${statusIcon} ${chalk.red(
+        "Backend connection failed: \n" + backendInfo.error
+      )}`;
+
+  console.log(
+    boxen("AskTable Code\n\n" + caption, {
+      padding: 1,
+      margin: 1,
+      borderStyle: "round",
+      borderColor: "cyan",
+      textAlignment: "center",
+    })
+  );
+}
+
+function displayConfigInfo(config) {
+  const configLines = [
+    chalk.bold.cyan("当前配置信息"),
+    "",
+    `${chalk.yellow("Base URL:")} ${config.base_url}`,
+    `${chalk.yellow("Datasource ID:")} ${config.ds_id}`,
+    `${chalk.yellow("API Key:")} ${config.apikey ? config.apikey : "未设置"}`,
+    `${chalk.yellow("Omit Single DB:")} ${
+      config.omit_single_db !== undefined
+        ? config.omit_single_db
+          ? chalk.green("true")
+          : chalk.red("false")
+        : chalk.gray("未设置")
+    }`,
+    `${chalk.yellow("Camel Alias:")} ${
+      config.camel_alias !== undefined
+        ? config.camel_alias
+          ? chalk.green("true")
+          : chalk.red("false")
+        : chalk.gray("未设置")
+    }`,
+  ];
+
+  console.log(
+    boxen(configLines.join("\n"), {
+      padding: 1,
+      margin: 1,
+      borderStyle: "round",
+      borderColor: "green",
+      title: "Debug Info",
+      titleAlignment: "center",
+    })
+  );
+}
+
+async function queryToSql(config, question) {
+  const url = `${config.base_url}/v1/q2m/`;
+
+  const headers = {
+    Authorization: `Bearer ${config.apikey}`,
+    "Content-Type": "application/json",
+  };
+
+  const payload = {
+    datasource_id: config.ds_id,
+    question: question,
+  };
+
+  // 添加可选的布尔配置项到请求体中
+  if (config.omit_single_db !== undefined) {
+    payload.omit_single_db = config.omit_single_db;
+  }
+
+  if (config.camel_alias !== undefined) {
+    payload.camel_alias = config.camel_alias;
+  }
+
+  try {
+    const response = await axios.post(url, payload, {
+      headers: headers,
+      timeout: 60000,
+    });
+    return {
+      success: true,
+      data: response.data,
+    };
+  } catch (error) {
+    if (error.response) {
+      return {
+        success: false,
+        error: `HTTP错误：${error.response.status} - ${
+          error.response.statusText
+        }\n错误详情：${JSON.stringify(error.response.data)}`,
+      };
+    } else {
+      return {
+        success: false,
+        error: `请求失败：${error.message}`,
+      };
+    }
+  }
+}
+
+function createReadlineInterface() {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  return rl;
+}
+
+function askQuestion(rl, question) {
+  return new Promise((resolve, reject) => {
+    if (rl.closed) {
+      reject(new Error("readline is closed"));
+      return;
+    }
+
+    rl.question(question, (answer) => {
+      resolve(answer);
+    });
+  });
+}
+
+async function main() {
+  // 加载配置
+  const config = loadConfig();
+
+  // 获取后端版本信息
+  const backendInfo = await fetchBackendVersion(config.base_url);
+
+  // 打印启动页
+  displayStartupScreen(backendInfo);
+
+  // 如果后端连接失败，直接退出
+  if (!backendInfo.connected) {
+    process.exit(1);
+  }
+
+  const rl = createReadlineInterface();
+
+  try {
+    while (true) {
+      const question = (
+        await askQuestion(rl, "请输入自然语言的查询语句：")
+      ).trim();
+
+      if (question.toLowerCase() === "exit") {
+        console.log("再见！");
+        break;
+      }
+
+      if (question.toLowerCase() === "/info") {
+        displayConfigInfo(config);
+        console.log("\n" + "=".repeat(50) + "\n");
+        continue;
+      }
+
+      if (!question) {
+        console.log("请输入有效的查询语句\n");
+        continue;
+      }
+
+      console.log("正在转换SQL...");
+      const result = await queryToSql(config, question);
+
+      if (result.success) {
+        console.log(chalk.green("✓ 转换成功"));
+        console.log(result.data);
+      } else {
+        console.log(chalk.red("✗ 转换失败"));
+        console.log(chalk.red(result.error));
+      }
+
+      console.log("\n" + "=".repeat(50) + "\n");
+    }
+  } catch (error) {
+    if (error.code === "SIGINT") {
+    } else {
+      console.log(`发生错误：${error.message}\n`);
+    }
+  } finally {
+    rl.close();
+  }
+}
+
+// 处理 Ctrl+C 中断
+process.on("SIGINT", () => {
+  console.log("\n\n再见！");
+  process.exit(0);
+});
+
+main().catch((error) => {
+  console.log(`发生错误：${error.message}`);
+  process.exit(1);
+});

package/package.json

@@ -0,0 +1,36 @@
+{
+  "name": "@rongchenglu/atc", 
+  "version": "1.0.6",
+  "description": "asktable 自然语言转SQL 工具",
+  "main": "index.js",
+  "type": "module",
+  "dependencies": {
+    "axios": "^1.11.0",
+    "boxen": "^8.0.1",
+    "chalk": "^5.6.2",
+    "js-yaml": "^4.1.0"
+  },
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/DataMini/gen_mybatis.git"
+  },
+  "keywords": [
+    "mybatis",
+    "nl2sql",
+    "asktable",
+    "datamini",
+    "atc"
+  ],
+  "author": "datamini",
+  "license": "ISC",
+  "bugs": {
+    "url": "https://github.com/DataMini/gen_mybatis/issues"
+  },
+  "homepage": "https://github.com/DataMini/gen_mybatis#readme",
+  "bin": {
+    "atc": "./index.js"
+  }
+}