auction/Roulette/Roulette.go

100 lines
2.8 KiB
Go

package Roulette
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"
)
const (
zaCoinHashKey = "zaCoinHash"
)
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 PlayRoulette(bet int, selectedNumber int) {
ctx := storage.GetContext()
playerOwner := runtime.GetScriptContainer().Sender
playerContract := runtime.GetExecutingScriptHash()
if bet <= 0 {
panic("Invalid bet amount")
}
zaCoinHash := storage.Get(ctx, zaCoinHashKey).(interop.Hash160)
playerBalance := contract.Call(zaCoinHash, "balanceOf", contract.ReadStates, playerOwner).(int)
if playerBalance < bet {
panic("Insufficient funds")
}
if selectedNumber < 1 || selectedNumber > 36 {
panic("Illegal number selected for roulette")
}
isWin := isWinner(selectedNumber)
if isWin {
winAmount := calculateWinAmount(bet, selectedNumber)
changePlayerBalance(playerContract, playerOwner, winAmount)
runtime.Notify("gameResult", int(1))
} else {
changePlayerBalance(playerOwner, playerContract, bet)
runtime.Notify("gameResult", int(0))
}
playerBalance = contract.Call(zaCoinHash, "balanceOf", contract.ReadStates, playerOwner).(int)
runtime.Notify("playerBalance", playerBalance)
}
func isWinner(selectedNumber int) bool {
rouletteNumber := (runtime.GetRandom() % 36) + 1
runtime.Notify("rouletteNumber", rouletteNumber)
runtime.Log("rouletteNumber " + string(rouletteNumber))
return rouletteNumber == selectedNumber
}
func calculateWinAmount(bet int, selectedNumber int) int {
coefs := []int{10, 20, 30, 2}
if selectedNumber == 36 {
return bet * coefs[0]
} else if selectedNumber == 18 {
return bet * coefs[1]
} else if selectedNumber == 2 {
return bet * coefs[2]
} else {
return bet * coefs[3]
}
}
func OnNEP17Payment(from interop.Hash160, amount int, data any) {
ctx := storage.GetContext()
zaCoinHash := storage.Get(ctx, zaCoinHashKey).(interop.Hash160)
callingHash := runtime.GetCallingScriptHash()
if !callingHash.Equals(zaCoinHash) {
panic("only ZC is accepted")
}
}
func changePlayerBalance(sender interop.Hash160, recipient interop.Hash160, balanceChange int) {
ctx := storage.GetContext()
zaCoinHash := storage.Get(ctx, zaCoinHashKey).(interop.Hash160)
transferred := contract.Call(zaCoinHash, "transfer", contract.All, sender, recipient, balanceChange, nil).(bool)
if !transferred {
panic("failed to transfer zaCoins")
}
}