OwnchatBot/ownchatbot/webhooks.py

from flask import Flask, request, json, Blueprint, current_app, render_template
from ownchatbot.db import get_db
from ownchatbot.owncast_com import send_chat
from ownchatbot.user_handlers import add_user_to_points, change_name, get_users_points, remove_duplicates
from ownchatbot.bot_messages import do_reward, help_message
from ownchatbot.reward_handlers import all_active_goals, all_active_votes


ocb = Blueprint('webhooks', __name__)


@ocb.route('/chatHook', methods=['POST'])
def chat_hook():
    prefix = current_app.config['PREFIX']
    data = request.json
    db = get_db()

    if data['type'] in ['CHAT', 'NAME_CHANGED', 'USER_JOINED']:  # Check if the viewer is in the chatbot database
        user_id = data['eventData']['user']['id']
        display_name = data['eventData']['user']['displayName']
        add_user_to_points(db, user_id, display_name)

    if data['type'] == 'USER_JOINED':  # Do username house cleaning when a viewer joins
        if data['eventData']['user']['authenticated']:
            remove_duplicates(db, user_id, display_name)
    elif data['type'] == 'NAME_CHANGE':
        user_id = data['eventData']['user']['id']
        new_name = data['eventData']['newName']
        change_name(db, user_id, new_name)
        if data['eventData']['user']['authenticated']:
            remove_duplicates(db, user_id, new_name)
    elif data['type'] == 'CHAT':  # If a chat message, sort out what command it is
        user_id = data['eventData']['user']['id']
        display_name = data['eventData']['user']['displayName']
        current_app.logger.debug(f'Chat message from {display_name}:')
        current_app.logger.debug(f'{data["eventData"]["rawBody"]}')
        
        lowercase_msg = data['eventData']['rawBody'].lower()  # Convert body to lower case to match reward case
        if lowercase_msg.startswith(f'{prefix}help'):  # Send the help message
            help_message()
            
        elif lowercase_msg.startswith(f'{prefix}points'):  # Get the viewer's current points
            points = get_users_points(db, user_id)
            if points is None:
                send_chat('Couldn\'t get your points, for some highly technical reason.')
            else:
                send_chat(f'{display_name}, you have {points} points.')
                
        elif lowercase_msg.startswith(f'{prefix}rewards'):  # Send rewards list
            if current_app.config['REWARDS']:
                rewards_msg = f'Currently active rewards:'
                for reward, details in current_app.config['REWARDS'].items():
                    if details.get('categories'):
                        if not (set(details['categories']) & set(current_app.config['ACTIVE_CAT'])):  # If there are no common categories, continue
                            continue
                    if 'type' in details and details['type'] == 'goal':
                        rewards_msg = f'{rewards_msg}<br>* {prefix}{reward} goal at {details["target"]} contributed points.'
                    else:
                        rewards_msg = f'{rewards_msg}<br>* {prefix}{reward} for {details["price"]} points.'
                    if 'info' in details:
                        rewards_msg = f'{rewards_msg}<br>{details["info"]}'
                    else:
                        rewards_msg = f'{rewards_msg}'
            else:
                rewards_msg = 'There are currently no active rewards.'
            send_chat(rewards_msg)
                    
        elif lowercase_msg.startswith(f'{prefix}'):  # Send to handle rewards
            do_reward(lowercase_msg, user_id)

    return data


@ocb.route('/goals', methods=['GET'])  # Route for goals overlay
def goals():
    db = get_db()
    return render_template('goals.html',
                           goals=all_active_goals(db))


@ocb.route('/votes', methods=['GET'])  # Route for votes overlay
def votes():
    db = get_db()
    return render_template('votes.html',
                           votes=all_active_votes(db))