"""User-owned GitHub App installations and per-project mirror targets."""
from __future__ import annotations

from datetime import datetime
from typing import Optional
from uuid import UUID, uuid4

import sqlalchemy as sa
from sqlalchemy import BigInteger, Boolean, ForeignKey, String, Text
from sqlalchemy.dialects.postgresql import UUID as PGUUID
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.sql import func
from sqlalchemy.types import DateTime

from src.core.database import Base


class WorkspaceUserGithubInstallation(Base):
    """GitHub App installation on the user's account/org (for pushing to their repos)."""

    __tablename__ = "workspace_user_github_installations"
    __table_args__ = (
        sa.UniqueConstraint(
            "workspace_id",
            "github_installation_id",
            name="uq_workspace_user_gh_installation_workspace_installation",
        ),
    )

    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True, default=uuid4)
    workspace_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True), ForeignKey("workspaces.id", ondelete="CASCADE"), nullable=False
    )
    github_installation_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    installation_id_ciphertext: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    account_login: Mapped[str] = mapped_column(String(255), nullable=False)
    account_id: Mapped[Optional[int]] = mapped_column(BigInteger, nullable=True)
    account_type: Mapped[str] = mapped_column(String(32), nullable=False)
    repository_selection: Mapped[str] = mapped_column(
        String(16),
        nullable=False,
        server_default="all",
    )
    created_by: Mapped[Optional[UUID]] = mapped_column(
        PGUUID(as_uuid=True), ForeignKey("users.id"), nullable=True
    )
    suspended_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now()
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
    )


class ProjectUserMirrorTarget(Base):
    """Optional mirror: push P2HS output to a user-selected repo under a linked installation."""

    __tablename__ = "project_user_mirror_targets"

    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True, default=uuid4)
    project_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("projects.id", ondelete="CASCADE"),
        nullable=False,
        unique=True,
    )
    workspace_user_github_installation_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("workspace_user_github_installations.id", ondelete="CASCADE"),
        nullable=False,
    )
    target_repo_full_name: Mapped[str] = mapped_column(String(255), nullable=False)
    target_default_branch: Mapped[str] = mapped_column(
        String(100),
        nullable=False,
        server_default="main",
    )
    enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, server_default=sa.true())
    last_push_sha: Mapped[Optional[str]] = mapped_column(String(40), nullable=True)
    last_error: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now()
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
    )