cosmos-sdk/app/genesis.go

100 lines
2.3 KiB
Go
Raw Normal View History

2017-01-28 09:29:32 -08:00
package app
import (
"encoding/json"
"github.com/pkg/errors"
2017-05-13 17:04:53 -07:00
2017-04-25 22:08:31 -07:00
cmn "github.com/tendermint/tmlibs/common"
2017-01-28 09:29:32 -08:00
)
2017-07-06 02:30:03 -07:00
// LoadGenesis - Load the genesis file into memory
2017-01-28 09:29:32 -08:00
func (app *Basecoin) LoadGenesis(path string) error {
2017-03-14 10:55:46 -07:00
genDoc, err := loadGenesis(path)
2017-01-28 09:29:32 -08:00
if err != nil {
return err
}
2017-03-14 00:11:49 -07:00
2017-03-14 10:55:46 -07:00
// set chain_id
2017-07-30 09:57:48 -07:00
app.InitState("base/chain_id", genDoc.ChainID)
2017-03-14 10:55:46 -07:00
// set accounts
2017-07-03 09:58:28 -07:00
for _, acct := range genDoc.AppOptions.Accounts {
2017-07-30 09:57:48 -07:00
_ = app.InitState("coin/account", string(acct))
2017-03-14 10:55:46 -07:00
}
// set plugin options
for _, kv := range genDoc.AppOptions.pluginOptions {
2017-07-30 09:57:48 -07:00
_ = app.InitState(kv.Key, kv.Value)
2017-01-28 09:29:32 -08:00
}
2017-01-28 09:29:32 -08:00
return nil
}
type keyValue struct {
Key string `json:"key"`
Value string `json:"value"`
}
2017-07-04 20:28:27 -07:00
// FullGenesisDoc - includes tendermint (in the json, we ignore here)
2017-03-14 00:11:49 -07:00
type FullGenesisDoc struct {
2017-03-14 10:55:46 -07:00
ChainID string `json:"chain_id"`
2017-03-14 00:11:49 -07:00
AppOptions *GenesisDoc `json:"app_options"`
}
2017-07-04 20:28:27 -07:00
// GenesisDoc - All genesis values
2017-03-14 00:11:49 -07:00
type GenesisDoc struct {
2017-07-03 09:58:28 -07:00
Accounts []json.RawMessage `json:"accounts"`
2017-03-14 10:55:46 -07:00
PluginOptions []json.RawMessage `json:"plugin_options"`
pluginOptions []keyValue // unmarshaled rawmessages
2017-03-14 00:11:49 -07:00
}
2017-03-14 10:55:46 -07:00
func loadGenesis(filePath string) (*FullGenesisDoc, error) {
2017-01-28 09:29:32 -08:00
bytes, err := cmn.ReadFile(filePath)
if err != nil {
2017-03-14 10:55:46 -07:00
return nil, errors.Wrap(err, "loading genesis file")
2017-01-28 09:29:32 -08:00
}
2017-03-14 00:11:49 -07:00
2017-04-25 22:08:31 -07:00
// the basecoin genesis go-wire/data :)
2017-03-14 10:55:46 -07:00
genDoc := new(FullGenesisDoc)
err = json.Unmarshal(bytes, genDoc)
if err != nil {
return nil, errors.Wrap(err, "unmarshaling genesis file")
}
if genDoc.AppOptions == nil {
genDoc.AppOptions = new(GenesisDoc)
}
2017-03-14 10:55:46 -07:00
pluginOpts, err := parseGenesisList(genDoc.AppOptions.PluginOptions)
2017-01-28 09:29:32 -08:00
if err != nil {
2017-03-14 10:55:46 -07:00
return nil, err
2017-01-28 09:29:32 -08:00
}
2017-03-14 10:55:46 -07:00
genDoc.AppOptions.pluginOptions = pluginOpts
return genDoc, nil
2017-03-14 00:11:49 -07:00
}
2017-07-04 20:28:27 -07:00
func parseGenesisList(kvzIn []json.RawMessage) (kvz []keyValue, err error) {
if len(kvzIn)%2 != 0 {
2017-01-28 09:29:32 -08:00
return nil, errors.New("genesis cannot have an odd number of items. Format = [key1, value1, key2, value2, ...]")
}
2017-02-24 14:25:48 -08:00
2017-07-04 20:28:27 -07:00
for i := 0; i < len(kvzIn); i += 2 {
2017-02-24 14:25:48 -08:00
kv := keyValue{}
2017-07-04 20:28:27 -07:00
rawK := []byte(kvzIn[i])
2017-02-24 14:25:48 -08:00
err := json.Unmarshal(rawK, &(kv.Key))
if err != nil {
return nil, errors.Errorf("Non-string key: %s", string(rawK))
2017-01-28 09:29:32 -08:00
}
2017-02-24 14:25:48 -08:00
// convert value to string if possible (otherwise raw json)
2017-07-04 20:28:27 -07:00
rawV := kvzIn[i+1]
2017-02-24 14:25:48 -08:00
err = json.Unmarshal(rawV, &(kv.Value))
if err != nil {
kv.Value = string(rawV)
2017-01-28 09:29:32 -08:00
}
2017-02-24 14:25:48 -08:00
kvz = append(kvz, kv)
2017-01-28 09:29:32 -08:00
}
return kvz, nil
}