[pkg/keys] Begin build out usage of keystore, issue about internal from main harmony-repo

client/cmd/keys.go

@@ -2,9 +2,8 @@ package cmd
 
 import (
 	"fmt"
-	"strings"
 
-	// homedir "github.com/mitchellh/go-homedir"
+	"github.com/harmony-one/go-sdk/pkg/keys"
 	"github.com/spf13/cobra"
 )
 
@@ -16,7 +15,7 @@ func init() {
 Manage your local keys
 `,
 		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Println("Print: " + strings.Join(args, " "))
+			cmd.Help()
 		},
 	}
 
@@ -30,9 +29,9 @@ Manage your local keys
 
 	cmdAdd := &cobra.Command{
 		Use:   "add",
-		Short: "Create a new key, or import from seed",
+		Short: "Create a new key with passphrase",
 		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Println(cmd)
+			keys.AddNewKey()
 		},
 	}
 
@@ -40,7 +39,7 @@ Manage your local keys
 		Use:   "list",
 		Short: "List all keys",
 		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Println(cmd)
+			keys.ListKeys()
 		},
 	}
 

pkg/keys/keys.go

@@ -0,0 +1,61 @@
+package keys
+
+import (
+	"fmt"
+	"os"
+	"path"
+
+	"github.com/harmony-one/harmony/accounts/keystore"
+	// common2 "github.com/harmony-one/harmony/internal/common"
+	// "github.com/harmony-one/harmony/internal/utils"
+
+	homedir "github.com/mitchellh/go-homedir"
+)
+
+func checkAndMakeKeyDirIfNeeded() string {
+	userDir, _ := homedir.Dir()
+	hmyCLIDir := path.Join(userDir, ".hmy_cli")
+	if _, err := os.Stat(hmyCLIDir); os.IsNotExist(err) {
+		// Double check with Leo what is right file persmission
+		os.Mkdir(hmyCLIDir, 0700)
+	}
+
+	return hmyCLIDir
+}
+
+func ListKeys() {
+	hmyCLIDir := checkAndMakeKeyDirIfNeeded()
+	scryptN := keystore.StandardScryptN
+	scryptP := keystore.StandardScryptP
+	ks := keystore.NewKeyStore(hmyCLIDir, scryptN, scryptP)
+	allAccounts := ks.Accounts()
+	for _, account := range allAccounts {
+		fmt.Printf("account: %s\n", account.Address)
+		fmt.Printf("URL: %s\n", account.URL)
+	}
+
+}
+
+func AddNewKey() {
+	hmyCLIDir := checkAndMakeKeyDirIfNeeded()
+	scryptN := keystore.StandardScryptN
+	scryptP := keystore.StandardScryptP
+	ks := keystore.NewKeyStore(hmyCLIDir, scryptN, scryptP)
+	password := "edgar"
+	// TODO Need to factor out some definitions from harmony/internal to something
+	// more public, like core or api
+	// password := utils.AskForPassphrase("Passphrase: ")
+	// password2 := utils.AskForPassphrase("Passphrase again: ")
+	// if password != password2 {
+	// 	fmt.Printf("Passphrase doesn't match. Please try again!\n")
+	// 	os.Exit(3)
+	// }
+	account, err := ks.NewAccount(password)
+	if err != nil {
+		fmt.Printf("new account error: %v\n", err)
+	}
+	// fmt.Printf("account: %s\n", common2.MustAddressToBech32(account.Address))
+	// fmt.Printf("URL: %s\n", account.URL)
+	fmt.Printf("account: %s\n", account.Address)
+	fmt.Printf("URL: %s\n", account.URL)
+}