from fastapi import APIRouter, Depends, HTTPException, Query, status
from sqlalchemy.orm import Session
from typing import List, Optional
from math import ceil
from app.db.session import get_db
from app.apps.auth.utils import get_current_user
from . import models, schemas

router = APIRouter(prefix="/students", tags=["Students"])


# ১. Create (তৈরি করা) - JWT Protection সহ
@router.post("/", response_model=schemas.StudentResponse, status_code=status.HTTP_201_CREATED)
def create_student(
    student: schemas.StudentCreate,
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """
    নতুন স্টুডেন্ট তৈরি করা (শুধুমাত্র লগইন করা ইউজাররা)
    """
    db_student = models.Student(**student.dict())
    db.add(db_student)
    db.commit()
    db.refresh(db_student)
    return db_student


# ২. Read All (সব দেখা) - JWT Protection সহ
@router.get("/", response_model=List[schemas.StudentResponse])
def get_all_students(
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db),
    search: Optional[str] = Query(None, description="নাম বা ইমেইল দিয়ে সার্চ করুন")
):
    """সব স্টুডেন্ট দেখা (শুধুমাত্র লগইন করা ইউজাররা)"""
    query = db.query(models.Student)
    
    if search:
        query = query.filter(
            (models.Student.name.ilike(f"%{search}%")) |
            (models.Student.email.ilike(f"%{search}%"))
        )
    
    return query.all()


# ৩. Read One (একটি দেখা) - JWT Protection সহ
@router.get("/{student_id}", response_model=schemas.StudentResponse)
def get_student(
    student_id: int,
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """নির্দিষ্ট স্টুডেন্ট দেখা (শুধুমাত্র লগইন করা ইউজাররা)"""
    db_student = db.query(models.Student).filter(models.Student.id == student_id).first()
    if not db_student:
        raise HTTPException(status_code=404, detail="Student not found")
    return db_student


# ৪. Search with Pagination - JWT Protection সহ
@router.get("/search/paginated", response_model=schemas.PaginatedStudentResponse)
def search_students_paginated(
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db),
    search: Optional[str] = Query(None),
    page: int = Query(1, ge=1),
    page_size: int = Query(10, ge=1, le=100)
):
    """Pagination সহ সার্চ (শুধুমাত্র লগইন করা ইউজাররা)"""
    query = db.query(models.Student)
    
    if search:
        query = query.filter(
            (models.Student.name.ilike(f"%{search}%")) |
            (models.Student.email.ilike(f"%{search}%"))
        )
    
    total = query.count()
    total_pages = ceil(total / page_size) if total > 0 else 0
    offset = (page - 1) * page_size
    students = query.offset(offset).limit(page_size).all()
    
    return {
        "total": total,
        "page": page,
        "page_size": page_size,
        "total_pages": total_pages,
        "data": students
    }


# ৫. Advanced Search - JWT Protection সহ
@router.get("/advanced-search", response_model=List[schemas.StudentResponse])
def advanced_search(
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db),
    name: Optional[str] = Query(None),
    email: Optional[str] = Query(None),
    age_min: Optional[int] = Query(None, ge=0),
    age_max: Optional[int] = Query(None, ge=0),
    sort_by: Optional[str] = Query("id"),
    sort_order: Optional[str] = Query("asc")
):
    """উন্নত সার্চ (শুধুমাত্র লগইন করা ইউজাররা)"""
    query = db.query(models.Student)
    
    if name:
        query = query.filter(models.Student.name.ilike(f"%{name}%"))
    if email:
        query = query.filter(models.Student.email.ilike(f"%{email}%"))
    if age_min is not None:
        query = query.filter(models.Student.age >= age_min)
    if age_max is not None:
        query = query.filter(models.Student.age <= age_max)
    
    if sort_by == "name":
        sort_column = models.Student.name
    elif sort_by == "email":
        sort_column = models.Student.email
    elif sort_by == "age":
        sort_column = models.Student.age
    else:
        sort_column = models.Student.id
    
    if sort_order == "desc":
        query = query.order_by(sort_column.desc())
    else:
        query = query.order_by(sort_column.asc())
    
    return query.all()


# ৬. Update (আপডেট করা) - JWT Protection সহ
@router.put("/{student_id}", response_model=schemas.StudentResponse)
def update_student(
    student_id: int,
    student_data: schemas.StudentUpdate,
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """স্টুডেন্ট আপডেট করা (শুধুমাত্র লগইন করা ইউজাররা)"""
    db_student = db.query(models.Student).filter(models.Student.id == student_id).first()
    
    if not db_student:
        raise HTTPException(status_code=404, detail="Student not found")
    
    updated_info = student_data.dict(exclude_unset=True)
    
    for key, value in updated_info.items():
        setattr(db_student, key, value)
    
    db.commit()
    db.refresh(db_student)
    return db_student


# ৭. Delete (মুছে ফেলা) - JWT Protection সহ
@router.delete("/{student_id}")
def delete_student(
    student_id: int,
    current_user: dict = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """স্টুডেন্ট মুছে ফেলা (শুধুমাত্র লগইন করা ইউজাররা)"""
    db_student = db.query(models.Student).filter(models.Student.id == student_id).first()
    if not db_student:
        raise HTTPException(status_code=404, detail="Student not found")
    
    db.delete(db_student)
    db.commit()
    return {"message": "Successfully Deleted"}

from app.apps.auth.utils import get_current_user

current_user: dict = Depends(get_current_user)

{
    "user_id": 1,
    "username": "ahmed",
    "type": "access"
}

curl -X POST http://localhost:8000/auth/login \
  -H "Content-Type: application/json" \
  -d '{
    "username": "ahmed",
    "password": "password123"
  }'

curl -X POST http://localhost:8000/students/ \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer YOUR_ACCESS_TOKEN" \
  -d '{
    "name": "ফাতিমা",
    "email": "fatima@example.com",
    "age": 20
  }'

{"detail":"Authorization header প্রয়োজন"}