Update tokenlist path getting (#69)

trustwallet · Jan 27, 2022 · 935a618 · 935a618
1 parent c8863a7
commit 935a618
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/path/assets.go b/path/assets.go
@@ -7,12 +7,22 @@ import (
 	"github.com/trustwallet/go-primitives/types"
 )
 
-func GetTokenListPath(chain string) string {
-	return fmt.Sprintf("blockchains/%s/tokenlist.json", chain)
-}
+type TokenListType int
+
+const (
+	TokenlistDefault = iota
+	TokenlistExtended
+)
+
+func GetTokenListPath(chain string, tokenlistType TokenListType) string {
+	switch tokenlistType {
+	case TokenlistDefault:
+		return fmt.Sprintf("blockchains/%s/tokenlist.json", chain)
+	case TokenlistExtended:
+		return fmt.Sprintf("blockchains/%s/tokenlist-extended.json", chain)
+	}
 
-func GetTokenListExtendedPath(chain string) string {
-	return fmt.Sprintf("blockchains/%s/tokenlist-extended.json", chain)
+	return ""
 }
 
 func GetAssetPath(chain, tokenID string) string {