import asyncio
from typing import TypedDict
from aiohttp import web
import asyncpg
import config
import api.route
import utils.web
from service.database import DatabaseService
from service.mediawiki_api import MediaWikiApi
from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker

from service.tiktoken import TikTokenService

async def index(request: web.Request):
    return utils.web.api_response(1, data={"message": "Isekai toolkit API"}, request=request)

async def init_mw_api(app: web.Application):
    mw_api = MediaWikiApi.create()
    if config.MW_BOT_LOGIN_USERNAME and config.MW_BOT_LOGIN_PASSWORD:
        await mw_api.robot_login(config.MW_BOT_LOGIN_USERNAME, config.MW_BOT_LOGIN_PASSWORD)

    site_meta = await mw_api.get_site_meta()

    print("Connected to Wiki %s, Robot username: %s" % (site_meta["sitename"], site_meta["user"]))

async def init_database(app: web.Application):
    dbs = await DatabaseService.create(app)
    print("Database connected.")

async def init_tiktoken(app: web.Application):
    await TikTokenService.create()
    print("Tiktoken model loaded.")

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    
    app = web.Application()

    if config.DATABASE:
        app.on_startup.append(init_database)

    if config.MW_API:
        app.on_startup.append(init_mw_api)

    if config.OPENAI_TOKEN:
        app.on_startup.append(init_tiktoken)
    
    app.router.add_route('*', '/', index)
    api.route.init(app)
    web.run_app(app, host='0.0.0.0', port=config.PORT, loop=loop)