"""Jira integration models — link Aegis entities to Jira issues."""

import enum
import uuid
from datetime import datetime

from sqlalchemy import Column, String, DateTime, ForeignKey, Enum as SQLEnum, Index
from sqlalchemy.dialects.postgresql import UUID, JSONB
from sqlalchemy.orm import relationship

from app.database import Base


class JiraLinkEntityType(str, enum.Enum):
    test = "test"
    technique = "technique"
    campaign = "campaign"
    evidence = "evidence"


class JiraSyncDirection(str, enum.Enum):
    aegis_to_jira = "aegis_to_jira"
    jira_to_aegis = "jira_to_aegis"
    bidirectional = "bidirectional"


class JiraLink(Base):
    """Associates an Aegis entity with a Jira issue for bidirectional sync."""

    __tablename__ = "jira_links"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    entity_type = Column(SQLEnum(JiraLinkEntityType), nullable=False)
    entity_id = Column(UUID(as_uuid=True), nullable=False)
    jira_issue_key = Column(String(50), nullable=False)
    jira_issue_id = Column(String(50))
    jira_project_key = Column(String(20))
    jira_status = Column(String(100))
    jira_priority = Column(String(50))
    jira_assignee = Column(String(255))
    jira_story_points = Column(String(10))
    sync_direction = Column(
        SQLEnum(JiraSyncDirection), default=JiraSyncDirection.bidirectional
    )
    last_synced_at = Column(DateTime)
    sync_metadata = Column(JSONB, default={})
    created_by = Column(UUID(as_uuid=True), ForeignKey("users.id"))
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)

    creator = relationship("User", foreign_keys=[created_by])

    __table_args__ = (
        Index("ix_jira_links_entity_id", "entity_id"),
        Index("ix_jira_links_issue_key", "jira_issue_key"),
        Index("ix_jira_links_entity_type_entity_id", "entity_type", "entity_id"),
    )