import json
import os
import asyncio
from bot import Bot
from log import getlogger

logger = getlogger()

async def main():
    
    if os.path.exists('config.json'):
        fp = open('config.json', 'r', encoding="utf8")
        config = json.load(fp)
        
        matrix_bot = Bot(homeserver=config.get('homeserver'),
                        user_id=config.get('user_id') ,
                        password=config.get('password'),
                        device_id=config.get('device_id'),
                        room_id=config.get('room_id'),
                        api_key=config.get('api_key'),
                        bing_api_endpoint=config.get('bing_api_endpoint'),
                        access_token=config.get('access_token'),
                        jailbreakEnabled=config.get('jailbreakEnabled'),
                        bing_auth_cookie=config.get('bing_auth_cookie'),
                        )                     

    else:
        matrix_bot = Bot(homeserver=os.environ.get('HOMESERVER'),
                        user_id=os.environ.get('USER_ID') ,
                        password=os.environ.get('PASSWORD'),
                        device_id=os.environ.get("DEVICE_ID"),
                        room_id=os.environ.get("ROOM_ID"),
                        api_key=os.environ.get("OPENAI_API_KEY"),
                        bing_api_endpoint=os.environ.get("BING_API_ENDPOINT"),
                        access_token=os.environ.get("ACCESS_TOKEN"),
                        jailbreakEnabled=os.environ.get("JAILBREAKENABLED"),
                        bing_auth_cookie=os.environ.get("BING_AUTH_COOKIE"),
                        )     

    await matrix_bot.login()
    await matrix_bot.sync_encryption_key()

    # await matrix_bot.trust_own_devices()

    await matrix_bot.sync_forever(timeout=30000, full_state=True)


if __name__ == "__main__":
    logger.info("matrix chatgpt bot start.....")
    asyncio.run(main())