main.py

import asyncio
import importlib
import os
import sys
import traceback

import aiohttp
from aiohttp import web
import cachetools
from gidgethub import aiohttp as gh_aiohttp
from gidgethub import routing
from gidgethub import sansio, apps

from routers import issue, stage, commands
from dotenv import load_dotenv

router = routing.Router(
    issue.router,
    stage.router,
    commands.router
)
cache = cachetools.LRUCache(maxsize=500)
load_dotenv()

async def main(request):
    try:
        body = await request.read()
        secret = os.getenv("WEBHOOK_SECRET")
        event = sansio.Event.from_http(request.headers, body, secret=secret)
        print('GH delivery ID', event.delivery_id, file=sys.stderr)
        
        if event.event == "ping":
            return web.Response(status=200)

        repo_full_name = event.data.get("repository", {}).get("full_name", "")
        if not repo_full_name.startswith("HytaleModding/"):
            print(f"Ignoring event from repository: {repo_full_name}")
            return web.Response(status=200)

        async with aiohttp.ClientSession() as session:
            gh = gh_aiohttp.GitHubAPI(session, "HytaleModding-Bot/1.0", cache=cache)
            
            if not event.data.get("installation"):
                return web.Response(text="Must be installed as an App.", status=400)
            
            installation_id = event.data["installation"]["id"]
            installation_access_token = await apps.get_installation_access_token(
                gh,
                installation_id=installation_id,
                app_id=os.getenv("GH_APP_ID"),
                private_key=os.getenv("GH_PRIVATE_KEY"),
            )
            gh.oauth_token = installation_access_token["token"]
            
            await asyncio.sleep(1)
            await router.dispatch(event, gh, session=session)
        
        try:
            print('GH requests remaining:', gh.rate_limit.remaining)
        except AttributeError:
            pass
        return web.Response(status=200)
    except Exception as exc:
        traceback.print_exc(file=sys.stderr)
        return web.Response(status=500)


if __name__ == "__main__":
    app = web.Application()
    app.router.add_post("/", main)
    port = int(os.getenv("PORT", 25574))
    
    print(f"Starting bot on port {port}")
    web.run_app(app, port=port, host="0.0.0.0")