initial commit

hooksy/app.py

@@ -0,0 +1,80 @@
+import datetime
+import importlib
+import json
+import logging
+import typing
+import uuid
+from typing import Any
+
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from starlette import status
+import sentry_sdk
+from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
+import seqlog
+
+app = FastAPI()
+
+
+class JsonDateEncoder(json.JSONEncoder):
+    def default(self, o: Any) -> Any:
+        if isinstance(o, datetime.datetime):
+            return o.isoformat()
+        return super().default(o)
+
+
+seqlog.log_to_seq(
+    server_url="http://log.abdus.dev:5341/",
+    api_key="ePewcJUxe8dST5EiC8Pw",
+    level=logging.INFO,
+    batch_size=10,
+    auto_flush_timeout=10,  # seconds
+    override_root_logger=True,
+    json_encoder_class=JsonDateEncoder,  # Optional; only specify this if you want to use a custom JSON encoder
+)
+sentry_sdk.init(
+    "https://ccb81025166c449d87e0646cede48448@o271257.ingest.sentry.io/6040020",
+    # Set traces_sample_rate to 1.0 to capture 100%
+    # of transactions for performance monitoring.
+    # We recommend adjusting this value in production.
+    traces_sample_rate=1.0,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class HookResponse(BaseModel):
+    id: str
+    result: typing.Optional[dict] = None
+    error: typing.Optional[str] = None
+
+
+@app.post("/{event}", response_model=HookResponse)
+def receive_hook(event: str, payload: dict):
+    logger.info(f"Received hook request for {event=} with {payload=}")
+    hook_id = uuid.uuid4().hex
+
+    try:
+        module = importlib.import_module(f"scripts.{event}")
+        func = getattr(module, "main")
+    except ModuleNotFoundError:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="No such handler")
+
+    if not callable(func):
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Invalid script interface")
+
+    try:
+        logger.info(f"Running hook handler {func=}")
+        result = func(payload)
+        return HookResponse(id=hook_id, result=result, error=None)
+    except Exception as e:
+        logger.error(f"Hook handler threw error", exc_info=True)
+        return HookResponse(id=hook_id, result=None, error=str(e))
+
+
+app = SentryAsgiMiddleware(app)
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app)

hooksy/scripts/handle_aria2_completed.py

@@ -0,0 +1,21 @@
+import logging
+import typing
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+
+Payload = typing.TypedDict(
+    "Payload",
+    {
+        "gid": str,
+        "num_files": int,
+        "path": str,
+        "files": list[typing.TypedDict("File", {"path": str, "size": int})],
+    },
+)
+
+
+def main(payload: Payload):
+    logging.info("got {torrent}", torrent=payload)
+    path = Path(payload["path"])

hooksy/scripts/handle_downloads.py

@@ -0,0 +1,26 @@
+import logging
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+
+def main(payload: dict = None):
+    if not payload:
+        payload = {}
+
+    links: list[dict] = payload.get("links", [])
+    if not links:
+        logger.info("No links, exiting")
+        return
+
+    logger.info("Got %d links", len(links))
+
+    for it in links:
+        download_path = Path(it["download_path"])
+        url: str = it.get("url", it.get("content_url"))
+        logger.info("Processing downloaded link {url} at {path}", path=download_path, url=url)
+
+        if download_path.suffix.lower() in {".zip", ".rar"}:
+            logger.info("Got an imageset!")
+
+    # raise Exception("oops!")