pump-parser/tx.go

package pump_parser

import (
	"github.com/gagliardetto/solana-go"
	"github.com/mr-tron/base58"
	"github.com/shopspring/decimal"
)

type Swap struct {
	Program string
	Event   string

	Pool      solana.PublicKey
	BaseMint  solana.PublicKey
	QuoteMint solana.PublicKey

	BaseTokenProgram  solana.PublicKey
	QuoteTokenProgram solana.PublicKey

	Creator solana.PublicKey

	BaseMintDecimals  uint8
	QuoteMintDecimals uint8

	User        solana.PublicKey
	BaseAmount  decimal.Decimal
	QuoteAmount decimal.Decimal

	BaseReserve  decimal.Decimal
	QuoteReserve decimal.Decimal
	Mayhem       bool

	UserBaseBalance  decimal.Decimal
	UserQuoteBalance decimal.Decimal
	EntryContract    solana.PublicKey
}

type platformInfo struct {
	Platform    string
	PlatformFee decimal.Decimal
}

type mevInfo struct {
	MevAgent    string
	MevAgentFee decimal.Decimal
}

type Tx struct {
	rawTx      *RawTx
	Signer     solana.PublicKey
	Err        interface{} `json:"err,omitempty"`
	Swaps      []Swap      `json:"swaps,omitempty"`
	Block      uint64      `json:"block"`
	BlockIndex uint64      `json:"index"`
	TxHash     *[64]byte   `json:"-"`
	BlockAt    int64       `json:"block_at"`

	cachedTxHash string

	Platform map[string]platformInfo `json:"platform"`
	MevAgent map[string]mevInfo      ` json:"tx_mev_agent"`
	CUPrice  decimal.Decimal         ` json:"tx_cu_price"`

	BeforeSolBalance decimal.Decimal `json:"-"`
	AfterSOLBalance  decimal.Decimal `json:"after_sol_balance"`

	// update tokenInfo
	Token map[solana.PublicKey]TokenMeta `gorm:"-"`

	// todo pool info ??
}

func (tx *Tx) SetRawTx(t *RawTx) {
	tx.rawTx = t
}

func (tx *Tx) GetSignerTokenBalanceAfterTx(tokenProgram, tokenMint solana.PublicKey) decimal.Decimal {
	return GetTokenBalanceAfterTx(tx.rawTx, 0, tokenProgram, tokenMint)
}

type TokenMeta struct {
	Mint         solana.PublicKey `json:"address"`
	TokenProgram solana.PublicKey `json:"token_program"`
	Decimals     uint8            `json:"decimals"`

	Name   string
	Symbol string
	Url    string

	TotalSupply *decimal.Decimal
}

func (tx *Tx) GetTxHash() string {
	if tx.cachedTxHash != "" {
		return tx.cachedTxHash
	}
	if tx.TxHash == nil {
		return ""
	}
	tx.cachedTxHash = base58.Encode(tx.TxHash[:])
	return tx.cachedTxHash
}

func (tx *Tx) CheckPlatform(swap Swap) (string, decimal.Decimal) {
	// hasSolProgramRaydiumLaunchLabBonk
	rawTx := tx.rawTx
	var platform string
	var platformFee decimal.Decimal
	if len(tx.Platform) == 0 {
		return PlatformNone, decimal.Zero
	}

	for p, info := range tx.Platform {
		platform = p
		platformFee = info.PlatformFee
		break
	}
	if swap.Event == "buy" && swap.Program == SolProgramRaydiumLaunchLabBonk {
		for _, p := range tx.Platform {
			switch p.Platform {
			case PlatformAxiom:
				if !checkBonkAxiomBuy(rawTx) {
					platform = PlatformFake
				}
			case PlatformGMGN:
				if !checkBonkGmgnBuy(rawTx) {
					platform = PlatformFake
				}
			}
		}

	}

	if platform != "" &&
		platform != PlatformFake {
		if (swap.QuoteMint.Equals(wSolMint) || swap.QuoteMint.IsZero()) &&
			platformFee.LessThan(swap.QuoteAmount.Div(decimal.New(1, int32(swap.QuoteMintDecimals))).Div(decimal.NewFromInt(10000)).Mul(decimal.NewFromInt(9))) {
			platform = PlatformFake
		} else if swap.BaseMint.Equals(wSolMint) &&
			platformFee.LessThan(swap.QuoteAmount.Div(decimal.New(1, int32(swap.QuoteMintDecimals))).Div(decimal.NewFromInt(10000)).Mul(decimal.NewFromInt(9))) {
			platform = PlatformFake
		}

	}
	if platform == "" {
		platform = PlatformNone
	}
	return platform, platformFee
}

func (tx *Tx) CheckMevAgent() (string, decimal.Decimal) {
	var mevAgent = MevAgentUnknown
	var mevAgentFee = decimal.Zero

	for m, info := range tx.MevAgent {
		if len(tx.MevAgent) > 1 && info.MevAgent == MevAgentUnknown {
			continue
		}
		mevAgent = m
		mevAgentFee = info.MevAgentFee
		break
	}
	if len(tx.MevAgent) == 0 && mevAgent == MevAgentUnknown {
		// set the mev agent to none if the platform does not exist
		return "", decimal.Zero
	}
	return mevAgent, mevAgentFee
}

func (s Swap) CheckEntryContract() string {
	name, ok := entryContractAddresses[s.EntryContract]
	if ok {
		return name
	}
	return EntryContractUnknown
}

func (s Swap) CheckEntryContractV2() string {
	name, ok := entryContractAddresses[s.EntryContract]
	if ok {
		return name
	}
	return s.EntryContract.String()
}