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