Add backend support for notifications

backend/backend.py

@@ -3,7 +3,7 @@ from fastapi import FastAPI
 from fastapi.exceptions import HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-from model import Log, SessionLocal, Script, Settings
+from model import Log, SessionLocal, Script, Settings, Subscription, Notification
 from run_scripts import run_scripts, update_requirements, update_environment
 import uvicorn
 
@@ -52,11 +52,183 @@ def hello():
     return {"message": "Welcome to the Project Monitor API"}
 
 
+# Subscriptions API Endpoints
+@app.get("/subscriptions")
+def list_subscriptions():
+    db = SessionLocal()
+    subscriptions = db.query(Subscription).all()
+    db.close()
+    return subscriptions
+
+
+class SubscriptionCreate(BaseModel):
+    topic: str
+
+
+class SubscriptionResponse(BaseModel):
+    id: int
+    topic: str
+    created_at: datetime
+
+    model_config = {"from_attributes": True}
+
+
+@app.get("/subscriptions/{subscription_id}", response_model=SubscriptionResponse)
+def get_subscription(subscription_id: int):
+    db = SessionLocal()
+    subscription = (
+        db.query(Subscription).filter(Subscription.id == subscription_id).first()
+    )
+    if not subscription:
+        db.close()
+        raise HTTPException(status_code=404, detail="Subscription not found")
+    db.close()
+    return subscription
+
+
+@app.post("/subscriptions")
+def add_subscription(subscription: SubscriptionCreate):
+    db = SessionLocal()
+    existing_subscription = (
+        db.query(Subscription).filter(Subscription.topic == subscription.topic).first()
+    )
+    if existing_subscription:
+        db.close()
+        raise HTTPException(status_code=400, detail="Subscription already exists")
+    new_subscription = Subscription(topic=subscription.topic)
+    db.add(new_subscription)
+    db.commit()
+    db.refresh(new_subscription)
+    db.close()
+    return new_subscription
+
+
+@app.delete("/subscriptions/{subscription_id}")
+def remove_subscription(subscription_id: int):
+    db = SessionLocal()
+    subscription = (
+        db.query(Subscription).filter(Subscription.id == subscription_id).first()
+    )
+    if not subscription:
+        db.close()
+        raise HTTPException(status_code=404, detail="Subscription not found")
+    db.delete(subscription)
+    db.commit()
+    db.close()
+    return {"message": "Subscription removed"}
+
+
+@app.get("/subscriptions/{subscription_id}/notifications")
+def list_subscription_notifications(subscription_id: int):
+    db = SessionLocal()
+    notifications = (
+        db.query(Notification)
+        .filter(Notification.subscription_id == subscription_id)
+        .all()
+    )
+    db.close()
+    return [
+        NotificationResponse.model_validate(notification)
+        for notification in notifications
+    ]
+
+
+@app.get("/notifications")
+def list_notifications():
+    db = SessionLocal()
+    notifications = db.query(Notification).all()
+    db.close()
+    return [
+        NotificationResponse.model_validate(notification)
+        for notification in notifications
+    ]
+
+
+@app.delete("/notifications/{notification_id}")
+def remove_notification(notification_id: int):
+    db = SessionLocal()
+    notification = (
+        db.query(Notification).filter(Notification.id == notification_id).first()
+    )
+    if not notification:
+        db.close()
+        raise HTTPException(status_code=404, detail="Notification not found")
+    db.delete(notification)
+    db.commit()
+    db.close()
+    return {"message": "Notification removed"}
+
+
+class NotificationCreate(BaseModel):
+    subscription_id: int
+    title: str
+    message: str
+    priority: int
+
+
+class NotificationUpdate(BaseModel):
+    subscription_id: int | None = None
+    title: str | None = None
+    message: str | None = None
+    priority: int | None = None
+    viewed: bool | None = None
+
+
+class NotificationResponse(NotificationCreate):
+    id: int
+    created_at: datetime
+    viewed: bool
+
+    model_config = {"from_attributes": True}
+
+
+@app.put("/notifications/{notification_id}", response_model=NotificationResponse)
+def update_notification(notification_id: int, notification: NotificationUpdate):
+    db = SessionLocal()
+    existing_notification = (
+        db.query(Notification).filter(Notification.id == notification_id).first()
+    )
+    if not existing_notification:
+        db.close()
+        raise HTTPException(status_code=404, detail="Notification not found")
+    if notification.subscription_id is not None:
+        existing_notification.subscription_id = notification.subscription_id
+    if notification.title is not None:
+        existing_notification.title = notification.title
+    if notification.message is not None:
+        existing_notification.message = notification.message
+    if notification.priority is not None:
+        existing_notification.priority = notification.priority
+    if notification.viewed is not None:
+        existing_notification.viewed = notification.viewed
+    db.commit()
+    db.refresh(existing_notification)
+    db.close()
+    return existing_notification
+
+
+@app.post("/notifications", response_model=NotificationResponse)
+def create_notification(notification: NotificationCreate):
+    db = SessionLocal()
+    new_notification = Notification(
+        subscription_id=notification.subscription_id,
+        title=notification.title,
+        message=notification.message,
+        priority=notification.priority,
+    )
+    db.add(new_notification)
+    db.commit()
+    db.refresh(new_notification)
+    db.close()
+    return new_notification
+
+
 # Define Pydantic models for Settings
 class SettingsBase(BaseModel):
     requirements: str
     environment: str
     user: str
+    ntfy_url: str
 
 
 class SettingsUpdate(SettingsBase):
@@ -107,14 +279,17 @@ def update_setting(settings_id: int, settings: SettingsUpdate):
     if not existing_setting:
         raise HTTPException(status_code=404, detail="Setting not found")
 
-    if existing_setting.requirements != settings.requirements:
+    if settings.requirements and existing_setting.requirements != settings.requirements:
         existing_setting.requirements = settings.requirements
         update_requirements(settings)
 
-    if existing_setting.environment != settings.environment:
+    if settings.environment and existing_setting.environment != settings.environment:
         existing_setting.environment = settings.environment
         update_environment(settings)
 
+    if settings.ntfy_url is not None:
+        existing_setting.ntfy_url = settings.ntfy_url
+
     db.commit()
     db.refresh(existing_setting)
     db.close()