mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-28 23:27:36 +00:00
5f13de3a76
If we need a positive number we can do `IsUint64()` instead of checking that `Int64()` is `> 0`.
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package state
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"math/big"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
)
|
|
|
|
// Deposit represents GAS deposit from Notary contract.
|
|
type Deposit struct {
|
|
Amount *big.Int
|
|
Till uint32
|
|
}
|
|
|
|
// ToStackItem implements stackitem.Convertible interface. It never returns an
|
|
// error.
|
|
func (d *Deposit) ToStackItem() (stackitem.Item, error) {
|
|
return stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.NewBigInteger(d.Amount),
|
|
stackitem.Make(d.Till),
|
|
}), nil
|
|
}
|
|
|
|
// FromStackItem implements stackitem.Convertible interface.
|
|
func (d *Deposit) FromStackItem(it stackitem.Item) error {
|
|
items, ok := it.Value().([]stackitem.Item)
|
|
if !ok {
|
|
return errors.New("not a struct")
|
|
}
|
|
if len(items) != 2 {
|
|
return errors.New("wrong number of elements")
|
|
}
|
|
amount, err := items[0].TryInteger()
|
|
if err != nil {
|
|
return fmt.Errorf("invalid amount: %w", err)
|
|
}
|
|
till, err := items[1].TryInteger()
|
|
if err != nil {
|
|
return fmt.Errorf("invalid till: %w", err)
|
|
}
|
|
tiu64 := till.Uint64()
|
|
if !till.IsUint64() || tiu64 > math.MaxUint32 {
|
|
return errors.New("wrong till value")
|
|
}
|
|
d.Amount = amount
|
|
d.Till = uint32(tiu64)
|
|
return nil
|
|
}
|