forked from TrueCloudLab/neoneo-go
c2b3ee3d8e
This allows to reuse it across different packages. testchain can't be used because of circular dependencies. Init() is not changed except for filepath.Join() use instead of direct string appends which is a better approach anyway. rootpath is required because current directory will change from package to package.
121 lines
2.6 KiB
Go
121 lines
2.6 KiB
Go
package testdata
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/contract"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/ledger"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/management"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/neo"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/util"
|
|
)
|
|
|
|
const (
|
|
totalSupply = 1000000
|
|
decimals = 2
|
|
)
|
|
|
|
var owner = util.FromAddress("NbrUYaZgyhSkNoRo9ugRyEMdUZxrhkNaWB")
|
|
|
|
func Init() bool {
|
|
ctx := storage.GetContext()
|
|
h := runtime.GetExecutingScriptHash()
|
|
amount := totalSupply
|
|
storage.Put(ctx, h, amount)
|
|
runtime.Notify("Transfer", interop.Hash160([]byte{}), h, amount)
|
|
return true
|
|
}
|
|
|
|
func OnNEP17Payment(from interop.Hash160, amount int, data interface{}) {
|
|
curr := runtime.GetExecutingScriptHash()
|
|
balance := neo.BalanceOf(curr)
|
|
if ledger.CurrentIndex() >= 100 {
|
|
ok := neo.Transfer(curr, owner, balance, nil)
|
|
if !ok {
|
|
panic("owner transfer failed")
|
|
}
|
|
ok = neo.Transfer(curr, owner, 0, nil)
|
|
if !ok {
|
|
panic("owner transfer failed")
|
|
}
|
|
}
|
|
}
|
|
|
|
// Verify always returns true and is aimed to serve the TestNEO_RecursiveGASMint.
|
|
func Verify() bool {
|
|
return true
|
|
}
|
|
|
|
func Transfer(from, to interop.Hash160, amount int, data interface{}) bool {
|
|
ctx := storage.GetContext()
|
|
if len(from) != 20 {
|
|
runtime.Log("invalid 'from' address")
|
|
return false
|
|
}
|
|
if len(to) != 20 {
|
|
runtime.Log("invalid 'to' address")
|
|
return false
|
|
}
|
|
if amount < 0 {
|
|
runtime.Log("invalid amount")
|
|
return false
|
|
}
|
|
|
|
var fromBalance int
|
|
val := storage.Get(ctx, from)
|
|
if val != nil {
|
|
fromBalance = val.(int)
|
|
}
|
|
if fromBalance < amount {
|
|
runtime.Log("insufficient funds")
|
|
return false
|
|
}
|
|
fromBalance -= amount
|
|
storage.Put(ctx, from, fromBalance)
|
|
|
|
var toBalance int
|
|
val = storage.Get(ctx, to)
|
|
if val != nil {
|
|
toBalance = val.(int)
|
|
}
|
|
toBalance += amount
|
|
storage.Put(ctx, to, toBalance)
|
|
|
|
runtime.Notify("Transfer", from, to, amount)
|
|
if management.GetContract(to) != nil {
|
|
contract.Call(to, "onNEP17Payment", contract.All, from, amount, data)
|
|
}
|
|
return true
|
|
}
|
|
|
|
func BalanceOf(account interop.Hash160) int {
|
|
ctx := storage.GetContext()
|
|
if len(account) != 20 {
|
|
panic("invalid address")
|
|
}
|
|
var amount int
|
|
val := storage.Get(ctx, account)
|
|
if val != nil {
|
|
amount = val.(int)
|
|
}
|
|
return amount
|
|
}
|
|
|
|
func Symbol() string {
|
|
return "RUB"
|
|
}
|
|
|
|
func Decimals() int {
|
|
return decimals
|
|
}
|
|
|
|
func TotalSupply() int {
|
|
return totalSupply
|
|
}
|
|
|
|
func PutValue(key []byte, value []byte) bool {
|
|
ctx := storage.GetContext()
|
|
storage.Put(ctx, key, value)
|
|
return true
|
|
}
|