@lasersell/lasersell-sdk 0.3.1 → 1.0.0

package/src/stream/proto.ts

@@ -48,11 +48,31 @@ export interface MarketContextMsg {
   raydium_cpmm?: RaydiumCpmmContextMsg;
 }
 
+export interface TakeProfitLevelMsg {
+  profit_pct: number;
+  sell_pct: number;
+  trailing_stop_pct: number;
+}
+
 export interface StrategyConfigMsg {
   target_profit_pct: number;
   stop_loss_pct: number;
   trailing_stop_pct?: number;
   sell_on_graduation?: boolean;
+  take_profit_levels?: TakeProfitLevelMsg[];
+  liquidity_guard?: boolean;
+  breakeven_trail_pct?: number;
+}
+
+export interface AutoBuyConfigMsg {
+  wallet_pubkey: string;
+  amount_quote_units: number;
+  amount_usd1_units?: number;
+}
+
+export interface WatchWalletEntryMsg {
+  pubkey: string;
+  auto_buy?: AutoBuyConfigMsg;
 }
 
 export interface LimitsMsg {
@@ -62,6 +82,7 @@ export interface LimitsMsg {
   max_wallets_per_session: number;
   max_positions_per_wallet: number;
   max_sessions_per_api_key: number;
+  max_watch_wallets_per_session: number;
 }
 
 export interface PingClientMessage {
@@ -75,6 +96,7 @@ export interface ConfigureClientMessage {
   strategy: StrategyConfigMsg;
   send_mode?: string;
   tip_lamports?: number;
+  watch_wallets?: WatchWalletEntryMsg[];
 }
 
 export interface UpdateStrategyClientMessage {
@@ -100,13 +122,26 @@ export interface UpdateWalletsClientMessage {
   wallet_pubkeys: string[];
 }
 
+export interface UpdateWatchWalletsClientMessage {
+  type: "update_watch_wallets";
+  watch_wallets: WatchWalletEntryMsg[];
+}
+
+export interface UpdatePositionStrategyClientMessage {
+  type: "update_position_strategy";
+  position_id: number;
+  strategy: StrategyConfigMsg;
+}
+
 export type ClientMessage =
   | PingClientMessage
   | ConfigureClientMessage
   | UpdateStrategyClientMessage
   | ClosePositionClientMessage
   | RequestExitSignalClientMessage
-  | UpdateWalletsClientMessage;
+  | UpdateWalletsClientMessage
+  | UpdateWatchWalletsClientMessage
+  | UpdatePositionStrategyClientMessage;
 
 export interface HelloOkServerMessage {
   type: "hello_ok";
@@ -132,6 +167,9 @@ export interface PnlUpdateServerMessage {
   profit_units: number;
   proceeds_units: number;
   server_time_ms: number;
+  token_price_quote?: number;
+  market_cap_quote?: number;
+  watched: boolean;
 }
 
 export interface SlippageBandMsg {
@@ -146,6 +184,7 @@ export interface LiquiditySnapshotServerMessage {
   bands: SlippageBandMsg[];
   liquidity_trend: string;
   server_time_ms: number;
+  watched: boolean;
 }
 
 export interface BalanceUpdateServerMessage {
@@ -169,6 +208,15 @@ export interface PositionOpenedServerMessage {
   entry_quote_units: number;
   market_context?: MarketContextMsg;
   slot: number;
+  token_name?: string;
+  token_symbol?: string;
+  token_decimals?: number;
+  token_price_quote?: number;
+  market_cap_quote?: number;
+  pool_liquidity_quote?: number;
+  opened_at_ms?: number;
+  mirror_source?: string;
+  watched: boolean;
 }
 
 export interface PositionClosedServerMessage {
@@ -179,6 +227,8 @@ export interface PositionClosedServerMessage {
   token_account?: string;
   reason: string;
   slot: number;
+  mirror_source?: string;
+  watched: boolean;
 }
 
 export interface ExitSignalWithTxServerMessage {
@@ -195,6 +245,23 @@ export interface ExitSignalWithTxServerMessage {
   triggered_at_ms: number;
   market_context?: MarketContextMsg;
   unsigned_tx_b64: string;
+  sell_tokens?: number;
+  level_index?: number;
+  mirror_source?: string;
+  watched: boolean;
+}
+
+export interface TradeTickServerMessage {
+  type: "trade_tick";
+  position_id: number;
+  time_ms: number;
+  side: string;
+  token_amount: number;
+  quote_amount: number;
+  price_quote: number;
+  maker?: string;
+  tx_signature?: string;
+  watched: boolean;
 }
 
 export type ServerMessage =
@@ -206,7 +273,8 @@ export type ServerMessage =
   | BalanceUpdateServerMessage
   | PositionOpenedServerMessage
   | PositionClosedServerMessage
-  | ExitSignalWithTxServerMessage;
+  | ExitSignalWithTxServerMessage
+  | TradeTickServerMessage;
 
 export function clientMessageFromText(text: string): ClientMessage {
   let parsed: unknown;
@@ -244,6 +312,10 @@ export function clientMessageFromUnknown(value: unknown): ClientMessage {
       if (tip_lamports !== undefined) {
         msg.tip_lamports = tip_lamports;
       }
+      const watch_wallets = optionalWatchWallets(obj.watch_wallets);
+      if (watch_wallets !== undefined && watch_wallets.length > 0) {
+        msg.watch_wallets = watch_wallets;
+      }
       return msg;
     }
     case "update_strategy": {
@@ -287,6 +359,20 @@ export function clientMessageFromUnknown(value: unknown): ClientMessage {
         wallet_pubkeys,
       };
     }
+    case "update_watch_wallets": {
+      const watch_wallets = parseWatchWallets(obj.watch_wallets);
+      return {
+        type: "update_watch_wallets",
+        watch_wallets,
+      };
+    }
+    case "update_position_strategy": {
+      return {
+        type: "update_position_strategy",
+        position_id: asNumber(obj.position_id, "update_position_strategy.position_id"),
+        strategy: parseStrategyConfig(obj.strategy),
+      };
+    }
     default:
       throw new Error(`unsupported client message type: ${type}`);
   }
@@ -333,13 +419,24 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
       };
     }
     case "pnl_update": {
-      return {
+      const msg: PnlUpdateServerMessage = {
         type: "pnl_update",
         position_id: asNumber(obj.position_id, "pnl_update.position_id"),
         profit_units: asNumber(obj.profit_units, "pnl_update.profit_units"),
         proceeds_units: asNumber(obj.proceeds_units, "pnl_update.proceeds_units"),
         server_time_ms: asNumber(obj.server_time_ms, "pnl_update.server_time_ms"),
+        watched: false,
       };
+      const token_price_quote = optionalNumber(obj.token_price_quote, "pnl_update.token_price_quote");
+      if (token_price_quote !== undefined) {
+        msg.token_price_quote = token_price_quote;
+      }
+      const market_cap_quote = optionalNumber(obj.market_cap_quote, "pnl_update.market_cap_quote");
+      if (market_cap_quote !== undefined) {
+        msg.market_cap_quote = market_cap_quote;
+      }
+      msg.watched = optionalBoolean(obj.watched, "pnl_update.watched") ?? false;
+      return msg;
     }
     case "liquidity_snapshot": {
       const rawBands = obj.bands;
@@ -362,6 +459,7 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
         bands,
         liquidity_trend: asString(obj.liquidity_trend, "liquidity_snapshot.liquidity_trend"),
         server_time_ms: asNumber(obj.server_time_ms, "liquidity_snapshot.server_time_ms"),
+        watched: optionalBoolean(obj.watched, "liquidity_snapshot.watched") ?? false,
       };
     }
     case "balance_update": {
@@ -382,7 +480,7 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
       };
     }
     case "position_opened": {
-      return {
+      const msg: PositionOpenedServerMessage = {
         type: "position_opened",
         position_id: asNumber(obj.position_id, "position_opened.position_id"),
         wallet_pubkey: asString(
@@ -405,10 +503,45 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
           "position_opened.market_context",
         ),
         slot: asNumber(obj.slot, "position_opened.slot"),
+        watched: false,
       };
+      const token_name = optionalString(obj.token_name, "position_opened.token_name");
+      if (token_name !== undefined) {
+        msg.token_name = token_name;
+      }
+      const token_symbol = optionalString(obj.token_symbol, "position_opened.token_symbol");
+      if (token_symbol !== undefined) {
+        msg.token_symbol = token_symbol;
+      }
+      const token_decimals = optionalNumber(obj.token_decimals, "position_opened.token_decimals");
+      if (token_decimals !== undefined) {
+        msg.token_decimals = token_decimals;
+      }
+      const token_price_quote = optionalNumber(obj.token_price_quote, "position_opened.token_price_quote");
+      if (token_price_quote !== undefined) {
+        msg.token_price_quote = token_price_quote;
+      }
+      const market_cap_quote = optionalNumber(obj.market_cap_quote, "position_opened.market_cap_quote");
+      if (market_cap_quote !== undefined) {
+        msg.market_cap_quote = market_cap_quote;
+      }
+      const pool_liquidity_quote = optionalNumber(obj.pool_liquidity_quote, "position_opened.pool_liquidity_quote");
+      if (pool_liquidity_quote !== undefined) {
+        msg.pool_liquidity_quote = pool_liquidity_quote;
+      }
+      const opened_at_ms = optionalNumber(obj.opened_at_ms, "position_opened.opened_at_ms");
+      if (opened_at_ms !== undefined) {
+        msg.opened_at_ms = opened_at_ms;
+      }
+      const mirror_source = optionalString(obj.mirror_source, "position_opened.mirror_source");
+      if (mirror_source !== undefined) {
+        msg.mirror_source = mirror_source;
+      }
+      msg.watched = optionalBoolean(obj.watched, "position_opened.watched") ?? false;
+      return msg;
     }
     case "position_closed": {
-      return {
+      const msg: PositionClosedServerMessage = {
         type: "position_closed",
         position_id: asNumber(obj.position_id, "position_closed.position_id"),
         wallet_pubkey: asString(
@@ -422,10 +555,17 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
         ),
         reason: asString(obj.reason, "position_closed.reason"),
         slot: asNumber(obj.slot, "position_closed.slot"),
+        watched: false,
       };
+      const mirror_source = optionalString(obj.mirror_source, "position_closed.mirror_source");
+      if (mirror_source !== undefined) {
+        msg.mirror_source = mirror_source;
+      }
+      msg.watched = optionalBoolean(obj.watched, "position_closed.watched") ?? false;
+      return msg;
     }
     case "exit_signal_with_tx": {
-      return {
+      const msg: ExitSignalWithTxServerMessage = {
         type: "exit_signal_with_tx",
         session_id: asNumber(obj.session_id, "exit_signal_with_tx.session_id"),
         position_id: asNumber(obj.position_id, "exit_signal_with_tx.position_id"),
@@ -460,6 +600,35 @@ export function serverMessageFromUnknown(value: unknown): ServerMessage {
           obj.unsigned_tx_b64,
           "exit_signal_with_tx.unsigned_tx_b64",
         ),
+        watched: false,
+      };
+      const sell_tokens = optionalNumber(obj.sell_tokens, "exit_signal_with_tx.sell_tokens");
+      if (sell_tokens !== undefined) {
+        msg.sell_tokens = sell_tokens;
+      }
+      const level_index = optionalNumber(obj.level_index, "exit_signal_with_tx.level_index");
+      if (level_index !== undefined) {
+        msg.level_index = level_index;
+      }
+      const mirror_source = optionalString(obj.mirror_source, "exit_signal_with_tx.mirror_source");
+      if (mirror_source !== undefined) {
+        msg.mirror_source = mirror_source;
+      }
+      msg.watched = optionalBoolean(obj.watched, "exit_signal_with_tx.watched") ?? false;
+      return msg;
+    }
+    case "trade_tick": {
+      return {
+        type: "trade_tick",
+        position_id: asNumber(obj.position_id, "trade_tick.position_id"),
+        time_ms: asNumber(obj.time_ms, "trade_tick.time_ms"),
+        side: asString(obj.side, "trade_tick.side"),
+        token_amount: asNumber(obj.token_amount, "trade_tick.token_amount"),
+        quote_amount: asNumber(obj.quote_amount, "trade_tick.quote_amount"),
+        price_quote: asNumber(obj.price_quote, "trade_tick.price_quote"),
+        maker: optionalString(obj.maker, "trade_tick.maker"),
+        tx_signature: optionalString(obj.tx_signature, "trade_tick.tx_signature"),
+        watched: optionalBoolean(obj.watched, "trade_tick.watched") ?? false,
       };
     }
     default:
@@ -518,6 +687,34 @@ function parseStrategyConfig(value: unknown): StrategyConfigMsg {
     result.sell_on_graduation = sellOnGraduation;
   }
 
+  const rawLevels = obj.take_profit_levels;
+  if (Array.isArray(rawLevels)) {
+    result.take_profit_levels = rawLevels.map(
+      (item: unknown, idx: number) => {
+        const level = asRecord(item, `strategy.take_profit_levels[${idx}]`);
+        return {
+          profit_pct: asNumber(level.profit_pct, `strategy.take_profit_levels[${idx}].profit_pct`),
+          sell_pct: asNumber(level.sell_pct, `strategy.take_profit_levels[${idx}].sell_pct`),
+          trailing_stop_pct: asNumber(level.trailing_stop_pct, `strategy.take_profit_levels[${idx}].trailing_stop_pct`),
+        };
+      },
+    );
+  } else {
+    result.take_profit_levels = [];
+  }
+
+  const liquidityGuard = optionalBoolean(
+    obj.liquidity_guard,
+    "strategy.liquidity_guard",
+  );
+  result.liquidity_guard = liquidityGuard ?? false;
+
+  const breakevenTrailPct = optionalNumber(
+    obj.breakeven_trail_pct,
+    "strategy.breakeven_trail_pct",
+  );
+  result.breakeven_trail_pct = breakevenTrailPct ?? 0;
+
   return result;
 }
 
@@ -545,6 +742,11 @@ function parseLimits(value: unknown): LimitsMsg {
         obj.max_sessions_per_api_key,
         "limits.max_sessions_per_api_key",
       ) ?? 0,
+    max_watch_wallets_per_session:
+      optionalNumber(
+        obj.max_watch_wallets_per_session,
+        "limits.max_watch_wallets_per_session",
+      ) ?? 0,
   };
 }
 
@@ -594,6 +796,38 @@ function parseMarketType(value: unknown, path: string): MarketTypeMsg {
   }
 }
 
+function parseWatchWallets(value: unknown): WatchWalletEntryMsg[] {
+  if (!Array.isArray(value)) {
+    throw new Error("watch_wallets must be an array");
+  }
+  return value.map((item: unknown, idx: number) => {
+    const entry = asRecord(item, `watch_wallets[${idx}]`);
+    const result: WatchWalletEntryMsg = {
+      pubkey: asString(entry.pubkey, `watch_wallets[${idx}].pubkey`),
+    };
+    if (entry.auto_buy !== undefined && entry.auto_buy !== null) {
+      const ab = asRecord(entry.auto_buy, `watch_wallets[${idx}].auto_buy`);
+      const autoBuy: AutoBuyConfigMsg = {
+        wallet_pubkey: asString(ab.wallet_pubkey, `watch_wallets[${idx}].auto_buy.wallet_pubkey`),
+        amount_quote_units: asNumber(ab.amount_quote_units, `watch_wallets[${idx}].auto_buy.amount_quote_units`),
+      };
+      const usd1 = optionalNumber(ab.amount_usd1_units, `watch_wallets[${idx}].auto_buy.amount_usd1_units`);
+      if (usd1 !== undefined) {
+        autoBuy.amount_usd1_units = usd1;
+      }
+      result.auto_buy = autoBuy;
+    }
+    return result;
+  });
+}
+
+function optionalWatchWallets(value: unknown): WatchWalletEntryMsg[] | undefined {
+  if (value === undefined || value === null) {
+    return undefined;
+  }
+  return parseWatchWallets(value);
+}
+
 function optionalBoolean(value: unknown, path: string): boolean | undefined {
   if (value === undefined || value === null) {
     return undefined;

package/src/stream/session.ts

@@ -23,6 +23,15 @@ export interface PositionHandle {
   token_program?: string;
   tokens: number;
   entry_quote_units: number;
+  token_name?: string;
+  token_symbol?: string;
+  token_decimals?: number;
+  market_type?: string;
+  launch_platform?: string;
+  token_price_quote?: number;
+  market_cap_quote?: number;
+  pool_liquidity_quote?: number;
+  opened_at_ms?: number;
 }
 
 export type StreamEvent =
@@ -54,6 +63,11 @@ export type StreamEvent =
       type: "liquidity_snapshot";
       handle?: PositionHandle;
       message: ServerMessage;
+    }
+  | {
+      type: "trade_tick";
+      handle?: PositionHandle;
+      message: ServerMessage;
     };
 
 export class StreamSession {
@@ -174,14 +188,24 @@ export class StreamSession {
   private applyMessage(message: ServerMessage): StreamEvent {
     switch (message.type) {
       case "position_opened": {
+        const msg = message;
         const handle: PositionHandle = {
-          position_id: message.position_id,
-          token_account: message.token_account,
-          wallet_pubkey: message.wallet_pubkey,
-          mint: message.mint,
-          token_program: message.token_program,
-          tokens: message.tokens,
-          entry_quote_units: message.entry_quote_units,
+          position_id: msg.position_id,
+          token_account: msg.token_account,
+          wallet_pubkey: msg.wallet_pubkey,
+          mint: msg.mint,
+          token_program: msg.token_program,
+          tokens: msg.tokens,
+          entry_quote_units: msg.entry_quote_units,
+          token_name: msg.token_name,
+          token_symbol: msg.token_symbol,
+          token_decimals: msg.token_decimals,
+          market_type: msg.market_context?.market_type?.toLowerCase(),
+          launch_platform: undefined,
+          token_price_quote: msg.token_price_quote,
+          market_cap_quote: msg.market_cap_quote,
+          pool_liquidity_quote: msg.pool_liquidity_quote,
+          opened_at_ms: msg.opened_at_ms,
         };
 
         this.positionsById.set(message.position_id, handle);
@@ -237,6 +261,14 @@ export class StreamSession {
           message,
         };
       }
+      case "trade_tick": {
+        const handle = this.findPosition(message.position_id);
+        return {
+          type: "trade_tick",
+          handle: handle ? { ...handle } : undefined,
+          message,
+        };
+      }
       default: {
         return {
           type: "message",
@@ -403,5 +435,8 @@ function defaultStrategy(): StrategyConfigMsg {
   return {
     target_profit_pct: 0,
     stop_loss_pct: 0,
+    take_profit_levels: [],
+    liquidity_guard: false,
+    breakeven_trail_pct: 0,
   };
 }