api add createAccount and createKey (#393)

[bytom/vapor.git] / toolbar / apinode / account.go

diff --git a/toolbar/apinode/account.go b/toolbar/apinode/account.go
new file mode 100644 (file)
index 0000000..818f7ee
--- /dev/null
+++ b/toolbar/apinode/account.go
@@ -0,0 +1,95 @@
+package apinode
+
+import (
+       "encoding/json"
+
+       "github.com/vapor/api"
+       "github.com/vapor/blockchain/pseudohsm"
+       "github.com/vapor/blockchain/query"
+       "github.com/vapor/blockchain/txbuilder"
+       "github.com/vapor/crypto/ed25519/chainkd"
+       "github.com/vapor/errors"
+)
+
+func (n *Node) CreateKey(alias, password string) (*api.CreateKeyResp, error) {
+       url := "/create-key"
+       payload, err := json.Marshal(api.CreateKeyReq{Alias: alias, Password: password})
+       if err != nil {
+               return nil, errors.Wrap(err, "json marshal")
+       }
+
+       res := &api.CreateKeyResp{}
+       return res, n.request(url, payload, res)
+}
+
+func (n *Node) ListKeys() (*[]pseudohsm.XPub, error) {
+       url := "/list-keys"
+       res := &[]pseudohsm.XPub{}
+       return res, n.request(url, nil, res)
+}
+
+//默认创建单签账户
+func (n *Node) CreateAccount(keyAlias, accountAlias string) (*query.AnnotatedAccount, error) {
+       xPub, err := n.ListKeys()
+       if err != nil {
+               return nil, err
+       }
+
+       var rootXPub *chainkd.XPub
+       for _, x := range *xPub {
+               if x.Alias == keyAlias {
+                       rootXPub = &x.XPub
+                       break
+               }
+       }
+
+       if rootXPub == nil {
+               return nil, errors.New("keyAlias not found!")
+       }
+
+       return n.postCreateAccount(accountAlias, 1, []chainkd.XPub{*rootXPub})
+}
+
+//多签账户
+func (n *Node) CreateMultiSignAccount(alias string, quorum int, rootXPubs []chainkd.XPub) (*query.AnnotatedAccount, error) {
+       return n.postCreateAccount(alias, quorum, rootXPubs)
+}
+
+func (n *Node) postCreateAccount(alias string, quorum int, rootXPubs []chainkd.XPub) (*query.AnnotatedAccount, error) {
+       url := "/create-account"
+       payload, err := json.Marshal(api.CreateAccountReq{
+               Alias:     alias,
+               Quorum:    quorum,
+               RootXPubs: rootXPubs,
+       })
+       if err != nil {
+               return nil, errors.Wrap(err, "json marshal")
+       }
+
+       res := &query.AnnotatedAccount{}
+       return res, n.request(url, payload, res)
+}
+
+func (n *Node) ListAccounts() (*[]query.AnnotatedAccount, error) {
+       url := "/list-accounts"
+       payload, err := json.Marshal(struct{}{})
+       if err != nil {
+               return nil, errors.Wrap(err, "json marshal")
+       }
+
+       res := &[]query.AnnotatedAccount{}
+       return res, n.request(url, payload, res)
+}
+
+func (n *Node) CreateAccountReceiver(alias string) (*txbuilder.Receiver, error) {
+       url := "/create-account-receiver"
+       payload, err := json.Marshal(api.AccountFilter{
+               AccountAlias: alias,
+       })
+       if err != nil {
+               return nil, errors.Wrap(err, "json marshal")
+       }
+
+       res := &txbuilder.Receiver{}
+       return res, n.request(url, payload, res)
+}