libbrc20-indexer/indexer/brc20_mint.go

package indexer

import (
	"errors"
	"fmt"
	"time"

	"github.com/unisat-wallet/libbrc20-indexer/constant"
	"github.com/unisat-wallet/libbrc20-indexer/decimal"
	"github.com/unisat-wallet/libbrc20-indexer/model"
	"github.com/unisat-wallet/libbrc20-indexer/utils"
)

func (g *BRC20ModuleIndexer) ProcessMint(data *model.InscriptionBRC20Data) error {
	body := new(model.InscriptionBRC20MintTransferContent)
	if err := body.Unmarshal(data.ContentBody); err != nil {
		return nil
	}

	// check tick
	uniqueLowerTicker, err := utils.GetValidUniqueLowerTickerTicker(body.BRC20Tick)
	if err != nil {
		return nil
		// return errors.New("mint, tick length not 4 or 5")
	}
	tokenInfo, ok := g.InscriptionsTickerInfoMap[uniqueLowerTicker]
	if !ok {
		return nil
		// return errors.New(fmt.Sprintf("mint %s, but tick not exist", body.BRC20Tick))
	}
	tinfo := tokenInfo.Deploy
	if tinfo.SelfMint {
		if utils.DecodeInscriptionFromBin(data.Parent) != tinfo.GetInscriptionId() {
			return errors.New(fmt.Sprintf("self mint %s, but parent invalid", body.BRC20Tick))
		}
	}

	// check mint amount
	amt, err := decimal.NewDecimalFromString(body.BRC20Amount, int(tinfo.Decimal))
	if err != nil {
		return errors.New(fmt.Sprintf("mint %s, but invalid amount(%s)", body.BRC20Tick, body.BRC20Amount))
	}
	if amt.Sign() <= 0 || amt.Cmp(tinfo.Limit) > 0 {
		return errors.New(fmt.Sprintf("mint %s, invalid amount(%s), limit(%s)", body.BRC20Tick, body.BRC20Amount, tinfo.Limit))
	}

	// get user's tokens to update
	tokenBalance := g.GetUserTokenBalance(tokenInfo.Ticker, string(data.PkScript))

	body.BRC20Tick = tokenInfo.Ticker
	mintInfo := model.NewInscriptionBRC20TickInfo(body.BRC20Tick, body.Operation, data)
	mintInfo.Data.BRC20Amount = body.BRC20Amount
	mintInfo.Data.BRC20Minted = amt.String()
	mintInfo.Decimal = tinfo.Decimal
	mintInfo.Amount = amt
	if tinfo.TotalMinted.Cmp(tinfo.Max) >= 0 {
		// invalid history
		if g.EnableHistory {
			historyObj := model.NewBRC20History(constant.BRC20_HISTORY_TYPE_N_INSCRIBE_MINT, false, false, mintInfo, tokenBalance, data)
			history := g.UpdateHistoryHeightAndGetHistoryIndex(historyObj)

			tokenBalance.History = append(tokenBalance.History, history)
			tokenBalance.HistoryMint = append(tokenBalance.HistoryMint, history)
			tokenInfo.History = append(tokenInfo.History, history)
			tokenInfo.HistoryMint = append(tokenInfo.HistoryMint, history)
		}
		return errors.New(fmt.Sprintf("mint %s, but mint out", body.BRC20Tick))
	}

	// update tinfo
	tokenInfo.UpdateHeight = data.Height

	// minted
	balanceMinted := decimal.NewDecimalCopy(amt)
	if tinfo.TotalMinted.Add(amt).Cmp(tinfo.Max) > 0 {
		balanceMinted = tinfo.Max.Sub(tinfo.TotalMinted)
	}
	tinfo.TotalMinted = tinfo.TotalMinted.Add(balanceMinted)
	if tinfo.TotalMinted.Cmp(tinfo.Max) >= 0 {
		tinfo.CompleteHeight = data.Height
		tinfo.CompleteBlockTime = data.BlockTime
	}
	// confirmed minted
	now := time.Now()
	if data.BlockTime > 0 {
		tinfo.ConfirmedMinted = tinfo.ConfirmedMinted.Add(balanceMinted)
		if data.BlockTime < uint32(now.Unix())-3600 {
			tinfo.ConfirmedMinted1h = tinfo.ConfirmedMinted1h.Add(balanceMinted)
		}
		if data.BlockTime < uint32(now.Unix())-86400 {
			tinfo.ConfirmedMinted24h = tinfo.ConfirmedMinted24h.Add(balanceMinted)
		}
	}
	// count
	tinfo.MintTimes++
	tinfo.Data.BRC20Minted = tinfo.TotalMinted.String()
	// valid mint inscriptionNumber range
	tinfo.InscriptionNumberEnd = data.InscriptionNumber

	// update mint info
	mintInfo.Data.BRC20Minted = balanceMinted.String()
	mintInfo.Amount = balanceMinted

	// update tokenBalance
	tokenBalance.UpdateHeight = data.Height

	if data.BlockTime > 0 {
		tokenBalance.AvailableBalanceSafe = tokenBalance.AvailableBalanceSafe.Add(balanceMinted)
	}
	tokenBalance.AvailableBalance = tokenBalance.AvailableBalance.Add(balanceMinted)

	// burn
	if len(data.PkScript) == 1 && data.PkScript[0] == 0x6a {
		tinfo.Burned = tinfo.Burned.Add(balanceMinted)
	}

	if g.EnableHistory {
		// history
		historyObj := model.NewBRC20History(constant.BRC20_HISTORY_TYPE_N_INSCRIBE_MINT, true, false, mintInfo, tokenBalance, data)
		history := g.UpdateHistoryHeightAndGetHistoryIndex(historyObj)

		// tick history
		tokenBalance.History = append(tokenBalance.History, history)
		tokenBalance.HistoryMint = append(tokenBalance.HistoryMint, history)
		tokenInfo.History = append(tokenInfo.History, history)
		tokenInfo.HistoryMint = append(tokenInfo.HistoryMint, history)
		// user address
		userHistory := g.GetBRC20HistoryByUser(string(data.PkScript))
		userHistory.History = append(userHistory.History, history)
		// all history
		g.AllHistory = append(g.AllHistory, history)
	}
	// g.InscriptionsValidBRC20DataMap[data.CreateIdxKey] = mintInfo.Data
	return nil
}