fix: cm

Author: harsh-98

ds/dc_wrapper/wrapper.go

@@ -176,7 +176,13 @@ func isCType(c CompressorType) bool {
 	}
 	return false
 }
-
+func compressors() []CompressorType {
+	compressors := []CompressorType{}
+	for _, v := range ContractNameToCompressortype {
+		compressors = append(compressors, v)
+	}
+	return compressors
+}
 func (dcw *DataCompressorWrapper) AddCompressorType(addr common.Address, cType CompressorType, discoveredAt int64) {
 	if !isCType(cType) {
 		log.Fatal("ctype is wrong, ", cType)
@@ -218,7 +224,8 @@ func (dcw *DataCompressorWrapper) LoadMultipleDC(multiDCs interface{}) {
 		sort.Slice(blockNums, func(i, j int) bool { return blockNums[i] < blockNums[j] })
 	}
 	for _, blockNum := range blockNums {
-		for _, suffix := range []CompressorType{CompressorType(""), CompressorType("300"), MARKET_COMPRESSOR, POOL_COMPRESSOR, CREDIT_ACCOUNT_COMPRESSOR} {
+		// TODO: NETWORK
+		for _, suffix := range append(compressors(), "", "300") {
 			key := fmt.Sprintf("%d", blockNum)
 			if suffix != "" {
 				key = fmt.Sprintf("%d_%s", blockNum, suffix)
@@ -240,6 +247,7 @@ func (dcw *DataCompressorWrapper) LoadMultipleDC(multiDCs interface{}) {
 type CompressorType string
 
 const (
+	// TODO : NETWORK
 	MARKET_COMPRESSOR         CompressorType = "MARKET"
 	POOL_COMPRESSOR           CompressorType = "POOL"
 	CREDIT_ACCOUNT_COMPRESSOR CompressorType = "ACCOUNT"
@@ -416,7 +424,14 @@ func (dcw *DataCompressorWrapper) GetCreditAccountData(version core.VersionType,
 			if err != nil {
 				return dc.CreditAccountCallData{}, err
 			}
-			accountData := *abi.ConvertType(out[0], new(creditAccountCompressor.CreditAccountData)).(*creditAccountCompressor.CreditAccountData)
+			var accountData creditAccountCompressor.CreditAccountData
+			switch compressorType {
+			case CREDIT_ACCOUNT_COMPRESSOR:
+				accountData = *abi.ConvertType(out[0], new(creditAccountCompressor.CreditAccountData)).(*creditAccountCompressor.CreditAccountData)
+			case GLOBAL_ACCOUNT_COMPRESSOR:
+				x := abi.ConvertType(out[0], new(globalAccountCompressor.CreditAccountData)).(*globalAccountCompressor.CreditAccountData)
+				accountData = dc.Convert(x)
+			}
 			return AddFieldsToAccountv310(dcw.client, blockNum, accountData)
 		case DCV3:
 			out, err := core.GetAbi("DataCompressorv3").Unpack("getCreditAccountData", bytes)
@@ -643,6 +658,7 @@ func (dcw *DataCompressorWrapper) GetPoolData(version core.VersionType, blockNum
 	resultFn func([]byte) (dc.PoolCallData, error),
 	errReturn error) {
 	//
+	// TODO : NETWORK
 	key, dcAddr, compressorType := dcw.GetKeyAndAddress(version, blockNum, []CompressorType{POOL_COMPRESSOR, GLOBAL_MARKET_COMPRESSOR})
 	switch key {
 	case NODC:

ds/debt.go

@@ -6,6 +6,7 @@ import (
 	dcv2 "github.com/Gearbox-protocol/sdk-go/artifacts/dataCompressor/dataCompressorv2"
 	"github.com/Gearbox-protocol/sdk-go/core"
 	"github.com/Gearbox-protocol/sdk-go/core/schemas"
+	"github.com/Gearbox-protocol/sdk-go/log"
 	"github.com/Gearbox-protocol/sdk-go/pkg/dc"
 	"github.com/Gearbox-protocol/sdk-go/utils"
 )
@@ -51,5 +52,11 @@ func (profile *DebtProfile) String() string {
 }
 
 func CallTraceAllowed(client core.ClientI) bool {
-	return utils.Contains([]int64{1, 42161, 10, 7878, 1337}, core.GetChainId(client))
+	network := log.TestNetOrMainnet()
+	if network == "" { // testnet
+		network = log.GetNetworkName(core.GetChainId(client))
+	}
+	chainid := log.GetNetworkToChainId(network)
+	testnet := log.GetTestNet(chainid)
+	return utils.Contains([]int64{7878, 1337}, core.GetChainId(client)) || (testnet != network)
 }

ds/repo.go

@@ -78,6 +78,7 @@ type RepositoryI interface {
 	AddAllowedTokenV2(logID uint, txHash, creditFilter string, atoken *schemas.AllowedToken)
 	UpdateLimits(logID uint, txHash, creditConfigurator string, params *schemas.Parameters)
 	UpdateEmergencyLiqDiscount(logID uint, txHash, creditConfigurator string, params *schemas.Parameters)
+	// saves in db and updates dao operations
 	UpdateFees(logID uint, txHash, creditConfigurator string, params *schemas.Parameters)
 	TransferAccountAllowed(*schemas.TransferAccountAllowed)
 	GetPricesInUSD(blockNum int64, pool string, tokenAddrs []string) core.JsonFloatMap

go.mod

@@ -5,7 +5,7 @@ go 1.23
 // toolchain go1.23.2
 
 require (
-	github.com/Gearbox-protocol/sdk-go v0.0.0-20250510115227-8bf87ea490e7
+	github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad
 	github.com/ethereum/go-ethereum v1.13.14
 	github.com/go-playground/validator/v10 v10.4.1
 	github.com/google/go-cmp v0.6.0

go.sum

@@ -6,8 +6,8 @@ github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/DataDog/zstd v1.4.5 h1:EndNeuB0l9syBZhut0wns3gV1hL8zX8LIu6ZiVHWLIQ=
 github.com/DataDog/zstd v1.4.5/go.mod h1:1jcaCB/ufaK+sKp1NBhlGmpz41jOoPQ35bpF36t7BBo=
-github.com/Gearbox-protocol/sdk-go v0.0.0-20250510115227-8bf87ea490e7 h1:A/M+HDbHhyEQmr6/90/A94Dj/RfbovEwi9DY5xeEsAY=
-github.com/Gearbox-protocol/sdk-go v0.0.0-20250510115227-8bf87ea490e7/go.mod h1:MZsiiSAwDdnHY6GEILYtHEqhb9wmbUUE34UBQgNlUyM=
+github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad h1:wRcGkN6RVzmxWD8lew22b4mjsddrWCCpmGGsy0xDZKQ=
+github.com/Gearbox-protocol/sdk-go v0.0.0-20250515050146-74e6c8e5dfad/go.mod h1:MZsiiSAwDdnHY6GEILYtHEqhb9wmbUUE34UBQgNlUyM=
 github.com/Microsoft/go-winio v0.6.1 h1:9/kr64B9VUZrLm5YYwbGtUJnMgqWVOdUAXu6Migciow=
 github.com/Microsoft/go-winio v0.6.1/go.mod h1:LRdKpFKfdobln8UmuiYcKPot9D2v6svN5+sAH+4kjUM=
 github.com/OffchainLabs/go-ethereum v1.13.4-0.20240313010929-e5d8587e7227 h1:+/3TrD+q+BP36jGj2Bycdmrc/joKLNbc5ImePQzKRLM=

migrations/000056_v310.up.sql

@@ -55,4 +55,6 @@ alter table price_feeds add PRIMARY KEY (block_num, feed);
 
 -- do after the legacy market configurator address is found.
 -- update legacy address in marketconfigurator.
--- update tvl_snapshots market=? where market='0x0000000000000000000000000000000000000000';
+-- update tvl_snapshots market=? where market='0x0000000000000000000000000000000000000000';
+
+insert into debt_sync values ('t', 0 ,0 );

models/credit_manager/cm_common/credit_session_state.go

@@ -314,11 +314,11 @@ func (mdl *CommonCMAdapter) updateSession(blockNum int64, session *schemas.Credi
 }
 
 func (mdl *CommonCMAdapter) addFloatValue(account string, blockNum int64, dcv2Balances []core.TokenBalanceCallData) *core.DBBalanceFormat {
+	var stETH string
 	if log.GetBaseNet(core.GetBaseChainId(mdl.Client)) == log.MAINNET {
-		return AddStETHBalance(account, blockNum, dcv2Balances, mdl.Client, mdl, core.GetToken(1, "stETH").Hex())
+		stETH = core.GetToken(1, "stETH").Hex()
 	}
-	log.Fatal("AddStETHBalance is not supported other networks than mainnet")
-	return nil
+	return AddStETHBalance(account, blockNum, dcv2Balances, mdl.Client, mdl, stETH)
 }
 
 func (mdl *CommonCMAdapter) GetDecimals(token string) int8 {

models/credit_manager/cm_v3/model.go

@@ -4,6 +4,7 @@ import (
 	"math/big"
 
 	"github.com/Gearbox-protocol/sdk-go/core"
+	"github.com/Gearbox-protocol/sdk-go/core/schemas"
 	"github.com/Gearbox-protocol/sdk-go/log"
 	"github.com/Gearbox-protocol/third-eye/ds"
 	"github.com/ethereum/go-ethereum/common"
@@ -23,7 +24,19 @@ func NewCMv3(addr string, client core.ClientI, repo ds.RepositoryI, discoveredAt
 		ds.NewSyncAdapter(addr, ds.CreditManager, discoveredAt, client, repo),
 	)
 	mdl.InitState()
+	params := &schemas.Parameters{
+		CreditManager:              addr,
+		BlockNum:                   discoveredAt,
+		FeeInterest:                5000,          // DEFAULT_FEE_INTEREST
+		FeeLiquidation:             150,           // DEFAULT_FEE_LIQUIDATION
+		LiquidationDiscount:        100_00 - 4_00, // DEFAULT_LIQUIDATION_PREMIUM
+		FeeLiquidationExpired:      1_00,          // DEFAULT_FEE_LIQUIDATION_EXPIRED
+		LiquidationDiscountExpired: 100_00 - 2_00, // DEFAULT_LIQUIDATION_PREMIUM_EXPIRED
+		EmergencyLiqDiscount:       0,
+	}
+	mdl.SetParams(params)
 	mdl.addCreditConfiguratorAdapter(mdl.GetDetailsByKey("configurator"))
+	mdl.Repo.UpdateFees(0, "", mdl.GetDetailsByKey("configurator"), params)
 	return mdl
 }
 

models/credit_manager/cm_v3/on_change.go

@@ -39,15 +39,22 @@ func (mdl *CMv3) OnLog(txLog types.Log) {
 		case core.Topic("SetExpirationDate(uint40)"):
 			mdl.expirationDate = uint64(new(big.Int).SetBytes(txLog.Data[:]).Int64())
 		case core.Topic("UpdateFees(uint16,uint16,uint16,uint16,uint16)"):
-			mdl.SetParams(&schemas.Parameters{
+			// feeInterest,
+			// uint16 feeLiquidation,
+			// uint16 liquidationPremium,
+			// uint16 feeLiquidationExpired,
+			// uint16 liquidationPremiumExpired
+			params := &schemas.Parameters{
 				BlockNum:                   int64(txLog.BlockNumber),
 				CreditManager:              mdl.Address,
 				FeeInterest:                bytesToUInt16(txLog.Data[:32]),
 				FeeLiquidation:             bytesToUInt16(txLog.Data[32:64]),
 				LiquidationDiscount:        10000 - bytesToUInt16(txLog.Data[64:96]), // 10000- liqPremium
 				FeeLiquidationExpired:      bytesToUInt16(txLog.Data[96:128]),
 				LiquidationDiscountExpired: 10000 - bytesToUInt16(txLog.Data[128:160]), // 10000- liqPremiumExpired
-			})
+			}
+			mdl.SetParams(params)
+			mdl.Repo.UpdateFees(txLog.Index, txLog.TxHash.Hex(), mdl.GetAddress(), params)
 		}
 		return
 	}

repository/save.go

@@ -8,7 +8,6 @@ import (
 func (repo *Repository) Flush(syncTill int64) error {
 	repo.mu.Lock()
 	defer repo.mu.Unlock()
-	// log.Fatal("")
 	// time.Sleep(time.Hour)
 	// preferred order (adapter | token) => pools => cm => credit session => blocks => allowedTokens