from sqlalchemy import Column, Integer, String, DateTime, Float, Boolean, Text, ForeignKey, Enum
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from datetime import datetime
import enum

Base = declarative_base()

class UserRole(str, enum.Enum):
    ADMIN = "admin"
    USER = "user"

class IssueStatus(str, enum.Enum):
    NEW = "new"
    PROGRESS = "progress"
    COMPLETE = "complete"

class IssueCategory(str, enum.Enum):
    MATERIAL_MISSING = "material_missing"
    DIMENSION_DEFECT = "dimension_defect"
    INCOMING_DEFECT = "incoming_defect"

class User(Base):
    __tablename__ = "users"
    
    id = Column(Integer, primary_key=True, index=True)
    username = Column(String, unique=True, index=True, nullable=False)
    hashed_password = Column(String, nullable=False)
    full_name = Column(String)
    role = Column(Enum(UserRole), default=UserRole.USER)
    is_active = Column(Boolean, default=True)
    created_at = Column(DateTime, default=datetime.utcnow)
    
    # Relationships
    issues = relationship("Issue", back_populates="reporter")
    daily_works = relationship("DailyWork", back_populates="created_by")

class Issue(Base):
    __tablename__ = "issues"
    
    id = Column(Integer, primary_key=True, index=True)
    photo_path = Column(String)
    category = Column(Enum(IssueCategory), nullable=False)
    description = Column(Text, nullable=False)
    status = Column(Enum(IssueStatus), default=IssueStatus.NEW)
    reporter_id = Column(Integer, ForeignKey("users.id"))
    report_date = Column(DateTime, default=datetime.utcnow)
    work_hours = Column(Float, default=0)
    detail_notes = Column(Text)
    
    # Relationships
    reporter = relationship("User", back_populates="issues")

class DailyWork(Base):
    __tablename__ = "daily_works"
    
    id = Column(Integer, primary_key=True, index=True)
    date = Column(DateTime, nullable=False, index=True)
    worker_count = Column(Integer, nullable=False)
    regular_hours = Column(Float, nullable=False)
    overtime_workers = Column(Integer, default=0)
    overtime_hours = Column(Float, default=0)
    overtime_total = Column(Float, default=0)
    total_hours = Column(Float, nullable=False)
    created_by_id = Column(Integer, ForeignKey("users.id"))
    created_at = Column(DateTime, default=datetime.utcnow)
    
    # Relationships
    created_by = relationship("User", back_populates="daily_works")