@thinkingai/ae-cli 6.1.19 → 6.1.21

package/dist/client-N7NGZUNI.js

@@ -1,204 +0,0 @@
-import { createRequire as __createRequire } from 'node:module'; const require = __createRequire(import.meta.url);
-import {
-  SecureStoreAuthError,
-  getToken,
-  invalidateAccessTokenForRetry,
-  resolveHost
-} from "./chunk-O6K2ODUE.js";
-import {
-  safeJsonParse
-} from "./chunk-KZ6ZHF5J.js";
-import {
-  logger
-} from "./chunk-FR2EDAQM.js";
-
-// src/core/client.ts
-import WebSocket from "ws";
-import { randomBytes } from "crypto";
-function genRequestId(prefix) {
-  const rand = randomBytes(4).toString("base64url").slice(0, 8);
-  return `${prefix}@@${rand}`;
-}
-function buildUrl(baseUrl, modulePath, params = {}) {
-  const base = baseUrl.replace(/\/$/, "");
-  const p = modulePath.startsWith("/") ? modulePath : `/${modulePath}`;
-  const url = new URL(`${base}${p}`);
-  for (const [k, v] of Object.entries(params)) {
-    if (v !== void 0 && v !== null) url.searchParams.set(k, String(v));
-  }
-  return url.toString();
-}
-function buildWsUrl(baseUrl, token) {
-  const base = baseUrl.replace(/\/$/, "");
-  const wsBase = base.replace(/^https:\/\//, "wss://").replace(/^http:\/\//, "ws://");
-  return `${wsBase}/v1/ta-websocket/query/${token}`;
-}
-async function request(method, modulePath, params = {}, body = null, retry = true, hostUrl) {
-  const resolvedHost = resolveHost(hostUrl);
-  const token = await getToken(resolvedHost);
-  const url = buildUrl(resolvedHost, modulePath, params);
-  const headers = {
-    "Authorization": `bearer ${token}`,
-    "Content-Type": "application/json"
-  };
-  const options = { method, headers };
-  if (body && method !== "GET") options.body = JSON.stringify(body);
-  const resp = await fetch(url, options);
-  const text = await resp.text();
-  const data = safeJsonParse(text);
-  logger.api(method, url, resp.status, body, data);
-  if (resp.status === 401) {
-    if (retry) {
-      invalidateAccessTokenForRetry(resolvedHost);
-      return request(method, modulePath, params, body, false, resolvedHost);
-    }
-    throw new SecureStoreAuthError(`Session expired or unauthorized for ${resolvedHost}. Please run: ae-cli auth login`);
-  }
-  if (resp.status === 403) {
-    const m = data && typeof data === "object" && typeof data.error === "string" ? data.error : "Permission denied";
-    throw new Error(m);
-  }
-  if (data.return_code === -1001 && retry) {
-    invalidateAccessTokenForRetry(resolvedHost);
-    return request(method, modulePath, params, body, false, resolvedHost);
-  }
-  if (data.return_code !== 0 && data.return_code !== void 0) {
-    throw new Error(`AE API error: ${data.return_message || "unknown"} (code: ${data.return_code})`);
-  }
-  return data.data !== void 0 ? data.data : data;
-}
-async function httpGet(modulePath, params = {}, hostUrl) {
-  return request("GET", modulePath, params, null, true, hostUrl);
-}
-async function httpPost(modulePath, params = {}, body, hostUrl) {
-  return request("POST", modulePath, params, body ?? {}, true, hostUrl);
-}
-async function httpRequest(method, modulePath, params = {}, body, hostUrl) {
-  return request(method.toUpperCase(), modulePath, params, body ?? {}, true, hostUrl);
-}
-async function httpDelete(modulePath, params = {}, body, hostUrl) {
-  return request("DELETE", modulePath, params, body ?? null, true, hostUrl);
-}
-async function uploadRequest(modulePath, form, params = {}, retry = true, hostUrl) {
-  const resolvedHost = resolveHost(hostUrl);
-  const token = await getToken(resolvedHost);
-  const url = buildUrl(resolvedHost, modulePath, params);
-  const headers = {
-    "Authorization": `bearer ${token}`
-  };
-  const resp = await fetch(url, { method: "POST", headers, body: form });
-  const text = await resp.text();
-  const data = safeJsonParse(text);
-  logger.api("POST", url, resp.status, { upload: true }, data);
-  if (resp.status === 401) {
-    if (retry) {
-      invalidateAccessTokenForRetry(resolvedHost);
-      return uploadRequest(modulePath, form, params, false, resolvedHost);
-    }
-    throw new SecureStoreAuthError(`Session expired or unauthorized for ${resolvedHost}. Please run: ae-cli auth login`);
-  }
-  if (resp.status === 403) {
-    const m = data && typeof data === "object" && typeof data.error === "string" ? data.error : "Permission denied";
-    throw new Error(m);
-  }
-  if (data.return_code === -1001 && retry) {
-    invalidateAccessTokenForRetry(resolvedHost);
-    return uploadRequest(modulePath, form, params, false, resolvedHost);
-  }
-  if (data.return_code !== 0 && data.return_code !== void 0) {
-    throw new Error(`AE API error: ${data.return_message || "unknown"} (code: ${data.return_code})`);
-  }
-  return data.data !== void 0 ? data.data : data;
-}
-async function httpUpload(modulePath, form, params = {}, hostUrl) {
-  return uploadRequest(modulePath, form, params, true, hostUrl);
-}
-async function wsQueryOnce(projectId, requestId, qp, eventModel, options = {}, token, hostUrl, timeoutMs = 3e4) {
-  const wsUrl = buildWsUrl(hostUrl, token);
-  return new Promise((resolve, reject) => {
-    const ws = new WebSocket(wsUrl);
-    const timer = setTimeout(() => {
-      ws.close();
-      reject(new Error(`WebSocket query timed out after ${timeoutMs}ms`));
-    }, timeoutMs);
-    ws.on("open", () => {
-      const payload = {
-        requestId,
-        projectId,
-        eventModel,
-        qp: typeof qp === "string" ? qp : JSON.stringify(qp),
-        searchSource: options.searchSource || "reportQuery",
-        querySource: options.querySource || "single_report",
-        contentTranslate: options.contentTranslate !== false,
-        ...options.extra || {}
-      };
-      ws.send(JSON.stringify(["data", payload, { channel: "ta" }]));
-    });
-    ws.on("message", (raw) => {
-      try {
-        const msg = safeJsonParse(raw.toString());
-        if (Array.isArray(msg) && msg.length >= 2) {
-          const data = msg[1];
-          if (data.requestId === requestId && data.progress === 100) {
-            clearTimeout(timer);
-            ws.close();
-            resolve(data);
-          }
-        }
-      } catch {
-      }
-    });
-    ws.on("error", (err) => {
-      clearTimeout(timer);
-      reject(err);
-    });
-    ws.on("close", () => {
-      clearTimeout(timer);
-    });
-  });
-}
-async function wsQuery(projectId, requestId, qp, eventModel, options = {}, hostUrl) {
-  const resolvedHost = resolveHost(hostUrl);
-  const token = await getToken(resolvedHost);
-  try {
-    return await wsQueryOnce(projectId, requestId, qp, eventModel, options, token, resolvedHost);
-  } catch (err) {
-    const msg = err.message || "";
-    if (msg.includes("401") || msg.includes("403") || msg.includes("auth")) {
-      invalidateAccessTokenForRetry(resolvedHost);
-      const newToken = await getToken(resolvedHost);
-      return wsQueryOnce(projectId, requestId, qp, eventModel, options, newToken, resolvedHost);
-    }
-    throw err;
-  }
-}
-async function querySql(projectId, sql, hostUrl) {
-  const requestId = genRequestId("sqlIde");
-  const qp = {
-    events: { sql },
-    eventView: { sqlViewParams: [] }
-  };
-  return wsQuery(projectId, requestId, qp, 10, {
-    searchSource: "sqlIde",
-    querySource: "sqlIde"
-  }, hostUrl);
-}
-async function queryReportData(projectId, reportId, qp, eventModel, options = {}, hostUrl) {
-  const dashboardId = options.dashboardId || 0;
-  const requestId = genRequestId(`${projectId}_${dashboardId}_${reportId}`);
-  return wsQuery(projectId, requestId, qp, eventModel, {
-    searchSource: options.searchSource || "model_search",
-    querySource: options.querySource || "module",
-    ...options
-  }, hostUrl);
-}
-export {
-  httpDelete,
-  httpGet,
-  httpPost,
-  httpRequest,
-  httpUpload,
-  queryReportData,
-  querySql,
-  wsQuery
-};