# -*- coding: utf-8 -*-
"""Repository für Employee-CRUD – einziger Ort der direkt mit der DB spricht."""

from typing import Optional

from sqlalchemy.orm import Session

from ..core.models import Employee
from ..core.schemas import EmployeeCreate, EmployeeUpdate


class EmployeeRepository:
    def __init__(self, db: Session):
        self.db = db

    def get_by_id(self, employee_id: str) -> Optional[Employee]:
        return self.db.query(Employee).filter(Employee.id == employee_id).first()

    def get_by_email(self, email: str) -> Optional[Employee]:
        return self.db.query(Employee).filter(Employee.email == email).first()

    def list_all(self, active_only: bool = True) -> list[Employee]:
        q = self.db.query(Employee)
        if active_only:
            q = q.filter(Employee.is_active == True)
        return q.order_by(Employee.name).all()

    def list_by_department(self, department: str) -> list[Employee]:
        return (
            self.db.query(Employee)
            .filter(Employee.department == department, Employee.is_active == True)
            .order_by(Employee.name)
            .all()
        )

    def create(self, data: EmployeeCreate) -> Employee:
        emp = Employee(**data.model_dump())
        self.db.add(emp)
        self.db.flush()
        return emp

    def update(self, employee_id: str, data: EmployeeUpdate) -> Optional[Employee]:
        emp = self.get_by_id(employee_id)
        if not emp:
            return None
        for field, value in data.model_dump(exclude_unset=True).items():
            setattr(emp, field, value)
        self.db.flush()
        return emp

    def delete(self, employee_id: str) -> bool:
        emp = self.get_by_id(employee_id)
        if not emp:
            return False
        self.db.delete(emp)
        self.db.flush()
        return True

    def count_active(self) -> int:
        return self.db.query(Employee).filter(Employee.is_active == True).count()