Add gitlab webhook support

ops_bot/main.py

@@ -1,21 +1,26 @@
 import asyncio
 import json
 import logging
+from pathlib import Path
 from typing import Any, Dict, Literal, Optional, Tuple, cast
-
+from dotenv import load_dotenv
 import uvicorn
-from fastapi import Depends, FastAPI, HTTPException, Request, status
+from fastapi import Depends, FastAPI, HTTPException, Request, status, Header
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
+import pydantic
 from pydantic import BaseSettings
 
 from ops_bot import aws, pagerduty
 from ops_bot.matrix import MatrixClient, MatrixClientSettings
+from ops_bot.gitlab import hook as gitlab_hook
 
+load_dotenv()
 
 class BotSettings(BaseSettings):
     bearer_token: str
     routing_keys: Dict[str, str]
     log_level: Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"] = "INFO"
+    matrix: MatrixClientSettings
 
     class Config:
         env_prefix = "BOT_"
@@ -40,11 +45,14 @@ async def matrix_main(matrix_client: MatrixClient) -> None:
 
 @app.on_event("startup")
 async def startup_event() -> None:
-    bot_settings = BotSettings(_env_file=".env", _env_file_encoding="utf-8")
+    # if "config.json" exists read it
+    if Path("config.json").exists():
+        bot_settings = BotSettings.parse_file("config.json")
+    else:
+        bot_settings = BotSettings(_env_file=".env", _env_file_encoding="utf-8")
     logging.getLogger().setLevel(bot_settings.log_level)
-    matrix_settings = MatrixClientSettings(_env_file=".env", _env_file_encoding="utf-8")
-    matrix_settings.join_rooms = list(bot_settings.routing_keys.values())
-    c = MatrixClient(settings=matrix_settings)
+    bot_settings.matrix.join_rooms = list(bot_settings.routing_keys.values())
+    c = MatrixClient(settings=bot_settings.matrix)
     app.state.matrix_client = c
     app.state.bot_settings = bot_settings
     asyncio.create_task(matrix_main(c))
@@ -120,6 +128,29 @@ async def aws_sns_hook(
     )
     return {"message": msg_plain, "message_formatted": msg_formatted}
 
+@app.post("/hook/gitlab/{routing_key}")
+async def gitlab_webhook(
+    request: Request,
+    x_gitlab_token: str = Header(default=""),
+    x_gitlab_event: str = Header(default=""),
+     matrix_client: MatrixClient = Depends(get_matrix_service)
+) -> Dict[str, str]:
+    bearer_token = request.app.state.bot_settings.bearer_token
+    if x_gitlab_token != bearer_token:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Incorrect X-Gitlab-Token"
+        )
+    room_id, payload = await receive_helper(request)
+    messages = await gitlab_hook.parse_event(x_gitlab_event, payload)
+    for msg_plain, msg_formatted in messages:
+        await matrix_client.room_send(
+            room_id,
+            msg_plain,
+            message_formatted=msg_formatted,
+        )
+    return {"status": "ok"}
+
 
 def start_dev() -> None:
     uvicorn.run("ops_bot.main:app", port=1111, host="127.0.0.1", reload=True)