diff --git a/app/main.py b/app/main.py new file mode 100644 index 0000000..5a5f2df --- /dev/null +++ b/app/main.py @@ -0,0 +1,112 @@ + +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())