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}
* {prefix}{reward} goal at {details["target"]} contributed points.' else: rewards_msg = f'{rewards_msg}
* {prefix}{reward} for {details["price"]} points.' if 'info' in details: rewards_msg = f'{rewards_msg}
{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))