From 57269ee77f119a0dfc99a6eb8c353e950e071f89 Mon Sep 17 00:00:00 2001
From: Daniil Kurepin <daniil.kurepin@gmail.com>
Date: Tue, 12 Dec 2023 17:50:57 +0300
Subject: [PATCH] feat: add roulette

---
 Roulette/Roulette.go | 97 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 Roulette/Roulette.go

diff --git a/Roulette/Roulette.go b/Roulette/Roulette.go
new file mode 100644
index 0000000..196936c
--- /dev/null
+++ b/Roulette/Roulette.go
@@ -0,0 +1,97 @@
+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 (
+	gasDecimals    = 1_0000_0000
+	initialBalance = 3000
+	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
+	isWin := isWinner(selectedNumber)
+	if isWin {
+		// Calculate win amount based on the selectedNumber
+		winAmount := calculateWinAmount(bet, selectedNumber)
+		changePlayerBalance(ctx, playerOwner, winAmount)
+	} else {
+		changePlayerBalance(ctx, playerOwner, -bet)
+	}
+}
+
+func isWinner(selectedNumber int) bool {
+	rouletteNumber := (runtime.GetRandom() % 36) + 1
+	runtime.Notify("Roulette number:", rouletteNumber)
+
+	return rouletteNumber == selectedNumber
+}
+
+func calculateWinAmount(bet int, selectedNumber int) int {
+	coefficients := map[int]int{
+		1:  36,
+		2:  18,
+		3:  2,
+	}
+
+	if coefficient, ok := coefficients[selectedNumber]; ok {
+		return bet * coefficient
+	}
+
+	return 0
+}
+
+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(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 {
+		from = playerContract
+		to = playerOwner
+		transferAmount = balanceChange
+	} else {
+		from = playerOwner
+		to = playerContract
+		transferAmount = -balanceChange
+	}
+
+	transferred := contract.Call(zaCoinHash, "transfer", contract.All, from, to, transferAmount, nil).(bool)
+	if !transferred {
+		panic("failed to transfer zaCoins")
+	}
+}