1use solana_sdk::pubkey::Pubkey;
13
14use crate::core::events::*;
15
16#[inline(always)]
28pub fn merge_events(base: &mut DexEvent, inner: DexEvent) {
29 use DexEvent::*;
30
31 match (base, inner) {
32 (PumpFunTrade(b), PumpFunTrade(i))
34 | (PumpFunTrade(b), PumpFunBuy(i))
35 | (PumpFunTrade(b), PumpFunSell(i))
36 | (PumpFunTrade(b), PumpFunBuyExactSolIn(i))
37 | (PumpFunBuy(b), PumpFunTrade(i))
38 | (PumpFunBuy(b), PumpFunBuy(i))
39 | (PumpFunSell(b), PumpFunTrade(i))
40 | (PumpFunSell(b), PumpFunSell(i))
41 | (PumpFunBuyExactSolIn(b), PumpFunTrade(i))
42 | (PumpFunBuyExactSolIn(b), PumpFunBuyExactSolIn(i)) => merge_pumpfun_trade(b, i),
43
44 (PumpFunCreate(b), PumpFunCreate(i)) => merge_pumpfun_create(b, i),
45 (PumpFunCreateV2(b), PumpFunCreateV2(i)) => merge_pumpfun_create_v2(b, i),
46 (PumpFunMigrate(b), PumpFunMigrate(i)) => merge_pumpfun_migrate(b, i),
47 (PumpFunMigrateBondingCurveCreator(b), PumpFunMigrateBondingCurveCreator(i)) => {
48 merge_generic(b, i)
49 }
50
51 (PumpFeesCreateFeeSharingConfig(b), PumpFeesCreateFeeSharingConfig(i)) => {
53 merge_generic(b, i)
54 }
55 (PumpFeesInitializeFeeConfig(b), PumpFeesInitializeFeeConfig(i)) => merge_generic(b, i),
56 (PumpFeesResetFeeSharingConfig(b), PumpFeesResetFeeSharingConfig(i)) => merge_generic(b, i),
57 (PumpFeesRevokeFeeSharingAuthority(b), PumpFeesRevokeFeeSharingAuthority(i)) => {
58 merge_generic(b, i)
59 }
60 (PumpFeesTransferFeeSharingAuthority(b), PumpFeesTransferFeeSharingAuthority(i)) => {
61 merge_generic(b, i)
62 }
63 (PumpFeesUpdateAdmin(b), PumpFeesUpdateAdmin(i)) => merge_generic(b, i),
64 (PumpFeesUpdateFeeConfig(b), PumpFeesUpdateFeeConfig(i)) => merge_generic(b, i),
65 (PumpFeesUpdateFeeShares(b), PumpFeesUpdateFeeShares(i)) => merge_generic(b, i),
66 (PumpFeesUpsertFeeTiers(b), PumpFeesUpsertFeeTiers(i)) => merge_generic(b, i),
67
68 (PumpSwapTrade(b), PumpSwapTrade(i)) => merge_generic(b, i),
70 (PumpSwapBuy(b), PumpSwapBuy(i)) => merge_pumpswap_buy(b, i),
71 (PumpSwapSell(b), PumpSwapSell(i)) => merge_pumpswap_sell(b, i),
72 (PumpSwapCreatePool(b), PumpSwapCreatePool(i)) => merge_generic(b, i),
73 (PumpSwapLiquidityAdded(b), PumpSwapLiquidityAdded(i)) => merge_generic(b, i),
74 (PumpSwapLiquidityRemoved(b), PumpSwapLiquidityRemoved(i)) => merge_generic(b, i),
75
76 (RaydiumClmmSwap(b), RaydiumClmmSwap(i)) => merge_generic(b, i),
78 (RaydiumClmmIncreaseLiquidity(b), RaydiumClmmIncreaseLiquidity(i)) => merge_generic(b, i),
79 (RaydiumClmmDecreaseLiquidity(b), RaydiumClmmDecreaseLiquidity(i)) => merge_generic(b, i),
80 (RaydiumClmmLiquidityChange(b), RaydiumClmmLiquidityChange(i)) => merge_generic(b, i),
81 (RaydiumClmmConfigChange(b), RaydiumClmmConfigChange(i)) => merge_generic(b, i),
82 (RaydiumClmmCreatePersonalPosition(b), RaydiumClmmCreatePersonalPosition(i)) => {
83 merge_generic(b, i)
84 }
85 (RaydiumClmmLiquidityCalculate(b), RaydiumClmmLiquidityCalculate(i)) => merge_generic(b, i),
86 (RaydiumClmmOpenLimitOrder(b), RaydiumClmmOpenLimitOrder(i)) => merge_generic(b, i),
87 (RaydiumClmmIncreaseLimitOrder(b), RaydiumClmmIncreaseLimitOrder(i)) => merge_generic(b, i),
88 (RaydiumClmmDecreaseLimitOrder(b), RaydiumClmmDecreaseLimitOrder(i)) => merge_generic(b, i),
89 (RaydiumClmmSettleLimitOrder(b), RaydiumClmmSettleLimitOrder(i)) => merge_generic(b, i),
90 (RaydiumClmmUpdateRewardInfos(b), RaydiumClmmUpdateRewardInfos(i)) => merge_generic(b, i),
91 (RaydiumClmmCreatePool(b), RaydiumClmmCreatePool(i)) => merge_generic(b, i),
92 (RaydiumClmmOpenPosition(b), RaydiumClmmOpenPosition(i)) => merge_generic(b, i),
93 (RaydiumClmmClosePosition(b), RaydiumClmmClosePosition(i)) => merge_generic(b, i),
94 (RaydiumClmmOpenPositionWithTokenExtNft(b), RaydiumClmmOpenPositionWithTokenExtNft(i)) => {
95 merge_generic(b, i)
96 }
97 (RaydiumClmmCollectFee(b), RaydiumClmmCollectFee(i)) => merge_generic(b, i),
98
99 (RaydiumCpmmSwap(b), RaydiumCpmmSwap(i)) => merge_generic(b, i),
101 (RaydiumCpmmDeposit(b), RaydiumCpmmDeposit(i)) => merge_generic(b, i),
102 (RaydiumCpmmWithdraw(b), RaydiumCpmmWithdraw(i)) => merge_generic(b, i),
103 (RaydiumCpmmInitialize(b), RaydiumCpmmInitialize(i)) => merge_generic(b, i),
104
105 (RaydiumAmmV4Swap(b), RaydiumAmmV4Swap(i)) => merge_generic(b, i),
107 (RaydiumAmmV4Deposit(b), RaydiumAmmV4Deposit(i)) => merge_generic(b, i),
108 (RaydiumAmmV4Withdraw(b), RaydiumAmmV4Withdraw(i)) => merge_generic(b, i),
109 (RaydiumAmmV4Initialize2(b), RaydiumAmmV4Initialize2(i)) => merge_generic(b, i),
110 (RaydiumAmmV4WithdrawPnl(b), RaydiumAmmV4WithdrawPnl(i)) => merge_generic(b, i),
111
112 (OrcaWhirlpoolSwap(b), OrcaWhirlpoolSwap(i)) => merge_generic(b, i),
114 (OrcaWhirlpoolLiquidityIncreased(b), OrcaWhirlpoolLiquidityIncreased(i)) => {
115 merge_generic(b, i)
116 }
117 (OrcaWhirlpoolLiquidityDecreased(b), OrcaWhirlpoolLiquidityDecreased(i)) => {
118 merge_generic(b, i)
119 }
120 (OrcaWhirlpoolPoolInitialized(b), OrcaWhirlpoolPoolInitialized(i)) => merge_generic(b, i),
121
122 (MeteoraPoolsSwap(b), MeteoraPoolsSwap(i)) => merge_generic(b, i),
124 (MeteoraPoolsAddLiquidity(b), MeteoraPoolsAddLiquidity(i)) => merge_generic(b, i),
125 (MeteoraPoolsRemoveLiquidity(b), MeteoraPoolsRemoveLiquidity(i)) => merge_generic(b, i),
126 (MeteoraPoolsBootstrapLiquidity(b), MeteoraPoolsBootstrapLiquidity(i)) => {
127 merge_generic(b, i)
128 }
129 (MeteoraPoolsPoolCreated(b), MeteoraPoolsPoolCreated(i)) => merge_generic(b, i),
130 (MeteoraPoolsSetPoolFees(b), MeteoraPoolsSetPoolFees(i)) => merge_generic(b, i),
131
132 (MeteoraDammV2Swap(b), MeteoraDammV2Swap(i)) => merge_generic(b, i),
134 (MeteoraDammV2AddLiquidity(b), MeteoraDammV2AddLiquidity(i)) => merge_generic(b, i),
135 (MeteoraDammV2RemoveLiquidity(b), MeteoraDammV2RemoveLiquidity(i)) => merge_generic(b, i),
136 (MeteoraDammV2InitializePool(b), MeteoraDammV2InitializePool(i)) => merge_generic(b, i),
137 (MeteoraDammV2CreatePosition(b), MeteoraDammV2CreatePosition(i)) => merge_generic(b, i),
138 (MeteoraDammV2ClosePosition(b), MeteoraDammV2ClosePosition(i)) => merge_generic(b, i),
139
140 (MeteoraDlmmSwap(b), MeteoraDlmmSwap(i)) => merge_generic(b, i),
142 (MeteoraDlmmAddLiquidity(b), MeteoraDlmmAddLiquidity(i)) => merge_generic(b, i),
143 (MeteoraDlmmRemoveLiquidity(b), MeteoraDlmmRemoveLiquidity(i)) => merge_generic(b, i),
144 (MeteoraDlmmInitializePool(b), MeteoraDlmmInitializePool(i)) => merge_generic(b, i),
145 (MeteoraDlmmInitializeBinArray(b), MeteoraDlmmInitializeBinArray(i)) => merge_generic(b, i),
146 (MeteoraDlmmCreatePosition(b), MeteoraDlmmCreatePosition(i)) => merge_generic(b, i),
147 (MeteoraDlmmClosePosition(b), MeteoraDlmmClosePosition(i)) => merge_generic(b, i),
148 (MeteoraDlmmClaimFee(b), MeteoraDlmmClaimFee(i)) => merge_generic(b, i),
149
150 (RaydiumLaunchlabTrade(b), RaydiumLaunchlabTrade(i)) => merge_generic(b, i),
152 (RaydiumLaunchlabPoolCreate(b), RaydiumLaunchlabPoolCreate(i)) => merge_generic(b, i),
153 (RaydiumLaunchlabMigrateAmm(b), RaydiumLaunchlabMigrateAmm(i)) => merge_generic(b, i),
154
155 _ => {}
157 }
158}
159
160#[inline(always)]
167fn merge_generic<T>(base: &mut T, inner: T) {
168 *base = inner;
169}
170
171#[inline(always)]
176fn put_pk_if_set(to: &mut Pubkey, from: Pubkey) {
177 if from != Pubkey::default() {
178 *to = from;
179 }
180}
181
182#[inline(always)]
183fn put_pumpfun_quote_mint_if_set(to: &mut Pubkey, from: Pubkey) {
184 let from = normalize_pumpfun_quote_mint(from);
185 if from != Pubkey::default()
186 && (*to == Pubkey::default()
187 || is_pumpfun_solscan_sol_quote_mint(*to)
188 || !is_pumpfun_solscan_sol_quote_mint(from))
189 {
190 *to = from;
191 }
192}
193
194#[inline(always)]
195fn put_u64_if_nonzero(to: &mut u64, from: u64) {
196 if from != 0 {
197 *to = from;
198 }
199}
200
201#[inline(always)]
202fn put_i64_if_nonzero(to: &mut i64, from: i64) {
203 if from != 0 {
204 *to = from;
205 }
206}
207
208#[inline(always)]
218fn merge_pumpfun_trade(base: &mut PumpFunTradeEvent, inner: PumpFunTradeEvent) {
219 let leg = inner.sol_amount != 0 || inner.token_amount != 0;
220
221 put_pk_if_set(&mut base.mint, inner.mint);
222 put_pk_if_set(&mut base.user, inner.user);
223 put_pk_if_set(&mut base.fee_recipient, inner.fee_recipient);
224 put_pk_if_set(&mut base.creator, inner.creator);
225
226 if leg {
227 base.sol_amount = inner.sol_amount;
228 base.token_amount = inner.token_amount;
229 base.is_buy = inner.is_buy;
230 base.timestamp = inner.timestamp;
231 base.virtual_sol_reserves = inner.virtual_sol_reserves;
232 base.virtual_token_reserves = inner.virtual_token_reserves;
233 base.real_sol_reserves = inner.real_sol_reserves;
234 base.real_token_reserves = inner.real_token_reserves;
235 base.fee_basis_points = inner.fee_basis_points;
236 base.fee = inner.fee;
237 base.creator_fee_basis_points = inner.creator_fee_basis_points;
238 base.creator_fee = inner.creator_fee;
239 base.track_volume |= inner.track_volume;
240 base.total_unclaimed_tokens = inner.total_unclaimed_tokens;
241 base.total_claimed_tokens = inner.total_claimed_tokens;
242 base.current_sol_volume = inner.current_sol_volume;
243 base.last_update_timestamp = inner.last_update_timestamp;
244 if !inner.ix_name.is_empty() {
245 base.ix_name = inner.ix_name;
246 }
247 base.mayhem_mode |= inner.mayhem_mode;
248 put_u64_if_nonzero(&mut base.cashback_fee_basis_points, inner.cashback_fee_basis_points);
249 put_u64_if_nonzero(&mut base.cashback, inner.cashback);
250 put_u64_if_nonzero(&mut base.buyback_fee_basis_points, inner.buyback_fee_basis_points);
251 put_u64_if_nonzero(&mut base.buyback_fee, inner.buyback_fee);
252 if base.shareholders.is_empty() && !inner.shareholders.is_empty() {
253 base.shareholders = inner.shareholders;
254 }
255 put_pumpfun_quote_mint_if_set(&mut base.quote_mint, inner.quote_mint);
256 put_u64_if_nonzero(&mut base.quote_amount, inner.quote_amount);
257 put_u64_if_nonzero(&mut base.virtual_quote_reserves, inner.virtual_quote_reserves);
258 put_u64_if_nonzero(&mut base.real_quote_reserves, inner.real_quote_reserves);
259 base.is_cashback_coin |= inner.is_cashback_coin;
260 } else {
261 put_u64_if_nonzero(&mut base.fee, inner.fee);
262 put_u64_if_nonzero(&mut base.creator_fee, inner.creator_fee);
263 put_u64_if_nonzero(&mut base.fee_basis_points, inner.fee_basis_points);
264 put_u64_if_nonzero(&mut base.creator_fee_basis_points, inner.creator_fee_basis_points);
265 put_u64_if_nonzero(&mut base.virtual_sol_reserves, inner.virtual_sol_reserves);
266 put_u64_if_nonzero(&mut base.virtual_token_reserves, inner.virtual_token_reserves);
267 put_u64_if_nonzero(&mut base.real_sol_reserves, inner.real_sol_reserves);
268 put_u64_if_nonzero(&mut base.real_token_reserves, inner.real_token_reserves);
269 put_u64_if_nonzero(&mut base.total_unclaimed_tokens, inner.total_unclaimed_tokens);
270 put_u64_if_nonzero(&mut base.total_claimed_tokens, inner.total_claimed_tokens);
271 put_u64_if_nonzero(&mut base.current_sol_volume, inner.current_sol_volume);
272 put_u64_if_nonzero(&mut base.cashback_fee_basis_points, inner.cashback_fee_basis_points);
273 put_u64_if_nonzero(&mut base.cashback, inner.cashback);
274 put_u64_if_nonzero(&mut base.buyback_fee_basis_points, inner.buyback_fee_basis_points);
275 put_u64_if_nonzero(&mut base.buyback_fee, inner.buyback_fee);
276 if base.shareholders.is_empty() && !inner.shareholders.is_empty() {
277 base.shareholders = inner.shareholders;
278 }
279 put_pumpfun_quote_mint_if_set(&mut base.quote_mint, inner.quote_mint);
280 put_u64_if_nonzero(&mut base.quote_amount, inner.quote_amount);
281 put_u64_if_nonzero(&mut base.virtual_quote_reserves, inner.virtual_quote_reserves);
282 put_u64_if_nonzero(&mut base.real_quote_reserves, inner.real_quote_reserves);
283 put_i64_if_nonzero(&mut base.timestamp, inner.timestamp);
284 put_i64_if_nonzero(&mut base.last_update_timestamp, inner.last_update_timestamp);
285 if !inner.ix_name.is_empty() {
286 base.ix_name = inner.ix_name;
287 }
288 base.track_volume |= inner.track_volume;
289 base.mayhem_mode |= inner.mayhem_mode;
290 base.is_cashback_coin |= inner.is_cashback_coin;
291 }
292 put_u64_if_nonzero(&mut base.amount, inner.amount);
293 put_u64_if_nonzero(&mut base.max_sol_cost, inner.max_sol_cost);
294 put_u64_if_nonzero(&mut base.min_sol_output, inner.min_sol_output);
295 put_u64_if_nonzero(&mut base.spendable_sol_in, inner.spendable_sol_in);
296 put_u64_if_nonzero(&mut base.spendable_quote_in, inner.spendable_quote_in);
297 put_u64_if_nonzero(&mut base.min_tokens_out, inner.min_tokens_out);
298 put_pk_if_set(&mut base.global, inner.global);
299 put_pk_if_set(&mut base.bonding_curve, inner.bonding_curve);
300 put_pk_if_set(&mut base.bonding_curve_v2, inner.bonding_curve_v2);
301 put_pk_if_set(&mut base.associated_bonding_curve, inner.associated_bonding_curve);
302 put_pk_if_set(&mut base.associated_user, inner.associated_user);
303 put_pk_if_set(&mut base.system_program, inner.system_program);
304 put_pk_if_set(&mut base.token_program, inner.token_program);
305 put_pk_if_set(&mut base.quote_token_program, inner.quote_token_program);
306 put_pk_if_set(&mut base.associated_token_program, inner.associated_token_program);
307 put_pk_if_set(&mut base.creator_vault, inner.creator_vault);
308 put_pk_if_set(&mut base.associated_quote_fee_recipient, inner.associated_quote_fee_recipient);
309 put_pk_if_set(&mut base.buyback_fee_recipient, inner.buyback_fee_recipient);
310 put_pk_if_set(
311 &mut base.associated_quote_buyback_fee_recipient,
312 inner.associated_quote_buyback_fee_recipient,
313 );
314 put_pk_if_set(&mut base.associated_quote_bonding_curve, inner.associated_quote_bonding_curve);
315 put_pk_if_set(&mut base.associated_quote_user, inner.associated_quote_user);
316 put_pk_if_set(&mut base.associated_creator_vault, inner.associated_creator_vault);
317 put_pk_if_set(&mut base.sharing_config, inner.sharing_config);
318 put_pk_if_set(&mut base.event_authority, inner.event_authority);
319 put_pk_if_set(&mut base.program, inner.program);
320 put_pk_if_set(&mut base.global_volume_accumulator, inner.global_volume_accumulator);
321 put_pk_if_set(&mut base.user_volume_accumulator, inner.user_volume_accumulator);
322 put_pk_if_set(
323 &mut base.associated_user_volume_accumulator,
324 inner.associated_user_volume_accumulator,
325 );
326 put_pk_if_set(&mut base.fee_config, inner.fee_config);
327 put_pk_if_set(&mut base.fee_program, inner.fee_program);
328 if base.account.is_none() {
329 base.account = inner.account;
330 }
331
332 base.is_created_buy |= inner.is_created_buy;
333 }
335
336#[inline(always)]
338fn merge_pumpfun_create(base: &mut PumpFunCreateTokenEvent, inner: PumpFunCreateTokenEvent) {
339 base.name = inner.name;
341 base.symbol = inner.symbol;
342 base.uri = inner.uri;
343 base.mint = inner.mint;
344 base.bonding_curve = inner.bonding_curve;
345 base.user = inner.user;
346 base.creator = inner.creator;
347 base.timestamp = inner.timestamp;
348 base.virtual_token_reserves = inner.virtual_token_reserves;
349 base.virtual_sol_reserves = inner.virtual_sol_reserves;
350 base.real_token_reserves = inner.real_token_reserves;
351 base.token_total_supply = inner.token_total_supply;
352 base.token_program = inner.token_program;
353 base.is_mayhem_mode = inner.is_mayhem_mode;
354 base.is_cashback_enabled = inner.is_cashback_enabled;
355 put_pumpfun_quote_mint_if_set(&mut base.quote_mint, inner.quote_mint);
356 put_u64_if_nonzero(&mut base.virtual_quote_reserves, inner.virtual_quote_reserves);
357}
358
359#[inline(always)]
361fn merge_pumpfun_create_v2(base: &mut PumpFunCreateV2TokenEvent, inner: PumpFunCreateV2TokenEvent) {
362 fill_str_if_empty(&mut base.name, &inner.name);
363 fill_str_if_empty(&mut base.symbol, &inner.symbol);
364 fill_str_if_empty(&mut base.uri, &inner.uri);
365 put_pk_if_set(&mut base.mint, inner.mint);
366 put_pk_if_set(&mut base.bonding_curve, inner.bonding_curve);
367 put_pk_if_set(&mut base.user, inner.user);
368 put_pk_if_set(&mut base.creator, inner.creator);
369 put_i64_if_nonzero(&mut base.timestamp, inner.timestamp);
370 put_u64_if_nonzero(&mut base.virtual_token_reserves, inner.virtual_token_reserves);
371 put_u64_if_nonzero(&mut base.virtual_sol_reserves, inner.virtual_sol_reserves);
372 put_u64_if_nonzero(&mut base.real_token_reserves, inner.real_token_reserves);
373 put_u64_if_nonzero(&mut base.token_total_supply, inner.token_total_supply);
374 put_pk_if_set(&mut base.token_program, inner.token_program);
375 base.is_mayhem_mode |= inner.is_mayhem_mode;
376 base.is_cashback_enabled |= inner.is_cashback_enabled;
377 put_pumpfun_quote_mint_if_set(&mut base.quote_mint, inner.quote_mint);
378 put_u64_if_nonzero(&mut base.virtual_quote_reserves, inner.virtual_quote_reserves);
379 put_pk_if_set(&mut base.mint_authority, inner.mint_authority);
380 put_pk_if_set(&mut base.associated_bonding_curve, inner.associated_bonding_curve);
381 put_pk_if_set(&mut base.global, inner.global);
382 put_pk_if_set(&mut base.system_program, inner.system_program);
383 put_pk_if_set(&mut base.associated_token_program, inner.associated_token_program);
384 put_pk_if_set(&mut base.mayhem_program_id, inner.mayhem_program_id);
385 put_pk_if_set(&mut base.global_params, inner.global_params);
386 put_pk_if_set(&mut base.sol_vault, inner.sol_vault);
387 put_pk_if_set(&mut base.mayhem_state, inner.mayhem_state);
388 put_pk_if_set(&mut base.mayhem_token_vault, inner.mayhem_token_vault);
389 put_pk_if_set(&mut base.event_authority, inner.event_authority);
390 put_pk_if_set(&mut base.program, inner.program);
391 put_pk_if_set(&mut base.observed_fee_recipient, inner.observed_fee_recipient);
392}
393
394#[inline(always)]
396fn merge_pumpfun_migrate(base: &mut PumpFunMigrateEvent, inner: PumpFunMigrateEvent) {
397 base.user = inner.user;
399 base.mint = inner.mint;
400 base.mint_amount = inner.mint_amount;
401 base.sol_amount = inner.sol_amount;
402 base.pool_migration_fee = inner.pool_migration_fee;
403 base.bonding_curve = inner.bonding_curve;
404 base.timestamp = inner.timestamp;
405 base.pool = inner.pool;
406}
407
408#[inline(always)]
409fn merge_pumpswap_buy(base: &mut PumpSwapBuyEvent, inner: PumpSwapBuyEvent) {
410 let ix = std::mem::take(base);
411 *base = inner;
412 merge_pumpswap_buy_log_preferred(base, ix);
413}
414
415#[inline(always)]
416fn merge_pumpswap_sell(base: &mut PumpSwapSellEvent, inner: PumpSwapSellEvent) {
417 let ix = std::mem::take(base);
418 *base = inner;
419 merge_pumpswap_sell_log_preferred(base, ix);
420}
421
422#[inline(always)]
433pub fn can_merge(base: &DexEvent, inner: &DexEvent) -> bool {
434 if base.metadata().signature != inner.metadata().signature {
436 return false;
437 }
438
439 match (base, inner) {
441 (DexEvent::PumpFunTrade(_), DexEvent::PumpFunTrade(_))
443 | (DexEvent::PumpFunTrade(_), DexEvent::PumpFunBuy(_))
444 | (DexEvent::PumpFunTrade(_), DexEvent::PumpFunSell(_))
445 | (DexEvent::PumpFunTrade(_), DexEvent::PumpFunBuyExactSolIn(_))
446 | (DexEvent::PumpFunBuy(_), DexEvent::PumpFunTrade(_))
447 | (DexEvent::PumpFunBuy(_), DexEvent::PumpFunBuy(_))
448 | (DexEvent::PumpFunSell(_), DexEvent::PumpFunTrade(_))
449 | (DexEvent::PumpFunSell(_), DexEvent::PumpFunSell(_))
450 | (DexEvent::PumpFunBuyExactSolIn(_), DexEvent::PumpFunTrade(_))
451 | (DexEvent::PumpFunBuyExactSolIn(_), DexEvent::PumpFunBuyExactSolIn(_)) => true,
452
453 (DexEvent::PumpFunCreate(_), DexEvent::PumpFunCreate(_)) => true,
455 (DexEvent::PumpFunCreateV2(_), DexEvent::PumpFunCreateV2(_)) => true,
456
457 (DexEvent::PumpFunMigrate(_), DexEvent::PumpFunMigrate(_)) => true,
459
460 _ => false,
462 }
463}
464
465#[inline(always)]
470fn fill_pk(to: &mut Pubkey, from: Pubkey) {
471 if *to == Pubkey::default() && from != Pubkey::default() {
472 *to = from;
473 }
474}
475
476#[inline(always)]
477fn fill_pumpfun_quote_mint(to: &mut Pubkey, from: Pubkey) {
478 let from = normalize_pumpfun_quote_mint(from);
479 if (*to == Pubkey::default() || is_pumpfun_solscan_sol_quote_mint(*to))
480 && from != Pubkey::default()
481 {
482 *to = from;
483 }
484}
485
486#[inline(always)]
487fn fill_str_if_empty(to: &mut String, from: &str) {
488 if to.is_empty() && !from.is_empty() {
489 to.push_str(from);
490 }
491}
492
493#[inline]
496fn merge_pumpfun_trade_log_preferred(log: &mut PumpFunTradeEvent, ix: PumpFunTradeEvent) {
497 fill_pk(&mut log.global, ix.global);
498 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
499 fill_pk(&mut log.bonding_curve_v2, ix.bonding_curve_v2);
500 fill_pk(&mut log.associated_bonding_curve, ix.associated_bonding_curve);
501 fill_pk(&mut log.associated_user, ix.associated_user);
502 fill_pk(&mut log.system_program, ix.system_program);
503 fill_pk(&mut log.token_program, ix.token_program);
504 fill_pk(&mut log.quote_token_program, ix.quote_token_program);
505 fill_pk(&mut log.associated_token_program, ix.associated_token_program);
506 fill_pk(&mut log.creator_vault, ix.creator_vault);
507 fill_pk(&mut log.fee_recipient, ix.fee_recipient);
508 fill_pk(&mut log.creator, ix.creator);
509 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
510 fill_pk(&mut log.associated_quote_fee_recipient, ix.associated_quote_fee_recipient);
511 fill_pk(&mut log.buyback_fee_recipient, ix.buyback_fee_recipient);
512 fill_pk(
513 &mut log.associated_quote_buyback_fee_recipient,
514 ix.associated_quote_buyback_fee_recipient,
515 );
516 fill_pk(&mut log.associated_quote_bonding_curve, ix.associated_quote_bonding_curve);
517 fill_pk(&mut log.associated_quote_user, ix.associated_quote_user);
518 fill_pk(&mut log.associated_creator_vault, ix.associated_creator_vault);
519 fill_pk(&mut log.sharing_config, ix.sharing_config);
520 fill_pk(&mut log.event_authority, ix.event_authority);
521 fill_pk(&mut log.program, ix.program);
522 fill_pk(&mut log.global_volume_accumulator, ix.global_volume_accumulator);
523 fill_pk(&mut log.user_volume_accumulator, ix.user_volume_accumulator);
524 fill_pk(&mut log.associated_user_volume_accumulator, ix.associated_user_volume_accumulator);
525 fill_pk(&mut log.fee_config, ix.fee_config);
526 fill_pk(&mut log.fee_program, ix.fee_program);
527 if log.account.is_none() {
528 log.account = ix.account;
529 }
530 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
531 log.ix_name = ix.ix_name;
532 }
533 put_u64_if_nonzero(&mut log.amount, ix.amount);
534 put_u64_if_nonzero(&mut log.max_sol_cost, ix.max_sol_cost);
535 put_u64_if_nonzero(&mut log.min_sol_output, ix.min_sol_output);
536 put_u64_if_nonzero(&mut log.spendable_sol_in, ix.spendable_sol_in);
537 put_u64_if_nonzero(&mut log.spendable_quote_in, ix.spendable_quote_in);
538 put_u64_if_nonzero(&mut log.min_tokens_out, ix.min_tokens_out);
539 put_u64_if_nonzero(&mut log.quote_amount, ix.quote_amount);
540 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
541 put_u64_if_nonzero(&mut log.real_quote_reserves, ix.real_quote_reserves);
542 if !log.is_created_buy && ix.is_created_buy {
543 log.is_created_buy = true;
544 }
545}
546
547#[inline]
548fn merge_pumpfun_create_log_preferred(
549 log: &mut PumpFunCreateTokenEvent,
550 ix: PumpFunCreateTokenEvent,
551) {
552 fill_str_if_empty(&mut log.name, &ix.name);
553 fill_str_if_empty(&mut log.symbol, &ix.symbol);
554 fill_str_if_empty(&mut log.uri, &ix.uri);
555 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
556 fill_pk(&mut log.user, ix.user);
557 fill_pk(&mut log.creator, ix.creator);
558 fill_pk(&mut log.token_program, ix.token_program);
559 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
560 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
561 log.is_mayhem_mode |= ix.is_mayhem_mode;
562 log.is_cashback_enabled |= ix.is_cashback_enabled;
563}
564
565#[inline]
566fn merge_pumpfun_create_v2_into_create_log_preferred(
567 log: &mut PumpFunCreateTokenEvent,
568 ix: PumpFunCreateV2TokenEvent,
569) {
570 fill_str_if_empty(&mut log.name, &ix.name);
571 fill_str_if_empty(&mut log.symbol, &ix.symbol);
572 fill_str_if_empty(&mut log.uri, &ix.uri);
573 fill_pk(&mut log.mint, ix.mint);
574 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
575 fill_pk(&mut log.user, ix.user);
576 fill_pk(&mut log.creator, ix.creator);
577 if log.timestamp == 0 && ix.timestamp != 0 {
578 log.timestamp = ix.timestamp;
579 }
580 put_u64_if_nonzero(&mut log.virtual_token_reserves, ix.virtual_token_reserves);
581 put_u64_if_nonzero(&mut log.virtual_sol_reserves, ix.virtual_sol_reserves);
582 put_u64_if_nonzero(&mut log.real_token_reserves, ix.real_token_reserves);
583 put_u64_if_nonzero(&mut log.token_total_supply, ix.token_total_supply);
584 fill_pk(&mut log.token_program, ix.token_program);
585 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
586 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
587 log.is_mayhem_mode |= ix.is_mayhem_mode;
588 log.is_cashback_enabled |= ix.is_cashback_enabled;
589}
590
591#[inline]
592fn merge_pumpfun_create_into_create_v2_log_preferred(
593 log: &mut PumpFunCreateV2TokenEvent,
594 ix: PumpFunCreateTokenEvent,
595) {
596 fill_str_if_empty(&mut log.name, &ix.name);
597 fill_str_if_empty(&mut log.symbol, &ix.symbol);
598 fill_str_if_empty(&mut log.uri, &ix.uri);
599 fill_pk(&mut log.mint, ix.mint);
600 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
601 fill_pk(&mut log.user, ix.user);
602 fill_pk(&mut log.creator, ix.creator);
603 if log.timestamp == 0 && ix.timestamp != 0 {
604 log.timestamp = ix.timestamp;
605 }
606 put_u64_if_nonzero(&mut log.virtual_token_reserves, ix.virtual_token_reserves);
607 put_u64_if_nonzero(&mut log.virtual_sol_reserves, ix.virtual_sol_reserves);
608 put_u64_if_nonzero(&mut log.real_token_reserves, ix.real_token_reserves);
609 put_u64_if_nonzero(&mut log.token_total_supply, ix.token_total_supply);
610 fill_pk(&mut log.token_program, ix.token_program);
611 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
612 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
613 log.is_mayhem_mode |= ix.is_mayhem_mode;
614 log.is_cashback_enabled |= ix.is_cashback_enabled;
615}
616
617#[inline]
618fn merge_pumpfun_create_v2_log_preferred(
619 log: &mut PumpFunCreateV2TokenEvent,
620 ix: PumpFunCreateV2TokenEvent,
621) {
622 fill_str_if_empty(&mut log.name, &ix.name);
623 fill_str_if_empty(&mut log.symbol, &ix.symbol);
624 fill_str_if_empty(&mut log.uri, &ix.uri);
625 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
626 fill_pk(&mut log.user, ix.user);
627 fill_pk(&mut log.creator, ix.creator);
628 fill_pk(&mut log.token_program, ix.token_program);
629 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
630 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
631 fill_pk(&mut log.mint_authority, ix.mint_authority);
632 fill_pk(&mut log.associated_bonding_curve, ix.associated_bonding_curve);
633 fill_pk(&mut log.global, ix.global);
634 fill_pk(&mut log.system_program, ix.system_program);
635 fill_pk(&mut log.associated_token_program, ix.associated_token_program);
636 fill_pk(&mut log.mayhem_program_id, ix.mayhem_program_id);
637 fill_pk(&mut log.global_params, ix.global_params);
638 fill_pk(&mut log.sol_vault, ix.sol_vault);
639 fill_pk(&mut log.mayhem_state, ix.mayhem_state);
640 fill_pk(&mut log.mayhem_token_vault, ix.mayhem_token_vault);
641 fill_pk(&mut log.event_authority, ix.event_authority);
642 fill_pk(&mut log.program, ix.program);
643 fill_pk(&mut log.observed_fee_recipient, ix.observed_fee_recipient);
644}
645
646#[inline]
647fn merge_pumpfun_migrate_log_preferred(log: &mut PumpFunMigrateEvent, ix: PumpFunMigrateEvent) {
648 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
649 fill_pk(&mut log.pool, ix.pool);
650 fill_pk(&mut log.user, ix.user);
651}
652
653#[inline]
654fn merge_pumpswap_trade_log_preferred(log: &mut PumpSwapTradeEvent, ix: PumpSwapTradeEvent) {
655 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
656 log.ix_name = ix.ix_name;
657 }
658}
659
660#[inline]
661fn merge_pumpswap_buy_log_preferred(log: &mut PumpSwapBuyEvent, ix: PumpSwapBuyEvent) {
662 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
663 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
664 fill_pk(&mut log.protocol_fee_recipient, ix.protocol_fee_recipient);
665 fill_pk(&mut log.protocol_fee_recipient_token_account, ix.protocol_fee_recipient_token_account);
666 fill_pk(&mut log.coin_creator, ix.coin_creator);
667 fill_pk(&mut log.base_mint, ix.base_mint);
668 fill_pk(&mut log.quote_mint, ix.quote_mint);
669 fill_pk(&mut log.pool_base_token_account, ix.pool_base_token_account);
670 fill_pk(&mut log.pool_quote_token_account, ix.pool_quote_token_account);
671 fill_pk(&mut log.coin_creator_vault_ata, ix.coin_creator_vault_ata);
672 fill_pk(&mut log.coin_creator_vault_authority, ix.coin_creator_vault_authority);
673 fill_pk(&mut log.base_token_program, ix.base_token_program);
674 fill_pk(&mut log.quote_token_program, ix.quote_token_program);
675 fill_pk(&mut log.pool_v2, ix.pool_v2);
676 fill_pk(&mut log.fee_recipient, ix.fee_recipient);
677 fill_pk(&mut log.fee_recipient_quote_token_account, ix.fee_recipient_quote_token_account);
678 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
679 log.ix_name = ix.ix_name;
680 }
681}
682
683#[inline]
684fn merge_pumpswap_sell_log_preferred(log: &mut PumpSwapSellEvent, ix: PumpSwapSellEvent) {
685 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
686 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
687 fill_pk(&mut log.protocol_fee_recipient, ix.protocol_fee_recipient);
688 fill_pk(&mut log.protocol_fee_recipient_token_account, ix.protocol_fee_recipient_token_account);
689 fill_pk(&mut log.coin_creator, ix.coin_creator);
690 fill_pk(&mut log.base_mint, ix.base_mint);
691 fill_pk(&mut log.quote_mint, ix.quote_mint);
692 fill_pk(&mut log.pool_base_token_account, ix.pool_base_token_account);
693 fill_pk(&mut log.pool_quote_token_account, ix.pool_quote_token_account);
694 fill_pk(&mut log.coin_creator_vault_ata, ix.coin_creator_vault_ata);
695 fill_pk(&mut log.coin_creator_vault_authority, ix.coin_creator_vault_authority);
696 fill_pk(&mut log.base_token_program, ix.base_token_program);
697 fill_pk(&mut log.quote_token_program, ix.quote_token_program);
698 fill_pk(&mut log.pool_v2, ix.pool_v2);
699 fill_pk(&mut log.fee_recipient, ix.fee_recipient);
700 fill_pk(&mut log.fee_recipient_quote_token_account, ix.fee_recipient_quote_token_account);
701}
702
703#[inline]
704fn merge_raydium_clmm_swap_log_preferred(log: &mut RaydiumClmmSwapEvent, ix: RaydiumClmmSwapEvent) {
705 fill_pk(&mut log.token_account_0, ix.token_account_0);
706 fill_pk(&mut log.token_account_1, ix.token_account_1);
707 fill_pk(&mut log.sender, ix.sender);
708}
709
710#[inline]
711fn merge_raydium_amm_v4_swap_log_preferred(
712 log: &mut RaydiumAmmV4SwapEvent,
713 ix: RaydiumAmmV4SwapEvent,
714) {
715 fill_pk(&mut log.token_program, ix.token_program);
716 fill_pk(&mut log.amm_authority, ix.amm_authority);
717 fill_pk(&mut log.amm_open_orders, ix.amm_open_orders);
718 if let Some(ref o) = ix.amm_target_orders {
719 if log.amm_target_orders.is_none() {
720 log.amm_target_orders = Some(*o);
721 }
722 }
723 fill_pk(&mut log.pool_coin_token_account, ix.pool_coin_token_account);
724 fill_pk(&mut log.pool_pc_token_account, ix.pool_pc_token_account);
725 fill_pk(&mut log.serum_program, ix.serum_program);
726 fill_pk(&mut log.serum_market, ix.serum_market);
727 fill_pk(&mut log.serum_bids, ix.serum_bids);
728 fill_pk(&mut log.serum_asks, ix.serum_asks);
729 fill_pk(&mut log.serum_event_queue, ix.serum_event_queue);
730 fill_pk(&mut log.serum_coin_vault_account, ix.serum_coin_vault_account);
731 fill_pk(&mut log.serum_pc_vault_account, ix.serum_pc_vault_account);
732 fill_pk(&mut log.serum_vault_signer, ix.serum_vault_signer);
733 fill_pk(&mut log.user_source_token_account, ix.user_source_token_account);
734 fill_pk(&mut log.user_destination_token_account, ix.user_destination_token_account);
735}
736
737#[inline]
738fn merge_pumpswap_create_pool_log_preferred(
739 log: &mut PumpSwapCreatePoolEvent,
740 ix: PumpSwapCreatePoolEvent,
741) {
742 fill_pk(&mut log.creator, ix.creator);
743 fill_pk(&mut log.pool, ix.pool);
744 fill_pk(&mut log.lp_mint, ix.lp_mint);
745 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
746 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
747 fill_pk(&mut log.coin_creator, ix.coin_creator);
748}
749
750#[inline]
751fn merge_pumpswap_liquidity_added_log_preferred(
752 log: &mut PumpSwapLiquidityAdded,
753 ix: PumpSwapLiquidityAdded,
754) {
755 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
756 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
757 fill_pk(&mut log.user_pool_token_account, ix.user_pool_token_account);
758}
759
760#[inline]
761fn merge_pumpswap_liquidity_removed_log_preferred(
762 log: &mut PumpSwapLiquidityRemoved,
763 ix: PumpSwapLiquidityRemoved,
764) {
765 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
766 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
767 fill_pk(&mut log.user_pool_token_account, ix.user_pool_token_account);
768}
769
770#[inline]
771fn merge_raydium_launchlab_pool_create_log_preferred(
772 log: &mut RaydiumLaunchlabPoolCreateEvent,
773 ix: RaydiumLaunchlabPoolCreateEvent,
774) {
775 fill_pk(&mut log.creator, ix.creator);
776 fill_str_if_empty(&mut log.base_mint_param.name, &ix.base_mint_param.name);
777 fill_str_if_empty(&mut log.base_mint_param.symbol, &ix.base_mint_param.symbol);
778 fill_str_if_empty(&mut log.base_mint_param.uri, &ix.base_mint_param.uri);
779}
780
781#[inline]
782fn merge_raydium_launchlab_migrate_amm_log_preferred(
783 log: &mut RaydiumLaunchlabMigrateAmmEvent,
784 ix: RaydiumLaunchlabMigrateAmmEvent,
785) {
786 fill_pk(&mut log.old_pool, ix.old_pool);
787 fill_pk(&mut log.new_pool, ix.new_pool);
788 fill_pk(&mut log.user, ix.user);
789}
790
791#[inline]
793fn merge_raydium_launchlab_trade_log_preferred(
794 _log: &mut RaydiumLaunchlabTradeEvent,
795 _ix: RaydiumLaunchlabTradeEvent,
796) {
797}
798
799#[inline]
800fn merge_meteora_dlmm_swap_log_preferred(
801 _log: &mut MeteoraDlmmSwapEvent,
802 _ix: MeteoraDlmmSwapEvent,
803) {
804}
805
806pub fn merge_grpc_instruction_into_log(log: &mut DexEvent, ix: DexEvent) {
812 use DexEvent::*;
813 match log {
814 PumpFunTrade(l) => {
815 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
816 merge_pumpfun_trade_log_preferred(l, i);
817 }
818 }
819 PumpFunBuy(l) => {
820 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
821 merge_pumpfun_trade_log_preferred(l, i);
822 }
823 }
824 PumpFunSell(l) => {
825 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
826 merge_pumpfun_trade_log_preferred(l, i);
827 }
828 }
829 PumpFunBuyExactSolIn(l) => {
830 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
831 merge_pumpfun_trade_log_preferred(l, i);
832 }
833 }
834 PumpFunCreate(l) => match ix {
835 DexEvent::PumpFunCreate(i) => merge_pumpfun_create_log_preferred(l, i),
836 DexEvent::PumpFunCreateV2(i) => merge_pumpfun_create_v2_into_create_log_preferred(l, i),
837 _ => {}
838 },
839 PumpFunCreateV2(l) => match ix {
840 DexEvent::PumpFunCreate(i) => merge_pumpfun_create_into_create_v2_log_preferred(l, i),
841 DexEvent::PumpFunCreateV2(i) => merge_pumpfun_create_v2_log_preferred(l, i),
842 _ => {}
843 },
844 PumpFunMigrate(l) => {
845 if let DexEvent::PumpFunMigrate(i) = ix {
846 merge_pumpfun_migrate_log_preferred(l, i);
847 }
848 }
849 PumpSwapTrade(l) => {
850 if let PumpSwapTrade(i) = ix {
851 merge_pumpswap_trade_log_preferred(l, i);
852 }
853 }
854 PumpSwapBuy(l) => {
855 if let PumpSwapBuy(i) = ix {
856 merge_pumpswap_buy_log_preferred(l, i);
857 }
858 }
859 PumpSwapSell(l) => {
860 if let PumpSwapSell(i) = ix {
861 merge_pumpswap_sell_log_preferred(l, i);
862 }
863 }
864 RaydiumClmmSwap(l) => {
865 if let RaydiumClmmSwap(i) = ix {
866 merge_raydium_clmm_swap_log_preferred(l, i);
867 }
868 }
869 RaydiumAmmV4Swap(l) => {
870 if let RaydiumAmmV4Swap(i) = ix {
871 merge_raydium_amm_v4_swap_log_preferred(l, i);
872 }
873 }
874 RaydiumLaunchlabTrade(l) => {
875 if let RaydiumLaunchlabTrade(i) = ix {
876 merge_raydium_launchlab_trade_log_preferred(l, i);
877 }
878 }
879 RaydiumLaunchlabPoolCreate(l) => {
880 if let RaydiumLaunchlabPoolCreate(i) = ix {
881 merge_raydium_launchlab_pool_create_log_preferred(l, i);
882 }
883 }
884 RaydiumLaunchlabMigrateAmm(l) => {
885 if let RaydiumLaunchlabMigrateAmm(i) = ix {
886 merge_raydium_launchlab_migrate_amm_log_preferred(l, i);
887 }
888 }
889 PumpSwapCreatePool(l) => {
890 if let PumpSwapCreatePool(i) = ix {
891 merge_pumpswap_create_pool_log_preferred(l, i);
892 }
893 }
894 PumpSwapLiquidityAdded(l) => {
895 if let PumpSwapLiquidityAdded(i) = ix {
896 merge_pumpswap_liquidity_added_log_preferred(l, i);
897 }
898 }
899 PumpSwapLiquidityRemoved(l) => {
900 if let PumpSwapLiquidityRemoved(i) = ix {
901 merge_pumpswap_liquidity_removed_log_preferred(l, i);
902 }
903 }
904 MeteoraDlmmSwap(l) => {
905 if let MeteoraDlmmSwap(i) = ix {
906 merge_meteora_dlmm_swap_log_preferred(l, i);
907 }
908 }
909 _ => {}
910 }
911}
912
913#[inline]
914fn pumpfun_trade_from_ix_variant(ix: DexEvent) -> Option<PumpFunTradeEvent> {
915 match ix {
916 DexEvent::PumpFunTrade(t)
917 | DexEvent::PumpFunBuy(t)
918 | DexEvent::PumpFunSell(t)
919 | DexEvent::PumpFunBuyExactSolIn(t) => Some(t),
920 _ => None,
921 }
922}
923
924#[cfg(test)]
925mod tests {
926 use super::*;
927 use solana_sdk::{pubkey::Pubkey, signature::Signature};
928
929 #[test]
930 fn test_merge_pumpfun_trade() {
931 let metadata = EventMetadata {
932 signature: Signature::default(),
933 slot: 100,
934 tx_index: 1,
935 block_time_us: 1000,
936 grpc_recv_us: 2000,
937 recent_blockhash: None,
938 };
939
940 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
942 metadata: metadata.clone(),
943 bonding_curve: Pubkey::new_unique(),
944 associated_bonding_curve: Pubkey::new_unique(),
945 ..Default::default()
946 });
947
948 let inner = DexEvent::PumpFunTrade(PumpFunTradeEvent {
950 metadata: metadata.clone(),
951 mint: Pubkey::new_unique(),
952 sol_amount: 1000,
953 token_amount: 2000,
954 is_buy: true,
955 user: Pubkey::new_unique(),
956 ..Default::default()
957 });
958
959 merge_events(&mut base, inner);
961
962 if let DexEvent::PumpFunTrade(trade) = base {
964 assert_eq!(trade.sol_amount, 1000);
965 assert_eq!(trade.token_amount, 2000);
966 assert!(trade.is_buy);
967 assert_ne!(trade.bonding_curve, Pubkey::default());
969 assert_ne!(trade.associated_bonding_curve, Pubkey::default());
970 } else {
971 panic!("Expected PumpFunTrade event");
972 }
973 }
974
975 #[test]
976 fn merge_preserves_instruction_context_when_log_tail_is_absent() {
977 let metadata = EventMetadata {
978 signature: Signature::default(),
979 slot: 100,
980 tx_index: 1,
981 block_time_us: 1000,
982 grpc_recv_us: 2000,
983 recent_blockhash: None,
984 };
985 let quote_mint = Pubkey::new_unique();
986 let associated_quote_user = Pubkey::new_unique();
987
988 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
989 metadata: metadata.clone(),
990 ix_name: "buy_exact_quote_in".to_string(),
991 quote_mint,
992 spendable_quote_in: 1_000,
993 min_tokens_out: 2_000,
994 associated_quote_user,
995 ..Default::default()
996 });
997
998 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent {
999 metadata,
1000 sol_amount: 1_000,
1001 token_amount: 2_000,
1002 is_buy: true,
1003 ..Default::default()
1004 });
1005
1006 merge_events(&mut base, inner);
1007
1008 if let DexEvent::PumpFunTrade(t) = base {
1009 assert_eq!(t.sol_amount, 1_000);
1010 assert_eq!(t.token_amount, 2_000);
1011 assert_eq!(t.ix_name, "buy_exact_quote_in");
1012 assert_eq!(t.quote_mint, quote_mint);
1013 assert_eq!(t.spendable_quote_in, 1_000);
1014 assert_eq!(t.min_tokens_out, 2_000);
1015 assert_eq!(t.associated_quote_user, associated_quote_user);
1016 } else {
1017 panic!("Expected PumpFunTrade event");
1018 }
1019 }
1020
1021 #[test]
1022 fn merge_replaces_sol_quote_sentinel_with_real_quote_mint() {
1023 let quote_mint = Pubkey::new_unique();
1024 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1025 quote_mint: PUMPFUN_SOLSCAN_SOL_QUOTE_MINT,
1026 ..Default::default()
1027 });
1028
1029 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent { quote_mint, ..Default::default() });
1030
1031 merge_events(&mut base, inner);
1032
1033 if let DexEvent::PumpFunTrade(t) = base {
1034 assert_eq!(t.quote_mint, quote_mint);
1035 } else {
1036 panic!("Expected PumpFunTrade event");
1037 }
1038 }
1039
1040 #[test]
1041 fn test_can_merge() {
1042 let metadata = EventMetadata {
1043 signature: Signature::default(),
1044 slot: 100,
1045 tx_index: 1,
1046 block_time_us: 1000,
1047 grpc_recv_us: 2000,
1048 recent_blockhash: None,
1049 };
1050
1051 let base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1052 metadata: metadata.clone(),
1053 ..Default::default()
1054 });
1055
1056 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent {
1057 metadata: metadata.clone(),
1058 ..Default::default()
1059 });
1060
1061 assert!(can_merge(&base, &inner));
1063
1064 let different_sig = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1066 metadata: EventMetadata { signature: Signature::new_unique(), ..metadata },
1067 ..Default::default()
1068 });
1069
1070 assert!(!can_merge(&base, &different_sig));
1071 }
1072
1073 #[test]
1074 fn grpc_merge_fills_fee_recipient_from_ix_when_log_default() {
1075 let metadata = EventMetadata {
1076 signature: Signature::default(),
1077 slot: 1,
1078 tx_index: 0,
1079 block_time_us: 0,
1080 grpc_recv_us: 0,
1081 recent_blockhash: None,
1082 };
1083 let fr = Pubkey::new_unique();
1084 let log_t =
1085 PumpFunTradeEvent { metadata: metadata.clone(), sol_amount: 50, ..Default::default() };
1086 let mut ix_t = log_t.clone();
1087 ix_t.fee_recipient = fr;
1088 ix_t.sol_amount = 777;
1089 let mut log_ev = DexEvent::PumpFunTrade(log_t);
1090 merge_grpc_instruction_into_log(&mut log_ev, DexEvent::PumpFunBuy(ix_t));
1091 match log_ev {
1092 DexEvent::PumpFunTrade(t) => {
1093 assert_eq!(t.fee_recipient, fr);
1094 assert_eq!(t.sol_amount, 50);
1095 }
1096 _ => panic!("expected trade"),
1097 }
1098 }
1099
1100 #[test]
1101 fn grpc_merge_keeps_log_trade_fields() {
1102 let metadata = EventMetadata {
1103 signature: Signature::default(),
1104 slot: 1,
1105 tx_index: 0,
1106 block_time_us: 0,
1107 grpc_recv_us: 0,
1108 recent_blockhash: None,
1109 };
1110 let log_t = PumpFunTradeEvent {
1111 metadata: metadata.clone(),
1112 mayhem_mode: true,
1113 sol_amount: 100,
1114 ..Default::default()
1115 };
1116 let mut ix_t = log_t.clone();
1117 ix_t.mayhem_mode = false;
1118 ix_t.sol_amount = 999;
1119
1120 let mut log_ev = DexEvent::PumpFunTrade(log_t);
1121 merge_grpc_instruction_into_log(&mut log_ev, DexEvent::PumpFunBuy(ix_t));
1122 match log_ev {
1123 DexEvent::PumpFunTrade(t) => {
1124 assert!(t.mayhem_mode);
1125 assert_eq!(t.sol_amount, 100);
1126 }
1127 _ => panic!("variant preserved"),
1128 }
1129 }
1130}