connectbase-client 3.0.1 → 3.2.0

package/dist/index.mjs

@@ -7632,6 +7632,176 @@ var AIAPI = class {
   }
 };
 
+// src/api/endpoint.ts
+var EndpointAPI = class {
+  constructor(http) {
+    this.http = http;
+  }
+  /**
+   * 라벨 + path 로 사용자 PC 모델 호출. fetch() 시그니처 호환.
+   *
+   * 동작:
+   *  - URL 조립: `${baseUrl}/v1/proxy/${label}${path}`
+   *  - X-Public-Key 헤더 자동 주입 (호출자가 명시하면 그 값 우선)
+   *  - body / method / 추가 헤더 / signal 그대로 전달
+   *  - 응답 그대로 반환 (Response 객체) — 스트리밍은 res.body 로 read
+   *
+   * @param label - 콘솔에서 등록한 endpoint 라벨 (예: "comfyui-main")
+   * @param init - fetch() 의 RequestInit + path. path 는 사용자 모델 서버의 엔드포인트 경로 (예: "/prompt", "/v1/chat/completions").
+   */
+  async call(label, init) {
+    if (!label) {
+      throw new Error("EndpointAPI.call: label required");
+    }
+    if (!init.path || !init.path.startsWith("/")) {
+      throw new Error(
+        `EndpointAPI.call: path must start with '/', got ${JSON.stringify(init.path)}`
+      );
+    }
+    const url = this.url(label, init.path);
+    const headers = new Headers(init.headers ?? {});
+    if (!headers.has("X-Public-Key")) {
+      const pk = this.http.getPublicKey();
+      if (!pk) {
+        throw new Error(
+          "EndpointAPI.call: publicKey not configured. Pass `publicKey` to ConnectBase constructor."
+        );
+      }
+      headers.set("X-Public-Key", pk);
+    }
+    return fetch(url, {
+      method: init.method ?? "GET",
+      headers,
+      body: init.body,
+      signal: init.signal,
+      // streaming 응답 (SSE / chunked) 을 바로 reader 로 받기 위해 'manual' redirect 회피.
+      redirect: "follow"
+    });
+  }
+  /**
+   * 라벨 + path 의 최종 호출 URL 을 조립해서 반환. fetch / WebSocket / new Image()
+   * 같이 SDK 의 `call()` 이 아닌 직접 호출이 필요할 때 사용.
+   *
+   * 주의: WebSocket 같이 `X-Public-Key` 헤더를 못 보내는 경로면, 모델 서버가
+   * 자체 토큰으로 별도 인증을 해야 합니다 (ConnectBase 는 인증 게이트가 아닌
+   * dumb pipe 라 path 만으로 인증하지 않음).
+   *
+   * @example
+   * ```typescript
+   * // 결과 이미지를 <img> 로 바로 그리기
+   * img.src = `${cb.endpoint.url("comfyui-main", `/view?filename=${name}`)}` +
+   *           `&__pk=${publicKey}` // 모델 서버가 직접 검증하는 토큰
+   * ```
+   */
+  url(label, path) {
+    if (!label) throw new Error("EndpointAPI.url: label required");
+    if (!path || !path.startsWith("/")) {
+      throw new Error(
+        `EndpointAPI.url: path must start with '/', got ${JSON.stringify(path)}`
+      );
+    }
+    const baseUrl = this.http.getBaseUrl().replace(/\/+$/, "");
+    return `${baseUrl}/v1/proxy/${encodeURIComponent(label)}${path}`;
+  }
+  /**
+   * 같은 endpoint 호출을 `predicate` 가 값을 반환할 때까지 주기적으로 반복.
+   * ComfyUI `/history/{id}`, A1111 `/sdapi/v1/progress`, 자체 큐 API 처럼
+   * "작업 제출 → 폴링" 패턴을 한 줄로 처리하기 위한 헬퍼.
+   *
+   * 동작:
+   *  - `call(label, init)` → predicate(response) 호출
+   *  - predicate 가 `undefined` 면 `intervalMs` 만큼 대기 후 재시도
+   *  - predicate 가 값을 반환하면 그 값을 즉시 resolve
+   *  - `timeoutMs` 초과 또는 `signal` abort 시 reject
+   *  - HTTP 5xx/네트워크 오류는 재시도, 4xx 는 즉시 reject (작업 자체가 잘못된 경우)
+   *
+   * predicate 는 같은 Response 를 한 번만 읽을 수 있으므로, 헬퍼 내부에서
+   * `res.clone().json()` 형태로 안전하게 파싱한 뒤 호출자에게 전달.
+   *
+   * @example ComfyUI 결과 폴링
+   * ```typescript
+   * type Hist = Record<string, { outputs: Record<string, { images?: { filename: string }[] }> }>
+   *
+   * const filename = await cb.endpoint.pollUntil<string>(
+   *     "comfyui-main",
+   *     { path: `/history/${promptId}` },
+   *     (data: Hist) => {
+   *         const entry = data[promptId]
+   *         if (!entry) return undefined // 아직 큐에 있음
+   *         for (const out of Object.values(entry.outputs)) {
+   *             const img = out.images?.[0]
+   *             if (img) return img.filename
+   *         }
+   *         return undefined
+   *     },
+   *     { intervalMs: 1000, timeoutMs: 5 * 60_000 },
+   * )
+   * ```
+   */
+  async pollUntil(label, init, predicate, opts = {}) {
+    const intervalMs = opts.intervalMs ?? 1500;
+    const timeoutMs = opts.timeoutMs ?? 5 * 6e4;
+    const parser = opts.parse ?? "json";
+    const start = Date.now();
+    while (true) {
+      if (opts.signal?.aborted) {
+        throw new DOMException("aborted", "AbortError");
+      }
+      if (Date.now() - start > timeoutMs) {
+        throw new Error(
+          `EndpointAPI.pollUntil: timeout after ${timeoutMs}ms (label=${label}, path=${init.path})`
+        );
+      }
+      let res;
+      try {
+        res = await this.call(label, { ...init, signal: opts.signal });
+      } catch (err) {
+        if (err.name === "AbortError") throw err;
+        await sleep(intervalMs, opts.signal);
+        continue;
+      }
+      if (res.status >= 500) {
+        await drainBody(res);
+        await sleep(intervalMs, opts.signal);
+        continue;
+      }
+      if (res.status >= 400) {
+        const text = await res.text().catch(() => "");
+        throw new Error(
+          `EndpointAPI.pollUntil: ${res.status} ${res.statusText} (label=${label}, path=${init.path})${text ? ` \u2014 ${text.slice(0, 200)}` : ""}`
+        );
+      }
+      let body;
+      if (parser === "text") body = await res.text();
+      else if (parser === "none") body = void 0;
+      else body = await res.json().catch(() => void 0);
+      const result = await predicate(body, res);
+      if (result !== void 0) return result;
+      await sleep(intervalMs, opts.signal);
+    }
+  }
+};
+async function sleep(ms, signal) {
+  if (signal?.aborted) throw new DOMException("aborted", "AbortError");
+  return new Promise((resolve, reject) => {
+    const t = setTimeout(() => {
+      signal?.removeEventListener("abort", onAbort);
+      resolve();
+    }, ms);
+    const onAbort = () => {
+      clearTimeout(t);
+      reject(new DOMException("aborted", "AbortError"));
+    };
+    signal?.addEventListener("abort", onAbort, { once: true });
+  });
+}
+async function drainBody(res) {
+  try {
+    await res.text();
+  } catch {
+  }
+}
+
 // src/api/queue.ts
 var QueueAPI = class {
   constructor(http) {
@@ -9170,6 +9340,7 @@ var ConnectBase = class {
     this.ai = new AIAPI(this.http);
     this.queue = new QueueAPI(this.http);
     this.analytics = new AnalyticsAPI(this.http);
+    this.endpoint = new EndpointAPI(this.http);
     this.auth._attachAnalytics(this.analytics);
   }
   /**
@@ -9198,6 +9369,7 @@ export {
   ApiError,
   AuthError,
   ConnectBase,
+  EndpointAPI,
   GameAPI,
   GameRoom,
   GameRoomTransport,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "connectbase-client",
-  "version": "3.0.1",
+  "version": "3.2.0",
   "description": "Connect Base JavaScript/TypeScript SDK for browser and Node.js",
   "repository": {
     "type": "git",