@upbit-official/upbit-sdk 1.0.0 → 1.1.0

package/src/resources/ws-info/index.ts

@@ -0,0 +1,44 @@
+export {
+  WsInfo,
+  type BollingerBandsIndicator,
+  type BollingerBandsIndicatorSignal,
+  type Indicator,
+  type IndicatorSignal,
+  type IndicatorTimeframe,
+  type IndicatorType,
+  type MaCrossoverIndicator,
+  type MaCrossoverIndicatorSignal,
+  type MacdIndicator,
+  type MacdIndicatorSignal,
+  type MovingAverageIndicator,
+  type MovingAverageIndicatorSignal,
+  type RsiIndicator,
+  type RsiIndicatorSignal,
+  type RsiOverboughtIndicatorSignal,
+  type RsiOverboughtIndicatorSignalRequest,
+  type RsiOversoldIndicatorSignal,
+  type RsiOversoldIndicatorSignalRequest,
+  type VolumeIndicator,
+  type VolumeIndicatorSignal,
+  type ConnectInfoClientEvent,
+  type ConnectInfoServerEvent,
+} from './ws-info';
+export { type WsInfoWSClientOptions } from './ws';
+export { type WsInfoWSReconnectOptions } from './ws';
+
+export type { SimpleInfoServerEvent } from '../../internal/ws-simple-types';
+export type {
+  SimpleVolumeIndicator,
+  SimpleRsiIndicator,
+  SimpleMovingAverageIndicator,
+  SimpleMaCrossoverIndicator,
+  SimpleBollingerBandsIndicator,
+  SimpleMacdIndicator,
+  SimpleVolumeIndicatorSignal,
+  SimpleRsiOverboughtIndicatorSignal,
+  SimpleRsiOversoldIndicatorSignal,
+  SimpleMovingAverageIndicatorSignal,
+  SimpleMaCrossoverIndicatorSignal,
+  SimpleBollingerBandsIndicatorSignal,
+  SimpleMacdIndicatorSignal,
+} from '../../internal/ws-simple-types';

package/src/resources/ws-info/internal-base.ts

@@ -0,0 +1,96 @@
+import * as WsInfoAPI from './ws-info';
+import { Upbit } from '../../client';
+import { EventEmitter } from '../../core/EventEmitter';
+import { UpbitError } from '../../core/error';
+import type { ReconnectingEvent } from '../../internal/ws';
+
+export type WsInfoServerMessage = WsInfoAPI.ConnectInfoServerEvent;
+
+export type WsInfoStreamMessage =
+  | { type: 'connecting' | 'open' | 'closing' }
+  | { type: 'close'; code: number; reason: string }
+  | { type: 'reconnecting'; reconnect: ReconnectingEvent }
+  | { type: 'reconnected' }
+  | { type: 'message'; message: WsInfoServerMessage }
+  | { type: 'error'; error: WebSocketError };
+
+export class WebSocketError extends UpbitError {
+  /**
+   * The error data that the API sent back in an error event.
+   */
+  error?: unknown | undefined;
+
+  constructor(message: string, event: unknown | null) {
+    super(message);
+
+    this.error = event ?? undefined;
+  }
+}
+
+type Simplify<T> = { [KeyType in keyof T]: T[KeyType] } & {};
+
+type WebSocketEvents = Simplify<
+  {
+    event: (event: WsInfoServerMessage) => void;
+    error: (error: WebSocketError) => void;
+    close: (code: number, reason: string) => void;
+    reconnecting: (event: ReconnectingEvent) => void;
+    reconnected: () => void;
+  } & {
+    [EventType in WsInfoAPI.ConnectInfoServerEvent['type']]: (
+      event: Extract<WsInfoAPI.ConnectInfoServerEvent, { type?: EventType }>,
+    ) => unknown;
+  }
+>;
+
+export abstract class WsInfoEmitter extends EventEmitter<WebSocketEvents> {
+  /**
+   * Send an event to the API.
+   */
+  abstract send(event: WsInfoAPI.ConnectInfoClientEvent): void;
+
+  /**
+   * Close the WebSocket connection.
+   */
+  abstract close(props?: { code: number; reason: string }): void;
+
+  protected _onError(event: null, message: string, cause: any): void;
+  protected _onError(event: unknown, message?: string | undefined): void;
+  protected _onError(event: unknown | null, message?: string | undefined, cause?: any): void {
+    message = message ?? safeJSONStringify(event) ?? 'unknown error';
+
+    if (!this._hasListener('error')) {
+      const error = new WebSocketError(
+        message +
+          `\n\nTo resolve these unhandled rejection errors you should bind an \`error\` callback, e.g. \`ws.on('error', (error) => ...)\` `,
+        event,
+      );
+      // @ts-ignore
+      error.cause = cause;
+      Promise.reject(error);
+      return;
+    }
+
+    const error = new WebSocketError(message, event);
+    // @ts-ignore
+    error.cause = cause;
+
+    this._emit('error', error);
+  }
+}
+
+export function buildURL(client: Upbit, parameters?: Record<string, unknown> | null): URL {
+  const { ...query } = parameters ?? {};
+  const endpoint = '/websocket/v1/info';
+  const url = new URL(client.buildURL(endpoint, query, undefined));
+  url.protocol = url.protocol === 'http:' || url.protocol === 'ws:' ? 'ws:' : 'wss:';
+  return url;
+}
+
+function safeJSONStringify(value: unknown): string | null {
+  try {
+    return JSON.stringify(value);
+  } catch {
+    return null;
+  }
+}