76 lines
3.6 KiB
Python
76 lines
3.6 KiB
Python
from flask import Flask, request, json, Blueprint, current_app
|
|
from datetime import datetime
|
|
from tlapbot.db import get_db, refresh_counters, clear_redeem_queue
|
|
from tlapbot.owncast_requests import send_chat
|
|
from tlapbot.owncast_helpers import (add_user_to_database, change_display_name,
|
|
read_users_points, remove_duplicate_usernames, get_last_online_time, delete_last_online_time)
|
|
from tlapbot.help_message import send_help
|
|
from tlapbot.redeems_handler import handle_redeem
|
|
|
|
# might need datetime timestamp
|
|
|
|
bp = Blueprint('owncast_webhooks', __name__)
|
|
|
|
|
|
@bp.route('/owncastWebhook', methods=['POST'])
|
|
def owncast_webhook():
|
|
data = request.json
|
|
db = get_db()
|
|
|
|
if data["type"] == "STREAM_STARTED":
|
|
# TODO: make this a function, import here and in init
|
|
delete_last_online_time(db)
|
|
last_online = get_last_online_time(db)
|
|
if last_online and current_app.config['AUTO_REFRESH']:
|
|
time_difference = datetime.now() - last_online
|
|
if time_difference.seconds//60 > current_app.config['RECONNECT_TIME']:
|
|
if refresh_counters() and clear_redeem_queue():
|
|
current_app.logger.debug(f'Counters refreshed, redeem queue cleared.')
|
|
else:
|
|
current_app.logger.error(
|
|
f'Error occured when automatically clearing queue and resetting counters.'
|
|
)
|
|
elif data["type"] == "STREAM_STOPPED":
|
|
save_last_online_time(db, datetime.now(), True)
|
|
|
|
# Make sure user is in db before doing anything else.
|
|
if data["type"] in ["CHAT", "NAME_CHANGED", "USER_JOINED"]:
|
|
user_id = data["eventData"]["user"]["id"]
|
|
display_name = data["eventData"]["user"]["displayName"]
|
|
add_user_to_database(db, user_id, display_name)
|
|
|
|
if data["type"] == "USER_JOINED":
|
|
if data["eventData"]["user"]["authenticated"]:
|
|
remove_duplicate_usernames(db, user_id, display_name)
|
|
elif data["type"] == "NAME_CHANGE":
|
|
user_id = data["eventData"]["user"]["id"]
|
|
new_name = data["eventData"]["newName"]
|
|
change_display_name(db, user_id, new_name)
|
|
if data["eventData"]["user"]["authenticated"]:
|
|
remove_duplicate_usernames(db, user_id, new_name)
|
|
elif data["type"] == "CHAT":
|
|
if not current_app.config['PASSIVE']:
|
|
prefix = current_app.config['PREFIX']
|
|
user_id = data["eventData"]["user"]["id"]
|
|
display_name = data["eventData"]["user"]["displayName"]
|
|
current_app.logger.debug(f'New chat message from {display_name}:')
|
|
current_app.logger.debug(f'{data["eventData"]["body"]}')
|
|
if data["eventData"]["body"].startswith(f"{prefix}help"):
|
|
send_help()
|
|
elif data["eventData"]["body"].startswith(f"{prefix}points"):
|
|
points = read_users_points(db, user_id)
|
|
if points is None:
|
|
send_chat("Error reading points.")
|
|
else:
|
|
send_chat(f"{display_name}'s points: {points}")
|
|
elif data["eventData"]["body"].startswith(f"{prefix}name_update"):
|
|
# Forces name update in case bot didn't catch the NAME_CHANGE
|
|
# event. Also removes saved usernames from users with same name
|
|
# if user is authenticated.
|
|
change_display_name(db, user_id, display_name)
|
|
if data["eventData"]["user"]["authenticated"]:
|
|
remove_duplicate_usernames(db, user_id, display_name)
|
|
elif data["eventData"]["body"].startswith(prefix):
|
|
handle_redeem(data["eventData"]["body"], user_id)
|
|
return data
|