import os

import simplematrixbotlib as botlib

from matrix_goal_bot.bot import get_bot, goal_bot
from matrix_chore_bot.bot import chore_bot

SERVER = os.environ.get('MATRIX_SERVER')
BOT_NAME = os.environ.get('MATRIX_BOT')
BOT_PWD = os.environ.get('MATRIX_BOT_PWD')
DATA_DIR = os.environ.get('MATRIX_DATA_DIR')

creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD)
bot = botlib.Bot(creds)


@bot.listener.on_message_event
async def do_goal_stuff(room, message):
    await goal_bot(room, message, bot, DATA_DIR)


@bot.listener.on_message_event
async def do_chore_stuff(room, message):
    await chore_bot(room, message, bot, DATA_DIR)


bot.run()