feat: add build crosschain input (#91)

[bytom/vapor.git] / blockchain / txbuilder / actions.go

diff --git a/blockchain/txbuilder/actions.go b/blockchain/txbuilder/actions.go
index 6c4ee84..99f35bc 100644 (file)
--- a/blockchain/txbuilder/actions.go
+++ b/blockchain/txbuilder/actions.go
@@ -5,6 +5,8 @@ import (
        stdjson "encoding/json"
        "errors"
 
+       "github.com/vapor/config"
+
        "github.com/vapor/common"
        "github.com/vapor/consensus"
        "github.com/vapor/encoding/json"
@@ -44,9 +46,9 @@ func (a *controlAddressAction) Build(ctx context.Context, b *TemplateBuilder) er
        if err != nil {
                return err
        }
+
        redeemContract := address.ScriptAddress()
        program := []byte{}
-
        switch address.(type) {
        case *common.AddressWitnessPubKeyHash:
                program, err = vmutil.P2WPKHProgram(redeemContract)
@@ -59,7 +61,7 @@ func (a *controlAddressAction) Build(ctx context.Context, b *TemplateBuilder) er
                return err
        }
 
-       out := types.NewTxOutput(*a.AssetId, a.Amount, program)
+       out := types.NewIntraChainOutput(*a.AssetId, a.Amount, program)
        return b.AddOutput(out)
 }
 
@@ -94,7 +96,7 @@ func (a *controlProgramAction) Build(ctx context.Context, b *TemplateBuilder) er
                return MissingFieldsError(missing...)
        }
 
-       out := types.NewTxOutput(*a.AssetId, a.Amount, a.Program)
+       out := types.NewIntraChainOutput(*a.AssetId, a.Amount, a.Program)
        return b.AddOutput(out)
 }
 
@@ -130,10 +132,164 @@ func (a *retireAction) Build(ctx context.Context, b *TemplateBuilder) error {
        if err != nil {
                return err
        }
-       out := types.NewTxOutput(*a.AssetId, a.Amount, program)
+       out := types.NewIntraChainOutput(*a.AssetId, a.Amount, program)
        return b.AddOutput(out)
 }
 
 func (a *retireAction) ActionType() string {
        return "retire"
 }
+
+// DecodeCrossOutAction convert input data to action struct
+func DecodeCrossOutAction(data []byte) (Action, error) {
+       a := new(crossOutAction)
+       err := stdjson.Unmarshal(data, a)
+       return a, err
+}
+
+type crossOutAction struct {
+       bc.AssetAmount
+       Address string `json:"address"`
+}
+
+func (a *crossOutAction) Build(ctx context.Context, b *TemplateBuilder) error {
+       var missing []string
+       if a.Address == "" {
+               missing = append(missing, "address")
+       }
+       if a.AssetId.IsZero() {
+               missing = append(missing, "asset_id")
+       }
+       if a.Amount == 0 {
+               missing = append(missing, "amount")
+       }
+       if len(missing) > 0 {
+               return MissingFieldsError(missing...)
+       }
+
+       address, err := common.DecodeAddress(a.Address, &consensus.MainNetParams)
+       if err != nil {
+               return err
+       }
+
+       redeemContract := address.ScriptAddress()
+       program := []byte{}
+       switch address.(type) {
+       case *common.AddressWitnessPubKeyHash:
+               program, err = vmutil.P2WPKHProgram(redeemContract)
+       case *common.AddressWitnessScriptHash:
+               program, err = vmutil.P2WSHProgram(redeemContract)
+       default:
+               return errors.New("unsupport address type")
+       }
+       if err != nil {
+               return err
+       }
+
+       out := types.NewCrossChainOutput(*a.AssetId, a.Amount, program)
+       return b.AddOutput(out)
+}
+
+func (a *crossOutAction) ActionType() string {
+       return "cross_chain_out"
+}
+
+// DecodeVoteOutputAction convert input data to action struct
+func DecodeVoteOutputAction(data []byte) (Action, error) {
+       a := new(voteOutputAction)
+       err := stdjson.Unmarshal(data, a)
+       return a, err
+}
+
+type voteOutputAction struct {
+       bc.AssetAmount
+       Address string        `json:"address"`
+       Vote    json.HexBytes `json:"vote"`
+}
+
+func (a *voteOutputAction) Build(ctx context.Context, b *TemplateBuilder) error {
+       var missing []string
+       if a.Address == "" {
+               missing = append(missing, "address")
+       }
+       if a.AssetId.IsZero() {
+               missing = append(missing, "asset_id")
+       }
+       if a.Amount == 0 {
+               missing = append(missing, "amount")
+       }
+       if len(a.Vote) == 0 {
+               missing = append(missing, "vote")
+       }
+       if len(missing) > 0 {
+               return MissingFieldsError(missing...)
+       }
+
+       address, err := common.DecodeAddress(a.Address, &consensus.ActiveNetParams)
+       if err != nil {
+               return err
+       }
+
+       redeemContract := address.ScriptAddress()
+       program := []byte{}
+       switch address.(type) {
+       case *common.AddressWitnessPubKeyHash:
+               program, err = vmutil.P2WPKHProgram(redeemContract)
+       case *common.AddressWitnessScriptHash:
+               program, err = vmutil.P2WSHProgram(redeemContract)
+       default:
+               return errors.New("unsupport address type")
+       }
+       if err != nil {
+               return err
+       }
+
+       out := types.NewVoteOutput(*a.AssetId, a.Amount, program, a.Vote)
+       return b.AddOutput(out)
+}
+
+func (a *voteOutputAction) ActionType() string {
+       return "vote_output"
+}
+
+// DecodeCrossInAction convert input data to action struct
+func DecodeCrossInAction(data []byte) (Action, error) {
+       a := new(crossInAction)
+       err := stdjson.Unmarshal(data, a)
+       return a, err
+}
+
+type crossInAction struct {
+       bc.AssetAmount
+       SourceID          bc.Hash       `json:"source_id"`
+       SourcePos         uint64        `json:"source_pos"`
+       RawDefinitionByte json.HexBytes `json:"raw_definition_byte"`
+}
+
+func (a *crossInAction) Build(ctx context.Context, builder *TemplateBuilder) error {
+       var missing []string
+       if a.SourceID.IsZero() {
+               missing = append(missing, "source_id")
+       }
+       if a.AssetId.IsZero() {
+               missing = append(missing, "asset_id")
+       }
+       if a.Amount == 0 {
+               missing = append(missing, "amount")
+       }
+       if len(missing) > 0 {
+               return MissingFieldsError(missing...)
+       }
+
+       // arguments will be set when materializeWitnesses
+       fedProg := config.FederationProgrom(config.CommonConfig)
+       txin := types.NewCrossChainInput(nil, a.SourceID, *a.AssetId, a.Amount, a.SourcePos, fedProg, a.RawDefinitionByte)
+       tplIn := &SigningInstruction{}
+       fed := config.CommonConfig.Federation
+       tplIn.AddRawWitnessKeys(fed.Xpubs, nil, fed.Quorum)
+       return builder.AddInput(txin, tplIn)
+}
+
+func (a *crossInAction) ActionType() string {
+       return "cross_chain_in"
+}