feat: add build crosschain input (#91)

[bytom/vapor.git] / wallet / annotated.go

diff --git a/wallet/annotated.go b/wallet/annotated.go
index 5f327df..5696e34 100644 (file)
--- a/wallet/annotated.go
+++ b/wallet/annotated.go
@@ -5,16 +5,15 @@ import (
        "fmt"
 
        log "github.com/sirupsen/logrus"
-       "github.com/tendermint/tmlibs/db"
 
        "github.com/vapor/account"
        "github.com/vapor/asset"
        "github.com/vapor/blockchain/query"
-       "github.com/vapor/blockchain/signers"
        "github.com/vapor/common"
        "github.com/vapor/consensus"
        "github.com/vapor/consensus/segwit"
        "github.com/vapor/crypto/sha3pool"
+       dbm "github.com/vapor/database/leveldb"
        "github.com/vapor/protocol/bc"
        "github.com/vapor/protocol/bc/types"
        "github.com/vapor/protocol/vm/vmutil"
@@ -51,11 +50,10 @@ func (w *Wallet) getExternalDefinition(assetID *bc.AssetID) json.RawMessage {
                Alias:             &alias,
                DefinitionMap:     definitionMap,
                RawDefinitionByte: definitionByte,
-               Signer:            &signers.Signer{Type: "external"},
        }
 
        if err := w.AssetReg.SaveAsset(externalAsset, alias); err != nil {
-               log.WithFields(log.Fields{"err": err, "assetID": alias}).Warning("fail on save external asset to internal asset DB")
+               log.WithFields(log.Fields{"module": logModule, "err": err, "assetID": alias}).Warning("fail on save external asset to internal asset DB")
        }
        return definitionByte
 }
@@ -85,7 +83,7 @@ func (w *Wallet) getAliasDefinition(assetID bc.AssetID) (string, json.RawMessage
 }
 
 // annotateTxs adds account data to transactions
-func annotateTxsAccount(txs []*query.AnnotatedTx, walletDB db.DB) {
+func annotateTxsAccount(txs []*query.AnnotatedTx, walletDB dbm.DB) {
        for i, tx := range txs {
                for j, input := range tx.Inputs {
                        //issue asset tx input SpentOutputID is nil
@@ -110,7 +108,7 @@ func annotateTxsAccount(txs []*query.AnnotatedTx, walletDB db.DB) {
        }
 }
 
-func getAccountFromACP(program []byte, walletDB db.DB) (*account.Account, error) {
+func getAccountFromACP(program []byte, walletDB dbm.DB) (*account.Account, error) {
        var hash common.Hash
        accountCP := account.CtrlProgram{}
        localAccount := account.Account{}
@@ -140,12 +138,6 @@ func getAccountFromACP(program []byte, walletDB db.DB) (*account.Account, error)
 
 var emptyJSONObject = json.RawMessage(`{}`)
 
-func isValidJSON(b []byte) bool {
-       var v interface{}
-       err := json.Unmarshal(b, &v)
-       return err == nil
-}
-
 func (w *Wallet) buildAnnotatedTransaction(orig *types.Tx, b *types.Block, statusFail bool, indexInBlock int) *query.AnnotatedTx {
        tx := &query.AnnotatedTx{
                ID:                     orig.ID,
@@ -183,39 +175,29 @@ func (w *Wallet) BuildAnnotatedInput(tx *types.Tx, i uint32) *query.AnnotatedInp
        in.InputID = id
        e := tx.Entries[id]
        switch e := e.(type) {
-       case *bc.Spend:
-               in.Type = "spend"
+       case *bc.CrossChainInput:
+               in.Type = "cross_chain_in"
                in.ControlProgram = orig.ControlProgram()
                in.Address = w.getAddressFromControlProgram(in.ControlProgram)
-               in.SpentOutputID = e.SpentOutputId
+               in.SpentOutputID = e.MainchainOutputId
                arguments := orig.Arguments()
                for _, arg := range arguments {
                        in.WitnessArguments = append(in.WitnessArguments, arg)
                }
-       case *bc.Issuance:
-               in.Type = "issue"
-               in.IssuanceProgram = orig.IssuanceProgram()
-               arguments := orig.Arguments()
-               for _, arg := range arguments {
-                       in.WitnessArguments = append(in.WitnessArguments, arg)
-               }
-               if assetDefinition := orig.AssetDefinition(); isValidJSON(assetDefinition) {
-                       assetDefinition := json.RawMessage(assetDefinition)
-                       in.AssetDefinition = &assetDefinition
-               }
-       case *bc.Coinbase:
-               in.Type = "coinbase"
-               in.Arbitrary = e.Arbitrary
-       case *bc.Claim:
-               in.Type = "claim"
+
+       case *bc.Spend:
+               in.Type = "spend"
                in.ControlProgram = orig.ControlProgram()
                in.Address = w.getAddressFromControlProgram(in.ControlProgram)
                in.SpentOutputID = e.SpentOutputId
-               in.Peginwitness = orig.Peginwitness
                arguments := orig.Arguments()
                for _, arg := range arguments {
                        in.WitnessArguments = append(in.WitnessArguments, arg)
                }
+
+       case *bc.Coinbase:
+               in.Type = "coinbase"
+               in.Arbitrary = e.Arbitrary
        }
        return in
 }
@@ -259,11 +241,11 @@ func (w *Wallet) BuildAnnotatedOutput(tx *types.Tx, idx int) *query.AnnotatedOut
        out := &query.AnnotatedOutput{
                OutputID:        *outid,
                Position:        idx,
-               AssetID:         *orig.AssetId,
+               AssetID:         *orig.AssetAmount().AssetId,
                AssetDefinition: &emptyJSONObject,
-               Amount:          orig.Amount,
-               ControlProgram:  orig.ControlProgram,
-               Address:         w.getAddressFromControlProgram(orig.ControlProgram),
+               Amount:          orig.AssetAmount().Amount,
+               ControlProgram:  orig.ControlProgram(),
+               Address:         w.getAddressFromControlProgram(orig.ControlProgram()),
        }
 
        if vmutil.IsUnspendable(out.ControlProgram) {