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 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
562 log.ix_name = ix.ix_name;
563 }
564 log.is_mayhem_mode |= ix.is_mayhem_mode;
565 log.is_cashback_enabled |= ix.is_cashback_enabled;
566}
567
568#[inline]
569fn merge_pumpfun_create_v2_into_create_log_preferred(
570 log: &mut PumpFunCreateTokenEvent,
571 ix: PumpFunCreateV2TokenEvent,
572) {
573 fill_str_if_empty(&mut log.name, &ix.name);
574 fill_str_if_empty(&mut log.symbol, &ix.symbol);
575 fill_str_if_empty(&mut log.uri, &ix.uri);
576 fill_pk(&mut log.mint, ix.mint);
577 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
578 fill_pk(&mut log.user, ix.user);
579 fill_pk(&mut log.creator, ix.creator);
580 if log.timestamp == 0 && ix.timestamp != 0 {
581 log.timestamp = ix.timestamp;
582 }
583 put_u64_if_nonzero(&mut log.virtual_token_reserves, ix.virtual_token_reserves);
584 put_u64_if_nonzero(&mut log.virtual_sol_reserves, ix.virtual_sol_reserves);
585 put_u64_if_nonzero(&mut log.real_token_reserves, ix.real_token_reserves);
586 put_u64_if_nonzero(&mut log.token_total_supply, ix.token_total_supply);
587 fill_pk(&mut log.token_program, ix.token_program);
588 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
589 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
590 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
591 log.ix_name = ix.ix_name;
592 }
593 log.is_mayhem_mode |= ix.is_mayhem_mode;
594 log.is_cashback_enabled |= ix.is_cashback_enabled;
595}
596
597#[inline]
598fn merge_pumpfun_create_into_create_v2_log_preferred(
599 log: &mut PumpFunCreateV2TokenEvent,
600 ix: PumpFunCreateTokenEvent,
601) {
602 fill_str_if_empty(&mut log.name, &ix.name);
603 fill_str_if_empty(&mut log.symbol, &ix.symbol);
604 fill_str_if_empty(&mut log.uri, &ix.uri);
605 fill_pk(&mut log.mint, ix.mint);
606 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
607 fill_pk(&mut log.user, ix.user);
608 fill_pk(&mut log.creator, ix.creator);
609 if log.timestamp == 0 && ix.timestamp != 0 {
610 log.timestamp = ix.timestamp;
611 }
612 put_u64_if_nonzero(&mut log.virtual_token_reserves, ix.virtual_token_reserves);
613 put_u64_if_nonzero(&mut log.virtual_sol_reserves, ix.virtual_sol_reserves);
614 put_u64_if_nonzero(&mut log.real_token_reserves, ix.real_token_reserves);
615 put_u64_if_nonzero(&mut log.token_total_supply, ix.token_total_supply);
616 fill_pk(&mut log.token_program, ix.token_program);
617 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
618 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
619 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
620 log.ix_name = ix.ix_name;
621 }
622 log.is_mayhem_mode |= ix.is_mayhem_mode;
623 log.is_cashback_enabled |= ix.is_cashback_enabled;
624}
625
626#[inline]
627fn merge_pumpfun_create_v2_log_preferred(
628 log: &mut PumpFunCreateV2TokenEvent,
629 ix: PumpFunCreateV2TokenEvent,
630) {
631 fill_str_if_empty(&mut log.name, &ix.name);
632 fill_str_if_empty(&mut log.symbol, &ix.symbol);
633 fill_str_if_empty(&mut log.uri, &ix.uri);
634 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
635 fill_pk(&mut log.user, ix.user);
636 fill_pk(&mut log.creator, ix.creator);
637 fill_pk(&mut log.token_program, ix.token_program);
638 fill_pumpfun_quote_mint(&mut log.quote_mint, ix.quote_mint);
639 put_u64_if_nonzero(&mut log.virtual_quote_reserves, ix.virtual_quote_reserves);
640 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
641 log.ix_name = ix.ix_name;
642 }
643 fill_pk(&mut log.mint_authority, ix.mint_authority);
644 fill_pk(&mut log.associated_bonding_curve, ix.associated_bonding_curve);
645 fill_pk(&mut log.global, ix.global);
646 fill_pk(&mut log.system_program, ix.system_program);
647 fill_pk(&mut log.associated_token_program, ix.associated_token_program);
648 fill_pk(&mut log.mayhem_program_id, ix.mayhem_program_id);
649 fill_pk(&mut log.global_params, ix.global_params);
650 fill_pk(&mut log.sol_vault, ix.sol_vault);
651 fill_pk(&mut log.mayhem_state, ix.mayhem_state);
652 fill_pk(&mut log.mayhem_token_vault, ix.mayhem_token_vault);
653 fill_pk(&mut log.event_authority, ix.event_authority);
654 fill_pk(&mut log.program, ix.program);
655 fill_pk(&mut log.observed_fee_recipient, ix.observed_fee_recipient);
656}
657
658#[inline]
659fn merge_pumpfun_migrate_log_preferred(log: &mut PumpFunMigrateEvent, ix: PumpFunMigrateEvent) {
660 fill_pk(&mut log.bonding_curve, ix.bonding_curve);
661 fill_pk(&mut log.pool, ix.pool);
662 fill_pk(&mut log.user, ix.user);
663}
664
665#[inline]
666fn merge_pumpswap_trade_log_preferred(log: &mut PumpSwapTradeEvent, ix: PumpSwapTradeEvent) {
667 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
668 log.ix_name = ix.ix_name;
669 }
670}
671
672#[inline]
673fn merge_pumpswap_buy_log_preferred(log: &mut PumpSwapBuyEvent, ix: PumpSwapBuyEvent) {
674 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
675 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
676 fill_pk(&mut log.protocol_fee_recipient, ix.protocol_fee_recipient);
677 fill_pk(&mut log.protocol_fee_recipient_token_account, ix.protocol_fee_recipient_token_account);
678 fill_pk(&mut log.coin_creator, ix.coin_creator);
679 fill_pk(&mut log.base_mint, ix.base_mint);
680 fill_pk(&mut log.quote_mint, ix.quote_mint);
681 fill_pk(&mut log.pool_base_token_account, ix.pool_base_token_account);
682 fill_pk(&mut log.pool_quote_token_account, ix.pool_quote_token_account);
683 fill_pk(&mut log.coin_creator_vault_ata, ix.coin_creator_vault_ata);
684 fill_pk(&mut log.coin_creator_vault_authority, ix.coin_creator_vault_authority);
685 fill_pk(&mut log.base_token_program, ix.base_token_program);
686 fill_pk(&mut log.quote_token_program, ix.quote_token_program);
687 fill_pk(&mut log.pool_v2, ix.pool_v2);
688 fill_pk(&mut log.fee_recipient, ix.fee_recipient);
689 fill_pk(&mut log.fee_recipient_quote_token_account, ix.fee_recipient_quote_token_account);
690 if log.ix_name.is_empty() && !ix.ix_name.is_empty() {
691 log.ix_name = ix.ix_name;
692 }
693}
694
695#[inline]
696fn merge_pumpswap_sell_log_preferred(log: &mut PumpSwapSellEvent, ix: PumpSwapSellEvent) {
697 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
698 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
699 fill_pk(&mut log.protocol_fee_recipient, ix.protocol_fee_recipient);
700 fill_pk(&mut log.protocol_fee_recipient_token_account, ix.protocol_fee_recipient_token_account);
701 fill_pk(&mut log.coin_creator, ix.coin_creator);
702 fill_pk(&mut log.base_mint, ix.base_mint);
703 fill_pk(&mut log.quote_mint, ix.quote_mint);
704 fill_pk(&mut log.pool_base_token_account, ix.pool_base_token_account);
705 fill_pk(&mut log.pool_quote_token_account, ix.pool_quote_token_account);
706 fill_pk(&mut log.coin_creator_vault_ata, ix.coin_creator_vault_ata);
707 fill_pk(&mut log.coin_creator_vault_authority, ix.coin_creator_vault_authority);
708 fill_pk(&mut log.base_token_program, ix.base_token_program);
709 fill_pk(&mut log.quote_token_program, ix.quote_token_program);
710 fill_pk(&mut log.pool_v2, ix.pool_v2);
711 fill_pk(&mut log.fee_recipient, ix.fee_recipient);
712 fill_pk(&mut log.fee_recipient_quote_token_account, ix.fee_recipient_quote_token_account);
713}
714
715#[inline]
716fn merge_raydium_clmm_swap_log_preferred(log: &mut RaydiumClmmSwapEvent, ix: RaydiumClmmSwapEvent) {
717 fill_pk(&mut log.token_account_0, ix.token_account_0);
718 fill_pk(&mut log.token_account_1, ix.token_account_1);
719 fill_pk(&mut log.sender, ix.sender);
720}
721
722#[inline]
723fn merge_raydium_amm_v4_swap_log_preferred(
724 log: &mut RaydiumAmmV4SwapEvent,
725 ix: RaydiumAmmV4SwapEvent,
726) {
727 fill_pk(&mut log.token_program, ix.token_program);
728 fill_pk(&mut log.amm_authority, ix.amm_authority);
729 fill_pk(&mut log.amm_open_orders, ix.amm_open_orders);
730 if let Some(ref o) = ix.amm_target_orders {
731 if log.amm_target_orders.is_none() {
732 log.amm_target_orders = Some(*o);
733 }
734 }
735 fill_pk(&mut log.pool_coin_token_account, ix.pool_coin_token_account);
736 fill_pk(&mut log.pool_pc_token_account, ix.pool_pc_token_account);
737 fill_pk(&mut log.serum_program, ix.serum_program);
738 fill_pk(&mut log.serum_market, ix.serum_market);
739 fill_pk(&mut log.serum_bids, ix.serum_bids);
740 fill_pk(&mut log.serum_asks, ix.serum_asks);
741 fill_pk(&mut log.serum_event_queue, ix.serum_event_queue);
742 fill_pk(&mut log.serum_coin_vault_account, ix.serum_coin_vault_account);
743 fill_pk(&mut log.serum_pc_vault_account, ix.serum_pc_vault_account);
744 fill_pk(&mut log.serum_vault_signer, ix.serum_vault_signer);
745 fill_pk(&mut log.user_source_token_account, ix.user_source_token_account);
746 fill_pk(&mut log.user_destination_token_account, ix.user_destination_token_account);
747}
748
749#[inline]
750fn merge_pumpswap_create_pool_log_preferred(
751 log: &mut PumpSwapCreatePoolEvent,
752 ix: PumpSwapCreatePoolEvent,
753) {
754 fill_pk(&mut log.creator, ix.creator);
755 fill_pk(&mut log.pool, ix.pool);
756 fill_pk(&mut log.lp_mint, ix.lp_mint);
757 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
758 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
759 fill_pk(&mut log.coin_creator, ix.coin_creator);
760}
761
762#[inline]
763fn merge_pumpswap_liquidity_added_log_preferred(
764 log: &mut PumpSwapLiquidityAdded,
765 ix: PumpSwapLiquidityAdded,
766) {
767 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
768 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
769 fill_pk(&mut log.user_pool_token_account, ix.user_pool_token_account);
770}
771
772#[inline]
773fn merge_pumpswap_liquidity_removed_log_preferred(
774 log: &mut PumpSwapLiquidityRemoved,
775 ix: PumpSwapLiquidityRemoved,
776) {
777 fill_pk(&mut log.user_base_token_account, ix.user_base_token_account);
778 fill_pk(&mut log.user_quote_token_account, ix.user_quote_token_account);
779 fill_pk(&mut log.user_pool_token_account, ix.user_pool_token_account);
780}
781
782#[inline]
783fn merge_raydium_launchlab_pool_create_log_preferred(
784 log: &mut RaydiumLaunchlabPoolCreateEvent,
785 ix: RaydiumLaunchlabPoolCreateEvent,
786) {
787 fill_pk(&mut log.creator, ix.creator);
788 fill_str_if_empty(&mut log.base_mint_param.name, &ix.base_mint_param.name);
789 fill_str_if_empty(&mut log.base_mint_param.symbol, &ix.base_mint_param.symbol);
790 fill_str_if_empty(&mut log.base_mint_param.uri, &ix.base_mint_param.uri);
791}
792
793#[inline]
794fn merge_raydium_launchlab_migrate_amm_log_preferred(
795 log: &mut RaydiumLaunchlabMigrateAmmEvent,
796 ix: RaydiumLaunchlabMigrateAmmEvent,
797) {
798 fill_pk(&mut log.old_pool, ix.old_pool);
799 fill_pk(&mut log.new_pool, ix.new_pool);
800 fill_pk(&mut log.user, ix.user);
801}
802
803#[inline]
805fn merge_raydium_launchlab_trade_log_preferred(
806 _log: &mut RaydiumLaunchlabTradeEvent,
807 _ix: RaydiumLaunchlabTradeEvent,
808) {
809}
810
811#[inline]
812fn merge_meteora_dlmm_swap_log_preferred(
813 _log: &mut MeteoraDlmmSwapEvent,
814 _ix: MeteoraDlmmSwapEvent,
815) {
816}
817
818pub fn merge_grpc_instruction_into_log(log: &mut DexEvent, ix: DexEvent) {
824 use DexEvent::*;
825 match log {
826 PumpFunTrade(l) => {
827 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
828 merge_pumpfun_trade_log_preferred(l, i);
829 }
830 }
831 PumpFunBuy(l) => {
832 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
833 merge_pumpfun_trade_log_preferred(l, i);
834 }
835 }
836 PumpFunSell(l) => {
837 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
838 merge_pumpfun_trade_log_preferred(l, i);
839 }
840 }
841 PumpFunBuyExactSolIn(l) => {
842 if let Some(i) = pumpfun_trade_from_ix_variant(ix) {
843 merge_pumpfun_trade_log_preferred(l, i);
844 }
845 }
846 PumpFunCreate(l) => match ix {
847 DexEvent::PumpFunCreate(i) => merge_pumpfun_create_log_preferred(l, i),
848 DexEvent::PumpFunCreateV2(i) => merge_pumpfun_create_v2_into_create_log_preferred(l, i),
849 _ => {}
850 },
851 PumpFunCreateV2(l) => match ix {
852 DexEvent::PumpFunCreate(i) => merge_pumpfun_create_into_create_v2_log_preferred(l, i),
853 DexEvent::PumpFunCreateV2(i) => merge_pumpfun_create_v2_log_preferred(l, i),
854 _ => {}
855 },
856 PumpFunMigrate(l) => {
857 if let DexEvent::PumpFunMigrate(i) = ix {
858 merge_pumpfun_migrate_log_preferred(l, i);
859 }
860 }
861 PumpSwapTrade(l) => {
862 if let PumpSwapTrade(i) = ix {
863 merge_pumpswap_trade_log_preferred(l, i);
864 }
865 }
866 PumpSwapBuy(l) => {
867 if let PumpSwapBuy(i) = ix {
868 merge_pumpswap_buy_log_preferred(l, i);
869 }
870 }
871 PumpSwapSell(l) => {
872 if let PumpSwapSell(i) = ix {
873 merge_pumpswap_sell_log_preferred(l, i);
874 }
875 }
876 RaydiumClmmSwap(l) => {
877 if let RaydiumClmmSwap(i) = ix {
878 merge_raydium_clmm_swap_log_preferred(l, i);
879 }
880 }
881 RaydiumAmmV4Swap(l) => {
882 if let RaydiumAmmV4Swap(i) = ix {
883 merge_raydium_amm_v4_swap_log_preferred(l, i);
884 }
885 }
886 RaydiumLaunchlabTrade(l) => {
887 if let RaydiumLaunchlabTrade(i) = ix {
888 merge_raydium_launchlab_trade_log_preferred(l, i);
889 }
890 }
891 RaydiumLaunchlabPoolCreate(l) => {
892 if let RaydiumLaunchlabPoolCreate(i) = ix {
893 merge_raydium_launchlab_pool_create_log_preferred(l, i);
894 }
895 }
896 RaydiumLaunchlabMigrateAmm(l) => {
897 if let RaydiumLaunchlabMigrateAmm(i) = ix {
898 merge_raydium_launchlab_migrate_amm_log_preferred(l, i);
899 }
900 }
901 PumpSwapCreatePool(l) => {
902 if let PumpSwapCreatePool(i) = ix {
903 merge_pumpswap_create_pool_log_preferred(l, i);
904 }
905 }
906 PumpSwapLiquidityAdded(l) => {
907 if let PumpSwapLiquidityAdded(i) = ix {
908 merge_pumpswap_liquidity_added_log_preferred(l, i);
909 }
910 }
911 PumpSwapLiquidityRemoved(l) => {
912 if let PumpSwapLiquidityRemoved(i) = ix {
913 merge_pumpswap_liquidity_removed_log_preferred(l, i);
914 }
915 }
916 MeteoraDlmmSwap(l) => {
917 if let MeteoraDlmmSwap(i) = ix {
918 merge_meteora_dlmm_swap_log_preferred(l, i);
919 }
920 }
921 _ => {}
922 }
923}
924
925#[inline]
926fn pumpfun_trade_from_ix_variant(ix: DexEvent) -> Option<PumpFunTradeEvent> {
927 match ix {
928 DexEvent::PumpFunTrade(t)
929 | DexEvent::PumpFunBuy(t)
930 | DexEvent::PumpFunSell(t)
931 | DexEvent::PumpFunBuyExactSolIn(t) => Some(t),
932 _ => None,
933 }
934}
935
936#[cfg(test)]
937mod tests {
938 use super::*;
939 use solana_sdk::{pubkey::Pubkey, signature::Signature};
940
941 #[test]
942 fn test_merge_pumpfun_trade() {
943 let metadata = EventMetadata {
944 signature: Signature::default(),
945 slot: 100,
946 tx_index: 1,
947 block_time_us: 1000,
948 grpc_recv_us: 2000,
949 recent_blockhash: None,
950 };
951
952 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
954 metadata: metadata.clone(),
955 bonding_curve: Pubkey::new_unique(),
956 associated_bonding_curve: Pubkey::new_unique(),
957 ..Default::default()
958 });
959
960 let inner = DexEvent::PumpFunTrade(PumpFunTradeEvent {
962 metadata: metadata.clone(),
963 mint: Pubkey::new_unique(),
964 sol_amount: 1000,
965 token_amount: 2000,
966 is_buy: true,
967 user: Pubkey::new_unique(),
968 ..Default::default()
969 });
970
971 merge_events(&mut base, inner);
973
974 if let DexEvent::PumpFunTrade(trade) = base {
976 assert_eq!(trade.sol_amount, 1000);
977 assert_eq!(trade.token_amount, 2000);
978 assert!(trade.is_buy);
979 assert_ne!(trade.bonding_curve, Pubkey::default());
981 assert_ne!(trade.associated_bonding_curve, Pubkey::default());
982 } else {
983 panic!("Expected PumpFunTrade event");
984 }
985 }
986
987 #[test]
988 fn merge_preserves_instruction_context_when_log_tail_is_absent() {
989 let metadata = EventMetadata {
990 signature: Signature::default(),
991 slot: 100,
992 tx_index: 1,
993 block_time_us: 1000,
994 grpc_recv_us: 2000,
995 recent_blockhash: None,
996 };
997 let quote_mint = Pubkey::new_unique();
998 let associated_quote_user = Pubkey::new_unique();
999
1000 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1001 metadata: metadata.clone(),
1002 ix_name: "buy_exact_quote_in".to_string(),
1003 quote_mint,
1004 spendable_quote_in: 1_000,
1005 min_tokens_out: 2_000,
1006 associated_quote_user,
1007 ..Default::default()
1008 });
1009
1010 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent {
1011 metadata,
1012 sol_amount: 1_000,
1013 token_amount: 2_000,
1014 is_buy: true,
1015 ..Default::default()
1016 });
1017
1018 merge_events(&mut base, inner);
1019
1020 if let DexEvent::PumpFunTrade(t) = base {
1021 assert_eq!(t.sol_amount, 1_000);
1022 assert_eq!(t.token_amount, 2_000);
1023 assert_eq!(t.ix_name, "buy_exact_quote_in");
1024 assert_eq!(t.quote_mint, quote_mint);
1025 assert_eq!(t.spendable_quote_in, 1_000);
1026 assert_eq!(t.min_tokens_out, 2_000);
1027 assert_eq!(t.associated_quote_user, associated_quote_user);
1028 } else {
1029 panic!("Expected PumpFunTrade event");
1030 }
1031 }
1032
1033 #[test]
1034 fn merge_replaces_sol_quote_sentinel_with_real_quote_mint() {
1035 let quote_mint = Pubkey::new_unique();
1036 let mut base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1037 quote_mint: PUMPFUN_SOLSCAN_SOL_QUOTE_MINT,
1038 ..Default::default()
1039 });
1040
1041 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent { quote_mint, ..Default::default() });
1042
1043 merge_events(&mut base, inner);
1044
1045 if let DexEvent::PumpFunTrade(t) = base {
1046 assert_eq!(t.quote_mint, quote_mint);
1047 } else {
1048 panic!("Expected PumpFunTrade event");
1049 }
1050 }
1051
1052 #[test]
1053 fn test_can_merge() {
1054 let metadata = EventMetadata {
1055 signature: Signature::default(),
1056 slot: 100,
1057 tx_index: 1,
1058 block_time_us: 1000,
1059 grpc_recv_us: 2000,
1060 recent_blockhash: None,
1061 };
1062
1063 let base = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1064 metadata: metadata.clone(),
1065 ..Default::default()
1066 });
1067
1068 let inner = DexEvent::PumpFunBuy(PumpFunTradeEvent {
1069 metadata: metadata.clone(),
1070 ..Default::default()
1071 });
1072
1073 assert!(can_merge(&base, &inner));
1075
1076 let different_sig = DexEvent::PumpFunTrade(PumpFunTradeEvent {
1078 metadata: EventMetadata { signature: Signature::new_unique(), ..metadata },
1079 ..Default::default()
1080 });
1081
1082 assert!(!can_merge(&base, &different_sig));
1083 }
1084
1085 #[test]
1086 fn grpc_merge_fills_fee_recipient_from_ix_when_log_default() {
1087 let metadata = EventMetadata {
1088 signature: Signature::default(),
1089 slot: 1,
1090 tx_index: 0,
1091 block_time_us: 0,
1092 grpc_recv_us: 0,
1093 recent_blockhash: None,
1094 };
1095 let fr = Pubkey::new_unique();
1096 let log_t =
1097 PumpFunTradeEvent { metadata: metadata.clone(), sol_amount: 50, ..Default::default() };
1098 let mut ix_t = log_t.clone();
1099 ix_t.fee_recipient = fr;
1100 ix_t.sol_amount = 777;
1101 let mut log_ev = DexEvent::PumpFunTrade(log_t);
1102 merge_grpc_instruction_into_log(&mut log_ev, DexEvent::PumpFunBuy(ix_t));
1103 match log_ev {
1104 DexEvent::PumpFunTrade(t) => {
1105 assert_eq!(t.fee_recipient, fr);
1106 assert_eq!(t.sol_amount, 50);
1107 }
1108 _ => panic!("expected trade"),
1109 }
1110 }
1111
1112 #[test]
1113 fn grpc_merge_keeps_log_trade_fields() {
1114 let metadata = EventMetadata {
1115 signature: Signature::default(),
1116 slot: 1,
1117 tx_index: 0,
1118 block_time_us: 0,
1119 grpc_recv_us: 0,
1120 recent_blockhash: None,
1121 };
1122 let log_t = PumpFunTradeEvent {
1123 metadata: metadata.clone(),
1124 mayhem_mode: true,
1125 sol_amount: 100,
1126 ..Default::default()
1127 };
1128 let mut ix_t = log_t.clone();
1129 ix_t.mayhem_mode = false;
1130 ix_t.sol_amount = 999;
1131
1132 let mut log_ev = DexEvent::PumpFunTrade(log_t);
1133 merge_grpc_instruction_into_log(&mut log_ev, DexEvent::PumpFunBuy(ix_t));
1134 match log_ev {
1135 DexEvent::PumpFunTrade(t) => {
1136 assert!(t.mayhem_mode);
1137 assert_eq!(t.sol_amount, 100);
1138 }
1139 _ => panic!("variant preserved"),
1140 }
1141 }
1142}