65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
|
import threading
|
||
|
import traceback
|
||
|
import queue
|
||
|
|
||
|
import structlog
|
||
|
|
||
|
from butterrobot.db import ChannelQuery
|
||
|
from butterrobot.platforms import get_available_platforms
|
||
|
from butterrobot.platforms.base import Platform
|
||
|
from butterrobot.plugins import get_available_plugins
|
||
|
|
||
|
logger = structlog.get_logger(__name__)
|
||
|
q = queue.Queue()
|
||
|
|
||
|
|
||
|
def handle_message(platform: str, request: dict):
|
||
|
try:
|
||
|
message = get_available_platforms()[platform].parse_incoming_message(
|
||
|
request=request
|
||
|
)
|
||
|
except Platform.PlatformAuthResponse as response:
|
||
|
return response.data, response.status_code
|
||
|
except Exception as error:
|
||
|
logger.error(
|
||
|
"Error parsing message",
|
||
|
platform=platform,
|
||
|
error=error,
|
||
|
traceback=traceback.format_exc(),
|
||
|
)
|
||
|
return
|
||
|
|
||
|
if not message or message.from_bot:
|
||
|
return
|
||
|
|
||
|
try:
|
||
|
channel = ChannelQuery.get_by_platform(platform, message.chat)
|
||
|
except ChannelQuery.NotFound:
|
||
|
# If channel is still not present on the database, create it (defaults to disabled)
|
||
|
channel = ChannelQuery.create(
|
||
|
platform, message.chat, channel_raw=message.channel.channel_raw
|
||
|
)
|
||
|
|
||
|
if not channel.enabled:
|
||
|
return
|
||
|
|
||
|
for plugin_id, channel_plugin in channel.plugins.items():
|
||
|
if not channel.has_enabled_plugin(plugin_id):
|
||
|
continue
|
||
|
|
||
|
for response_message in get_available_plugins()[plugin_id].on_message(
|
||
|
message, plugin_config=channel_plugin.config
|
||
|
):
|
||
|
get_available_platforms()[platform].methods.send_message(response_message)
|
||
|
|
||
|
|
||
|
def worker_thread():
|
||
|
while True:
|
||
|
item = q.get()
|
||
|
handle_message(item["platform"], item["request"])
|
||
|
q.task_done()
|
||
|
|
||
|
|
||
|
# turn-on the worker thread
|
||
|
worker = threading.Thread(target=worker_thread, daemon=True).start()
|