status-go/src/gethdep.go

package main

import (
	"errors"
	"fmt"
	"io/ioutil"
	"time"

	"github.com/ethereum/go-ethereum/accounts"
	"github.com/ethereum/go-ethereum/cmd/utils"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	errextra "github.com/pkg/errors"
)

var (
	scryptN = 4096
	scryptP = 6
)

// createAccount creates an internal geth account
func createAccount(password string) (string, string, error) {

	if currentNode != nil {

		w := true
		keydir := datadir + "/testnet/keystore"
		accman := accounts.NewManager(keydir, scryptN, scryptP, accountSync)

		// generate the account
		account, err := accman.NewAccount(password, w)
		if err != nil {
			return "", "", errextra.Wrap(err, "Account manager could not create the account")
		}
		address := fmt.Sprintf("%x", account.Address)

		// recover the public key to return
		keyContents, err := ioutil.ReadFile(account.File)
		if err != nil {
			return address, "", errextra.Wrap(err, "Could not load the key contents")
		}
		key, err := accounts.DecryptKey(keyContents, password)
		if err != nil {
			return address, "", errextra.Wrap(err, "Could not recover the key")
		}
		pubKey := common.ToHex(crypto.FromECDSAPub(&key.PrivateKey.PublicKey))

		return address, pubKey, nil

	}

	return "", "", errors.New("No running node detected for account creation")

}

// unlockAccount unlocks an existing account for a certain duration and
// inject the account as a whisper identity if the account was created as
// a whisper enabled account
func unlockAccount(address, password string, seconds int) error {

	if currentNode != nil {

		accman := utils.MakeAccountManager(c, accountSync)
		account, err := utils.MakeAddress(accman, address)
		if err != nil {
			return errextra.Wrap(err, "Could not retrieve account from address")
		}

		err = accman.TimedUnlock(account, password, time.Duration(seconds)*time.Second)
		if err != nil {
			return errextra.Wrap(err, "Could not decrypt account")
		}

		return nil

	}

	return errors.New("No running node detected for account unlock")

}

// createAndStartNode creates a node entity and starts the
// node running locally
func createAndStartNode(inputDir string) error {

	currentNode = MakeNode(inputDir)
	if currentNode != nil {
		RunNode(currentNode)
		return nil
	}

	return errors.New("Could not create the in-memory node object")

}