103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
package Exchanger
|
|
|
|
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/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/gas"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/lib/address"
|
|
//"github.com/nspcc-dev/neo-go/pkg/interop/util"
|
|
)
|
|
|
|
const (
|
|
zaCoinHashKey = "zaCoinHash"
|
|
)
|
|
|
|
var walletAddress = address.ToHash160("NXbLSnHA8dNuMUPUSNNivx7XFucN1w5bRq")
|
|
|
|
func _deploy(data interface{}, isUpdate bool) {
|
|
if isUpdate {
|
|
return
|
|
}
|
|
|
|
args := data.(struct {
|
|
zaCoinHash interop.Hash160
|
|
})
|
|
|
|
if len(args.zaCoinHash) != interop.Hash160Len {
|
|
panic("Invalid hash of zaCoin contract")
|
|
}
|
|
|
|
ctx := storage.GetContext()
|
|
storage.Put(ctx, zaCoinHashKey, args.zaCoinHash)
|
|
}
|
|
|
|
func BuyZaCoin(count int) {
|
|
ctx := storage.GetContext()
|
|
playerOwner := runtime.GetScriptContainer().Sender
|
|
|
|
//balanceBefore := gas.BalanceOf(runtime.GetExecutingScriptHash())
|
|
gasTransfer(playerOwner, count)
|
|
//balanceAfter := gas.BalanceOf(runtime.GetExecutingScriptHash())
|
|
|
|
changePlayerBalance(ctx, playerOwner, count)
|
|
//if (balanceBefore - balanceAfter == count) {
|
|
// changePlayerBalance(ctx, playerOwner, count)
|
|
//} else {
|
|
// util.Abort()
|
|
//}
|
|
}
|
|
|
|
func OnNEP17Payment(from interop.Hash160, amount int, data any) {
|
|
zacoinStr := "zacoin"
|
|
ctx := storage.GetContext()
|
|
zaCoinHash := storage.Get(ctx, zaCoinHashKey).(interop.Hash160)
|
|
|
|
callingHash := runtime.GetCallingScriptHash()
|
|
if data == zacoinStr {
|
|
if !callingHash.Equals(zaCoinHash) {
|
|
panic("only ZC is accepted")
|
|
}
|
|
} else {
|
|
if !callingHash.Equals(gas.Hash) {
|
|
panic("only GAS is accepted")
|
|
}
|
|
}
|
|
}
|
|
|
|
func gasTransfer(playerOwner interop.Hash160, gasCount int) {
|
|
contractHash := runtime.GetExecutingScriptHash()
|
|
|
|
|
|
transferredGas := gas.Transfer(playerOwner, contractHash, gasCount, nil)
|
|
|
|
if !transferredGas {
|
|
panic("Failed to transfer gas")
|
|
}
|
|
}
|
|
|
|
|
|
func changePlayerBalance(ctx storage.Context, playerOwner interop.Hash160, balanceChange int) {
|
|
zaCoinHash := storage.Get(ctx, zaCoinHashKey).(interop.Hash160)
|
|
playerContract := runtime.GetExecutingScriptHash()
|
|
|
|
var from, to interop.Hash160
|
|
var transferAmount int
|
|
if balanceChange > 0 {
|
|
// Transfer funds from contract to player owner
|
|
from = playerContract
|
|
to = playerOwner
|
|
transferAmount = balanceChange
|
|
} else {
|
|
// Transfer funds from player owner to contract
|
|
from = playerOwner
|
|
to = playerContract
|
|
transferAmount = -balanceChange // We flip sender/receiver, but keep amount positive
|
|
}
|
|
|
|
transferred := contract.Call(zaCoinHash, "transfer", contract.All, from, to, transferAmount, "zacoin").(bool)
|
|
if !transferred {
|
|
panic("failed to transfer zaCoins")
|
|
}
|
|
}
|