feat: support anvil fork

Gearbox-protocol · Oct 8, 2023 · f40cbfb · f40cbfb
1 parent 5b88927
commit f40cbfb
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/core/schemas/contract.go b/core/schemas/contract.go
@@ -96,7 +96,6 @@ func (c *Contract) DiscoverFirstLog() int64 {
 }
 
 func (c *Contract) findFirstLogBound(fromBlock, toBlock int64) (int64, error) {
-
 	query := ethereum.FilterQuery{
 		FromBlock: big.NewInt(fromBlock),
 		ToBlock:   big.NewInt(toBlock),

diff --git a/core/symbol.go b/core/symbol.go
@@ -38,15 +38,15 @@ func GetAddrToSymbol(fileName string) map[common.Address]Symbol {
 }
 
 func GetSymToAddrByChainId(chainId int64) *SymTOAddrStore {
-	if chainId == 1337 {
+	if chainId == 1337 || chainId == 7878 {
 		chainId = 1
 	}
 	fileName := strings.ToLower(log.GetNetworkName(chainId)) + ".jsonnet"
 	return GetSymToAddrStore(fileName)
 }
 
 func GetAddrToSymbolByChainId(chainId int64) map[common.Address]Symbol {
-	if chainId == 1337 {
+	if chainId == 1337 || chainId == 7878 {
 		chainId = 1
 	}
 	fileName := strings.ToLower(log.GetNetworkName(chainId)) + ".jsonnet"

diff --git a/core/version.go b/core/version.go
@@ -36,10 +36,12 @@ type VersionType struct {
 }
 
 func NewVersion(v int16) VersionType {
-	if v == 1 || v == 2 || v == 3 {
+	if v == 1 || v == 2 || v == 300 {
 		return VersionType{v: v}
 	} else if v == 210 || v == 220 {
 		return VersionType{v: 2}
+	} else if v == 10_000 { // for testing
+		return VersionType{v: 10000}
 	}
 	log.Fatal("version not supported", v)
 	panic("")
@@ -49,7 +51,7 @@ func (v VersionType) Decimals() int8 {
 	switch v.v {
 	case 1:
 		return 18 // eth decimals
-	case 2:
+	case 2, 300:
 		return 8 // USD decimals
 	default:
 		log.Fatal("version not supported")

diff --git a/utils/string.go b/utils/string.go
@@ -16,7 +16,7 @@ import (
 	"github.com/ethereum/go-ethereum/common"
 )
 
-func Contains(s []string, e string) bool {
+func Contains[T comparable](s []T, e T) bool {
 	for _, a := range s {
 		if a == e {
 			return true