import asyncio
from aiohttp import web
from damination import DaminationAPI


routes = web.RouteTableDef()
toonat = DaminationAPI()
app = web.Application()


@routes.get('/subscribe/{widget_id}/{url}')
async def subscribe(request):
    widget_id = request.match_info['widget_id']
    url = request.match_info['url']
    success = toonat.add_subscriber(widget_id, url)
    if success:
        return web.Response(text=f"Subscribed to widget ID {widget_id} with URL {url}.")
    else:
        return web.Response(text=f"Already subscribed to widget ID {widget_id} with URL {url}.")

@routes.get('/unsubscribe/{widget_id}/{url}')
async def unsubscribe(request):
    widget_id = request.match_info['widget_id']
    url = request.match_info['url']
    success = toonat.remove_subscriber(widget_id, url)
    if success:
        return web.Response(text=f"Unsubscribed from widget ID {widget_id} with URL {url}.")
    else:
        return web.Response(text=f"Not subscribed to widget ID {widget_id} with URL {url}.")

@routes.get('/init_widget_id')
async def init_widget_id(request):
    if not toonat.WIDGET_IDS:
        widget_id = request.query.get("widget_id")
        if widget_id:
            toonat.WIDGET_IDS.append(widget_id)
            return web.Response(text=f"Widget ID {widget_id} added.")
        else:
            return web.Response(text="Invalid widget ID. Please try again.")
    else:
        return web.Response(text="Widget ID already exists.")


@routes.get('/get_widgets')
async def get_widgets(request):
    """
    현재 설정된 모든 위젯 ID 목록을 반환하는 웹 API 엔드포인트입니다.
    """
    widget_ids = toonat.get_widget_ids()
    return web.json_response(widget_ids)


@routes.get('/add_widget/{widget_id}')
async def add_widget(request):
    """
    주어진 widget_id를 위젯 ID 목록에 추가하고 알림을 가져오는 작업을 시작하는 웹 API 엔드포인트입니다.
    widget_id: 추가할 위젯 ID
    """
    widget_id = request.match_info['widget_id']
    await toonat.add_widget_id_and_start_fetching(widget_id)
    return web.Response(text=f"Widget ID {widget_id} added and fetching started.")

@routes.get('/remove_widget/{widget_id}')
async def remove_widget(request):
    """
    주어진 widget_id를 위젯 ID 목록에서 제거하고 알림을 가져오는 작업을 중단하는 웹 API 엔드포인트입니다.
    widget_id: 제거할 위젯 ID
    """
    widget_id = request.match_info['widget_id']
    await toonat.remove_widget_id_and_stop_fetching(widget_id)
    return web.Response(text=f"Widget ID {widget_id} removed and fetching stopped.")

@routes.get('/status')
async def status(request):
    """
    서버 상태를 확인하는 웹 API 엔드포인트입니다. 이 엔드포인트는 서버가 정상적으로 작동 중임을 확인하는 데 사용됩니다.
    """
    return web.HTTPOk()


async def web_server(app):
    """
    aiohttp 웹 서버를 설정하고 시작하는 비동기 함수입니다.
    """
    app.add_routes(routes)
    runner = web.AppRunner(app)
    await runner.setup()
    site = web.TCPSite(runner, '0.0.0.0', 80)
    await site.start()


async def main():

    # DaminationAPI를 실행합니다.
    api_task = asyncio.create_task(toonat.run())

    # 웹 서버를 실행합니다.
    web_server_task = asyncio.create_task(web_server(app))

    # 두 작업을 병렬로 실행하지 않고 순서대로 처리합니다.
    await api_task
    await web_server_task



if __name__ == "__main__":
    if not toonat.WIDGET_IDS:
        print("No widget ID found in settings.ini.")
        print("Please enter the widget ID via web endpoint (http://localhost/init_widget_id?widget_id=YOUR_WIDGET_ID)")

    asyncio.run(main())