@2oolkit/kiwoom-cli 0.1.0 → 0.1.2

package/dist/mcp.js

@@ -84,6 +84,14 @@ var ORDER_TYPES = {
   "81": "\uC7A5\uB9C8\uAC10\uD6C4\uC2DC\uAC04\uC678"
 };
 var MARKET_ORDER_TYPES = /* @__PURE__ */ new Set(["3", "13", "23"]);
+var CHART_PER_PAGE_CAP = {
+  tick: 900,
+  minute: 900,
+  day: 600,
+  week: 300,
+  month: 240,
+  year: 30
+};
 
 // src/config/store.ts
 var fs = __toESM(require("fs"));
@@ -374,7 +382,7 @@ var KiwoomClient = class {
    */
   async callEndpoint(def, body = {}, opts = {}) {
     if (opts.paginate && def.listKey) {
-      const data = await this.requestAll(def.apiId, def.path, body, def.listKey);
+      const data = await this.requestAll(def.apiId, def.path, body, def.listKey, opts.maxPages);
       return { data, contYn: false, nextKey: "" };
     }
     return this.request(def.apiId, def.path, body, {
@@ -384,9 +392,11 @@ var KiwoomClient = class {
   }
   /**
    * Fetch all pages of a TR, concatenating the array under `listKey`.
-   * Caps at `maxPages` to avoid runaway loops.
+   * Caps at `maxPages` to avoid runaway loops (default 100 — high enough for
+   * large chart pulls; callers pass a tighter bound when they know how many
+   * pages a target row count needs).
    */
-  async requestAll(apiId, path2, body, listKey, maxPages = 20) {
+  async requestAll(apiId, path2, body, listKey, maxPages = 100) {
     let page = await this.request(apiId, path2, body);
     const acc = Array.isArray(page.data[listKey]) ? [...page.data[listKey]] : [];
     let pages = 1;
@@ -508,6 +518,7 @@ var ENDPOINTS = {
   rankTradeAmount: { apiId: "ka10032", path: PATHS.rkinfo, korean: "\uAC70\uB798\uB300\uAE08\uC0C1\uC704\uC694\uCCAD", listKey: "trde_prica_upper" },
   rankVolumeSurge: { apiId: "ka10023", path: PATHS.rkinfo, korean: "\uAC70\uB798\uB7C9\uAE09\uC99D\uC694\uCCAD", listKey: "trde_qty_sdnin" },
   rankPrevVolume: { apiId: "ka10031", path: PATHS.rkinfo, korean: "\uC804\uC77C\uAC70\uB798\uB7C9\uC0C1\uC704\uC694\uCCAD", listKey: "pred_trde_qty_upper" },
+  rankForeignInst: { apiId: "ka90009", path: PATHS.rkinfo, korean: "\uC678\uAD6D\uC778\uAE30\uAD00\uB9E4\uB9E4\uC0C1\uC704\uC694\uCCAD", listKey: "frgnr_orgn_trde_upper" },
   // ── Sector / industry (업종) ───────────────────────────────────────────────
   sectorPrice: { apiId: "ka20001", path: PATHS.sect, korean: "\uC5C5\uC885\uD604\uC7AC\uAC00\uC694\uCCAD", listKey: "inds_cur_prc_tm" },
   sectorStocks: { apiId: "ka20002", path: PATHS.sect, korean: "\uC5C5\uC885\uBCC4\uC8FC\uAC00\uC694\uCCAD", listKey: "inds_stkpc" },
@@ -526,6 +537,15 @@ function unpad(value) {
   const intPart = m[2].replace(/^0+(?=\d)/, "");
   return `${sign}${intPart}${m[3] ?? ""}`;
 }
+function withCommas(value) {
+  const m = value.match(/^([+-]?)(\d+)(\.\d+)?$/);
+  if (!m) return value;
+  const grouped = m[2].replace(/\B(?=(\d{3})+(?!\d))/g, ",");
+  return `${m[1]}${grouped}${m[3] ?? ""}`;
+}
+function won(value) {
+  return withCommas(unpad(value));
+}
 function formatStamp(value) {
   if (!value) return "";
   const s = String(value).trim();
@@ -687,25 +707,30 @@ var PERIOD_EP = {
   month: ENDPOINTS.monthlyChart,
   year: ENDPOINTS.yearlyChart
 };
+var DEFAULT_COUNT = 50;
+var MAX_COUNT = 1e5;
 function registerChartTools(server2) {
   tool(
     server2,
     "get_chart",
     {
-      description: "Get OHLC chart data for a stock. Period charts (day/week/month/year) end at base date; tick/minute use an aggregation scope. Returns latest-first; use count to cap rows.",
+      description: "Get OHLC chart data for a stock. Period charts (day/week/month/year) end at base date; tick/minute use an aggregation scope. Returns latest-first; use count to cap rows. Per-request caps: tick/minute 900, day 600, week 300, month 240, year 30. When count exceeds the cap the tool auto-paginates (cont-yn/next-key) and returns up to count rows.",
       inputSchema: {
         code: import_zod2.z.string().describe("6-digit stock code"),
         timeframe: import_zod2.z.enum(["tick", "minute", "day", "week", "month", "year"]).describe("Chart timeframe"),
         scope: import_zod2.z.string().optional().describe("Tick units (1/3/5/10/30) or minute interval (1/3/5/10/15/30/45/60); default 1"),
         date: import_zod2.z.string().optional().describe("Base date YYYYMMDD for period charts (default today)"),
         adjusted: import_zod2.z.boolean().optional().describe("Adjust for splits/rights (default true)"),
-        count: import_zod2.z.number().min(1).max(900).optional().describe("Max rows to return (default 50)")
+        count: import_zod2.z.number().min(1).max(MAX_COUNT).optional().describe(
+          "Max rows to return (default 50). Exceeding the per-request cap (tick/min 900, day 600, week 300, month 240, year 30) auto-paginates."
+        )
       }
     },
     async ({ code, timeframe, scope, date, adjusted, count }) => withErrorHandling(async () => {
       const client = clientOrThrow();
       const stk = normalizeStockCode(code);
       const upd = adjusted === false ? "0" : "1";
+      const want = count ?? DEFAULT_COUNT;
       let def;
       let body;
       if (timeframe === "tick") {
@@ -718,12 +743,15 @@ function registerChartTools(server2) {
         def = PERIOD_EP[timeframe];
         body = { stk_cd: stk, base_dt: date ?? todayKst(), upd_stkpc_tp: upd };
       }
-      const { data } = await client.callEndpoint(def, body);
+      const cap = CHART_PER_PAGE_CAP[timeframe];
+      const paginate = want > cap;
+      const maxPages = paginate ? Math.max(1, Math.ceil(want / cap)) : 1;
+      const { data } = await client.callEndpoint(def, body, { paginate, maxPages });
       const rows = Array.isArray(data[def.listKey]) ? data[def.listKey] : [];
       return mcpJson({
         code: stk,
         timeframe,
-        rows: rows.slice(0, count ?? 50)
+        rows: rows.slice(0, want)
       });
     })
   );
@@ -968,6 +996,28 @@ ${JSON.stringify(preview, null, 2)}`);
 
 // src/mcp/tools/ranking.ts
 var import_zod5 = require("zod");
+
+// src/utils/ranking.ts
+var NETTRADE_FIELDS = {
+  foreign: { buy: "for_netprps", sell: "for_netslmt" },
+  institution: { buy: "orgn_netprps", sell: "orgn_netslmt" }
+};
+function extractNetTrade(rows, prefix, n) {
+  return rows.slice(0, n).map((r, i) => ({
+    rank: i + 1,
+    code: r[`${prefix}_stk_cd`],
+    name: r[`${prefix}_stk_nm`],
+    \uAE08\uC561: won(unpad(r[`${prefix}_amt`] ?? "")),
+    \uC218\uB7C9: won(unpad(r[`${prefix}_qty`] ?? ""))
+  })).filter((x) => x.code);
+}
+function buildNetTradeResult(rows, investors, side, n) {
+  const result = {};
+  for (const inv of investors) result[inv] = extractNetTrade(rows, NETTRADE_FIELDS[inv][side], n);
+  return result;
+}
+
+// src/mcp/tools/ranking.ts
 function registerRankingTools(server2) {
   tool(
     server2,
@@ -1040,6 +1090,39 @@ function registerRankingTools(server2) {
       return mcpJson(data);
     })
   );
+  tool(
+    server2,
+    "get_net_buy_ranking",
+    {
+      description: "Get the foreign/institution net-buy (\uC218\uAE09) ranking (ka90009): top stocks by foreign and/or institutional net buying or selling. One call returns parallel rankings for both investor types.",
+      inputSchema: {
+        by: import_zod5.z.enum(["foreign", "institution", "both"]).optional().describe("Investor type: foreign=\uC678\uAD6D\uC778, institution=\uAE30\uAD00, both (default both)"),
+        side: import_zod5.z.enum(["buy", "sell"]).optional().describe("buy=\uC21C\uB9E4\uC218 (default), sell=\uC21C\uB9E4\uB3C4"),
+        market: import_zod5.z.enum(["000", "001", "101"]).optional().describe("000=all, 001=KOSPI, 101=KOSDAQ"),
+        exchange: import_zod5.z.enum(["1", "2", "3"]).optional().describe("1=KRX (default), 2=NXT, 3=unified"),
+        count: import_zod5.z.number().int().min(1).max(50).optional().describe("Top N (default 10)"),
+        rankBy: import_zod5.z.enum(["1", "2"]).optional().describe("Rank by 1=amount \uAE08\uC561 (default), 2=quantity \uC218\uB7C9"),
+        date: import_zod5.z.string().optional().describe("Query date YYYYMMDD (default: latest)")
+      }
+    },
+    async ({ by, side, market, exchange, count, rankBy, date }) => withErrorHandling(async () => {
+      const investors = by === "foreign" ? ["foreign"] : by === "institution" ? ["institution"] : ["foreign", "institution"];
+      const netSide = side === "sell" ? "sell" : "buy";
+      const { data } = await clientOrThrow().callEndpoint(ENDPOINTS.rankForeignInst, {
+        mrkt_tp: market ?? "000",
+        amt_qty_tp: rankBy === "2" ? "2" : "1",
+        qry_dt_tp: date ? "1" : "0",
+        date: date ?? "",
+        stex_tp: exchange ?? "1"
+      });
+      const rows = Array.isArray(data?.[ENDPOINTS.rankForeignInst.listKey]) ? data[ENDPOINTS.rankForeignInst.listKey] : [];
+      return mcpJson({
+        side: netSide,
+        rankBy: rankBy === "2" ? "quantity" : "amount",
+        ...buildNetTradeResult(rows, investors, netSide, count ?? 10)
+      });
+    })
+  );
   tool(
     server2,
     "get_sector",