from dataclasses import dataclass
from enum import Enum
from typing import TYPE_CHECKING
from uuid import UUID

from jinja2 import Environment
from pydantic import BaseModel

if TYPE_CHECKING:
    from integrations.models import Message

    from openhands.app_server.integrations.provider import PROVIDER_TOKEN_TYPE
    from openhands.app_server.user_auth.user_auth import UserAuth


class GitLabResourceType(Enum):
    GROUP = 'group'
    SUBGROUP = 'subgroup'
    PROJECT = 'project'


class PRStatus(Enum):
    CLOSED = 'CLOSED'
    MERGED = 'MERGED'


class UserData(BaseModel):
    # int for GitHub/GitLab numeric user ids; str for providers like Bitbucket
    # whose user identifier is a UUID. Always pass through ``str()`` at use sites.
    user_id: int | str
    username: str
    keycloak_user_id: str


@dataclass
class SummaryExtractionTracker:
    conversation_id: str
    should_extract: bool
    send_summary_instruction: bool


@dataclass
class ResolverViewInterface(SummaryExtractionTracker):
    # installation_id type varies by provider:
    # - GitHub: int (GitHub App installation ID)
    # - GitLab: str (webhook installation ID from our DB)
    installation_id: int | str
    user_info: UserData
    issue_number: int
    full_repo_name: str
    is_public_repo: bool
    raw_payload: 'Message'

    async def _get_instructions(self, jinja_env: Environment) -> tuple[str, str]:
        """Instructions passed when conversation is first initialized."""
        raise NotImplementedError()

    async def initialize_new_conversation(self) -> UUID:
        """Initialize a new conversation and return the conversation ID.

        This method resolves the target organization and generates a new
        conversation ID.
        """
        raise NotImplementedError()

    async def create_new_conversation(
        self,
        jinja_env: Environment,
        git_provider_tokens: 'PROVIDER_TOKEN_TYPE',
        conversation_id: UUID,
        saas_user_auth: 'UserAuth',
    ) -> None:
        """Create a new conversation.

        Args:
            jinja_env: Jinja2 environment for template rendering
            git_provider_tokens: Token mapping for git providers
            conversation_id: The UUID of the conversation to create
            saas_user_auth: User authentication for SaaS
        """
        raise NotImplementedError()