list-addresses sort addressResp by create index

[bytom/bytom.git] / api / accounts.go

diff --git a/api/accounts.go b/api/accounts.go
index 7644c64..61ec5bc 100644 (file)
--- a/api/accounts.go
+++ b/api/accounts.go
@@ -2,6 +2,7 @@ package api
 
 import (
        "context"
+       "sort"
 
        log "github.com/sirupsen/logrus"
 
@@ -29,7 +30,7 @@ func (a *API) createAccount(ctx context.Context, ins struct {
        return NewSuccessResponse(annotatedAccount)
 }
 
-// AccountInfo
+// AccountInfo is request struct for deleteAccount
 type AccountInfo struct {
        Info string `json:"account_info"`
 }
@@ -43,7 +44,7 @@ func (a *API) deleteAccount(ctx context.Context, in AccountInfo) Response {
 }
 
 type validateAddressResp struct {
-       Vaild   bool `json:"vaild"`
+       Valid   bool `json:"valid"`
        IsLocal bool `json:"is_local"`
 }
 
@@ -52,10 +53,10 @@ func (a *API) validateAddress(ctx context.Context, ins struct {
        Address string `json:"address"`
 }) Response {
        resp := &validateAddressResp{
-               Vaild:   false,
+               Valid:   false,
                IsLocal: false,
        }
-       address, err := common.DecodeAddress(ins.Address, consensus.ActiveNetParams)
+       address, err := common.DecodeAddress(ins.Address, &consensus.ActiveNetParams)
        if err != nil {
                return NewSuccessResponse(resp)
        }
@@ -74,7 +75,7 @@ func (a *API) validateAddress(ctx context.Context, ins struct {
                return NewSuccessResponse(resp)
        }
 
-       resp.Vaild = true
+       resp.Valid = true
        resp.IsLocal = a.wallet.AccountMgr.IsLocalControlProgram(program)
        return NewSuccessResponse(resp)
 }
@@ -86,6 +87,18 @@ type addressResp struct {
        Change       bool   `json:"change"`
 }
 
+type addressRespIndex struct {
+       KeyIndex    uint64
+       AddressResp addressResp
+}
+
+// SortByIndex implements sort.Interface for addressRespIndex slices
+type SortByIndex []addressRespIndex
+
+func (a SortByIndex) Len() int           { return len(a) }
+func (a SortByIndex) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
+func (a SortByIndex) Less(i, j int) bool { return a[i].KeyIndex < a[j].KeyIndex }
+
 func (a *API) listAddresses(ctx context.Context, ins struct {
        AccountID    string `json:"account_id"`
        AccountAlias string `json:"account_alias"`
@@ -106,19 +119,35 @@ func (a *API) listAddresses(ctx context.Context, ins struct {
                target = acc
        }
 
-       cps, err := a.wallet.AccountMgr.ListCtrlProgramsByAccountId(ctx, target.ID)
+       cps, err := a.wallet.AccountMgr.ListControlProgram()
        if err != nil {
                return NewErrorResponse(err)
        }
 
-       var addresses []*addressResp
+       addressRespIndexes := []addressRespIndex{}
        for _, cp := range cps {
-               addresses = append(addresses, &addressResp{
-                       AccountAlias: target.Alias,
-                       AccountID:    cp.AccountID,
-                       Address:      cp.Address,
-                       Change:       cp.Change,
-               })
+               if cp.Address == "" || cp.AccountID != target.ID {
+                       continue
+               }
+
+               addressIndex := addressRespIndex{
+                       KeyIndex: cp.KeyIndex,
+                       AddressResp: addressResp{
+                               AccountAlias: target.Alias,
+                               AccountID:    cp.AccountID,
+                               Address:      cp.Address,
+                               Change:       cp.Change,
+                       },
+               }
+               addressRespIndexes = append(addressRespIndexes, addressIndex)
        }
+
+       // sort AddressResp by KeyIndex
+       addresses := []addressResp{}
+       sort.Sort(SortByIndex(addressRespIndexes))
+       for _, addr := range addressRespIndexes {
+               addresses = append(addresses, addr.AddressResp)
+       }
+
        return NewSuccessResponse(addresses)
 }