rDrama/files/routes/casino.py

from files.__main__ import app
from files.helpers.wrappers import *
from files.helpers.alerts import *
from files.helpers.get import *
from files.helpers.const import *
from files.helpers.wrappers import *
from files.helpers.slots import *
from files.helpers.lottery import *
from files.helpers.casino import *
from files.helpers.twentyone import *


@app.get("/casino")
@auth_required
def casino(v):
    if v.rehab: return render_template("casino/rehab.html", v=v)
    return render_template("casino.html", v=v)


@app.get("/lottershe")
@auth_required
def lottershe(v):
    if v.rehab: return render_template("casino/rehab.html", v=v)
    participants = get_users_participating_in_lottery()
    return render_template("lottery.html", v=v, participants=participants)



@app.get("/casino/<game>")
@auth_required
def casino_game_page(v, game):
    if v.rehab: return render_template("casino/rehab.html", v=v)
    
    feed = json.dumps(get_game_feed(game))
    leaderboard = json.dumps(get_game_leaderboard(game))

    return render_template(
        f"casino/{game}_screen.html",
        v=v,
        game=game,
        feed=feed,
        leaderboard=leaderboard
    )


@app.get("/casino/<game>/feed")
@limiter.limit("1/second;50/minute;600/hour;12000/day")
@auth_required
def casino_game_feed(v, game):
    feed = get_game_feed(game)
    return {"feed": feed}


@app.post("/casino/slots")
@limiter.limit("3/second;30/minute;600/hour;12000/day")
@auth_required
def pull_slots(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}

    try:
        wager = int(request.values.get("wager"))
    except:
        return {"error": "Invalid wager."}

    try:
        currency = request.values.get("currency")
    except:
        return {"error": "Invalid currency (expected 'dramacoin' or 'marseybux')."}

    if (currency == "dramacoin" and wager > v.coins) or (currency == "marseybux" and wager > v.procoins):
        return {"error": f"Not enough {currency} to make that bet."}

    success, game_state = casino_slot_pull(v, wager, currency)

    if success:
        return {"game_state": game_state, "gambler": {"coins": v.coins, "procoins": v.procoins}}
    else:
        return {"error": f"Wager must be more than 5 {currency}."}


@app.post("/casino/twentyone/deal")
@auth_required
def blackjack_deal_to_player(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}

    wager = int(request.values.get("wager"))
    currency = request.values.get("currency")
    create_new_game(v, wager, currency)
    state = dispatch_action(v, BlackjackAction.DEAL)
    feed = get_game_feed('blackjack')

    return {"success": True, "state": state, "feed": feed}


@app.post("/casino/twentyone/hit")
@auth_required
def blackjack_player_hit(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}
    
    try:
        state = dispatch_action(v, BlackjackAction.HIT)
        feed = get_game_feed('blackjack')
        return {"success": True, "state": state, "feed": feed, "gambler": {"coins": v.coins, "procoins": v.procoins}}
    except:
        return {"error": "Unable to hit."}


@app.post("/casino/twentyone/stay")
@auth_required
def blackjack_player_stay(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}

    try:
        state = dispatch_action(v, BlackjackAction.STAY)
        feed = get_game_feed('blackjack')
        return {"success": True, "state": state, "feed": feed, "gambler": {"coins": v.coins, "procoins": v.procoins}}
    except:
        return {"error": "Unable to stay."}


@app.post("/casino/twentyone/double-down")
@auth_required
def blackjack_player_doubled_down(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}

    try:
        state = dispatch_action(v, BlackjackAction.DOUBLE_DOWN)
        feed = get_game_feed('blackjack')
        return {"success": True, "state": state, "feed": feed, "gambler": {"coins": v.coins, "procoins": v.procoins}}
    except:
        return {"error": "Unable to double down."}


@app.post("/casino/twentyone/buy-insurance")
@auth_required
def blackjack_player_bought_insurance(v):
    if v.rehab: return {"error": "You are under Rehab award effect!"}

    try:
        state = dispatch_action(v, BlackjackAction.BUY_INSURANCE)
        feed = get_game_feed('blackjack')
        return {"success": True, "state": state, "feed": feed, "gambler": {"coins": v.coins, "procoins": v.procoins}}
    except:
        return {"error": "Unable to buy insurance."}