import uuid
from sqlalchemy import Column, String, Boolean, ForeignKey, DateTime, Text, Enum as SqlEnum
from sqlalchemy.orm import relationship
from sqlalchemy.dialects.postgresql import JSONB
from utils.utils import Base, utc_now, UserRole


class User(Base):
    __tablename__ = "users"
    
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    email = Column(String(255), nullable=False, unique=True, index=True)
    password_hash = Column(String(255), nullable=False)
    first_name = Column(String(100), nullable=False)
    last_name = Column(String(100), nullable=False)
    role = Column(SqlEnum(UserRole, name="userrole_enum"), nullable=False)
    phone = Column(String(20), nullable=True)
    department_id = Column(String(36), ForeignKey("departments.id", ondelete="SET NULL"), nullable=True, index=True)
    is_active = Column(Boolean, nullable=False, default=True)
    created_at = Column(DateTime(timezone=True), default=utc_now, nullable=False)
    updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now, nullable=False)
    
    # Relationships
    department = relationship("Department", back_populates="users")
    appointments_as_doctor = relationship("Appointment", foreign_keys="[Appointment.doctor_id]", back_populates="doctor")
    vital_signs_recorded = relationship("VitalSign", foreign_keys="[VitalSign.recorded_by_user_id]", back_populates="recorded_by")
    consultations = relationship("Consultation", foreign_keys="[Consultation.doctor_id]", back_populates="doctor")
    prescriptions = relationship("Prescription", foreign_keys="[Prescription.doctor_id]", back_populates="doctor")
    lab_tests_ordered = relationship("LabTest", foreign_keys="[LabTest.ordered_by_doctor_id]", back_populates="ordered_by_doctor")
    invoices_created = relationship("Invoice", foreign_keys="[Invoice.created_by_user_id]", back_populates="created_by_user")
    payments_processed = relationship("Payment", foreign_keys="[Payment.processed_by_user_id]", back_populates="processed_by_user")
    reports_generated = relationship("Report", foreign_keys="[Report.generated_by_user_id]", back_populates="generated_by")
    audit_logs = relationship("Auditlog", back_populates="user")


class Auditlog(Base):
    __tablename__ = "audit_logs"
    
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = Column(String(36), ForeignKey("users.id", ondelete="SET NULL"), nullable=True, index=True)
    action = Column(String(100), nullable=False)
    entity_type = Column(String(100), nullable=False)
    entity_id = Column(String(36), nullable=True)
    old_values = Column(JSONB, nullable=True, default=dict)
    new_values = Column(JSONB, nullable=True, default=dict)
    ip_address = Column(String(45), nullable=True)
    user_agent = Column(Text, nullable=True)
    created_at = Column(DateTime(timezone=True), default=utc_now, nullable=False)
    updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now, nullable=False)
    
    # Relationships
    user = relationship("User", back_populates="audit_logs")