122 lines
3.4 KiB
Go
122 lines
3.4 KiB
Go
package e2e
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
comettypes "github.com/cometbft/cometbft/types"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
"github.com/cosmos/cosmos-sdk/x/genutil"
|
|
genutiltypes "github.com/cosmos/cosmos-sdk/x/genutil/types"
|
|
"github.com/skip-mev/pob/x/builder/types"
|
|
)
|
|
|
|
func getGenDoc(path string) (*comettypes.GenesisDoc, error) {
|
|
serverCtx := server.NewDefaultContext()
|
|
config := serverCtx.Config
|
|
config.SetRoot(path)
|
|
|
|
genFile := config.GenesisFile()
|
|
doc := &comettypes.GenesisDoc{}
|
|
|
|
if _, err := os.Stat(genFile); err != nil {
|
|
if !os.IsNotExist(err) {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
var err error
|
|
|
|
doc, err = comettypes.GenesisDocFromFile(genFile)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to read genesis doc from file: %w", err)
|
|
}
|
|
}
|
|
|
|
return doc, nil
|
|
}
|
|
|
|
func initGenesisFile(path, moniker, amountStr string, accAddr sdk.AccAddress, params types.Params) error {
|
|
serverCtx := server.NewDefaultContext()
|
|
config := serverCtx.Config
|
|
|
|
config.SetRoot(path)
|
|
config.Moniker = moniker
|
|
|
|
coins, err := sdk.ParseCoinsNormalized(amountStr)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to parse coins: %w", err)
|
|
}
|
|
|
|
balances := banktypes.Balance{Address: accAddr.String(), Coins: coins.Sort()}
|
|
genAccount := authtypes.NewBaseAccount(accAddr, nil, 0, 0)
|
|
|
|
genFile := config.GenesisFile()
|
|
appState, genDoc, err := genutiltypes.GenesisStateFromGenFile(genFile)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to unmarshal genesis state: %w", err)
|
|
}
|
|
|
|
authGenState := authtypes.GetGenesisStateFromAppState(cdc, appState)
|
|
|
|
accs, err := authtypes.UnpackAccounts(authGenState.Accounts)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get accounts from any: %w", err)
|
|
}
|
|
|
|
if accs.Contains(accAddr) {
|
|
return fmt.Errorf("failed to add account to genesis state; account already exists: %s", accAddr)
|
|
}
|
|
|
|
// Add the new account to the set of genesis accounts and sanitize the
|
|
// accounts afterwards.
|
|
accs = append(accs, genAccount)
|
|
accs = authtypes.SanitizeGenesisAccounts(accs)
|
|
|
|
genAccs, err := authtypes.PackAccounts(accs)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to convert accounts into any's: %w", err)
|
|
}
|
|
|
|
authGenState.Accounts = genAccs
|
|
|
|
authGenStateBz, err := cdc.MarshalJSON(&authGenState)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal auth genesis state: %w", err)
|
|
}
|
|
|
|
appState[authtypes.ModuleName] = authGenStateBz
|
|
|
|
bankGenState := banktypes.GetGenesisStateFromAppState(cdc, appState)
|
|
bankGenState.Balances = append(bankGenState.Balances, balances)
|
|
bankGenState.Balances = banktypes.SanitizeGenesisBalances(bankGenState.Balances)
|
|
|
|
bankGenStateBz, err := cdc.MarshalJSON(bankGenState)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal bank genesis state: %w", err)
|
|
}
|
|
|
|
appState[banktypes.ModuleName] = bankGenStateBz
|
|
|
|
builderGenState := types.GetGenesisStateFromAppState(cdc, appState)
|
|
builderGenState.Params = params
|
|
|
|
builderGenStateBz, err := cdc.MarshalJSON(&builderGenState)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal builder genesis state: %w", err)
|
|
}
|
|
|
|
appState[types.ModuleName] = builderGenStateBz
|
|
|
|
appStateJSON, err := json.Marshal(appState)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal application genesis state: %w", err)
|
|
}
|
|
|
|
genDoc.AppState = appStateJSON
|
|
return genutil.ExportGenesisFile(genDoc, genFile)
|
|
}
|