-
Notifications
You must be signed in to change notification settings - Fork 2
/
withdraw.go
89 lines (69 loc) · 2.13 KB
/
withdraw.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
package cmds
import (
"context"
"github.com/ProtoconNet/mitum-currency/v3/operation/extension"
"github.com/ProtoconNet/mitum-currency/v3/types"
"github.com/pkg/errors"
"github.com/ProtoconNet/mitum2/base"
)
type WithdrawCommand struct {
BaseCommand
OperationFlags
Sender AddressFlag `arg:"" name:"sender" help:"sender address" required:"true"`
Target AddressFlag `arg:"" name:"target" help:"target contract account address" required:"true"`
Amount CurrencyAmountFlag `arg:"" name:"currency-amount" help:"amount (ex: \"<currency>,<amount>\")"`
sender base.Address
target base.Address
}
func (cmd *WithdrawCommand) Run(pctx context.Context) error {
if _, err := cmd.prepare(pctx); err != nil {
return err
}
if err := cmd.parseFlags(); err != nil {
return err
}
op, err := cmd.createOperation()
if err != nil {
return err
}
PrettyPrint(cmd.Out, op)
return nil
}
func (cmd *WithdrawCommand) parseFlags() error {
if err := cmd.OperationFlags.IsValid(nil); err != nil {
return err
}
if sender, err := cmd.Sender.Encode(enc); err != nil {
return errors.Wrapf(err, "invalid sender format, %v", cmd.Sender.String())
} else if target, err := cmd.Target.Encode(enc); err != nil {
return errors.Wrapf(err, "invalid target format, %v", cmd.Target.String())
} else {
cmd.sender = sender
cmd.target = target
}
return nil
}
func (cmd *WithdrawCommand) createOperation() (base.Operation, error) { // nolint:dupl
var items []extension.WithdrawItem
ams := make([]types.Amount, 1)
am := types.NewAmount(cmd.Amount.Big, cmd.Amount.CID)
if err := am.IsValid(nil); err != nil {
return nil, err
}
ams[0] = am
item := extension.NewWithdrawItemMultiAmounts(cmd.target, ams)
if err := item.IsValid(nil); err != nil {
return nil, err
}
items = append(items, item)
fact := extension.NewWithdrawFact([]byte(cmd.Token), cmd.sender, items)
op, err := extension.NewWithdraw(fact)
if err != nil {
return nil, errors.Wrap(err, "create withdraw operation")
}
err = op.HashSign(cmd.Privatekey, cmd.NetworkID.NetworkID())
if err != nil {
return nil, errors.Wrap(err, "create withdraw operation")
}
return op, nil
}