neo-go/cli/wallet/wallet.go
Anthony De Meulemeester f3f6662fc9
Base wallet implementation (#35)
* Initial draft of the neo-go wallet

* Cleanup + more test for util package

* integrated wallet into neo-cli partially

* base wallet implementation + smartcontract code.
2018-03-02 16:24:09 +01:00

116 lines
2.5 KiB
Go

package wallet
import (
"bufio"
"errors"
"fmt"
"os"
"strings"
"github.com/CityOfZion/neo-go/pkg/wallet"
"github.com/urfave/cli"
)
var (
errNoPath = errors.New("Target path where the wallet should be stored is mandatory and should be passed using (--path, -p) flags.")
errPhraseMissmatch = errors.New("The entered passphrases do not match. Maybe you have misspelled them?")
)
// NewComand creates a new Wallet command.
func NewCommand() cli.Command {
return cli.Command{
Name: "wallet",
Usage: "create, open and manage a NEO wallet",
Subcommands: []cli.Command{
{
Name: "create",
Usage: "create a new wallet",
Action: createWallet,
Flags: []cli.Flag{
cli.StringFlag{
Name: "path, p",
Usage: "Target location of the wallet file.",
},
cli.BoolFlag{
Name: "account, a",
Usage: "Create a new account",
},
},
},
{
Name: "open",
Usage: "open a existing NEO wallet",
Action: openWallet,
Flags: []cli.Flag{
cli.StringFlag{
Name: "path, p",
Usage: "Target location of the wallet file.",
},
},
},
},
}
}
func openWallet(ctx *cli.Context) error {
return nil
}
func createWallet(ctx *cli.Context) error {
path := ctx.String("path")
if len(path) == 0 {
return cli.NewExitError(errNoPath, 1)
}
wall, err := wallet.NewWallet(path)
if err != nil {
return cli.NewExitError(err, 1)
}
if err := wall.Save(); err != nil {
return cli.NewExitError(err, 1)
}
if ctx.Bool("account") {
if err := createAccount(ctx, wall); err != nil {
return cli.NewExitError(err, 1)
}
}
dumpWallet(wall)
fmt.Printf("wallet succesfully created, file location is %s\n", wall.Path())
return nil
}
func createAccount(ctx *cli.Context, wall *wallet.Wallet) error {
var (
rawName,
rawPhrase,
rawPhraseCheck []byte
)
buf := bufio.NewReader(os.Stdin)
fmt.Print("Enter the name of the account > ")
rawName, _ = buf.ReadBytes('\n')
fmt.Print("Enter passphrase > ")
rawPhrase, _ = buf.ReadBytes('\n')
fmt.Print("Confirm passphrase > ")
rawPhraseCheck, _ = buf.ReadBytes('\n')
// Clean data
var (
name = strings.TrimRight(string(rawName), "\n")
phrase = strings.TrimRight(string(rawPhrase), "\n")
phraseCheck = strings.TrimRight(string(rawPhraseCheck), "\n")
)
if phrase != phraseCheck {
return errPhraseMissmatch
}
return wall.CreateAccount(name, phrase)
}
func dumpWallet(wall *wallet.Wallet) {
b, _ := wall.JSON()
fmt.Println("")
fmt.Println(string(b))
fmt.Println("")
}