ci: add new modules types (#528)

elys-network · May 25, 2024 · 6160846 · 6160846
1 parent 4bd62d6
commit 6160846
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/cmd/upgrade-assure/types.go b/cmd/upgrade-assure/types.go
@@ -36,7 +36,10 @@ import (
 	stablestaketypes "github.com/elys-network/elys/x/stablestake/types"
 	tokenomicstypes "github.com/elys-network/elys/x/tokenomics/types"
 	transferhooktypes "github.com/elys-network/elys/x/transferhook/types"
+
 	// genutiltypes "github.com/cosmos/cosmos-sdk/x/genutil/types"
+	estakingtypes "github.com/elys-network/elys/x/estaking/types"
+	mastercheftypes "github.com/elys-network/elys/x/masterchef/types"
 )
 
 type Genesis struct {
@@ -88,13 +91,15 @@ type AppState struct {
 	Crisis        crisistypes.GenesisState        `json:"crisis"`
 	Distribution  Distribution                    `json:"distribution"`
 	Epochs        Epochs                          `json:"epochs"`
+	Estaking      Estaking                        `json:"estaking"`
 	Evidence      EvidenceState                   `json:"evidence"`
 	Genutil       Genutil                         `json:"genutil"`
 	Gov           Gov                             `json:"gov"`
 	Ibc           Ibc                             `json:"ibc"`
 	Incentive     Incentive                       `json:"incentive"`
 	LeverageLP    LeverageLP                      `json:"leveragelp"`
 	Perpetual     Perpetual                       `json:"perpetual"`
+	Masterchef    Masterchef                      `json:"masterchef"`
 	Mint          Mint                            `json:"mint"`
 	Oracle        Oracle                          `json:"oracle"`
 	Parameter     Parameter                       `json:"parameter"`
@@ -111,6 +116,38 @@ type AppState struct {
 	// Include other fields as needed
 }
 
+type Masterchef struct {
+	mastercheftypes.GenesisState
+
+	ExternalIncentiveIndex json.Number                `json:"external_incentive_index"`
+	PoolInfos              []MasterchefPoolInfo       `json:"pool_infos"`
+	PoolRewardInfos        []MasterchefPoolRewardInfo `json:"pool_reward_infos"`
+	UserRewardInfos        []MasterchefUserRewardInfo `json:"user_reward_infos"`
+}
+
+type MasterchefUserRewardInfo struct {
+	mastercheftypes.UserRewardInfo
+
+	PoolId json.Number `json:"pool_id"`
+}
+
+type MasterchefPoolRewardInfo struct {
+	mastercheftypes.PoolRewardInfo
+
+	PoolId           json.Number `json:"pool_id"`
+	LastUpdatedBlock json.Number `json:"last_updated_block"`
+}
+
+type MasterchefPoolInfo struct {
+	mastercheftypes.PoolInfo
+
+	PoolId json.Number `json:"pool_id"`
+}
+
+type Estaking struct {
+	estakingtypes.GenesisState
+}
+
 type Tokenomics struct {
 	tokenomicstypes.GenesisState