Sentry Monitoring (#75)

* loading sentry dsn from env file

* only call init on sentry sdk if enabled

* removed sentry debug endpoint

pyproject.toml

@@ -14,6 +14,7 @@ dependencies = [
     "hypercorn>=0.17.3",
     "pyjwt[crypto]>=2.9.0",
     "bcrypt>=4.2.0",
+    "sentry-sdk[fastapi]>=2.14.0",
 ]
 
 [build-system]

src/smolvault/config.py

@@ -13,6 +13,8 @@ class Settings(BaseSettings):
     user_whitelist: str
     users_limit: int
     daily_upload_limit_bytes: int
+    sentry_enabled: bool
+    sentry_dsn: str
 
     model_config = SettingsConfigDict(env_file=".env")
 

src/smolvault/main.py

@@ -6,6 +6,7 @@
 from logging.handlers import RotatingFileHandler
 from typing import Annotated
 
+import sentry_sdk
 from fastapi import BackgroundTasks, Depends, FastAPI, File, Form, HTTPException, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.gzip import GZipMiddleware
@@ -31,6 +32,14 @@
 )
 logger = logging.getLogger(__name__)
 
+settings: Settings = get_settings()
+if settings.sentry_enabled:
+    sentry_sdk.init(
+        dsn=settings.sentry_dsn,
+        traces_sample_rate=1.0,  # Set traces_sample_rate to 1.0 to capture 100%
+        profiles_sample_rate=1.0,
+    )
+
 app = FastAPI(title="smolvault", docs_url=None, redoc_url=None)
 
 app.add_middleware(GZipMiddleware, minimum_size=1000)
@@ -43,7 +52,6 @@
 )
 
 
-settings: Settings = get_settings()
 s3_client = S3Client(bucket_name=settings.smolvault_bucket)
 cache = CacheManager(cache_dir=settings.smolvault_cache)
 

tests/testing.env

@@ -5,3 +5,5 @@ SMOLVAULT_CACHE="./uploads/"
 USER_WHITELIST="1,2"
 USERS_LIMIT="3"
 DAILY_UPLOAD_LIMIT_BYTES="50000"
+SENTRY_ENABLED="false"
+SENTRY_DSN="none"