# Copyright (c) 2025 Patrick Motsch
# All rights reserved.

"""
Export Tickets As JSON action for JIRA operations.
Exports tickets from JIRA as JSON list.
"""

import logging
import json
from typing import Dict, Any
from modules.workflows.methods.methodBase import action
from modules.datamodels.datamodelChat import ActionResult, ActionDocument

logger = logging.getLogger(__name__)

@action
async def exportTicketsAsJson(self, parameters: Dict[str, Any]) -> ActionResult:
    """
    Export tickets from JIRA as JSON list.
    
    Parameters:
    - connectionId (str, required): Connection ID from connectJira action result
    - taskSyncDefinition (str or dict, optional): Field mapping definition (if not provided, uses stored definition)
    
    Returns:
    - ActionResult with ActionDocument containing list of tickets as JSON
    """
    try:
        connectionIdParam = parameters.get("connectionId")
        if not connectionIdParam:
            return ActionResult.isFailure(error="connectionId parameter is required")
        
        # Get connection ID from document if it's a reference
        connectionId = None
        if isinstance(connectionIdParam, str):
            # Try to parse from document reference
            connectionInfo = self.documentParsing.parseJsonFromDocument(connectionIdParam)
            if connectionInfo and "connectionId" in connectionInfo:
                connectionId = connectionInfo["connectionId"]
            else:
                # Assume it's the connection ID directly
                connectionId = connectionIdParam
        
        if not connectionId or connectionId not in self._connections:
            return ActionResult.isFailure(error=f"Connection ID {connectionIdParam} not found. Ensure connectJira was called first.")
        
        connection = self._connections[connectionId]
        syncInterface = connection["interface"]
        
        # Export tickets
        dataList = await syncInterface.exportTicketsAsList()
        
        logger.info(f"Exported {len(dataList)} tickets from JIRA")
        
        # Generate filename
        workflowContext = self.services.chat.getWorkflowContext() if hasattr(self.services, 'chat') else None
        filename = self._generateMeaningfulFileName(
            "jira_tickets_export",
            "json",
            workflowContext,
            "exportTicketsAsJson"
        )
        
        validationMetadata = self._createValidationMetadata(
            "exportTicketsAsJson",
            connectionId=connectionId,
            ticketCount=len(dataList)
        )
        
        document = ActionDocument(
            documentName=filename,
            documentData=json.dumps(dataList, indent=2, ensure_ascii=False),
            mimeType="application/json",
            validationMetadata=validationMetadata
        )
        
        return ActionResult.isSuccess(documents=[document])
        
    except Exception as e:
        errorMsg = f"Error exporting tickets from JIRA: {str(e)}"
        logger.error(errorMsg)
        return ActionResult.isFailure(error=errorMsg)