forked from TrueCloudLab/frostfs-node
141 lines
3.7 KiB
Go
141 lines
3.7 KiB
Go
package morph
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/nspcc-dev/neo-go/cli/input"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native/nativenames"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// defaultNotaryDepositLifetime is an amount of blocks notary deposit stays valid.
|
|
// https://github.com/nspcc-dev/neo-go/blob/master/pkg/core/native/notary.go#L48
|
|
const defaultNotaryDepositLifetime = 5760
|
|
|
|
func depositNotary(cmd *cobra.Command, _ []string) error {
|
|
p, err := cmd.Flags().GetString(storageWalletFlag)
|
|
if err != nil {
|
|
return err
|
|
} else if p == "" {
|
|
return fmt.Errorf("missing wallet path (use '--%s <out.json>')", storageWalletFlag)
|
|
}
|
|
|
|
w, err := wallet.NewWalletFromFile(p)
|
|
if err != nil {
|
|
return fmt.Errorf("can't open wallet: %v", err)
|
|
}
|
|
|
|
accHash := w.GetChangeAddress()
|
|
if addr, err := cmd.Flags().GetString(walletAccountFlag); err == nil {
|
|
accHash, err = address.StringToUint160(addr)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid address: %s", addr)
|
|
}
|
|
}
|
|
|
|
acc := w.GetAccount(accHash)
|
|
if acc == nil {
|
|
return fmt.Errorf("can't find account for %s", accHash)
|
|
}
|
|
|
|
prompt := fmt.Sprintf("Enter password for %s >", address.Uint160ToString(accHash))
|
|
pass, err := input.ReadPassword(prompt)
|
|
if err != nil {
|
|
return fmt.Errorf("can't get password: %v", err)
|
|
}
|
|
|
|
err = acc.Decrypt(pass, keys.NEP2ScryptParams())
|
|
if err != nil {
|
|
return fmt.Errorf("can't unlock account: %v", err)
|
|
}
|
|
|
|
gasStr, err := cmd.Flags().GetString(refillGasAmountFlag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
gasAmount, err := parseGASAmount(gasStr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
till := int64(defaultNotaryDepositLifetime)
|
|
tillStr, err := cmd.Flags().GetString(notaryDepositTillFlag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if tillStr != "" {
|
|
till, err = strconv.ParseInt(tillStr, 10, 64)
|
|
if err != nil || till <= 0 {
|
|
return fmt.Errorf("notary deposit lifetime must be a positive integer")
|
|
}
|
|
}
|
|
|
|
c, err := getN3Client(viper.GetViper())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
nhs, err := getNativeHashes(c)
|
|
if err != nil {
|
|
return fmt.Errorf("can't get native contract hashes: %w", err)
|
|
}
|
|
|
|
gasHash, ok := nhs[nativenames.Gas]
|
|
if !ok {
|
|
return fmt.Errorf("can't retrieve %s contract hash", nativenames.Gas)
|
|
}
|
|
notaryHash, ok := nhs[nativenames.Notary]
|
|
if !ok {
|
|
return fmt.Errorf("can't retrieve %s contract hash", nativenames.Notary)
|
|
}
|
|
|
|
height, err := c.GetBlockCount()
|
|
if err != nil {
|
|
return fmt.Errorf("can't get current height: %v", err)
|
|
}
|
|
|
|
bw := io.NewBufBinWriter()
|
|
emit.AppCall(bw.BinWriter, gasHash, "transfer", callflag.All,
|
|
accHash, notaryHash.BytesBE(), int64(gasAmount), []interface{}{nil, int64(height) + till})
|
|
emit.Opcodes(bw.BinWriter, opcode.ASSERT)
|
|
if bw.Err != nil {
|
|
return fmt.Errorf("BUG: invalid transfer arguments: %w", bw.Err)
|
|
}
|
|
|
|
tx, err := c.CreateTxFromScript(bw.Bytes(), acc, -1, 0, []rpcclient.SignerAccount{{
|
|
Signer: transaction.Signer{
|
|
Account: acc.Contract.ScriptHash(),
|
|
Scopes: transaction.Global,
|
|
},
|
|
Account: acc,
|
|
}})
|
|
if err != nil {
|
|
return fmt.Errorf("can't create tx: %w", err)
|
|
}
|
|
|
|
mn, err := c.GetNetwork()
|
|
if err != nil {
|
|
// error appears only if client
|
|
// has not been initialized
|
|
panic(err)
|
|
}
|
|
|
|
err = acc.SignTx(mn, tx)
|
|
if err != nil {
|
|
return fmt.Errorf("can't sign tx: %w", err)
|
|
}
|
|
|
|
cc := defaultClientContext(c)
|
|
return cc.sendTx(tx, cmd, true)
|
|
}
|