Merge pull request #76 from Team-Kujira/fix-matic

Add rewrites from MATIC to POL
Team-Kujira · Sep 14, 2024 · 2d9c7b5 · 2d9c7b5
2 parents 41c5264 + 94085ae
commit 2d9c7b5
Show file tree

Hide file tree

Showing 12 changed files with 101 additions and 16 deletions.
diff --git a/oracle/provider/binance.go b/oracle/provider/binance.go
@@ -79,7 +79,7 @@ func NewBinanceProvider(
 	}
 
 	availablePairs, _ := provider.GetAvailablePairs()
-	provider.setPairs(pairs, availablePairs, nil)
+	provider.setPairs(pairs, availablePairs, currencyPairToBinanceSymbol)
 
 	go startPolling(provider, provider.endpoints.PollInterval, logger)
 	return provider, nil
@@ -140,3 +140,21 @@ func (p *BinanceProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 	return symbols, nil
 }
+
+func currencyPairToBinanceSymbol(pair types.CurrencyPair) string {
+	mapping := map[string]string{
+		"MATIC": "POL",
+	}
+
+	base, found := mapping[pair.Base]
+	if !found {
+		base = pair.Base
+	}
+
+	quote, found := mapping[pair.Quote]
+	if !found {
+		quote = pair.Quote
+	}
+
+	return base + quote
+}
diff --git a/oracle/provider/bitfinex.go b/oracle/provider/bitfinex.go
@@ -134,8 +134,9 @@ func (p *BitfinexProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToBitfinexSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"LUNC": "LUNA",
-		"LUNA": "LUNA2",
+		"LUNC":  "LUNA",
+		"LUNA":  "LUNA2",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/bitget.go b/oracle/provider/bitget.go
@@ -129,7 +129,8 @@ func (p *BitgetProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToBitgetSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"AXL": "WAXL",
+		"AXL":   "WAXL",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/bybit.go b/oracle/provider/bybit.go
@@ -60,7 +60,7 @@ func NewBybitProvider(
 	)
 
 	availablePairs, _ := provider.GetAvailablePairs()
-	provider.setPairs(pairs, availablePairs, nil)
+	provider.setPairs(pairs, availablePairs, currencyPairToBybitSymbol)
 
 	go startPolling(provider, provider.endpoints.PollInterval, logger)
 	return provider, nil
@@ -121,3 +121,21 @@ func (p *BybitProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 	return symbols, nil
 }
+
+func currencyPairToBybitSymbol(pair types.CurrencyPair) string {
+	mapping := map[string]string{
+		"MATIC": "POL",
+	}
+
+	base, found := mapping[pair.Base]
+	if !found {
+		base = pair.Base
+	}
+
+	quote, found := mapping[pair.Quote]
+	if !found {
+		quote = pair.Quote
+	}
+
+	return base + quote
+}
diff --git a/oracle/provider/gate.go b/oracle/provider/gate.go
@@ -114,7 +114,8 @@ func (p *GateProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToGateSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"AXL": "WAXL",
+		"AXL":   "WAXL",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/huobi.go b/oracle/provider/huobi.go
@@ -120,7 +120,8 @@ func (p *HuobiProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToHuobiSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"AXL": "WAXL",
+		"AXL":   "WAXL",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/kraken.go b/oracle/provider/kraken.go
@@ -176,10 +176,11 @@ func currencyPairToKrakenSymbol(pair types.CurrencyPair) string {
 		"USDJPY":  "ZUSDZJPY",
 	}
 	mapping := map[string]string{
-		"AXL":  "WAXL",
-		"BTC":  "XBT",
-		"LUNC": "LUNA",
-		"LUNA": "LUNA2",
+		"AXL":   "WAXL",
+		"BTC":   "XBT",
+		"LUNC":  "LUNA",
+		"LUNA":  "LUNA2",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/kucoin.go b/oracle/provider/kucoin.go
@@ -124,7 +124,8 @@ func (p *KucoinProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToKucoinSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"AXL": "WAXL",
+		"AXL":   "WAXL",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/lbank.go b/oracle/provider/lbank.go
@@ -123,5 +123,19 @@ func (p *LbankProvider) GetAvailablePairs() (map[string]struct{}, error) {
 }
 
 func currencyPairToLbankSymbol(pair types.CurrencyPair) string {
-	return strings.ToLower(pair.Join("_"))
+	mapping := map[string]string{
+		"MATIC": "POL",
+	}
+
+	base, found := mapping[pair.Base]
+	if !found {
+		base = pair.Base
+	}
+
+	quote, found := mapping[pair.Quote]
+	if !found {
+		quote = pair.Quote
+	}
+
+	return strings.ToLower(base + "_" + quote)
 }
diff --git a/oracle/provider/mexc.go b/oracle/provider/mexc.go
@@ -114,7 +114,8 @@ func (p *MexcProvider) GetAvailablePairs() (map[string]struct{}, error) {
 
 func currencyPairToMexcSymbol(pair types.CurrencyPair) string {
 	mapping := map[string]string{
-		"AXL": "WAXL",
+		"AXL":   "WAXL",
+		"MATIC": "POL",
 	}
 
 	base, found := mapping[pair.Base]

diff --git a/oracle/provider/okx.go b/oracle/provider/okx.go
@@ -128,5 +128,19 @@ func (p *OkxProvider) GetAvailablePairs() (map[string]struct{}, error) {
 }
 
 func currencyPairToOkxSymbol(pair types.CurrencyPair) string {
-	return pair.Join("-")
+	mapping := map[string]string{
+		"MATIC": "POL",
+	}
+
+	base, found := mapping[pair.Base]
+	if !found {
+		base = pair.Base
+	}
+
+	quote, found := mapping[pair.Quote]
+	if !found {
+		quote = pair.Quote
+	}
+
+	return base + "-" + quote
 }
diff --git a/oracle/provider/phemex.go b/oracle/provider/phemex.go
@@ -214,5 +214,19 @@ func (p *PhemexProvider) GetAvailablePairs() (map[string]struct{}, error) {
 }
 
 func currencyPairToPhemexSymbol(pair types.CurrencyPair) string {
-	return "s" + pair.String()
+	mapping := map[string]string{
+		"MATIC": "POL",
+	}
+
+	base, found := mapping[pair.Base]
+	if !found {
+		base = pair.Base
+	}
+
+	quote, found := mapping[pair.Quote]
+	if !found {
+		quote = pair.Quote
+	}
+
+	return "s" + base + quote
 }