package account import ( "fmt" "github.com/ethereum/go-ethereum/crypto" "github.com/pborman/uuid" "github.com/status-im/status-go/account/generator" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/extkeys" ) // OnboardingAccount is returned during onboarding and contains its ID and the mnemonic to re-generate the same account Info keys. type OnboardingAccount struct { ID string `json:"id"` mnemonic string Info Info `json:"info"` } // Onboarding is a struct contains a slice of OnboardingAccount. type Onboarding struct { accounts map[string]*OnboardingAccount } // NewOnboarding returns a new onboarding struct generating n accounts. func NewOnboarding(n, mnemonicPhraseLength int) (*Onboarding, error) { onboarding := &Onboarding{ accounts: make(map[string]*OnboardingAccount), } for i := 0; i < n; i++ { account, err := onboarding.generateAccount(mnemonicPhraseLength) if err != nil { return nil, err } onboarding.accounts[account.ID] = account } return onboarding, nil } // Accounts return the list of OnboardingAccount generated. func (o *Onboarding) Accounts() []*OnboardingAccount { accounts := make([]*OnboardingAccount, 0) for _, a := range o.accounts { accounts = append(accounts, a) } return accounts } // Account returns an OnboardingAccount by id. func (o *Onboarding) Account(id string) (*OnboardingAccount, error) { account, ok := o.accounts[id] if !ok { return nil, ErrOnboardingAccountNotFound } return account, nil } func (o *Onboarding) generateAccount(mnemonicPhraseLength int) (*OnboardingAccount, error) { entropyStrength, err := generator.MnemonicPhraseLengthToEntropyStrength(mnemonicPhraseLength) if err != nil { return nil, err } mnemonic := extkeys.NewMnemonic() mnemonicPhrase, err := mnemonic.MnemonicPhrase(entropyStrength, extkeys.EnglishLanguage) if err != nil { return nil, fmt.Errorf("can not create mnemonic seed: %v", err) } masterExtendedKey, err := extkeys.NewMaster(mnemonic.MnemonicSeed(mnemonicPhrase, "")) if err != nil { return nil, fmt.Errorf("can not create master extended key: %v", err) } walletAddress, walletPubKey, err := o.deriveAccount(masterExtendedKey, extkeys.KeyPurposeWallet, 0) if err != nil { return nil, err } info := Info{ WalletAddress: walletAddress, WalletPubKey: walletPubKey, ChatAddress: walletAddress, ChatPubKey: walletPubKey, } uuid := uuid.NewRandom().String() account := &OnboardingAccount{ ID: uuid, mnemonic: mnemonicPhrase, Info: info, } return account, nil } func (o *Onboarding) deriveAccount(masterExtendedKey *extkeys.ExtendedKey, purpose extkeys.KeyPurpose, index uint32) (string, string, error) { extendedKey, err := masterExtendedKey.ChildForPurpose(purpose, index) if err != nil { return "", "", err } privateKeyECDSA := extendedKey.ToECDSA() address := crypto.PubkeyToAddress(privateKeyECDSA.PublicKey) publicKeyHex := types.EncodeHex(crypto.FromECDSAPub(&privateKeyECDSA.PublicKey)) return address.Hex(), publicKeyHex, nil }