@pionex/pionex-ai-kit 0.2.19 → 0.2.21

package/README.md

@@ -13,6 +13,10 @@ npm install -g @pionex/pionex-ai-kit
 - **pionex-ai-kit onboard** — Interactive wizard: API Key, API Secret, profile name. Writes `~/.pionex/config.toml`.
 - **pionex-ai-kit setup** — Register MCP servers for supported clients.
 - **pionex-ai-kit help** — Show help.
+- **pionex market ...** — Public market data commands.
+- **pionex account ...** — Account balance command.
+- **pionex orders ...** — Spot order lifecycle commands.
+- **pionex bot ...** — Futures grid bot lifecycle commands (`get/create/adjust/reduce/cancel`).
 
 The MCP server (**@pionex/pionex-trade-mcp**) reads from the same file. After init, run:
 

package/dist/index.js

@@ -987,8 +987,8 @@ function runSetup(options) {
   );
 }
 var PIONEX_API_DEFAULT_BASE_URL = "https://api.pionex.com";
-var MODULES = ["market", "account", "orders"];
-var DEFAULT_MODULES = ["market", "account", "orders"];
+var MODULES = ["market", "account", "orders", "bot"];
+var DEFAULT_MODULES = ["market", "account", "orders", "bot"];
 var ConfigError = class extends Error {
   suggestion;
   constructor(message, suggestion) {
@@ -1480,8 +1480,318 @@ function registerOrdersTools() {
     }
   ];
 }
+function asNonEmptyString(value, field) {
+  if (typeof value !== "string" || value.trim().length === 0) {
+    throw new Error(`Invalid "${field}": expected non-empty string.`);
+  }
+  return value.trim();
+}
+function asFiniteNumber(value, field) {
+  if (typeof value !== "number" || !Number.isFinite(value)) {
+    throw new Error(`Invalid "${field}": expected finite number.`);
+  }
+  return value;
+}
+function asPositiveNumber(value, field) {
+  const n = asFiniteNumber(value, field);
+  if (n <= 0) throw new Error(`Invalid "${field}": expected number > 0.`);
+  return n;
+}
+function asPositiveInteger(value, field) {
+  const n = asPositiveNumber(value, field);
+  if (!Number.isInteger(n)) {
+    throw new Error(`Invalid "${field}": expected positive integer.`);
+  }
+  return n;
+}
+function asBoolean(value, field) {
+  if (typeof value !== "boolean") {
+    throw new Error(`Invalid "${field}": expected boolean.`);
+  }
+  return value;
+}
+function assertEnum(value, field, allowed) {
+  if (!allowed.includes(value)) {
+    throw new Error(`Invalid "${field}": expected one of ${allowed.join(", ")}.`);
+  }
+}
+function asObject(value, field) {
+  if (!value || typeof value !== "object" || Array.isArray(value)) {
+    throw new Error(`Invalid "${field}": expected JSON object.`);
+  }
+  return value;
+}
+function asPositiveDecimalString(value, field) {
+  const s = asNonEmptyString(value, field);
+  if (!/^\d+(\.\d+)?$/.test(s)) {
+    throw new Error(`Invalid "${field}": expected positive decimal string.`);
+  }
+  const n = Number(s);
+  if (!Number.isFinite(n) || n <= 0) {
+    throw new Error(`Invalid "${field}": expected positive decimal string.`);
+  }
+  return s;
+}
+function registerBotTools() {
+  return [
+    {
+      name: "pionex_bot_get_futures_grid_order",
+      module: "bot",
+      isWrite: false,
+      description: "Get one futures grid bot order by buOrderId.",
+      inputSchema: {
+        type: "object",
+        additionalProperties: false,
+        properties: {
+          buOrderId: { type: "string", description: "Futures grid bot order id." },
+          lang: { type: "string", description: "Optional language code." }
+        },
+        required: ["buOrderId"]
+      },
+      async handler(args, { client }) {
+        const buOrderId = String(args.buOrderId);
+        const q = { buOrderId };
+        if (args.lang != null) q.lang = String(args.lang);
+        return (await client.signedGet("/api/v1/bot/orders/futuresGrid/order", q)).data;
+      }
+    },
+    {
+      name: "pionex_bot_create_futures_grid_order",
+      module: "bot",
+      isWrite: true,
+      description: "Create a futures grid bot order.",
+      inputSchema: {
+        type: "object",
+        additionalProperties: false,
+        properties: {
+          keyId: { type: "string" },
+          exchange: { type: "string", description: "e.g. pionex.v2" },
+          base: { type: "string", description: "e.g. BTC" },
+          quote: { type: "string", description: "e.g. USDT" },
+          copyFrom: { type: "string" },
+          copyType: { type: "string" },
+          groupId: { type: "string" },
+          copyBotOrderId: { type: "string" },
+          lang: { type: "string" },
+          buOrderData: {
+            type: "object",
+            additionalProperties: true,
+            description: "CreateFuturesGridOrderData payload from openapi_bot.yaml."
+          }
+        },
+        required: ["keyId", "exchange", "base", "quote", "buOrderData"]
+      },
+      async handler(args, { client, config }) {
+        if (config.readOnly) {
+          throw new Error("Server is running in --read-only mode; bot create is disabled.");
+        }
+        const keyId = asNonEmptyString(args.keyId, "keyId");
+        const exchange = asNonEmptyString(args.exchange, "exchange");
+        const base = asNonEmptyString(args.base, "base");
+        const quote = asNonEmptyString(args.quote, "quote");
+        const buOrderData = asObject(args.buOrderData, "buOrderData");
+        const top = asPositiveDecimalString(buOrderData.top, "buOrderData.top");
+        const bottom = asPositiveDecimalString(buOrderData.bottom, "buOrderData.bottom");
+        if (Number(top) <= Number(bottom)) {
+          throw new Error('Invalid "buOrderData.top": expected top > bottom.');
+        }
+        const row = asPositiveInteger(buOrderData.row, "buOrderData.row");
+        const gridType = asNonEmptyString(buOrderData.grid_type, "buOrderData.grid_type");
+        assertEnum(gridType, "buOrderData.grid_type", ["arithmetic", "geometric"]);
+        const openPrice = asPositiveDecimalString(buOrderData.openPrice, "buOrderData.openPrice");
+        const trend = asNonEmptyString(buOrderData.trend, "buOrderData.trend");
+        assertEnum(trend, "buOrderData.trend", ["long", "short", "no_trend"]);
+        const leverage = asPositiveNumber(buOrderData.leverage, "buOrderData.leverage");
+        const extraMargin = asPositiveDecimalString(buOrderData.extraMargin, "buOrderData.extraMargin");
+        const quoteInvestment = asPositiveDecimalString(buOrderData.quoteInvestment, "buOrderData.quoteInvestment");
+        const body = {
+          keyId,
+          exchange,
+          base,
+          quote,
+          buOrderData: {
+            ...buOrderData,
+            top,
+            bottom,
+            row,
+            grid_type: gridType,
+            openPrice,
+            trend,
+            leverage,
+            extraMargin,
+            quoteInvestment
+          }
+        };
+        if (args.copyFrom != null) body.copyFrom = String(args.copyFrom);
+        if (args.copyType != null) body.copyType = String(args.copyType);
+        if (args.groupId != null) body.groupId = String(args.groupId);
+        if (args.copyBotOrderId != null) body.copyBotOrderId = String(args.copyBotOrderId);
+        if (args.lang != null) body.lang = String(args.lang);
+        return (await client.signedPost("/api/v1/bot/orders/futuresGrid/create", body)).data;
+      }
+    },
+    {
+      name: "pionex_bot_adjust_futures_grid_params",
+      module: "bot",
+      isWrite: true,
+      description: "Adjust futures grid bot params (invest_in / adjust_params / invest_in_trigger).",
+      inputSchema: {
+        type: "object",
+        additionalProperties: false,
+        properties: {
+          buOrderId: { type: "string" },
+          type: { type: "string", enum: ["invest_in", "adjust_params", "invest_in_trigger"] },
+          quoteInvestment: { type: "number" },
+          extraMargin: { type: "boolean" },
+          openPrice: { type: "number" },
+          bottom: { type: "string" },
+          top: { type: "string" },
+          row: { type: "number" },
+          extraMarginAmount: { type: "number" },
+          isRecommend: { type: "boolean" },
+          isReinvest: { type: "boolean" },
+          investCoin: { type: "string" },
+          investmentFrom: { type: "string", enum: ["USER", "LOCK_ACTIVITY"] },
+          condition: { type: "string" },
+          conditionDirection: { type: "string", enum: ["1", "-1"] },
+          slippage: { type: "string" },
+          adjustParamsSence: { type: "string" }
+        },
+        required: ["buOrderId", "type", "extraMargin", "openPrice"]
+      },
+      async handler(args, { client, config }) {
+        if (config.readOnly) {
+          throw new Error("Server is running in --read-only mode; bot adjust is disabled.");
+        }
+        const buOrderId = asNonEmptyString(args.buOrderId, "buOrderId");
+        const type = asNonEmptyString(args.type, "type");
+        assertEnum(type, "type", ["invest_in", "adjust_params", "invest_in_trigger"]);
+        const extraMargin = asBoolean(args.extraMargin, "extraMargin");
+        const openPrice = asFiniteNumber(args.openPrice, "openPrice");
+        if (type === "invest_in" && args.quoteInvestment != null) {
+          asPositiveNumber(args.quoteInvestment, "quoteInvestment");
+        }
+        if (type === "adjust_params") {
+          const bottom = asPositiveDecimalString(args.bottom, "bottom");
+          const top = asPositiveDecimalString(args.top, "top");
+          if (Number(top) <= Number(bottom)) {
+            throw new Error('Invalid "top": expected top > bottom.');
+          }
+          asPositiveInteger(args.row, "row");
+        }
+        if (type === "invest_in_trigger") {
+          asPositiveDecimalString(args.condition, "condition");
+          const conditionDirection = asNonEmptyString(args.conditionDirection, "conditionDirection");
+          assertEnum(conditionDirection, "conditionDirection", ["1", "-1"]);
+        }
+        const body = {
+          buOrderId,
+          type,
+          extraMargin,
+          openPrice
+        };
+        if (args.quoteInvestment != null) body.quoteInvestment = asFiniteNumber(args.quoteInvestment, "quoteInvestment");
+        if (args.bottom != null) body.bottom = asPositiveDecimalString(args.bottom, "bottom");
+        if (args.top != null) body.top = asPositiveDecimalString(args.top, "top");
+        if (args.row != null) body.row = asPositiveInteger(args.row, "row");
+        if (args.extraMarginAmount != null) body.extraMarginAmount = asFiniteNumber(args.extraMarginAmount, "extraMarginAmount");
+        if (args.isRecommend != null) body.isRecommend = asBoolean(args.isRecommend, "isRecommend");
+        if (args.isReinvest != null) body.isReinvest = asBoolean(args.isReinvest, "isReinvest");
+        if (args.investCoin != null) body.investCoin = String(args.investCoin);
+        if (args.investmentFrom != null) {
+          const investmentFrom = asNonEmptyString(args.investmentFrom, "investmentFrom");
+          assertEnum(investmentFrom, "investmentFrom", ["USER", "LOCK_ACTIVITY"]);
+          body.investmentFrom = investmentFrom;
+        }
+        if (args.condition != null) body.condition = asPositiveDecimalString(args.condition, "condition");
+        if (args.conditionDirection != null) {
+          const conditionDirection = asNonEmptyString(args.conditionDirection, "conditionDirection");
+          assertEnum(conditionDirection, "conditionDirection", ["1", "-1"]);
+          body.conditionDirection = conditionDirection;
+        }
+        if (args.slippage != null) body.slippage = String(args.slippage);
+        if (args.adjustParamsSence != null) body.adjustParamsSence = String(args.adjustParamsSence);
+        return (await client.signedPost("/api/v1/bot/orders/futuresGrid/adjustParams", body)).data;
+      }
+    },
+    {
+      name: "pionex_bot_reduce_futures_grid_position",
+      module: "bot",
+      isWrite: true,
+      description: "Reduce futures grid bot position.",
+      inputSchema: {
+        type: "object",
+        additionalProperties: false,
+        properties: {
+          buOrderId: { type: "string" },
+          openPrice: { type: "string" },
+          reduceNum: { type: "number" },
+          slippage: { type: "string" },
+          condition: { type: "string" },
+          conditionDirection: { type: "string", enum: ["1", "-1"] }
+        },
+        required: ["buOrderId", "openPrice", "reduceNum"]
+      },
+      async handler(args, { client, config }) {
+        if (config.readOnly) {
+          throw new Error("Server is running in --read-only mode; bot reduce is disabled.");
+        }
+        const buOrderId = asNonEmptyString(args.buOrderId, "buOrderId");
+        const openPrice = asPositiveDecimalString(args.openPrice, "openPrice");
+        const reduceNum = asPositiveInteger(args.reduceNum, "reduceNum");
+        const body = {
+          buOrderId,
+          openPrice,
+          reduceNum
+        };
+        if (args.slippage != null) body.slippage = String(args.slippage);
+        if (args.condition != null) body.condition = String(args.condition);
+        if (args.conditionDirection != null) {
+          const conditionDirection = asNonEmptyString(args.conditionDirection, "conditionDirection");
+          assertEnum(conditionDirection, "conditionDirection", ["1", "-1"]);
+          body.conditionDirection = conditionDirection;
+        }
+        return (await client.signedPost("/api/v1/bot/orders/futuresGrid/reduce", body)).data;
+      }
+    },
+    {
+      name: "pionex_bot_cancel_futures_grid_order",
+      module: "bot",
+      isWrite: true,
+      description: "Cancel and close a futures grid bot order.",
+      inputSchema: {
+        type: "object",
+        additionalProperties: false,
+        properties: {
+          buOrderId: { type: "string" },
+          closeNote: { type: "string" },
+          closeSellModel: { type: "string", enum: ["TO_QUOTE", "TO_USDT"] },
+          immediate: { type: "boolean" },
+          closeSlippage: { type: "string" }
+        },
+        required: ["buOrderId"]
+      },
+      async handler(args, { client, config }) {
+        if (config.readOnly) {
+          throw new Error("Server is running in --read-only mode; bot cancel is disabled.");
+        }
+        const buOrderId = asNonEmptyString(args.buOrderId, "buOrderId");
+        const body = { buOrderId };
+        if (args.closeNote != null) body.closeNote = String(args.closeNote);
+        if (args.closeSellModel != null) {
+          const closeSellModel = asNonEmptyString(args.closeSellModel, "closeSellModel");
+          assertEnum(closeSellModel, "closeSellModel", ["TO_QUOTE", "TO_USDT"]);
+          body.closeSellModel = closeSellModel;
+        }
+        if (args.immediate != null) body.immediate = asBoolean(args.immediate, "immediate");
+        if (args.closeSlippage != null) body.closeSlippage = String(args.closeSlippage);
+        return (await client.signedPost("/api/v1/bot/orders/futuresGrid/cancel", body)).data;
+      }
+    }
+  ];
+}
 function allToolSpecs() {
-  return [...registerMarketTools(), ...registerAccountTools(), ...registerOrdersTools()];
+  return [...registerMarketTools(), ...registerAccountTools(), ...registerOrdersTools(), ...registerBotTools()];
 }
 function createToolRunner(client, config) {
   const fullConfig = { ...config, modules: [...MODULES] };
@@ -1636,20 +1946,23 @@ function parseFlags(argv) {
 }
 function printPionexHelp() {
   process.stdout.write(`
-Usage: pionex-trade-cli <group> <command> [args] [--flags]
+Usage: pionex <group> <command> [args] [--flags]
 
 Groups:
   market   Market data (public)
   account  Account data (requires auth)
   orders   Spot orders (requires auth)
+  bot      Futures grid bot (requires auth)
 
 Examples:
-  pionex-trade-cli market depth BTC_USDT --limit 5
-  pionex-trade-cli market tickers --symbol BTC_USDT
-  pionex-trade-cli market symbols --symbols BTC_USDT
-  pionex-trade-cli account balance
-  pionex-trade-cli orders new --symbol BTC_USDT --side BUY --type MARKET --amount 10
-  pionex-trade-cli orders cancel --symbol BTC_USDT --order-id 123
+  pionex market depth BTC_USDT --limit 5
+  pionex market tickers --symbol BTC_USDT
+  pionex market symbols --symbols BTC_USDT
+  pionex account balance
+  pionex orders new --symbol BTC_USDT --side BUY --type MARKET --amount 10
+  pionex orders cancel --symbol BTC_USDT --order-id 123
+  pionex bot get --bu-order-id <id>
+  pionex bot create --key-id <keyId> --exchange pionex.v2 --base BTC --quote USDT --bu-order-data-json '{"top":"110000","bottom":"90000","row":100,"grid_type":"arithmetic","openPrice":"100000","trend":"long","leverage":5,"extraMargin":"0","quoteInvestment":"100"}'
 
 Global flags:
   --profile <name>     Profile in ~/.pionex/config.toml
@@ -1659,6 +1972,21 @@ Global flags:
   --dry-run            Print the tool call payload without executing (write ops only)
 `);
 }
+function parseJsonFlag(raw, flagName) {
+  if (typeof raw !== "string") {
+    throw new Error(`Missing required flag: --${flagName}`);
+  }
+  try {
+    const parsed = JSON.parse(raw);
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+      throw new Error("must be a JSON object");
+    }
+    return parsed;
+  } catch (e) {
+    const msg = e instanceof Error ? e.message : String(e);
+    throw new Error(`Invalid --${flagName}: ${msg}`);
+  }
+}
 async function runPionexCommand(argv) {
   const { positionals, flags } = parseFlags(argv);
   const group = positionals[0];
@@ -1679,7 +2007,7 @@ async function runPionexCommand(argv) {
   if (group === "market") {
     if (command === "depth") {
       const symbol = positionals[2];
-      if (!symbol) throw new Error("Missing symbol. Example: pionex-trade-cli market depth BTC_USDT");
+      if (!symbol) throw new Error("Missing symbol. Example: pionex market depth BTC_USDT");
       const limit = flags.limit != null ? Number(flags.limit) : void 0;
       const out = await runTool("pionex_market_get_depth", { symbol, limit });
       process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
@@ -1687,7 +2015,7 @@ async function runPionexCommand(argv) {
     }
     if (command === "trades") {
       const symbol = positionals[2];
-      if (!symbol) throw new Error("Missing symbol. Example: pionex-trade-cli market trades BTC_USDT");
+      if (!symbol) throw new Error("Missing symbol. Example: pionex market trades BTC_USDT");
       const limit = flags.limit != null ? Number(flags.limit) : void 0;
       const out = await runTool("pionex_market_get_trades", { symbol, limit });
       process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
@@ -1710,8 +2038,7 @@ async function runPionexCommand(argv) {
     if (command === "klines") {
       const symbol = typeof flags.symbol === "string" ? flags.symbol : positionals[2];
       const interval = typeof flags.interval === "string" ? flags.interval : positionals[3];
-      if (!symbol || !interval)
-        throw new Error("Missing symbol/interval. Example: pionex-trade-cli market klines BTC_USDT 60M");
+      if (!symbol || !interval) throw new Error("Missing symbol/interval. Example: pionex market klines BTC_USDT 60M");
       const endTime = flags.endTime != null ? Number(flags.endTime) : void 0;
       const limit = flags.limit != null ? Number(flags.limit) : void 0;
       const out = await runTool("pionex_market_get_klines", { symbol, interval, endTime, limit });
@@ -1807,6 +2134,76 @@ async function runPionexCommand(argv) {
     }
     throw new Error(`Unknown orders command: ${command}`);
   }
+  if (group === "bot") {
+    if (command === "get") {
+      const buOrderId = typeof flags["bu-order-id"] === "string" ? flags["bu-order-id"] : typeof flags.buOrderId === "string" ? flags.buOrderId : void 0;
+      const lang = typeof flags.lang === "string" ? flags.lang : void 0;
+      if (!buOrderId) throw new Error("Missing required flag: --bu-order-id");
+      const out = await runTool("pionex_bot_get_futures_grid_order", { buOrderId, lang });
+      process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
+      return;
+    }
+    if (command === "create") {
+      const keyId = typeof flags["key-id"] === "string" ? flags["key-id"] : typeof flags.keyId === "string" ? flags.keyId : void 0;
+      const exchange = typeof flags.exchange === "string" ? flags.exchange : void 0;
+      const base = typeof flags.base === "string" ? flags.base : void 0;
+      const quote = typeof flags.quote === "string" ? flags.quote : void 0;
+      const copyFrom = typeof flags["copy-from"] === "string" ? flags["copy-from"] : typeof flags.copyFrom === "string" ? flags.copyFrom : void 0;
+      const copyType = typeof flags["copy-type"] === "string" ? flags["copy-type"] : typeof flags.copyType === "string" ? flags.copyType : void 0;
+      const groupId = typeof flags["group-id"] === "string" ? flags["group-id"] : typeof flags.groupId === "string" ? flags.groupId : void 0;
+      const copyBotOrderId = typeof flags["copy-bot-order-id"] === "string" ? flags["copy-bot-order-id"] : typeof flags.copyBotOrderId === "string" ? flags.copyBotOrderId : void 0;
+      const lang = typeof flags.lang === "string" ? flags.lang : void 0;
+      const buOrderData = parseJsonFlag(flags["bu-order-data-json"] ?? flags.buOrderDataJson, "bu-order-data-json");
+      if (!keyId || !exchange || !base || !quote) {
+        throw new Error("Missing required flags: --key-id --exchange --base --quote --bu-order-data-json");
+      }
+      const payload = { keyId, exchange, base, quote, copyFrom, copyType, groupId, copyBotOrderId, lang, buOrderData };
+      if (dryRun) {
+        process.stdout.write(JSON.stringify({ tool: "pionex_bot_create_futures_grid_order", args: payload }, null, 2) + "\n");
+        return;
+      }
+      const out = await runTool("pionex_bot_create_futures_grid_order", payload);
+      process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
+      return;
+    }
+    if (command === "adjust") {
+      const payload = parseJsonFlag(flags["body-json"] ?? flags.bodyJson, "body-json");
+      if (dryRun) {
+        process.stdout.write(JSON.stringify({ tool: "pionex_bot_adjust_futures_grid_params", args: payload }, null, 2) + "\n");
+        return;
+      }
+      const out = await runTool("pionex_bot_adjust_futures_grid_params", payload);
+      process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
+      return;
+    }
+    if (command === "reduce") {
+      const payload = parseJsonFlag(flags["body-json"] ?? flags.bodyJson, "body-json");
+      if (dryRun) {
+        process.stdout.write(JSON.stringify({ tool: "pionex_bot_reduce_futures_grid_position", args: payload }, null, 2) + "\n");
+        return;
+      }
+      const out = await runTool("pionex_bot_reduce_futures_grid_position", payload);
+      process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
+      return;
+    }
+    if (command === "cancel") {
+      const buOrderId = typeof flags["bu-order-id"] === "string" ? flags["bu-order-id"] : typeof flags.buOrderId === "string" ? flags.buOrderId : void 0;
+      const closeNote = typeof flags["close-note"] === "string" ? flags["close-note"] : typeof flags.closeNote === "string" ? flags.closeNote : void 0;
+      const closeSellModel = typeof flags["close-sell-model"] === "string" ? flags["close-sell-model"] : typeof flags.closeSellModel === "string" ? flags.closeSellModel : void 0;
+      const immediate = typeof flags.immediate === "boolean" ? flags.immediate : void 0;
+      const closeSlippage = typeof flags["close-slippage"] === "string" ? flags["close-slippage"] : typeof flags.closeSlippage === "string" ? flags.closeSlippage : void 0;
+      if (!buOrderId) throw new Error("Missing required flag: --bu-order-id");
+      const payload = { buOrderId, closeNote, closeSellModel, immediate, closeSlippage };
+      if (dryRun) {
+        process.stdout.write(JSON.stringify({ tool: "pionex_bot_cancel_futures_grid_order", args: payload }, null, 2) + "\n");
+        return;
+      }
+      const out = await runTool("pionex_bot_cancel_futures_grid_order", payload);
+      process.stdout.write(JSON.stringify(out.data, null, 2) + "\n");
+      return;
+    }
+    throw new Error(`Unknown bot command: ${command}`);
+  }
   throw new Error(`Unknown group: ${group}`);
 }
 function main() {