Add models router

app/routers/models.py

@@ -0,0 +1,73 @@
+"""Model repository routes."""
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+
+from .. import models, schemas
+from ..database import get_db
+from ..deps import get_current_user
+
+router = APIRouter(prefix="/models", tags=["models"])
+
+
+@router.post("", response_model=schemas.ModelOut, status_code=201)
+def create_model(body: schemas.ModelCreate, user=Depends(get_current_user), db: Session = Depends(get_db)):
+    m = models.ModelRepo(owner=user.username, **body.model_dump())
+    db.add(m)
+    db.commit()
+    db.refresh(m)
+    return m
+
+
+@router.get("", response_model=list[schemas.ModelOut])
+def list_models(owner: str | None = None, search: str | None = None, db: Session = Depends(get_db)):
+    q = db.query(models.ModelRepo)
+    if owner:
+        q = q.filter(models.ModelRepo.owner == owner)
+    if search:
+        like = f"%{search}%"
+        q = q.filter(models.ModelRepo.name.ilike(like) | models.ModelRepo.description.ilike(like))
+    return q.order_by(models.ModelRepo.downloads.desc()).all()
+
+
+@router.get("/{model_id}", response_model=schemas.ModelOut)
+def get_model(model_id: str, db: Session = Depends(get_db)):
+    m = db.get(models.ModelRepo, model_id)
+    if m is None:
+        raise HTTPException(404, "Model not found")
+    return m
+
+
+@router.patch("/{model_id}", response_model=schemas.ModelOut)
+def update_model(model_id: str, body: schemas.ModelUpdate, user=Depends(get_current_user), db: Session = Depends(get_db)):
+    m = db.get(models.ModelRepo, model_id)
+    if m is None:
+        raise HTTPException(404, "Model not found")
+    if m.owner != user.username:
+        raise HTTPException(403, "Not your model")
+    for k, v in body.model_dump(exclude_unset=True).items():
+        setattr(m, k, v)
+    db.commit()
+    db.refresh(m)
+    return m
+
+
+@router.delete("/{model_id}", status_code=204)
+def delete_model(model_id: str, user=Depends(get_current_user), db: Session = Depends(get_db)):
+    m = db.get(models.ModelRepo, model_id)
+    if m is None:
+        raise HTTPException(404, "Model not found")
+    if m.owner != user.username:
+        raise HTTPException(403, "Not your model")
+    db.delete(m)
+    db.commit()
+
+
+@router.post("/{model_id}/like", response_model=schemas.ModelOut)
+def like_model(model_id: str, db: Session = Depends(get_db)):
+    m = db.get(models.ModelRepo, model_id)
+    if m is None:
+        raise HTTPException(404, "Model not found")
+    m.likes += 1
+    db.commit()
+    db.refresh(m)
+    return m