init.go

// Package bridge init router bridge and load / reload configs.
package bridge

import (
	"fmt"
	"math/big"
	"strings"
	"sync"

	"github.com/deltaswapio/swaprouter/v3/log"
	"github.com/deltaswapio/swaprouter/v3/mongodb"
	"github.com/deltaswapio/swaprouter/v3/mpc"
	"github.com/deltaswapio/swaprouter/v3/params"
	"github.com/deltaswapio/swaprouter/v3/router"
	"github.com/deltaswapio/swaprouter/v3/tokens"
)

var (
	routerInfoIsLoaded = new(sync.Map) // key is chainID and router contract address
)

func getRouterInfoLoadedKey(chainID, routerContract string) string {
	return strings.ToLower(fmt.Sprintf("%s:%s", chainID, routerContract))
}

func isRouterInfoLoaded(chainID, routerContract string) bool {
	key := getRouterInfoLoadedKey(chainID, routerContract)
	_, exist := routerInfoIsLoaded.Load(key)
	return exist
}

func setRouterInfoLoaded(chainID, routerContract string) {
	key := getRouterInfoLoadedKey(chainID, routerContract)
	routerInfoIsLoaded.Store(key, struct{}{})
}

// InitRouterBridges init router bridges
//
//nolint:funlen,gocyclo // ok
func InitRouterBridges(isServer bool) {
	log.Info("start init router bridges", "isServer", isServer)
	var success bool
	router.IsIniting = true
	defer func() {
		router.IsIniting = false
		routerInfoIsLoaded = new(sync.Map)
		log.Info("init router bridges finished", "isServer", isServer, "success", success)
	}()

	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)

	router.InitRouterConfigClients()

	log.Info("start get all chain ids")
	allChainIDs, err := router.GetAllChainIDs()
	if err != nil {
		logErrFunc("call GetAllChainIDs failed", "err", err)
		return
	}
	// get rid of blacked chainIDs
	chainIDs := make([]*big.Int, 0, len(allChainIDs))
	for _, chainID := range allChainIDs {
		if params.IsChainIDInBlackList(chainID.String()) {
			log.Debugf("ingore chainID %v in black list", chainID)
			continue
		}
		chainIDs = append(chainIDs, chainID)
	}
	log.Info("get all chain ids success", "chainIDs", chainIDs)
	if len(chainIDs) == 0 {
		logErrFunc("empty chain IDs")
		return
	}

	log.Info("start get all token ids")
	allTokenIDs, err := router.GetAllTokenIDs()
	if err != nil {
		logErrFunc("call GetAllTokenIDs failed", "err", err)
		return
	}
	// get rid of blacked tokenIDs
	tokenIDs := make([]string, 0, len(allTokenIDs))
	for _, tokenID := range allTokenIDs {
		if params.IsTokenIDInBlackList(tokenID) {
			log.Debugf("ingore tokenID %v in black list", tokenID)
			continue
		}
		tokenIDs = append(tokenIDs, tokenID)
	}
	log.Info("get all token ids success", "tokenIDs", tokenIDs)
	if len(tokenIDs) == 0 && !tokens.IsAnyCallRouter() {
		logErrFunc("empty token IDs")
		return
	}

	wg := new(sync.WaitGroup)
	wg.Add(len(chainIDs))
	for _, chainID := range chainIDs {
		go func(wg *sync.WaitGroup, chainID *big.Int) {
			defer wg.Done()

			bridge := NewCrossChainBridge(chainID)

			InitGatewayConfig(bridge, chainID)
			if bridge.GetGatewayConfig().IsEmpty() {
				logErrFunc("bridge has no gateway config", "chainID", chainID)
				return
			}

			AdjustGatewayOrder(bridge, chainID.String())
			InitChainConfig(bridge, chainID)

			bridge.InitAfterConfig()
			router.SetBridge(chainID.String(), bridge)

			latestBlock, err := bridge.GetLatestBlockNumber()
			if err != nil {
				log.Warn("get lastest block number failed", "chainID", chainID, "err", err)
			} else {
				log.Infof("[%5v] lastest block number is %v", chainID, latestBlock)
			}

			if params.GetLocalChainConfig(chainID.String()).ForbidParallelLoading {
				for _, tokenID := range tokenIDs {
					log.Info("start load token config", "tokenID", tokenID, "chainID", chainID)
					InitTokenConfig(bridge, tokenID, chainID)
				}
			} else {
				wg2 := new(sync.WaitGroup)
				wg2.Add(len(tokenIDs))
				for _, tokenID := range tokenIDs {
					go func(wg2 *sync.WaitGroup, tokenID string, chainID *big.Int) {
						defer wg2.Done()
						log.Info("start load token config", "tokenID", tokenID, "chainID", chainID)
						InitTokenConfig(bridge, tokenID, chainID)
					}(wg2, tokenID, chainID)
				}
				wg2.Wait()
			}
		}(wg, chainID)
	}
	wg.Wait()

	initSwapNonces()

	router.AllChainIDs = chainIDs
	router.AllTokenIDs = tokenIDs

	router.PrintMultichainTokens()

	loadSwapAndFeeConfigs()

	mpc.Init(isServer)

	success = true

	go WatchGatewayConfig()
}

func loadSwapAndFeeConfigs() {
	if !tokens.IsERC20Router() {
		return
	}

	swapConfigs := new(sync.Map)
	feeConfigs := new(sync.Map)

	wg := new(sync.WaitGroup)
	for _, tokenID := range router.AllTokenIDs {
		supportChainIDs := make([]*big.Int, 0, len(router.AllChainIDs))
		for _, chainID := range router.AllChainIDs {
			multichainToken := router.GetCachedMultichainToken(tokenID, chainID.String())
			if multichainToken != "" {
				supportChainIDs = append(supportChainIDs, chainID)
			}
		}
		if len(supportChainIDs) == 0 {
			continue
		}

		tokenIDSwapConfig := new(sync.Map)
		swapConfigs.Store(tokenID, tokenIDSwapConfig)

		tokenIDFeeConfig := new(sync.Map)
		feeConfigs.Store(tokenID, tokenIDFeeConfig)

		wg.Add(2)
		go loadSwapConfigs(wg, tokenIDSwapConfig, tokenID, supportChainIDs)
		go loadFeeConfigs(wg, tokenIDFeeConfig, tokenID, supportChainIDs)
	}
	wg.Wait()

	tokens.SetSwapConfigs(swapConfigs)
	tokens.SetFeeConfigs(feeConfigs)

	log.Info("load all swap and fee config success")
}

//nolint:dupl // allow duplicate
func loadSwapConfigs(wg *sync.WaitGroup, swapConfigs *sync.Map, tokenID string, supportChainIDs []*big.Int) {
	defer wg.Done()

	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)

	swapCfgs, err := router.GetSwapConfigs(tokenID)
	if err != nil {
		logErrFunc("get swap configs failed", "tokenID", tokenID, "err", err)
		return
	}

	swapCfgMap := make(map[string]map[string]*tokens.SwapConfig)
	for _, cfg := range swapCfgs {
		fromChainID := cfg.FromChainID.String()
		toChainID := cfg.ToChainID.String()
		innerMap, exist := swapCfgMap[fromChainID]
		if !exist {
			innerMap = make(map[string]*tokens.SwapConfig)
			swapCfgMap[fromChainID] = innerMap
		}
		swapCfg := &tokens.SwapConfig{
			MaximumSwap:       cfg.MaximumSwap,
			MinimumSwap:       cfg.MinimumSwap,
			BigValueThreshold: cfg.BigValueThreshold,
		}
		err = swapCfg.CheckConfig()
		if err != nil {
			logErrFunc("check swap config failed", "tokenID", tokenID, "fromChainID", fromChainID, "toChainID", toChainID, "config", swapCfg, "err", err)
			return
		}
		innerMap[toChainID] = swapCfg
		log.Info("load swap config success", "tokenID", tokenID, "fromChainID", fromChainID, "toChainID", toChainID, "config", swapCfg)
	}

	for i, fromChainID := range supportChainIDs {
		innerMap := new(sync.Map)
		swapConfigs.Store(fromChainID.String(), innerMap)
		fmap, fexist := swapCfgMap[fromChainID.String()]
		for j, toChainID := range supportChainIDs {
			if i == j {
				continue
			}
			var swapCfg *tokens.SwapConfig
			var exist bool
			if fexist {
				// 1. if _swapConfig[tokenID][srcChainID][dstChainID] exist, then use it.
				swapCfg, exist = fmap[toChainID.String()]
				if !exist {
					// 2. else if _swapConfig[tokenID][srcChainID][0] exist, then use it.
					swapCfg, exist = fmap["0"]
				}
			}
			if !exist {
				if zmap, zexist := swapCfgMap["0"]; zexist {
					// 3. else if _swapConfig[tokenID][0][dstChainID] exist, then use it.
					swapCfg, exist = zmap[toChainID.String()]
					if !exist {
						// 4. else use _swapConfig[tokenID][0][0].
						swapCfg, exist = zmap["0"]
					}
				}
			}
			if !exist {
				continue
			}
			innerMap.Store(toChainID.String(), swapCfg)
		}
	}
}

//nolint:dupl // allow duplicate
func loadFeeConfigs(wg *sync.WaitGroup, feeConfigs *sync.Map, tokenID string, supportChainIDs []*big.Int) {
	defer wg.Done()

	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)

	feeCfgs, err := router.GetFeeConfigs(tokenID)
	if err != nil {
		logErrFunc("get fee configs failed", "tokenID", tokenID, "err", err)
		return
	}

	feeCfgMap := make(map[string]map[string]*tokens.FeeConfig)
	for _, cfg := range feeCfgs {
		fromChainID := cfg.FromChainID.String()
		toChainID := cfg.ToChainID.String()
		innerMap, exist := feeCfgMap[fromChainID]
		if !exist {
			innerMap = make(map[string]*tokens.FeeConfig)
			feeCfgMap[fromChainID] = innerMap
		}
		feeCfg := &tokens.FeeConfig{
			MaximumSwapFee:        cfg.MaximumSwapFee,
			MinimumSwapFee:        cfg.MinimumSwapFee,
			SwapFeeRatePerMillion: cfg.SwapFeeRatePerMillion,
		}
		err = feeCfg.CheckConfig()
		if err != nil {
			logErrFunc("check fee config failed", "tokenID", tokenID, "fromChainID", fromChainID, "toChainID", toChainID, "config", feeCfg, "err", err)
			return
		}
		innerMap[toChainID] = feeCfg
		log.Info("load fee config success", "tokenID", tokenID, "fromChainID", fromChainID, "toChainID", toChainID, "config", feeCfg)
	}

	for i, fromChainID := range supportChainIDs {
		innerMap := new(sync.Map)
		feeConfigs.Store(fromChainID.String(), innerMap)
		fmap, fexist := feeCfgMap[fromChainID.String()]
		for j, toChainID := range supportChainIDs {
			if i == j {
				continue
			}
			var feeCfg *tokens.FeeConfig
			var exist bool
			if fexist {
				// 1. if _feeConfig[tokenID][srcChainID][dstChainID] exist, then use it.
				feeCfg, exist = fmap[toChainID.String()]
				if !exist {
					// 2. else if _feeConfig[tokenID][srcChainID][0] exist, then use it.
					feeCfg, exist = fmap["0"]
				}
			}
			if !exist {
				if zmap, zexist := feeCfgMap["0"]; zexist {
					// 3. else if _feeConfig[tokenID][0][dstChainID] exist, then use it.
					feeCfg, exist = zmap[toChainID.String()]
					if !exist {
						// 4. else use _feeConfig[tokenID][0][0].
						feeCfg, exist = zmap["0"]
					}
				}
			}
			if !exist {
				continue
			}
			innerMap.Store(toChainID.String(), feeCfg)
		}
	}
}

// InitGatewayConfig impl
func InitGatewayConfig(b tokens.IBridge, chainID *big.Int) {
	isReload := router.IsReloading
	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)
	if chainID == nil || chainID.Sign() == 0 {
		logErrFunc("init gateway with zero chain ID")
		return
	}
	SetGatewayConfig(b, chainID.String())
	log.Info(fmt.Sprintf("[%5v] init gateway config success", chainID), "isReload", isReload)
}

// SetGatewayConfig set gateway config
func SetGatewayConfig(b tokens.IBridge, chainID string) {
	cfg := params.GetRouterConfig()
	apiAddrs := cfg.Gateways[chainID]
	apiAddrsExt := cfg.GatewaysExt[chainID]
	evmapiext := cfg.EVMGatewaysExt[chainID]
	finalizeAPIs := cfg.FinalizeGateways[chainID]
	grpcAPIs := cfg.GRPCGateways[chainID]
	b.SetGatewayConfig(&tokens.GatewayConfig{
		APIAddress:         apiAddrs,
		APIAddressExt:      apiAddrsExt,
		EVMAPIAddress:      evmapiext,
		FinalizeAPIAddress: finalizeAPIs,
		GRPCAPIAddress:     grpcAPIs,
	})
}

// InitChainConfig impl
func InitChainConfig(b tokens.IBridge, chainID *big.Int) {
	isReload := router.IsReloading
	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)
	chainCfg, err := router.GetChainConfig(chainID)
	if err != nil {
		logErrFunc("get chain config failed", "chainID", chainID, "err", err)
		return
	}
	if chainCfg == nil {
		logErrFunc("chain config not found", "chainID", chainID)
		return
	}
	if chainID.String() != chainCfg.ChainID {
		logErrFunc("verify chain ID mismatch", "inconfig", chainCfg.ChainID, "inchainids", chainID)
		return
	}
	if err = chainCfg.CheckConfig(); err != nil {
		logErrFunc("check chain config failed", "chainID", chainID, "err", err)
		return
	}
	b.SetChainConfig(chainCfg)
	log.Info("init chain config success", "blockChain", chainCfg.BlockChain, "chainID", chainID, "isReload", isReload, "chainCfg", chainCfg)

	routerContract := chainCfg.RouterContract
	if routerContract != "" && !isRouterInfoLoaded(chainID.String(), routerContract) {
		err = b.InitRouterInfo(routerContract, chainCfg.RouterVersion)
		if err == nil {
			setRouterInfoLoaded(chainID.String(), routerContract)
		} else {
			logErrFunc("init chain router info failed", "chainID", chainID, "routerContract", routerContract, "err", err)
			return
		}
	}
}

// InitTokenConfig impl
//
//nolint:funlen,gocyclo // allow long init token config method
func InitTokenConfig(b tokens.IBridge, tokenID string, chainID *big.Int) {
	isReload := router.IsReloading
	logErrFunc := log.GetLogFuncOr(router.DontPanicInLoading(), log.Error, log.Fatal)
	if tokenID == "" {
		logErrFunc("empty token ID")
		return
	}
	tokenAddr, err := router.GetMultichainToken(tokenID, chainID)
	if err != nil {
		logErrFunc("get token address failed", "tokenID", tokenID, "chainID", chainID, "err", err)
		return
	}
	if tokenAddr == "" {
		log.Debugf("[%5v] '%v' token address is empty", chainID, tokenID)
		return
	}
	tokenCfg, err := router.GetTokenConfig(chainID, tokenID)
	if err != nil {
		logErrFunc("get token config failed", "chainID", chainID, "tokenID", tokenID, "err", err)
		return
	}
	if tokenCfg == nil {
		log.Debug("token config not found", "tokenID", tokenID, "chainID", chainID, "tokenAddr", tokenAddr)
		return
	}
	if !strings.EqualFold(tokenAddr, tokenCfg.ContractAddress) {
		logErrFunc("verify token address mismach", "tokenID", tokenID, "chainID", chainID, "inconfig", tokenCfg.ContractAddress, "inmultichain", tokenAddr)
		return
	}
	if tokenID != tokenCfg.TokenID {
		logErrFunc("verify token ID mismatch", "chainID", chainID, "inconfig", tokenCfg.TokenID, "intokenids", tokenID)
		return
	}
	if err = tokenCfg.CheckConfig(); err != nil {
		logErrFunc("check token config failed", "tokenID", tokenID, "chainID", chainID, "tokenAddr", tokenAddr, "err", err)
		return
	}
	router.InitOnchainCustomConfig(chainID, tokenID)
	b.SetTokenConfig(tokenAddr, tokenCfg)

	router.SetMultichainToken(tokenID, chainID.String(), tokenAddr)

	log.Info(fmt.Sprintf("[%5v] init '%v' token config success", chainID, tokenID), "tokenAddr", tokenAddr, "decimals", tokenCfg.Decimals, "isReload", isReload, "tokenCfg", tokenCfg)

	routerContract := tokenCfg.RouterContract
	if routerContract != "" && !isRouterInfoLoaded(chainID.String(), routerContract) {
		err = b.InitRouterInfo(routerContract, tokenCfg.RouterVersion)
		if err == nil {
			setRouterInfoLoaded(chainID.String(), routerContract)
		} else {
			logErrFunc("init token router info failed", "chainID", chainID, "routerContract", routerContract, "err", err)
			return
		}
	}
}

func initSwapNonces() {
	if !mongodb.HasClient() {
		return
	}

	routerInfoIsLoaded.Range(func(k, v interface{}) bool {
		key := k.(string)
		parts := strings.Split(key, ":")
		if len(parts) != 2 {
			return true
		}

		chainID := parts[0]
		routerContract := parts[1]

		br := router.GetBridgeByChainID(chainID)
		if br == nil {
			return true
		}
		nonceSetter, ok := br.(tokens.NonceSetter)
		if !ok {
			return true
		}

		routerInfo := router.GetRouterInfo(routerContract, chainID)
		if routerInfo == nil {
			return true
		}
		routerMPC := routerInfo.RouterMPC

		for i := 0; i < 3; i++ {
			nextSwapNonce, err := mongodb.FindNextSwapNonce(chainID, routerMPC)
			if err == nil {
				nonceSetter.InitSwapNonce(nonceSetter, routerMPC, nextSwapNonce)
				break
			}
		}

		return true
	})
}