Tlapbot/tlapbot/redeems_handler.py

from flask import current_app
from tlapbot.db import get_db
from tlapbot.owncast_requests import send_chat
from tlapbot.owncast_helpers import (use_points, add_to_redeem_queue,
        add_to_counter, read_users_points, remove_emoji,
        add_to_milestone, check_apply_milestone_completion, milestone_complete)


def handle_redeem(message, user_id):
    split_message = message[1:].split(maxsplit=1)
    redeem = split_message[0]
    if len(split_message) == 1:
        note = None
    else:
        note = split_message[1]

    if redeem not in current_app.config['REDEEMS']:
        send_chat("Can't redeem, redeem not found.")
        return

    db = get_db()
    price = current_app.config['REDEEMS'][redeem]["price"]
    redeem_type = current_app.config['REDEEMS'][redeem]["type"]
    points = read_users_points(db, user_id)

    if not points or points < price:
        send_chat(f"Can't redeem {redeem}, you don't have enough points.")
        return

    if redeem_type == "counter":
        if add_to_counter(db, redeem) and use_points(db, user_id, price):
            send_chat(f"{redeem} redeemed for {price} points.")
        else:
            send_chat(f"Redeeming {redeem} failed.")
    elif redeem_type == "list":
        if (add_to_redeem_queue(db, user_id, redeem) and
                use_points(db, user_id, price)):
            send_chat(f"{redeem} redeemed for {price} points.")
        else:
            send_chat(f"Redeeming {redeem} failed.")
    elif redeem_type == "note":
        if not note:
            send_chat(f"Cannot redeem {redeem}, no note included.")
            return
        if (add_to_redeem_queue(db, user_id, redeem, remove_emoji(note)) and
                use_points(db, user_id, price)):
            send_chat(f"{redeem} redeemed for {price} points.")
        else:
            send_chat(f"Redeeming {redeem} failed.")
    elif redeem_type == "milestone":
        if milestone_complete(db, redeem):
            send_chat(f"Can't redeem {redeem}, that milestone was already completed!")
        elif not note:
            send_chat(f"Cannot redeem {redeem}, no amount of points specified.")
        elif not note.isdigit():
            send_chat(f"Cannot redeem {redeem}, amount of points is not an integer.")
        elif int(note) > points:
            send_chat(f"Can't redeem {redeem}, you don't have enough points.")
        elif add_to_milestone(db, user_id, redeem, int(note)):
            send_chat(f"Succesfully donated to {redeem} milestone!")
            if check_apply_milestone_completion(db, redeem):
                send_chat(f"Milestone goal {redeem} complete!")
    else:
        send_chat(f"{redeem} not redeemed, type of redeem not found.")