Binance Withdrawals

client/mm/config.go

@@ -38,15 +38,18 @@ type CEXConfig struct {
 // The balance fields are the initial amounts that will be reserved to use for
 // this bot. As the bot trades, the amounts reserved for it will be updated.
 type BotConfig struct {
-	Host       string `json:"host"`
-	BaseAsset  uint32 `json:"baseAsset"`
-	QuoteAsset uint32 `json:"quoteAsset"`
-
-	BaseBalanceType BalanceType `json:"baseBalanceType"`
-	BaseBalance     uint64      `json:"baseBalance"`
-
+	Host             string      `json:"host"`
+	BaseAsset        uint32      `json:"baseAsset"`
+	QuoteAsset       uint32      `json:"quoteAsset"`
+	BaseBalanceType  BalanceType `json:"baseBalanceType"`
+	BaseBalance      uint64      `json:"baseBalance"`
 	QuoteBalanceType BalanceType `json:"quoteBalanceType"`
 	QuoteBalance     uint64      `json:"quoteBalance"`
+	// CEX balance types are only applicable for arb bots.
+	CEXBaseBalanceType  BalanceType `json:"cexBaseBalanceType"`
+	CEXBaseBalance      uint64      `json:"cexBaseBalance"`
+	CEXQuoteBalanceType BalanceType `json:"cexQuoteBalanceType"`
+	CEXQuoteBalance     uint64      `json:"cexQuoteBalance"`
 
 	// Only one of the following configs should be set
 	BasicMMConfig   *BasicMarketMakingConfig   `json:"basicMarketMakingConfig,omitempty"`
@@ -63,6 +66,20 @@ func (c *BotConfig) requiresPriceOracle() bool {
 	return false
 }
 
+func (c *BotConfig) requiresCEX() bool {
+	return c.SimpleArbConfig != nil || c.MMWithCEXConfig != nil
+}
+
+func (c *BotConfig) cexName() string {
+	switch {
+	case c.SimpleArbConfig != nil:
+		return c.SimpleArbConfig.CEXName
+	case c.MMWithCEXConfig != nil:
+		return ""
+	}
+	return ""
+}
+
 func dexMarketID(host string, base, quote uint32) string {
 	return fmt.Sprintf("%s-%d-%d", host, base, quote)
 }