package cc.unitmesh.devins.ui.compose.agent

import androidx.compose.foundation.clickable
import androidx.compose.foundation.hoverable
import androidx.compose.foundation.interaction.MutableInteractionSource
import androidx.compose.foundation.interaction.collectIsHoveredAsState
import androidx.compose.foundation.layout.Arrangement
import androidx.compose.foundation.layout.Column
import androidx.compose.foundation.layout.ExperimentalLayoutApi
import androidx.compose.foundation.layout.Row
import androidx.compose.foundation.layout.Spacer
import androidx.compose.foundation.layout.fillMaxWidth
import androidx.compose.foundation.layout.height
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.layout.size
import androidx.compose.animation.AnimatedVisibility
import androidx.compose.animation.expandVertically
import androidx.compose.animation.fadeIn
import androidx.compose.animation.shrinkVertically
import androidx.compose.animation.fadeOut
import androidx.compose.foundation.shape.RoundedCornerShape
import androidx.compose.material3.Surface
import androidx.compose.material3.Icon
import androidx.compose.material3.IconButton
import androidx.compose.material3.MaterialTheme
import androidx.compose.material3.Text
import androidx.compose.material3.TextButton
import androidx.compose.runtime.Composable
import androidx.compose.runtime.LaunchedEffect
import androidx.compose.runtime.getValue
import androidx.compose.runtime.mutableStateOf
import androidx.compose.runtime.remember
import androidx.compose.runtime.setValue
import androidx.compose.ui.Alignment
import androidx.compose.ui.Modifier
import androidx.compose.ui.platform.LocalClipboardManager
import androidx.compose.ui.text.AnnotatedString
import androidx.compose.ui.text.font.FontFamily
import androidx.compose.ui.text.font.FontWeight
import androidx.compose.ui.unit.dp
import cc.unitmesh.agent.Platform
import cc.unitmesh.agent.render.RendererUtils
import cc.unitmesh.agent.tool.ToolType
import cc.unitmesh.agent.tool.impl.docql.DocQLSearchStats
import cc.unitmesh.devins.ui.compose.agent.knowledge.DocQLDetailDialog
import cc.unitmesh.devins.ui.compose.agent.knowledge.DocQLStatsSection
import cc.unitmesh.devins.ui.compose.agent.codereview.FileViewerDialog
import cc.unitmesh.devins.ui.compose.icons.AutoDevComposeIcons
import cc.unitmesh.devins.ui.compose.sketch.TextBlockRenderer
import cc.unitmesh.devins.ui.compose.theme.AutoDevColors
import cc.unitmesh.devins.ui.compose.terminal.PlatformTerminalDisplay

/**
 * Combined tool call and result display - shows both in a single compact row
 * Similar to TerminalOutputItem but for general tools (ReadFile, WriteFile, Glob, etc.)
 */
@OptIn(ExperimentalLayoutApi::class)
@Composable
fun ToolItem(
    toolName: String,
    details: String?,
    fullParams: String? = null,
    filePath: String? = null,
    toolType: ToolType? = null,
    success: Boolean? = null, // null means still executing
    summary: String? = null,
    output: String? = null,
    fullOutput: String? = null,
    executionTimeMs: Long? = null,
    docqlStats: DocQLSearchStats? = null,
    onOpenFileViewer: ((String) -> Unit)? = null,
    onExpand: () -> Unit = {}
) {
    var expanded by remember { mutableStateOf(success == false) } // Auto-expand on error

    LaunchedEffect(expanded) {
        if (expanded) {
            onExpand()
        }
    }
    var showFullOutput by remember { mutableStateOf(success == false) }
    var showFileViewerDialog by remember { mutableStateOf(false) }
    var showStats by remember { mutableStateOf(false) }
    var showDetailDialog by remember { mutableStateOf(false) }
    val clipboardManager = LocalClipboardManager.current

    val isDesktop = Platform.isJvm && !Platform.isAndroid
    val sectionGap = if (isDesktop) 6.dp else 8.dp
    val blockPadding = if (isDesktop) 6.dp else 8.dp

    // Parameters are debug-only most of the time (details vs raw).
    // Default to showing params only when there's no output to show.
    val paramsText = fullParams ?: details
    val hasParams = !paramsText.isNullOrBlank()
    val hasAnyOutput = !output.isNullOrBlank() || !fullOutput.isNullOrBlank()
    var showParams by remember(toolName, fullParams, details) {
        mutableStateOf(!hasAnyOutput && hasParams)
    }

    // Determine which output to display
    val displayOutput = if (showFullOutput) fullOutput else output
    val hasFullOutput = fullOutput != null && fullOutput != output

    // Check if this is a file operation that can be viewed
    val isFileOperation =
        toolType in
            listOf(
                ToolType.ReadFile,
                ToolType.WriteFile
            )

    val isExecuting = success == null
    val hasStats = docqlStats != null

    val isRetrievalToolCall = remember(toolName, toolType) {
        RendererUtils.isRetrievalToolCall(toolName = toolName, toolType = toolType)
    }

    val containerPadding = if (isRetrievalToolCall) 6.dp else 8.dp
    val headerGap = if (isRetrievalToolCall) 6.dp else 8.dp
    val statusIconSize = if (isRetrievalToolCall) 14.dp else 16.dp
    val chevronIconSize = if (isRetrievalToolCall) 18.dp else 20.dp

    val mutedTextColor = MaterialTheme.colorScheme.onSurfaceVariant.copy(alpha = 0.75f)
    val containerColor = if (isRetrievalToolCall) {
        MaterialTheme.colorScheme.surfaceVariant.copy(alpha = 0.25f)
    } else {
        MaterialTheme.colorScheme.surface
    }

    val statusTint = when {
        success == false -> MaterialTheme.colorScheme.error
        isRetrievalToolCall -> MaterialTheme.colorScheme.onSurfaceVariant.copy(alpha = 0.55f)
        isExecuting -> MaterialTheme.colorScheme.primary
        success == true -> AutoDevColors.Signal.success
        else -> MaterialTheme.colorScheme.onSurfaceVariant
    }

    Surface(
        color = containerColor,
        shape = RoundedCornerShape(4.dp),
        modifier = Modifier
    ) {
        Column(modifier = Modifier.padding(containerPadding)) {
            val hoverInteractionSource = remember { MutableInteractionSource() }
            val isHovered by hoverInteractionSource.collectIsHoveredAsState()
            val showDebugActions = !isDesktop || isHovered || showParams

            Row(
                modifier =
                    Modifier
                        .fillMaxWidth()
                        .hoverable(hoverInteractionSource)
                        .clickable(
                            indication = null,
                            interactionSource = remember { MutableInteractionSource() }
                        ) { if (hasAnyOutput || hasParams) expanded = !expanded },
                verticalAlignment = Alignment.Companion.CenterVertically,
                horizontalArrangement = Arrangement.spacedBy(headerGap)
            ) {
                Icon(
                    imageVector = when {
                        isExecuting -> AutoDevComposeIcons.PlayArrow
                        success -> AutoDevComposeIcons.CheckCircle
                        else -> AutoDevComposeIcons.Error
                    },
                    contentDescription = when {
                        isExecuting -> "Executing"
                        success -> "Success"
                        else -> "Failed"
                    },
                    tint = statusTint,
                    modifier = Modifier.Companion.size(statusIconSize)
                )

                val displayToolName = when {
                    toolName.lowercase() == "docql" && docqlStats != null -> {
                        val query = docqlStats.query.take(30).let {
                            if (docqlStats.query.length > 30) "$it..." else it
                        }
                        if (query.isNotEmpty()) "DocQL - $query" else "DocQL"
                    }

                    else -> toolName
                }

                Text(
                    text = displayToolName,
                    fontWeight = if (isRetrievalToolCall) FontWeight.Companion.Medium else FontWeight.Companion.Bold,
                    color = if (isRetrievalToolCall) mutedTextColor else MaterialTheme.colorScheme.onSurface,
                    style = if (isRetrievalToolCall) MaterialTheme.typography.labelMedium else MaterialTheme.typography.bodyMedium,
                    modifier = Modifier.Companion.weight(1f)
                )

                val displaySummary = when {
                    toolName.lowercase() == "docql" && docqlStats?.smartSummary != null ->
                        "${docqlStats.totalRawResults} results"

                    else -> summary
                }

                if (displaySummary != null) {
                    Text(
                        text = "-> $displaySummary",
                        color = when (success) {
                            false -> MaterialTheme.colorScheme.error
                            true -> if (isRetrievalToolCall) mutedTextColor else AutoDevColors.Signal.success
                            null -> if (isRetrievalToolCall) mutedTextColor else MaterialTheme.colorScheme.onSurfaceVariant
                        },
                        style = if (isRetrievalToolCall) MaterialTheme.typography.labelSmall else MaterialTheme.typography.bodyMedium,
                        fontWeight = if (isRetrievalToolCall) FontWeight.Companion.Normal else FontWeight.Companion.Medium,
                        maxLines = 1
                    )
                }

                if (executionTimeMs != null && executionTimeMs > 0) {
                    Text(
                        text = "${executionTimeMs}ms",
                        color = MaterialTheme.colorScheme.onSurfaceVariant.copy(alpha = if (isRetrievalToolCall) 0.45f else 0.6f),
                        style = MaterialTheme.typography.labelSmall
                    )
                }

                if (hasStats) {
                    IconButton(
                        onClick = { showDetailDialog = true },
                        modifier = Modifier.Companion.size(24.dp)
                    ) {
                        Icon(
                            imageVector = AutoDevComposeIcons.Description,
                            contentDescription = "View Details",
                            tint = MaterialTheme.colorScheme.primary,
                            modifier = Modifier.Companion.size(18.dp)
                        )
                    }

                    IconButton(
                        onClick = { showStats = !showStats },
                        modifier = Modifier.Companion.size(24.dp)
                    ) {
                        Icon(
                            imageVector = AutoDevComposeIcons.Analytics,
                            contentDescription = if (showStats) "Hide Stats" else "Show Stats",
                            tint = if (showStats) MaterialTheme.colorScheme.primary else MaterialTheme.colorScheme.onSurfaceVariant,
                            modifier = Modifier.Companion.size(18.dp)
                        )
                    }
                }

                if (isFileOperation && !filePath.isNullOrEmpty()) {
                    IconButton(
                        onClick = {
                            if (onOpenFileViewer != null) {
                                onOpenFileViewer(filePath)
                            } else {
                                showFileViewerDialog = true
                            }
                        },
                        modifier = Modifier.Companion.size(24.dp)
                    ) {
                        Icon(
                            imageVector = AutoDevComposeIcons.Visibility,
                            contentDescription = "View File",
                            tint = MaterialTheme.colorScheme.primary,
                            modifier = Modifier.Companion.size(18.dp)
                        )
                    }
                }

                // Debug params: show a compact icon only when hovered (desktop) or always (mobile).
                if (hasParams && showDebugActions) {
                    IconButton(
                        onClick = {
                            if (!expanded) expanded = true
                            showParams = !showParams
                        },
                        modifier = Modifier.Companion.size(24.dp)
                    ) {
                        Icon(
                            imageVector = AutoDevComposeIcons.BugReport,
                            contentDescription = if (showParams) "Hide parameters" else "Show parameters",
                            tint = if (showParams) MaterialTheme.colorScheme.primary else MaterialTheme.colorScheme.onSurfaceVariant,
                            modifier = Modifier.Companion.size(18.dp)
                        )
                    }
                }

                if (hasAnyOutput || hasParams) {
                    Icon(
                        imageVector = if (expanded) AutoDevComposeIcons.ExpandLess else AutoDevComposeIcons.ExpandMore,
                        contentDescription = if (expanded) "Collapse" else "Expand",
                        tint = (if (isRetrievalToolCall) mutedTextColor else MaterialTheme.colorScheme.onSurface.copy(alpha = 0.7f)),
                        modifier = Modifier.Companion.size(chevronIconSize)
                    )
                }
            }

            AnimatedVisibility(
                visible = expanded,
                enter = expandVertically() + fadeIn(),
                exit = shrinkVertically() + fadeOut()
            ) {
                Column {
                    if (showParams && hasParams) {
                        Spacer(modifier = Modifier.Companion.height(sectionGap))
                        Surface(
                            color = MaterialTheme.colorScheme.surfaceVariant.copy(alpha = 0.5f),
                            shape = RoundedCornerShape(4.dp),
                            modifier = Modifier.Companion.fillMaxWidth()
                        ) {
                            Text(
                                text = paramsText.orEmpty(),
                                modifier = Modifier.Companion.padding(blockPadding),
                                color = MaterialTheme.colorScheme.onSurface,
                                style = MaterialTheme.typography.bodySmall,
                                fontFamily = FontFamily.Companion.Monospace
                            )
                        }
                    }

                    if (displayOutput != null) {
                        Spacer(modifier = Modifier.Companion.height(sectionGap))

                        Row(
                            modifier = Modifier.Companion.fillMaxWidth(),
                            horizontalArrangement = Arrangement.SpaceBetween,
                            verticalAlignment = Alignment.Companion.Top
                        ) {
                            Row(
                                verticalAlignment = Alignment.Companion.CenterVertically,
                                horizontalArrangement = Arrangement.spacedBy(8.dp)
                            ) {
                                Text(
                                    text = "Output:",
                                    fontWeight = FontWeight.Companion.Medium,
                                    color = MaterialTheme.colorScheme.onSurface,
                                    style = MaterialTheme.typography.labelSmall
                                )

                                if (hasFullOutput) {
                                    TextButton(
                                        onClick = { showFullOutput = !showFullOutput },
                                        modifier = Modifier.Companion.height(28.dp)
                                    ) {
                                        Text(
                                            text = if (showFullOutput) "Show Less" else "Show Full",
                                            style = MaterialTheme.typography.labelSmall,
                                            color = MaterialTheme.colorScheme.primary
                                        )
                                    }
                                }

                                if (hasStats && docqlStats.detailedResults != null) {
                                    TextButton(
                                        onClick = { showDetailDialog = true },
                                        modifier = Modifier.Companion.height(28.dp)
                                    ) {
                                        Icon(
                                            imageVector = AutoDevComposeIcons.OpenInNew,
                                            contentDescription = "View All",
                                            tint = MaterialTheme.colorScheme.primary,
                                            modifier = Modifier.Companion.size(14.dp)
                                        )
                                        Text(
                                            text = "View All",
                                            style = MaterialTheme.typography.labelSmall,
                                            color = MaterialTheme.colorScheme.primary,
                                            modifier = Modifier.Companion.padding(start = 4.dp)
                                        )
                                    }
                                }
                            }

                            IconButton(
                                onClick = {
                                    clipboardManager.setText(
                                        androidx.compose.ui.text.AnnotatedString(
                                            displayOutput
                                        )
                                    )
                                },
                                modifier = Modifier.Companion.size(24.dp)
                            ) {
                                Icon(
                                    imageVector = AutoDevComposeIcons.ContentCopy,
                                    contentDescription = "Copy output",
                                    tint = MaterialTheme.colorScheme.onSurface.copy(alpha = 0.7f),
                                    modifier = Modifier.Companion.size(16.dp)
                                )
                            }
                        }

                        if (toolName.lowercase().contains("docql")) {
                            Surface(
                                color = MaterialTheme.colorScheme.surfaceVariant.copy(alpha = 0.3f),
                                shape = RoundedCornerShape(4.dp),
                                modifier = Modifier.Companion.fillMaxWidth()
                            ) {
                                TextBlockRenderer(displayOutput)
                            }
                        } else {
                            Surface(
                                color = MaterialTheme.colorScheme.surfaceVariant.copy(alpha = 0.5f),
                                shape = RoundedCornerShape(4.dp),
                                modifier = Modifier.Companion.fillMaxWidth()
                            ) {
                                Text(
                                    text = formatOutput(displayOutput),
                                    modifier = Modifier.Companion.padding(blockPadding),
                                    color = MaterialTheme.colorScheme.onSurface,
                                    style = MaterialTheme.typography.bodySmall,
                                    fontFamily = FontFamily.Companion.Monospace
                                )
                            }
                        }
                    }
                }
            }

            if (showStats && docqlStats != null) {
                Spacer(modifier = Modifier.Companion.height(8.dp))
                DocQLStatsSection(stats = docqlStats)
            }
        }
    }

    // Show file viewer dialog when onOpenFileViewer is null
    if (showFileViewerDialog && !filePath.isNullOrEmpty()) {
        FileViewerDialog(
            filePath = filePath,
            onClose = { showFileViewerDialog = false }
        )
    }

    // Show DocQL detail dialog
    if (showDetailDialog && docqlStats != null) {
        // Extract keyword from details or use a default
        val keyword = details?.substringAfter("query: ")?.substringBefore(",")?.trim('"') ?: "search"
        DocQLDetailDialog(
            keyword = keyword,
            stats = docqlStats,
            onDismiss = { showDetailDialog = false }
        )
    }
}

fun formatOutput(output: String): String {
    return when {
        output.trim().startsWith("{") || output.trim().startsWith("[") -> {
            try {
                output.replace(",", ",\n")
                    .replace("{", "{\n  ")
                    .replace("}", "\n}")
                    .replace("[", "[\n  ")
                    .replace("]", "\n]")
            } catch (e: Exception) {
                output
            }
        }

        output.contains("│") -> output
        output.contains("\n") -> output
        output.length > 100 -> "${output.take(100)}..."
        else -> output
    }
}

@Composable
fun TerminalOutputItem(
    output: String,
    exitCode: Int,
    executionTimeMs: Long
) {
    var expanded by remember {
        mutableStateOf(
            when {
                Platform.isJvm -> false
                exitCode != 0 -> true
                else -> false
            }
        )
    } // Auto-expand on error
    val isSuccess = exitCode == 0

    Surface(
        color = MaterialTheme.colorScheme.surface,
        shape = RoundedCornerShape(4.dp),
        modifier = Modifier
    ) {
        Column(modifier = Modifier.Companion.padding(8.dp)) {
            Row(
                modifier =
                    Modifier.Companion
                        .fillMaxWidth()
                        .clickable(
                            indication = null,
                            interactionSource = remember { MutableInteractionSource() }
                        ) { expanded = !expanded },
                verticalAlignment = Alignment.Companion.CenterVertically,
                horizontalArrangement = Arrangement.spacedBy(8.dp)
            ) {
                Row(
                    verticalAlignment = Alignment.Companion.CenterVertically,
                    horizontalArrangement = Arrangement.spacedBy(4.dp),
                    modifier = Modifier.Companion.weight(1f)
                ) {
                    Icon(
                        imageVector = if (isSuccess) AutoDevComposeIcons.CheckCircle else AutoDevComposeIcons.Error,
                        contentDescription = if (isSuccess) "Success" else "Error",
                        tint = if (isSuccess) AutoDevColors.Signal.success else MaterialTheme.colorScheme.error,
                        modifier = Modifier.Companion.size(16.dp)
                    )
                    Text(
                        text = if (isSuccess) "Exit 0" else "Exit $exitCode",
                        color = if (isSuccess) AutoDevColors.Signal.success else MaterialTheme.colorScheme.error,
                        style = MaterialTheme.typography.bodyMedium,
                        fontWeight = FontWeight.Companion.Medium
                    )
                }
                Text(
                    text = "${executionTimeMs}ms",
                    color = MaterialTheme.colorScheme.onSurfaceVariant.copy(alpha = 0.6f),
                    style = MaterialTheme.typography.labelSmall
                )
                Icon(
                    imageVector = if (expanded) AutoDevComposeIcons.ExpandLess else AutoDevComposeIcons.ExpandMore,
                    contentDescription = if (expanded) "Collapse" else "Expand",
                    tint = MaterialTheme.colorScheme.onSurface.copy(alpha = 0.7f),
                    modifier = Modifier.Companion.size(20.dp)
                )
            }

            AnimatedVisibility(
                visible = expanded && output.isNotEmpty(),
                enter = expandVertically() + fadeIn(),
                exit = shrinkVertically() + fadeOut()
            ) {
                PlatformTerminalDisplay(
                    output = output,
                    modifier = Modifier.Companion.padding(8.dp)
                )
            }
        }
    }
}