Remove Li.fi bridge and optimize quote process

Author: perrornet

README.md

@@ -25,7 +25,6 @@ Omni Balance is an intelligent liquidity management tool designed to achieve sea
 | [Darwinia Bridge](https://bridge.darwinia.network/)                              |      | ✅           |
 | [OKX web3](https://www.okx.com/zh-hans/web3/build/docs/waas/dex-crosschain-swap) | ✅    | ✅           |
 | [Router Nitro](https://app.routernitro.com/swap)                                 | ✅    | ✅           |
-| [Li.FI](https://li.fi/)                                                          | ✅    | ✅           |
 
 
 ### Installation

cmd/main.go

@@ -17,7 +17,6 @@ import (
 	"omni-balance/utils/provider"
 	_ "omni-balance/utils/provider/bridge/darwinia"
 	_ "omni-balance/utils/provider/bridge/helix_liquidity_claim"
-	_ "omni-balance/utils/provider/bridge/li"
 	_ "omni-balance/utils/provider/bridge/okx"
 	_ "omni-balance/utils/provider/bridge/routernitro"
 	_ "omni-balance/utils/provider/cex/gate"

internal/daemons/bot/util.go

@@ -68,6 +68,7 @@ func createOrder(ctx context.Context, tasks []bot.Task, processType bot.ProcessT
 			TaskId:           taskId,
 			ProcessType:      string(processType),
 			Remark:           v.Remark,
+			CurrentBalance:   v.CurrentBalance,
 		}
 
 		if err = txn.Create(o).Error; err != nil {

internal/daemons/market/init.go

@@ -11,6 +11,6 @@ func init() {
 		Description:     "Look for tasks in the database that are not being processed and process them.",
 		TaskFunc:        Run,
 		DefaultInterval: time.Minute * 3,
-		RunOnStart:      true,
+		RunOnStart:      false,
 	})
 }

internal/daemons/market/util.go

@@ -96,7 +96,7 @@ func getBestProvider(ctx context.Context, order models.Order, conf configs.Confi
 			if !ok || len(tokenInCosts) == 0 {
 				continue
 			}
-			log.Debugf("provider %s can use %s on %s. The tokenInCosts is %+v",
+			log.Infof("provider %s can use %s on %s. The tokenInCosts is %+v",
 				p.Name(), order.TokenOutName, order.TargetChainName, tokenInCosts)
 			canUseProviders = append(canUseProviders, canUseProvider{
 				provider:     p,
@@ -131,16 +131,27 @@ func getBestProvider(ctx context.Context, order models.Order, conf configs.Confi
 		}
 		tokenName2Price, err := models.FindTokenPrice(db.DB(), tokenNames)
 		if err != nil {
-			log.Warnf("find token price error: %s", err.Error())
+			log.Warnf("find %+v token price error: %s", tokenNames, err.Error())
 			continue
 		}
 		// if token out is RING and provider is darwinia, then use darwinia as the provider
 		if order.TokenOutName == "RING" && canUseProvider.provider.Name() == new(darwinia.Bridge).Name() {
 			return canUseProvider.provider, nil
 		}
+		if len(tokenName2Price) == 0 {
+			log.Warnf("not found %+v price, set to 0", tokenNames)
+			minPrice = decimal.Zero
+			providerObj = canUseProvider.provider
+			continue
+		}
 
 		for name, v := range tokenName2Price {
 			log.Debugf("token %s price %s on %s", name, v.String(), canUseProvider.provider.Name())
+			if v.LessThanOrEqual(decimal.Zero) {
+				log.Warnf("token %s price %s is less than or equal to zero, set to 1",
+					name, v.String())
+				v = decimal.RequireFromString("1")
+			}
 			price := v.Mul(tokenInCostsMap[name])
 			if price.IsZero() {
 				continue
@@ -173,12 +184,12 @@ func providerSupportsOrder(ctx context.Context, p provider.Provider, order model
 		TargetChain:      order.TargetChainName,
 		Amount:           order.Amount,
 		SourceChainNames: order.TokenInChainNames,
+		CurrentBalance:   order.CurrentBalance,
 	})
 	if err != nil {
 		log.Debugf("token %s on %s cannot use %s provider, source chain is '%s', source token is '%s'", order.TokenOutName, order.TargetChainName, p.Name(), order.SourceChainName, order.TokenInName)
 		return nil, false
 	}
-	log.Debugf("check token %s on %s use %s success, the tokenInCosts is %+v", order.TokenOutName, order.TargetChainName, p.Name(), tokenInCosts)
 	return tokenInCosts, true
 }
 

internal/daemons/token_price/token_price.go

@@ -21,7 +21,7 @@ func init() {
 		Description:     "Responsible for obtaining the token price, denominated in USDT.",
 		TaskFunc:        Run,
 		DefaultInterval: time.Minute * 3,
-		// RunOnStart:      true,
+		RunOnStart:      true,
 	})
 }
 
@@ -39,7 +39,6 @@ func Run(ctx context.Context, conf configs.Config) error {
 			defer w.Done()
 			result, err := provider.GetTokenPriceInUSDT(ctx, conf.SourceTokens...)
 			if err != nil {
-
 				log.Warnf("%s get token price error: %s", provider.Name(), err)
 				return
 			}

internal/daemons/utils.go

@@ -39,6 +39,7 @@ func CreateSwapParams(order models.Order, orderProcess models.OrderProcess, wall
 		RecordFn:         createRecordFunction(order),
 		Order:            order.Order,
 		Remark:           order.Remark,
+		CurrentBalance:   order.CurrentBalance,
 	}
 }
 

internal/models/token_price.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"context"
+
 	"github.com/shopspring/decimal"
 	"gorm.io/gorm"
 )
@@ -56,7 +57,7 @@ func FindTokenPrice(db *gorm.DB, tokenName []string) (result map[string]decimal.
 		tokenPrices     []TokenPrice
 		tokenName2Price = make(map[string]TokenPrices)
 	)
-	err = db.Where("token_name in (?)", tokenName).First(&tokenPrices).Error
+	err = db.Where("token_name in (?)", tokenName).Find(&tokenPrices).Error
 	if err != nil {
 		return
 	}

utils/bot/balance_on_chain/monitor.go

@@ -56,6 +56,7 @@ func (b BalanceOnChain) Check(ctx context.Context, args bot.Params) ([]bot.Task,
 		Wallet:            info.Wallet.GetAddress().Hex(),
 		TokenOutName:      info.TokenName,
 		Amount:            amount,
+		CurrentBalance:    balance,
 	})
 	return tasks, bot.Queue, nil
 }

utils/bot/bot.go

@@ -41,6 +41,7 @@ type Config struct {
 
 type Task struct {
 	Wallet            string               `json:"wallet" gorm:"type:varchar(64)"`
+	CurrentBalance    decimal.Decimal      `json:"current_balance"`
 	TokenInName       string               `json:"token_in_name"`
 	TokenOutName      string               `json:"token_out_name" gorm:"type:varchar(64)"`
 	TokenInChainName  string               `json:"source_chain_name"`

utils/bot/gate_liquidity/gate_liquidity.go

@@ -50,6 +50,7 @@ func (b GateLiquidity) Check(ctx context.Context, args bot.Params) ([]bot.Task,
 			ProviderType:      configs.CEX,
 			ProviderName:      "gate.io",
 			Remark:            b.Name(),
+			CurrentBalance:    task.CurrentBalance,
 		})
 	}
 	return result, processType, nil

utils/bot/helix_liquidity/helix_liquidity.go

@@ -71,8 +71,8 @@ func (h HelixLiquidity) Check(ctx context.Context, args bot.Params) ([]bot.Task,
 
 	if total.Add(purchaseAmount).LessThanOrEqual(threshold) {
 		newAmount := threshold.Add(threshold.Mul(decimal.RequireFromString("0.3")))
-		log.Infof("The %s current balance is %s, amount in config is %s, balance(%s) + amount(%s) <= threshold(%s), so set amount to %s",
-			args.Info.Wallet.GetAddress(), total, purchaseAmount, total, purchaseAmount, threshold, newAmount)
+		log.Infof("The %s %s on %s current balance is %s, amount in config is %s, balance(%s) + amount(%s) <= threshold(%s), so set amount to %s",
+			args.Info.Wallet.GetAddress(), args.Info.TokenName, args.Info.Chain, total, purchaseAmount, total, purchaseAmount, threshold, newAmount)
 		purchaseAmount = newAmount
 	}
 	return []bot.Task{
@@ -82,6 +82,7 @@ func (h HelixLiquidity) Check(ctx context.Context, args bot.Params) ([]bot.Task,
 			TokenOutChainName: args.Info.Chain,
 			Amount:            purchaseAmount,
 			Status:            provider.TxStatusPending,
+			CurrentBalance:    total,
 		},
 	}, bot.Queue, nil
 }

utils/configs/config.go

@@ -332,7 +332,7 @@ func (c *Config) Check() error {
 				}
 			}
 			if !ok {
-				return errors.Errorf("source_token[%d] token name not in chainsMap", index)
+				return errors.Errorf("token %s, source_token[%d]  token name not in chainsMap", v.Name, index)
 			}
 		}
 	}