fix: restore canonical token lookup for market caches

internal/data/market/futures_pairs.go

@@ -25,17 +25,16 @@ func (ms *MarketData) refreshFuturePairCache() error {
 			continue
 		}
 		futurePairs[s.Symbol] = true
-		token := parseTokenFromSymbolByQuotePriority(s.Symbol)
+		token := tokenKeyFromSymbol(s.Symbol)
 		if token == "" {
 			continue
 		}
-		token = futureCacheTokenKey(token)
 		futureTokenCandidates[token] = append(futureTokenCandidates[token], s.Symbol)
 	}
 
 	futureToken2Symbol := make(map[string]string, len(futureTokenCandidates))
 	for token, candidates := range futureTokenCandidates {
-		futureToken2Symbol[token] = selectCanonicalSymbolByQuotePriority(token, candidates)
+		futureToken2Symbol[token] = selectCanonicalFutureSymbol(token, candidates)
 	}
 
 	ms.pairCacheMutex.Lock()
@@ -47,10 +46,6 @@ func (ms *MarketData) refreshFuturePairCache() error {
 	return nil
 }
 
-func futureCacheTokenKey(token string) string {
-	return strings.ToUpper(token)
-}
-
 func (ms *MarketData) IsFuturesPair(symbol string) bool {
 	ms.pairCacheMutex.RLock()
 	defer ms.pairCacheMutex.RUnlock()

internal/data/market/pairs_test.go

@@ -26,16 +26,39 @@ func TestSelectCanonicalSymbolByQuotePriority_NoPreferredQuote(t *testing.T) {
 	}
 }
 
-func TestFutureCacheTokenKey_PreservesRawFutureToken(t *testing.T) {
-	got := futureCacheTokenKey("LUNA2")
+func TestSelectCanonicalFutureSymbol_PrefersPrefixThenQuote(t *testing.T) {
+	pairs := []string{"PEPEUSDC", "PEPEUSDT", "1000PEPEUSDC", "1000PEPEUSDT"}
+	got := selectCanonicalFutureSymbol("PEPE", pairs)
+	if got != "1000PEPEUSDT" {
+		t.Fatalf("expected 1000PEPEUSDT, got %q", got)
+	}
+}
+
+func TestSelectCanonicalFutureSymbol_UsesQuoteAbbreviation(t *testing.T) {
+	pairs := []string{"PEPEUSDC"}
+	got := selectCanonicalFutureSymbol("PEPEC", pairs)
+	if got != "PEPEUSDC" {
+		t.Fatalf("expected PEPEUSDC, got %q", got)
+	}
+}
+
+func TestTokenKeyFromSymbol_PreservesUnmappedFutureToken(t *testing.T) {
+	got := tokenKeyFromSymbol("LUNA2USDT")
 	if got != "LUNA2" {
 		t.Fatalf("expected LUNA2, got %q", got)
 	}
 }
 
-func TestFutureCacheTokenKey_NoOverride(t *testing.T) {
-	got := futureCacheTokenKey("PEPE")
+func TestTokenKeyFromSymbol_StripsKnownPrefix(t *testing.T) {
+	got := tokenKeyFromSymbol("1000PEPEUSDT")
 	if got != "PEPE" {
 		t.Fatalf("expected PEPE, got %q", got)
 	}
 }
+
+func TestTokenKeyFromSymbol_UsesQuoteAbbreviation(t *testing.T) {
+	got := tokenKeyFromSymbol("PEPEUSDC")
+	if got != "PEPEC" {
+		t.Fatalf("expected PEPEC, got %q", got)
+	}
+}

internal/data/market/spot_pairs.go

@@ -27,7 +27,7 @@ func (ms *MarketData) refreshSpotPairCache() error {
 			continue
 		}
 		spotPairs[s.Symbol] = true
-		token := parseTokenFromSymbolByQuotePriority(s.Symbol)
+		token := tokenKeyFromSymbol(s.Symbol)
 		if token == "" {
 			continue
 		}
@@ -36,7 +36,7 @@ func (ms *MarketData) refreshSpotPairCache() error {
 
 	spotToken2Symbol := make(map[string]string, len(spotTokenCandidates))
 	for token, candidates := range spotTokenCandidates {
-		spotToken2Symbol[token] = selectCanonicalSymbolByQuotePriority(token, candidates)
+		spotToken2Symbol[token] = selectCanonicalSpotSymbol(token, candidates)
 	}
 
 	ms.pairCacheMutex.Lock()

internal/data/market/token_lookup.go

@@ -0,0 +1,78 @@
+package market
+
+import (
+	"sort"
+	"strings"
+
+	"me.thuanle/bbot/internal/configs/binance"
+	"me.thuanle/bbot/internal/utils/stringx"
+)
+
+var futureSymbolPrefixPriority = append(append([]string{}, binance.SymbolPrefixList...), "")
+
+func tokenKeyFromSymbol(symbol string) string {
+	token := strings.ToUpper(symbol)
+
+	for _, prefix := range binance.SymbolPrefixList {
+		token, _ = strings.CutPrefix(token, strings.ToUpper(prefix))
+	}
+
+	for _, suffix := range binance.SymbolSuffixList {
+		replaced, ok := stringx.ReplaceSuffix(token, strings.ToUpper(suffix), strings.ToUpper(binance.SymbolSuffixMap[suffix]))
+		if ok {
+			return replaced
+		}
+	}
+
+	return ""
+}
+
+func selectCanonicalSpotSymbol(token string, candidates []string) string {
+	return selectCanonicalSymbolByTokenOrder(token, candidates, []string{""})
+}
+
+func selectCanonicalFutureSymbol(token string, candidates []string) string {
+	return selectCanonicalSymbolByTokenOrder(token, candidates, futureSymbolPrefixPriority)
+}
+
+func selectCanonicalSymbolByTokenOrder(token string, candidates []string, prefixes []string) string {
+	if len(candidates) == 0 {
+		return ""
+	}
+	if len(candidates) == 1 {
+		return strings.ToUpper(candidates[0])
+	}
+
+	token = strings.ToUpper(token)
+	normalized := make([]string, 0, len(candidates))
+	candidateSet := make(map[string]struct{}, len(candidates))
+	for _, candidate := range candidates {
+		candidate = strings.ToUpper(candidate)
+		normalized = append(normalized, candidate)
+		candidateSet[candidate] = struct{}{}
+	}
+
+	for _, prefix := range prefixes {
+		base := strings.ToUpper(prefix) + token
+		if _, ok := candidateSet[base]; ok {
+			return base
+		}
+
+		for _, suffix := range binance.SymbolSuffixList {
+			target := base + strings.ToUpper(suffix)
+			if _, ok := candidateSet[target]; ok {
+				return target
+			}
+
+			replaced, ok := stringx.ReplaceSuffix(base, strings.ToUpper(binance.SymbolSuffixMap[suffix]), strings.ToUpper(suffix))
+			if ok {
+				if _, exists := candidateSet[replaced]; exists {
+					return replaced
+				}
+			}
+		}
+	}
+
+	sort.Strings(normalized)
+	return normalized[0]
+}