import uuid
from typing import TYPE_CHECKING

from sqlalchemy import ForeignKey, String, Text
from sqlalchemy import Uuid as UUID
from sqlalchemy.orm import Mapped, mapped_column, relationship

from src.models.base import Base, TimestampMixin, UUIDPrimaryKeyMixin

if TYPE_CHECKING:
    from src.models.project import Project
    from src.models.tool_type import ToolType
    from src.models.user import User


class ToolConfig(UUIDPrimaryKeyMixin, TimestampMixin, Base):
    __tablename__ = "tool_configs"

    user_id: Mapped[uuid.UUID] = mapped_column(
        UUID(), ForeignKey("users.id"), nullable=False
    )
    tool_type_id: Mapped[uuid.UUID] = mapped_column(
        UUID(), ForeignKey("tool_types.id"), nullable=False
    )
    project_id: Mapped[uuid.UUID | None] = mapped_column(
        UUID(), ForeignKey("projects.id"), nullable=True
    )
    key: Mapped[str] = mapped_column(String(255), nullable=False)
    value: Mapped[str] = mapped_column(Text, nullable=False)
    config_type: Mapped[str] = mapped_column(
        String(20), nullable=False, default="env"
    )  # "env" or "file"
    file_path: Mapped[str | None] = mapped_column(
        String(1024), nullable=True
    )  # Only for file type

    user: Mapped["User"] = relationship()
    tool_type: Mapped["ToolType"] = relationship()
    project: Mapped["Project | None"] = relationship()