frostfs-node/pkg/innerring/invoke/balance.go
Alex Vanin 91a1896b8b [#486] innerring: Use fee provider interface in invoke package
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
2021-05-05 12:45:16 +03:00

118 lines
2.3 KiB
Go

package invoke
import (
"github.com/nspcc-dev/neo-go/pkg/util"
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
)
type (
// LockParams for LockAsset invocation.
LockParams struct {
ID []byte
User util.Uint160
LockAccount util.Uint160
Amount int64 // in Fixed16
Until uint64 // epochs
}
// MintBurnParams for Mint and Burn invocations.
MintBurnParams struct {
ScriptHash []byte
Amount int64 // in Fixed16
Comment []byte
}
)
const (
transferXMethod = "transferX"
lockMethod = "lock"
mintMethod = "mint"
burnMethod = "burn"
precisionMethod = "decimals"
)
// Mint assets in contract.
func Mint(cli *client.Client, con util.Uint160, fee SideFeeProvider, p *MintBurnParams) error {
if cli == nil {
return client.ErrNilClient
}
if !cli.NotaryEnabled() {
return cli.Invoke(con, fee.SideChainFee(), mintMethod,
p.ScriptHash,
p.Amount,
p.Comment,
)
}
return cli.NotaryInvoke(con, mintMethod,
p.ScriptHash,
p.Amount,
p.Comment,
)
}
// Burn minted assets.
func Burn(cli *client.Client, con util.Uint160, fee SideFeeProvider, p *MintBurnParams) error {
if cli == nil {
return client.ErrNilClient
}
if !cli.NotaryEnabled() {
return cli.Invoke(con, fee.SideChainFee(), burnMethod,
p.ScriptHash,
p.Amount,
p.Comment,
)
}
return cli.NotaryInvoke(con, burnMethod,
p.ScriptHash,
p.Amount,
p.Comment,
)
}
// LockAsset invokes Lock method.
func LockAsset(cli *client.Client, con util.Uint160, fee SideFeeProvider, p *LockParams) error {
if cli == nil {
return client.ErrNilClient
}
if !cli.NotaryEnabled() {
return cli.Invoke(con, fee.SideChainFee(), lockMethod,
p.ID,
p.User.BytesBE(),
p.LockAccount.BytesBE(),
p.Amount,
int64(p.Until),
)
}
return cli.NotaryInvoke(con, lockMethod,
p.ID,
p.User.BytesBE(),
p.LockAccount.BytesBE(),
p.Amount,
int64(p.Until),
)
}
// BalancePrecision invokes Decimal method that returns precision of NEP-5 contract.
func BalancePrecision(cli *client.Client, con util.Uint160) (uint32, error) {
if cli == nil {
return 0, client.ErrNilClient
}
v, err := cli.TestInvoke(con, precisionMethod)
if err != nil {
return 0, err
}
precision, err := client.IntFromStackItem(v[0])
if err != nil {
return 0, err
}
return uint32(precision), nil
}