from sqlalchemy import (
    Column, Integer, String, Text, ForeignKey, DateTime, Table
)
from sqlalchemy.orm import relationship, Mapped, mapped_column
from datetime import datetime

from .db import Base

work_entry_servers = Table(
    "work_entry_servers",
    Base.metadata,
    Column("work_entry_id", ForeignKey("work_entries.id"), primary_key=True),
    Column("server_id", ForeignKey("servers.id"), primary_key=True),
)

class User(Base):
    __tablename__ = "users"

    id: Mapped[int] = mapped_column(primary_key=True, index=True)
    username: Mapped[str] = mapped_column(String(50), unique=True, index=True)
    password_hash: Mapped[str] = mapped_column(String(255))
    is_active: Mapped[bool] = mapped_column(default=True)

class Owner(Base):
    __tablename__ = "owners"

    id: Mapped[int] = mapped_column(primary_key=True)
    name: Mapped[str] = mapped_column(String(100), unique=True)
    contact: Mapped[str | None] = mapped_column(String(255), nullable=True)

    servers: Mapped[list["Server"]] = relationship("Server", back_populates="owner")

class Server(Base):
    __tablename__ = "servers"

    id: Mapped[int] = mapped_column(primary_key=True)
    hostname: Mapped[str] = mapped_column(String(100), unique=True, index=True)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    owner_id: Mapped[int | None] = mapped_column(ForeignKey("owners.id"))

    owner: Mapped[Owner | None] = relationship("Owner", back_populates="servers")
    work_entries: Mapped[list["WorkEntry"]] = relationship(
        "WorkEntry",
        secondary=work_entry_servers,
        back_populates="servers",
    )

class WorkEntry(Base):
    __tablename__ = "work_entries"

    id: Mapped[int] = mapped_column(primary_key=True)
    title: Mapped[str] = mapped_column(String(200))
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    created_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
    duration_minutes: Mapped[int | None] = mapped_column(nullable=True)
    created_by_user_id: Mapped[int] = mapped_column(ForeignKey("users.id"))

    servers: Mapped[list[Server]] = relationship(
        "Server",
        secondary=work_entry_servers,
        back_populates="work_entries",
    )