Skip to content

Commit

Permalink
feat(x/genutil): Allow creation of AppGenesis without a file lookup (#…
Browse files Browse the repository at this point in the history
…17571)

Co-authored-by: Julien Robert <[email protected]>
  • Loading branch information
chatton and julienrbrt authored Aug 29, 2023
1 parent aabcfb2 commit 0a253f3
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 5 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ Ref: https://keepachangelog.com/en/1.0.0/

* (client) [#17513](https://github.com/cosmos/cosmos-sdk/pull/17513) Allow overwritting `client.toml`. Use `client.CreateClientConfig` in place of `client.ReadFromClientConfig` and provide a custom template and a custom config.
* (x/bank) [#14224](https://github.com/cosmos/cosmos-sdk/pull/14224) Allow injection of restrictions on transfers using `AppendSendRestriction` or `PrependSendRestriction`.
* (genutil) [#17571](https://github.com/cosmos/cosmos-sdk/pull/17571) Allow creation of `AppGenesis` without a file lookup.

### Improvements

Expand Down
32 changes: 27 additions & 5 deletions x/genutil/types/genesis.go
Original file line number Diff line number Diff line change
@@ -1,11 +1,14 @@
package types

import (
"bufio"
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"os"
"path/filepath"
"time"

cmtjson "github.com/cometbft/cometbft/libs/json"
Expand Down Expand Up @@ -85,19 +88,19 @@ func (ag *AppGenesis) SaveAs(file string) error {
return os.WriteFile(file, appGenesisBytes, 0o600)
}

// AppGenesisFromFile reads the AppGenesis from the provided file.
func AppGenesisFromFile(genFile string) (*AppGenesis, error) {
jsonBlob, err := os.ReadFile(genFile)
// AppGenesisFromReader reads the AppGenesis from the reader.
func AppGenesisFromReader(reader io.Reader) (*AppGenesis, error) {
jsonBlob, err := io.ReadAll(reader)
if err != nil {
return nil, fmt.Errorf("couldn't read AppGenesis file (%s): %w", genFile, err)
return nil, err
}

var appGenesis AppGenesis
if err := json.Unmarshal(jsonBlob, &appGenesis); err != nil {
// fallback to CometBFT genesis
var ctmGenesis cmttypes.GenesisDoc
if err2 := cmtjson.Unmarshal(jsonBlob, &ctmGenesis); err2 != nil {
return nil, fmt.Errorf("error unmarshalling AppGenesis at %s: %w\n failed fallback to CometBFT GenDoc: %w", genFile, err, err2)
return nil, fmt.Errorf("error unmarshalling AppGenesis: %w\n failed fallback to CometBFT GenDoc: %w", err, err2)
}

appGenesis = AppGenesis{
Expand All @@ -118,6 +121,25 @@ func AppGenesisFromFile(genFile string) (*AppGenesis, error) {
return &appGenesis, nil
}

// AppGenesisFromFile reads the AppGenesis from the provided file.
func AppGenesisFromFile(genFile string) (*AppGenesis, error) {
file, err := os.Open(filepath.Clean(genFile))
if err != nil {
return nil, err
}

appGenesis, err := AppGenesisFromReader(bufio.NewReader(file))
if err != nil {
return nil, fmt.Errorf("failed to read genesis from file %s: %w", genFile, err)
}

if err := file.Close(); err != nil {
return nil, err
}

return appGenesis, nil
}

// --------------------------
// CometBFT Genesis Handling
// --------------------------
Expand Down

0 comments on commit 0a253f3

Please sign in to comment.