mcp-ticketer 0.4.11__py3-none-any.whl → 2.0.1__py3-none-any.whl

mcp_ticketer/adapters/jira.py

@@ -1,5 +1,7 @@
 """JIRA adapter implementation using REST API v3."""
 
+from __future__ import annotations
+
 import asyncio
 import builtins
 import logging
@@ -13,7 +15,15 @@ from httpx import AsyncClient, HTTPStatusError, TimeoutException
 
 from ..core.adapter import BaseAdapter
 from ..core.env_loader import load_adapter_config, validate_adapter_config
-from ..core.models import Comment, Epic, Priority, SearchQuery, Task, TicketState
+from ..core.models import (
+    Attachment,
+    Comment,
+    Epic,
+    Priority,
+    SearchQuery,
+    Task,
+    TicketState,
+)
 from ..core.registry import AdapterRegistry
 
 logger = logging.getLogger(__name__)
@@ -51,9 +61,11 @@ def extract_text_from_adf(adf_content: str | dict[str, Any]) -> str:
     """Extract plain text from Atlassian Document Format (ADF).
 
     Args:
+    ----
         adf_content: Either a string (already plain text) or ADF document dict
 
     Returns:
+    -------
         Plain text string extracted from the ADF content
 
     """
@@ -115,6 +127,7 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         """Initialize JIRA adapter.
 
         Args:
+        ----
             config: Configuration with:
                 - server: JIRA server URL (e.g., https://company.atlassian.net)
                 - email: User email for authentication
@@ -175,6 +188,7 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         """Validate that required credentials are present.
 
         Returns:
+        -------
             (is_valid, error_message) - Tuple of validation result and error message
 
         """
@@ -228,6 +242,7 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         """Make HTTP request to JIRA API with retry logic.
 
         Args:
+        ----
             method: HTTP method
             endpoint: API endpoint
             data: Request body data
@@ -235,9 +250,11 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
             retry_count: Current retry attempt
 
         Returns:
+        -------
             Response data
 
         Raises:
+        ------
             HTTPStatusError: On API errors
             TimeoutException: On timeout
 
@@ -882,10 +899,12 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         """Execute a raw JQL query.
 
         Args:
+        ----
             jql: JIRA Query Language string
             limit: Maximum number of results
 
         Returns:
+        -------
             List of matching tickets
 
         """
@@ -909,9 +928,11 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         """Get active sprints for a board (requires JIRA Software).
 
         Args:
+        ----
             board_id: Agile board ID
 
         Returns:
+        -------
             List of sprint information
 
         """
@@ -995,6 +1016,756 @@ class JiraAdapter(BaseAdapter[Union[Epic, Task]]):
         except Exception:
             return None
 
+    async def list_labels(self) -> builtins.list[dict[str, Any]]:
+        """List all labels used in the project.
+
+        JIRA doesn't have a direct "list all labels" endpoint, so we query
+        recent issues and extract unique labels from them.
+
+        Returns:
+        -------
+            List of label dictionaries with 'id' and 'name' fields
+
+        """
+        try:
+            # Query recent issues to get labels in use
+            jql = f"project = {self.project_key} ORDER BY updated DESC"
+            data = await self._make_request(
+                "GET",
+                "search/jql",
+                params={
+                    "jql": jql,
+                    "maxResults": 100,  # Sample from recent 100 issues
+                    "fields": "labels",
+                },
+            )
+
+            # Collect unique labels
+            unique_labels = set()
+            for issue in data.get("issues", []):
+                labels = issue.get("fields", {}).get("labels", [])
+                for label in labels:
+                    if isinstance(label, dict):
+                        unique_labels.add(label.get("name", ""))
+                    else:
+                        unique_labels.add(str(label))
+
+            # Transform to standardized format
+            return [
+                {"id": label, "name": label} for label in sorted(unique_labels) if label
+            ]
+
+        except Exception:
+            # Fallback: return empty list if query fails
+            return []
+
+    async def create_issue_label(
+        self, name: str, color: str | None = None
+    ) -> dict[str, Any]:
+        """Create a new issue label in JIRA.
+
+        Note: JIRA doesn't have a dedicated label creation API. Labels are
+        created automatically when first used on an issue. This method
+        validates the label name and returns a success response.
+
+        Args:
+        ----
+            name: Label name to create
+            color: Optional color (JIRA doesn't support colors natively, ignored)
+
+        Returns:
+        -------
+            Dict with label details:
+                - id: Label name (same as name in JIRA)
+                - name: Label name
+                - status: "ready" indicating the label can be used
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid or label name is invalid
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Validate label name
+        if not name or not name.strip():
+            raise ValueError("Label name cannot be empty")
+
+        # JIRA label names must not contain spaces
+        if " " in name:
+            raise ValueError(
+                "JIRA label names cannot contain spaces. Use underscores or hyphens instead."
+            )
+
+        # Return success response
+        # The label will be created automatically when first used on an issue
+        return {"id": name, "name": name, "status": "ready"}
+
+    async def list_project_labels(
+        self, project_key: str | None = None, limit: int = 100
+    ) -> builtins.list[dict[str, Any]]:
+        """List all labels used in a JIRA project.
+
+        JIRA doesn't have a dedicated endpoint for listing project labels.
+        This method queries recent issues and extracts unique labels.
+
+        Args:
+        ----
+            project_key: JIRA project key (e.g., 'PROJ'). If None, uses configured project.
+            limit: Maximum number of labels to return (default: 100)
+
+        Returns:
+        -------
+            List of label dictionaries with 'id', 'name', and 'usage_count' fields
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid or project key not available
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Use configured project if not specified
+        key = project_key or self.project_key
+        if not key:
+            raise ValueError("Project key is required")
+
+        try:
+            # Query recent issues to get labels in use
+            jql = f"project = {key} ORDER BY updated DESC"
+            data = await self._make_request(
+                "GET",
+                "search/jql",
+                params={
+                    "jql": jql,
+                    "maxResults": 500,  # Sample from more issues for better coverage
+                    "fields": "labels",
+                },
+            )
+
+            # Collect labels with usage count
+            label_counts: dict[str, int] = {}
+            for issue in data.get("issues", []):
+                labels = issue.get("fields", {}).get("labels", [])
+                for label in labels:
+                    label_name = (
+                        label.get("name", "") if isinstance(label, dict) else str(label)
+                    )
+                    if label_name:
+                        label_counts[label_name] = label_counts.get(label_name, 0) + 1
+
+            # Transform to standardized format with usage counts
+            result = [
+                {"id": label, "name": label, "usage_count": count}
+                for label, count in sorted(
+                    label_counts.items(), key=lambda x: x[1], reverse=True
+                )
+            ]
+
+            return result[:limit]
+
+        except Exception as e:
+            logger.error(f"Failed to list project labels: {e}")
+            raise ValueError(f"Failed to list project labels: {e}") from e
+
+    async def list_cycles(
+        self, board_id: str | None = None, state: str | None = None, limit: int = 50
+    ) -> builtins.list[dict[str, Any]]:
+        """List JIRA sprints (cycles) for a board.
+
+        Requires JIRA Agile/Software. Falls back to empty list if not available.
+
+        Args:
+        ----
+            board_id: JIRA Agile board ID. If None, finds first board for project.
+            state: Filter by state ('active', 'closed', 'future'). If None, returns all.
+            limit: Maximum number of sprints to return (default: 50)
+
+        Returns:
+        -------
+            List of sprint dictionaries with fields:
+                - id: Sprint ID
+                - name: Sprint name
+                - state: Sprint state (active, closed, future)
+                - startDate: Start date (ISO format)
+                - endDate: End date (ISO format)
+                - completeDate: Completion date (ISO format, None if not completed)
+                - goal: Sprint goal
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        try:
+            # If no board_id provided, try to find a board for the project
+            if not board_id:
+                boards_data = await self._make_request(
+                    "GET",
+                    "/rest/agile/1.0/board",
+                    params={"projectKeyOrId": self.project_key, "maxResults": 1},
+                )
+                boards = boards_data.get("values", [])
+                if not boards:
+                    logger.warning(
+                        f"No Agile boards found for project {self.project_key}"
+                    )
+                    return []
+                board_id = str(boards[0]["id"])
+
+            # Get sprints for the board
+            params = {"maxResults": limit}
+            if state:
+                params["state"] = state
+
+            sprints_data = await self._make_request(
+                "GET", f"/rest/agile/1.0/board/{board_id}/sprint", params=params
+            )
+
+            sprints = sprints_data.get("values", [])
+
+            # Transform to standardized format
+            return [
+                {
+                    "id": sprint.get("id"),
+                    "name": sprint.get("name"),
+                    "state": sprint.get("state"),
+                    "startDate": sprint.get("startDate"),
+                    "endDate": sprint.get("endDate"),
+                    "completeDate": sprint.get("completeDate"),
+                    "goal": sprint.get("goal", ""),
+                }
+                for sprint in sprints
+            ]
+
+        except HTTPStatusError as e:
+            if e.response.status_code == 404:
+                logger.warning("JIRA Agile API not available (404)")
+                return []
+            logger.error(f"Failed to list sprints: {e}")
+            raise ValueError(f"Failed to list sprints: {e}") from e
+        except Exception as e:
+            logger.warning(f"JIRA Agile may not be available: {e}")
+            return []
+
+    async def list_issue_statuses(
+        self, project_key: str | None = None
+    ) -> builtins.list[dict[str, Any]]:
+        """List all workflow statuses in JIRA.
+
+        Args:
+        ----
+            project_key: Optional project key to filter statuses.
+                        If None, returns all statuses.
+
+        Returns:
+        -------
+            List of status dictionaries with fields:
+                - id: Status ID
+                - name: Status name (e.g., "To Do", "In Progress", "Done")
+                - category: Status category key (e.g., "new", "indeterminate", "done")
+                - categoryName: Human-readable category name
+                - description: Status description
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        try:
+            # Use project-specific statuses if project key provided
+            if project_key:
+                # Get statuses for the project
+                data = await self._make_request(
+                    "GET", f"project/{project_key}/statuses"
+                )
+
+                # Extract unique statuses from all issue types
+                status_map: dict[str, dict[str, Any]] = {}
+                for issue_type_data in data:
+                    for status in issue_type_data.get("statuses", []):
+                        status_id = status.get("id")
+                        if status_id not in status_map:
+                            status_map[status_id] = status
+
+                statuses = list(status_map.values())
+            else:
+                # Get all statuses
+                statuses = await self._make_request("GET", "status")
+
+            # Transform to standardized format
+            return [
+                {
+                    "id": status.get("id"),
+                    "name": status.get("name"),
+                    "category": status.get("statusCategory", {}).get("key", ""),
+                    "categoryName": status.get("statusCategory", {}).get("name", ""),
+                    "description": status.get("description", ""),
+                }
+                for status in statuses
+            ]
+
+        except Exception as e:
+            logger.error(f"Failed to list issue statuses: {e}")
+            raise ValueError(f"Failed to list issue statuses: {e}") from e
+
+    async def get_issue_status(self, issue_key: str) -> dict[str, Any] | None:
+        """Get rich status information for an issue.
+
+        Args:
+        ----
+            issue_key: JIRA issue key (e.g., 'PROJ-123')
+
+        Returns:
+        -------
+            Dict with status details and available transitions:
+                - id: Status ID
+                - name: Status name
+                - category: Status category key
+                - categoryName: Human-readable category name
+                - description: Status description
+                - transitions: List of available transitions with:
+                    - id: Transition ID
+                    - name: Transition name
+                    - to: Target status info (id, name, category)
+            Returns None if issue not found.
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        try:
+            # Get issue with status field
+            issue = await self._make_request(
+                "GET", f"issue/{issue_key}", params={"fields": "status"}
+            )
+
+            if not issue:
+                return None
+
+            status = issue.get("fields", {}).get("status", {})
+
+            # Get available transitions
+            transitions_data = await self._make_request(
+                "GET", f"issue/{issue_key}/transitions"
+            )
+            transitions = transitions_data.get("transitions", [])
+
+            # Transform transitions to simplified format
+            transition_list = [
+                {
+                    "id": trans.get("id"),
+                    "name": trans.get("name"),
+                    "to": {
+                        "id": trans.get("to", {}).get("id"),
+                        "name": trans.get("to", {}).get("name"),
+                        "category": trans.get("to", {})
+                        .get("statusCategory", {})
+                        .get("key", ""),
+                    },
+                }
+                for trans in transitions
+            ]
+
+            return {
+                "id": status.get("id"),
+                "name": status.get("name"),
+                "category": status.get("statusCategory", {}).get("key", ""),
+                "categoryName": status.get("statusCategory", {}).get("name", ""),
+                "description": status.get("description", ""),
+                "transitions": transition_list,
+            }
+
+        except HTTPStatusError as e:
+            if e.response.status_code == 404:
+                return None
+            logger.error(f"Failed to get issue status: {e}")
+            raise ValueError(f"Failed to get issue status: {e}") from e
+        except Exception as e:
+            logger.error(f"Failed to get issue status: {e}")
+            raise ValueError(f"Failed to get issue status: {e}") from e
+
+    async def create_epic(
+        self,
+        title: str,
+        description: str = "",
+        priority: Priority = Priority.MEDIUM,
+        tags: list[str] | None = None,
+        **kwargs: Any,
+    ) -> Epic:
+        """Create a new JIRA Epic.
+
+        Args:
+        ----
+            title: Epic title
+            description: Epic description
+            priority: Priority level
+            tags: List of labels
+            **kwargs: Additional fields (reserved for future use)
+
+        Returns:
+        -------
+            Created Epic with ID populated
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid or creation fails
+
+        """
+        # Validate credentials
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Build epic input
+        epic = Epic(
+            id="",  # Will be populated by JIRA
+            title=title,
+            description=description,
+            priority=priority,
+            tags=tags or [],
+            state=TicketState.OPEN,
+        )
+
+        # Create using base create method with Epic type
+        created_epic = await self.create(epic)
+
+        if not isinstance(created_epic, Epic):
+            raise ValueError("Created ticket is not an Epic")
+
+        return created_epic
+
+    async def get_epic(self, epic_id: str) -> Epic | None:
+        """Get a JIRA Epic by key or ID.
+
+        Args:
+        ----
+            epic_id: Epic identifier (key like PROJ-123)
+
+        Returns:
+        -------
+            Epic object if found and is an Epic type, None otherwise
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid
+
+        """
+        # Validate credentials
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Read issue
+        ticket = await self.read(epic_id)
+
+        if not ticket:
+            return None
+
+        # Verify it's an Epic
+        if not isinstance(ticket, Epic):
+            return None
+
+        return ticket
+
+    async def list_epics(
+        self, limit: int = 50, offset: int = 0, state: str | None = None, **kwargs: Any
+    ) -> builtins.list[Epic]:
+        """List JIRA Epics with pagination.
+
+        Args:
+        ----
+            limit: Maximum number of epics to return (default: 50)
+            offset: Number of epics to skip for pagination (default: 0)
+            state: Filter by state/status name (e.g., "To Do", "In Progress", "Done")
+            **kwargs: Additional filter parameters (reserved for future use)
+
+        Returns:
+        -------
+            List of Epic objects
+
+        Raises:
+        ------
+            ValueError: If credentials are invalid or query fails
+
+        """
+        # Validate credentials
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Build JQL query for epics
+        jql_parts = [f"project = {self.project_key}", 'issuetype = "Epic"']
+
+        # Add state filter if provided
+        if state:
+            jql_parts.append(f'status = "{state}"')
+
+        jql = " AND ".join(jql_parts) + " ORDER BY updated DESC"
+
+        try:
+            # Execute search
+            data = await self._make_request(
+                "GET",
+                "search/jql",
+                params={
+                    "jql": jql,
+                    "startAt": offset,
+                    "maxResults": limit,
+                    "fields": "*all",
+                    "expand": "renderedFields",
+                },
+            )
+
+            # Convert issues to tickets
+            issues = data.get("issues", [])
+            epics = []
+
+            for issue in issues:
+                ticket = self._issue_to_ticket(issue)
+                # Only include if it's actually an Epic
+                if isinstance(ticket, Epic):
+                    epics.append(ticket)
+
+            return epics
+
+        except Exception as e:
+            raise ValueError(f"Failed to list JIRA epics: {e}") from e
+
+    async def update_epic(self, epic_id: str, updates: dict[str, Any]) -> Epic | None:
+        """Update a JIRA Epic with epic-specific field handling.
+
+        Args:
+        ----
+            epic_id: Epic identifier (key like PROJ-123 or ID)
+            updates: Dictionary with fields to update:
+                - title: Epic title (maps to summary)
+                - description: Epic description (auto-converted to ADF)
+                - state: TicketState value (transitions via workflow)
+                - tags: List of labels
+                - priority: Priority level
+
+        Returns:
+        -------
+            Updated Epic object or None if not found
+
+        Raises:
+        ------
+            ValueError: If no fields provided for update
+            HTTPStatusError: If update fails
+
+        """
+        fields = {}
+
+        # Map title to summary
+        if "title" in updates:
+            fields["summary"] = updates["title"]
+
+        # Convert description to ADF format
+        if "description" in updates:
+            fields["description"] = self._convert_to_adf(updates["description"])
+
+        # Map tags to labels
+        if "tags" in updates:
+            fields["labels"] = updates["tags"]
+
+        # Map priority (some JIRA configs allow priority on Epics)
+        if "priority" in updates:
+            priority_value = updates["priority"]
+            if isinstance(priority_value, Priority):
+                fields["priority"] = {
+                    "name": self._map_priority_to_jira(priority_value)
+                }
+            else:
+                # String priority passed directly
+                fields["priority"] = {"name": priority_value}
+
+        if not fields and "state" not in updates:
+            raise ValueError("At least one field must be updated")
+
+        # Apply field updates if any
+        if fields:
+            await self._make_request("PUT", f"issue/{epic_id}", data={"fields": fields})
+
+        # Handle state transitions separately (JIRA uses workflow transitions)
+        if "state" in updates:
+            await self.transition_state(epic_id, updates["state"])
+
+        # Fetch and return updated epic
+        return await self.read(epic_id)
+
+    async def add_attachment(
+        self, ticket_id: str, file_path: str, description: str | None = None
+    ) -> Attachment:
+        """Attach file to JIRA issue (including Epic).
+
+        Args:
+        ----
+            ticket_id: Issue key (e.g., PROJ-123) or ID
+            file_path: Path to file to attach
+            description: Optional description (stored in metadata, not used by JIRA directly)
+
+        Returns:
+        -------
+            Attachment object with metadata
+
+        Raises:
+        ------
+            FileNotFoundError: If file doesn't exist
+            ValueError: If credentials invalid
+            HTTPStatusError: If upload fails
+
+        """
+        from pathlib import Path
+
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        file_path_obj = Path(file_path)
+        if not file_path_obj.exists():
+            raise FileNotFoundError(f"File not found: {file_path}")
+
+        # JIRA requires special header for attachment upload
+        headers = {
+            "X-Atlassian-Token": "no-check",
+            # Don't set Content-Type - let httpx handle multipart
+        }
+
+        # Prepare multipart file upload
+        with open(file_path_obj, "rb") as f:
+            files = {"file": (file_path_obj.name, f, "application/octet-stream")}
+
+            url = f"{self.api_base}/issue/{ticket_id}/attachments"
+
+            # Use existing client infrastructure
+            async with await self._get_client() as client:
+                response = await client.post(
+                    url, files=files, headers={**self.headers, **headers}
+                )
+                response.raise_for_status()
+
+                # JIRA returns array with single attachment
+                attachment_data = response.json()[0]
+
+                return Attachment(
+                    id=attachment_data["id"],
+                    ticket_id=ticket_id,
+                    filename=attachment_data["filename"],
+                    url=attachment_data["content"],
+                    content_type=attachment_data["mimeType"],
+                    size_bytes=attachment_data["size"],
+                    created_at=parse_jira_datetime(attachment_data["created"]),
+                    created_by=attachment_data["author"]["displayName"],
+                    description=description,
+                    metadata={"jira": attachment_data},
+                )
+
+    async def get_attachments(self, ticket_id: str) -> builtins.list[Attachment]:
+        """Get all attachments for a JIRA issue.
+
+        Args:
+        ----
+            ticket_id: Issue key or ID
+
+        Returns:
+        -------
+            List of Attachment objects
+
+        Raises:
+        ------
+            ValueError: If credentials invalid
+            HTTPStatusError: If request fails
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        # Fetch issue with attachment field
+        issue = await self._make_request(
+            "GET", f"issue/{ticket_id}", params={"fields": "attachment"}
+        )
+
+        attachments = []
+        for att_data in issue.get("fields", {}).get("attachment", []):
+            attachments.append(
+                Attachment(
+                    id=att_data["id"],
+                    ticket_id=ticket_id,
+                    filename=att_data["filename"],
+                    url=att_data["content"],
+                    content_type=att_data["mimeType"],
+                    size_bytes=att_data["size"],
+                    created_at=parse_jira_datetime(att_data["created"]),
+                    created_by=att_data["author"]["displayName"],
+                    metadata={"jira": att_data},
+                )
+            )
+
+        return attachments
+
+    async def delete_attachment(self, ticket_id: str, attachment_id: str) -> bool:
+        """Delete an attachment from a JIRA issue.
+
+        Args:
+        ----
+            ticket_id: Issue key or ID (for validation/context)
+            attachment_id: Attachment ID to delete
+
+        Returns:
+        -------
+            True if deleted successfully, False otherwise
+
+        Raises:
+        ------
+            ValueError: If credentials invalid
+
+        """
+        # Validate credentials before attempting operation
+        is_valid, error_message = self.validate_credentials()
+        if not is_valid:
+            raise ValueError(error_message)
+
+        try:
+            await self._make_request("DELETE", f"attachment/{attachment_id}")
+            return True
+        except HTTPStatusError as e:
+            if e.response.status_code == 404:
+                logger.warning(f"Attachment {attachment_id} not found")
+                return False
+            logger.error(
+                f"Failed to delete attachment {attachment_id}: {e.response.status_code} - {e.response.text}"
+            )
+            return False
+        except Exception as e:
+            logger.error(f"Unexpected error deleting attachment {attachment_id}: {e}")
+            return False
+
     async def close(self) -> None:
         """Close the adapter and cleanup resources."""
         # Clear caches