@k256/sdk 0.4.0 → 0.5.0

package/src/ws/decoder.ts

@@ -9,7 +9,7 @@
 
 import { base58Encode } from '../utils/base58';
 import type { BlockMiniStats, TrendDirection } from '../types';
-import { MessageType, type DecodedMessage, type PoolUpdateMessage, type FeeMarketMessage } from './types';
+import { MessageType, type DecodedMessage, type PoolUpdateMessage, type FeeMarketMessage, type PriceEntry, type PriceBatchMessage, type PriceSnapshotMessage } from './types';
 
 /**
  * Decode a binary WebSocket message from K2
@@ -232,11 +232,66 @@ export function decodeMessage(data: ArrayBuffer): DecodedMessage | null {
       };
     }
 
+    case MessageType.PriceUpdate: {
+      // Single price entry (56 bytes, no count prefix):
+      // [mint:32B][usd_price:u64 LE][slot:u64 LE][timestamp_ms:u64 LE]
+      // K2 currently doesn't send 0x11, but handle for forward compat.
+      if (payload.byteLength < 56) return null;
+      const mintBytes = new Uint8Array(payload, 0, 32);
+      return {
+        type: 'price_update',
+        data: {
+          mint: base58Encode(mintBytes),
+          usdPrice: Number(payloadView.getBigUint64(32, true)) / 1e12,
+          slot: Number(payloadView.getBigUint64(40, true)),
+          timestampMs: Number(payloadView.getBigUint64(48, true)),
+        },
+      };
+    }
+
+    case MessageType.PriceBatch:
+    case MessageType.PriceSnapshot: {
+      // Both use identical binary format from serialize_price_message():
+      // [count:u16 LE][entry₁:56B]...[entryₙ:56B]
+      // Each entry: [mint:32B][usd_price:u64 LE][slot:u64 LE][timestamp_ms:u64 LE]
+      if (payload.byteLength < 2) return null;
+      const entries = decodePriceEntries(payload);
+      const priceType = msgType === MessageType.PriceSnapshot ? 'price_snapshot' : 'price_batch';
+      return { type: priceType, data: entries } as PriceBatchMessage | PriceSnapshotMessage;
+    }
+
     default:
       return null;
   }
 }
 
+/**
+ * Decode price entries from a PriceBatch or PriceSnapshot payload.
+ *
+ * Wire format: [count:u16 LE][entry₁:56B]...[entryₙ:56B]
+ */
+export function decodePriceEntries(payload: ArrayBuffer): PriceEntry[] {
+  const view = new DataView(payload);
+  if (payload.byteLength < 2) return [];
+
+  const count = view.getUint16(0, true);
+  const entries: PriceEntry[] = [];
+  let offset = 2;
+
+  for (let i = 0; i < count && offset + 56 <= payload.byteLength; i++) {
+    const mintBytes = new Uint8Array(payload, offset, 32);
+    entries.push({
+      mint: base58Encode(mintBytes),
+      usdPrice: Number(view.getBigUint64(offset + 32, true)) / 1e12,
+      slot: Number(view.getBigUint64(offset + 40, true)),
+      timestampMs: Number(view.getBigUint64(offset + 48, true)),
+    });
+    offset += 56;
+  }
+
+  return entries;
+}
+
 /**
  * Decode a batch of pool updates
  * 

package/src/ws/index.ts

@@ -33,10 +33,11 @@ export type {
   ConnectionState,
   SubscribeOptions,
   SubscribeQuoteOptions,
+  SubscribePriceOptions,
 } from './client';
 
 // Decoder (for advanced usage)
-export { decodeMessage, decodePoolUpdateBatch } from './decoder';
+export { decodeMessage, decodePoolUpdateBatch, decodePriceEntries } from './decoder';
 
 // Types
 export { MessageType } from './types';
@@ -48,6 +49,10 @@ export type {
   BlockStatsMessage,
   BlockhashMessage,
   QuoteMessage,
+  PriceEntry,
+  PriceUpdateMessage,
+  PriceBatchMessage,
+  PriceSnapshotMessage,
   HeartbeatMessage,
   ErrorMessage,
   SubscribedMessage,

package/src/ws/types.ts

@@ -44,6 +44,16 @@ export const MessageType = {
   PoolUpdateBatch: 0x0e,
   /** Block-level statistics (v3) */
   BlockStats: 0x0f,
+  /** Subscribe to price updates (JSON) - Client → Server */
+  SubscribePrice: 0x10,
+  /** Single price update (bincode 56B) - Server → Client */
+  PriceUpdate: 0x11,
+  /** Batched price updates (bincode [u16 count][entries...]) - Server → Client */
+  PriceBatch: 0x12,
+  /** Initial price snapshot (same binary format as PriceBatch) - Server → Client */
+  PriceSnapshot: 0x13,
+  /** Unsubscribe from prices (no payload) - Client → Server */
+  UnsubscribePrice: 0x14,
   /** Error message (UTF-8 string) */
   Error: 0xff,
 } as const;
@@ -214,6 +224,42 @@ export interface QuoteSubscribedMessage {
   };
 }
 
+/**
+ * Single decoded price entry (shared by PriceUpdate, PriceBatch, PriceSnapshot)
+ */
+export interface PriceEntry {
+  mint: string;
+  usdPrice: number;
+  slot: number;
+  timestampMs: number;
+}
+
+/**
+ * Single price update from binary message (0x11)
+ * Note: K2 currently sends all updates via PriceBatch (0x12)
+ */
+export interface PriceUpdateMessage {
+  type: 'price_update';
+  data: PriceEntry;
+}
+
+/**
+ * Batched price updates from binary message (0x12)
+ */
+export interface PriceBatchMessage {
+  type: 'price_batch';
+  data: PriceEntry[];
+}
+
+/**
+ * Initial price snapshot from binary message (0x13)
+ * Same binary format as PriceBatch — sent once after subscribe_price
+ */
+export interface PriceSnapshotMessage {
+  type: 'price_snapshot';
+  data: PriceEntry[];
+}
+
 /**
  * Error message from server
  */
@@ -243,6 +289,9 @@ export type DecodedMessage =
   | BlockStatsMessage
   | BlockhashMessage
   | QuoteMessage
+  | PriceUpdateMessage
+  | PriceBatchMessage
+  | PriceSnapshotMessage
   | HeartbeatMessage
   | SubscribedMessage
   | QuoteSubscribedMessage