main.py

from fastapi import FastAPI, HTTPException
from fastapi.responses import Response

from config import get_database_url, get_redis_url
from repositories import PostgresTaskRepository
from schemas import TaskCreate, TaskUpdate
from services import TaskService


app = FastAPI()
repository = PostgresTaskRepository(get_database_url())
task_service = TaskService(repository)


@app.on_event("startup")
def startup():
    repository.seed_if_empty()


# Root endpoint
@app.get("/")
def root():
    return {
        "name": "Task API",
        "version": "1.0",
        "endpoints": ["/tasks"],
    }


# Health endpoint
@app.get("/health")
def health():
    checks = {
        "status": "ok",
        "database": "ok" if repository.ping() else "error",
    }

    redis_url = get_redis_url()

    if redis_url:
        checks["redis"] = "ok" if task_service.ping_redis(redis_url) else "error"

    if "error" in checks.values():
        raise HTTPException(status_code=503, detail=checks)

    return checks


# Get all tasks
@app.get("/tasks")
def get_tasks():
    return task_service.get_tasks()


# Get one task by ID
@app.get("/tasks/{task_id}")
def get_task(task_id: int):
    task = task_service.get_task(task_id)

    if task is None:
        raise HTTPException(
            status_code=404,
            detail=f"Task {task_id} not found",
        )

    return task


# Create a new task
@app.post("/tasks", status_code=201)
def create_task(task: TaskCreate):
    try:
        return task_service.create_task(task.title)
    except ValueError as error:
        raise HTTPException(
            status_code=400,
            detail=str(error),
        ) from error


# Update a task
@app.put("/tasks/{task_id}")
def update_task(task_id: int, updated: TaskUpdate):
    try:
        task = task_service.update_task(task_id, updated.title, updated.done)
    except ValueError as error:
        raise HTTPException(
            status_code=400,
            detail=str(error),
        ) from error

    if task is None:
        raise HTTPException(
            status_code=404,
            detail=f"Task {task_id} not found",
        )

    return task


# Delete a task
@app.delete("/tasks/{task_id}", status_code=204)
def delete_task(task_id: int):
    if not task_service.delete_task(task_id):
        raise HTTPException(
            status_code=404,
            detail=f"Task {task_id} not found",
        )

    return Response(status_code=204)