package cc.unitmesh.devins.ui.compose.document

import androidx.compose.foundation.layout.Box
import androidx.compose.foundation.layout.fillMaxSize
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.layout.size
import androidx.compose.material3.Icon
import androidx.compose.material3.IconButton
import androidx.compose.material3.MaterialTheme
import androidx.compose.material3.Scaffold
import androidx.compose.runtime.*
import androidx.compose.ui.Modifier
import androidx.compose.ui.unit.dp
import cc.unitmesh.agent.Platform
import cc.unitmesh.devins.document.DocumentRegistry
import cc.unitmesh.devins.document.docql.DocQLResult
import cc.unitmesh.devins.document.docql.executeDocQL
import cc.unitmesh.devins.ui.compose.agent.AgentTopAppBar
import cc.unitmesh.devins.ui.compose.agent.AgentTopAppBarActions
import cc.unitmesh.devins.ui.base.ResizableSplitPane
import cc.unitmesh.devins.ui.base.VerticalResizableSplitPane
import cc.unitmesh.devins.ui.compose.icons.AutoDevComposeIcons
import cc.unitmesh.devins.workspace.WorkspaceManager

/**
 * 文档阅读器主页面
 * 采用查询优先的 3 栏布局：
 * 1. 左侧：AI 聊天 (DocumentChatPane) - 主要交互区域，显示索引状态
 * 2. 中间：文档导航 (DocumentNavigationPane) - 文件列表
 * 3. 右侧：文档查看 (DocumentViewerPane) + 结构化信息 (StructuredInfoPane) - 垂直分割
 *
 * 工作流：用户在左侧查询 → 中间看到文件列表 → 右侧查看详细内容和结构
 */
@Composable
fun DocumentReaderPage(
    modifier: Modifier = Modifier,
    onBack: () -> Unit = {}
) {
    val workspace = remember { WorkspaceManager.getCurrentOrEmpty() }
    val viewModel = remember(workspace) { DocumentReaderViewModel(workspace) }

    val notMobile = (Platform.isAndroid || Platform.isIOS).not()

    // State for domain dictionary dialog
    var showDomainDictDialog by remember { mutableStateOf(false) }

    Scaffold(
        modifier = modifier.fillMaxSize(),
        topBar = {
            if (notMobile) {
                AgentTopAppBar(
                    title = "Knowledge Agent",
                    subtitle = workspace.name.takeIf { it.isNotBlank() },
                    onBack = onBack,
                    actions = {
                        // Domain Dictionary button - for managing terminology
                        IconButton(
                            onClick = { showDomainDictDialog = true },
                            modifier = Modifier.size(36.dp)
                        ) {
                            Icon(
                                imageVector = AutoDevComposeIcons.MenuBook,
                                contentDescription = "Domain Dictionary",
                                tint = MaterialTheme.colorScheme.onSurfaceVariant,
                                modifier = Modifier.size(20.dp)
                            )
                        }

                        AgentTopAppBarActions.DeleteButton(
                            onClick = { viewModel.clearChatHistory() },
                            contentDescription = "Clear Chat"
                        )
                        AgentTopAppBarActions.RefreshButton(
                            onClick = { viewModel.refreshDocuments() }
                        )
                    }
                )
            }
        }
    ) { paddingValues ->
        Box(modifier = Modifier.padding(paddingValues).fillMaxSize()) {
            ResizableSplitPane(
                modifier = Modifier.fillMaxSize(),
                initialSplitRatio = 0.65f,
                minRatio = 0.35f,
                maxRatio = 0.85f,
                first = {
                    DocumentChatPane(
                        viewModel = viewModel
                    )
                },
                second = {
                    ResizableSplitPane(
                        modifier = Modifier.fillMaxSize(),
                        initialSplitRatio = 0.5f,
                        minRatio = 0.3f,
                        maxRatio = 0.8f,
                        first = {
                            DocumentNavigationPane(
                                documentLoadState = viewModel.documentLoadState,
                                documents = viewModel.filteredDocuments,
                                indexingStatus = viewModel.indexingStatus.collectAsState().value,
                                searchQuery = viewModel.searchQuery,
                                onSearchQueryChange = { viewModel.updateSearchQuery(it) },
                                onDocumentSelected = { viewModel.selectDocument(it) },
                                onRefresh = { viewModel.refreshDocuments() },
                                onStartIndexing = { viewModel.startIndexing() },
                                onResetIndexing = { viewModel.resetIndexingStatus() }
                            )
                        },
                        second = {
                            // Conditional rendering:
                            // - When no document is selected, show StructuredInfoPane globally
                            // - When a document is selected, show DocumentViewerPane with StructuredInfoPane below
                            if (viewModel.selectedDocument == null) {
                                // No document selected - show global StructuredInfoPane
                                StructuredInfoPane(
                                    toc = emptyList(),
                                    entities = emptyList(),
                                    onTocSelected = { /* No action when no document selected */ },
                                    onEntitySelected = { /* No action when no document selected */ },
                                    onDocQLQuery = { query ->
                                        // Global query across all indexed documents
                                        try {
                                            DocumentRegistry.queryDocuments(query)
                                        } catch (e: Exception) {
                                            DocQLResult.Error("全局查询失败: ${e.message}")
                                        }
                                    }
                                )
                            } else {
                                // Document selected - show DocumentViewerPane with StructuredInfoPane
                                VerticalResizableSplitPane(
                                    modifier = Modifier.fillMaxSize(),
                                    initialSplitRatio = 0.5f,
                                    minRatio = 0.2f,
                                    maxRatio = 0.8f,
                                    top = {
                                        DocumentViewerPane(
                                            document = viewModel.selectedDocument,
                                            rawContent = viewModel.documentContent,
                                            parsedContent = viewModel.parsedContent,
                                            isLoading = viewModel.isLoading,
                                            indexStatus = viewModel.selectedDocumentIndexStatus,
                                            targetLineNumber = viewModel.targetLineNumber,
                                            highlightedText = viewModel.highlightedText
                                        )
                                    },
                                    bottom = {
                                        val toc = viewModel.selectedDocument?.toc ?: emptyList()
                                        val entities = viewModel.selectedDocument?.entities ?: emptyList()

                                        StructuredInfoPane(
                                            toc = toc,
                                            entities = entities,
                                            onTocSelected = { tocItem ->
                                                viewModel.navigateToTocItem(tocItem)
                                            },
                                            onEntitySelected = { entity ->
                                                viewModel.navigateToEntity(entity)
                                            },
                                            onDocQLQuery = { query ->
                                                val document = viewModel.selectedDocument
                                                if (document != null) {
                                                    // 查询当前选中的文档
                                                    executeDocQL(query, document, null)
                                                } else {
                                                    // 全局查询所有已索引的文档
                                                    try {
                                                        DocumentRegistry.queryDocuments(query)
                                                    } catch (e: Exception) {
                                                        DocQLResult.Error("全局查询失败: ${e.message}")
                                                    }
                                                }
                                            }
                                        )
                                    }
                                )
                            }
                        }
                    )
                }
            )
        }

        // Domain Dictionary Dialog
        if (showDomainDictDialog) {
            DomainDictDialog(
                workspace = workspace,
                onDismiss = { showDomainDictDialog = false }
            )
        }
    }
}