-
Notifications
You must be signed in to change notification settings - Fork 662
/
visitor.go
92 lines (80 loc) · 1.98 KB
/
visitor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
// Copyright (C) 2019-2022, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package txs
import "github.com/ava-labs/avalanchego/vms/components/avax"
var _ Visitor = (*utxoGetter)(nil)
// Allow vm to execute custom logic against the underlying transaction types.
type Visitor interface {
BaseTx(*BaseTx) error
CreateAssetTx(*CreateAssetTx) error
OperationTx(*OperationTx) error
ImportTx(*ImportTx) error
ExportTx(*ExportTx) error
}
// utxoGetter returns the UTXOs transaction is producing.
type utxoGetter struct {
tx *Tx
utxos []*avax.UTXO
}
func (u *utxoGetter) BaseTx(tx *BaseTx) error {
txID := u.tx.ID()
u.utxos = make([]*avax.UTXO, len(tx.Outs))
for i, out := range tx.Outs {
u.utxos[i] = &avax.UTXO{
UTXOID: avax.UTXOID{
TxID: txID,
OutputIndex: uint32(i),
},
Asset: avax.Asset{ID: out.AssetID()},
Out: out.Out,
}
}
return nil
}
func (u *utxoGetter) ImportTx(tx *ImportTx) error {
return u.BaseTx(&tx.BaseTx)
}
func (u *utxoGetter) ExportTx(tx *ExportTx) error {
return u.BaseTx(&tx.BaseTx)
}
func (u *utxoGetter) CreateAssetTx(t *CreateAssetTx) error {
if err := u.BaseTx(&t.BaseTx); err != nil {
return err
}
txID := u.tx.ID()
for _, state := range t.States {
for _, out := range state.Outs {
u.utxos = append(u.utxos, &avax.UTXO{
UTXOID: avax.UTXOID{
TxID: txID,
OutputIndex: uint32(len(u.utxos)),
},
Asset: avax.Asset{
ID: txID,
},
Out: out,
})
}
}
return nil
}
func (u *utxoGetter) OperationTx(t *OperationTx) error {
// The error is explicitly dropped here because no error is ever returned
// from the utxoGetter.
_ = u.BaseTx(&t.BaseTx)
txID := u.tx.ID()
for _, op := range t.Ops {
asset := op.AssetID()
for _, out := range op.Op.Outs() {
u.utxos = append(u.utxos, &avax.UTXO{
UTXOID: avax.UTXOID{
TxID: txID,
OutputIndex: uint32(len(u.utxos)),
},
Asset: avax.Asset{ID: asset},
Out: out,
})
}
}
return nil
}