// InsightFlow Frontend - Phase 4 (Agent Assistant + Provenance)
const API_BASE = '/api/v1';

let currentProject = null;
let currentData = null;
let selectedEntity = null;
let projectRelations = [];
let projectEntities = [];
let entityDetailsCache = {};

// Init
document.addEventListener('DOMContentLoaded', () => {
    const isWorkbench = window.location.pathname.includes('workbench');
    if (isWorkbench) {
        initWorkbench();
    }
});

// Initialize workbench
async function initWorkbench() {
    const projectId = localStorage.getItem('currentProject');
    if (!projectId) {
        window.location.href = '/';
        return;
    }
    
    try {
        const projects = await fetchProjects();
        currentProject = projects.find(p => p.id === projectId);
        
        if (!currentProject) {
            localStorage.removeItem('currentProject');
            window.location.href = '/';
            return;
        }
        
        const nameEl = document.getElementById('projectName');
        if (nameEl) nameEl.textContent = currentProject.name;
        
        initUpload();
        initAgentPanel();
        initEntityCard();
        await loadProjectData();
        
    } catch (err) {
        console.error('Init failed:', err);
        alert('加载失败，请返回项目列表');
    }
}

// ==================== API Calls ====================

async function fetchProjects() {
    const res = await fetch(`${API_BASE}/projects`);
    if (!res.ok) throw new Error('Failed to fetch projects');
    return await res.json();
}

async function uploadAudio(file) {
    const formData = new FormData();
    formData.append('file', file);
    
    const res = await fetch(`${API_BASE}/projects/${currentProject.id}/upload`, {
        method: 'POST',
        body: formData
    });
    
    if (!res.ok) throw new Error('Upload failed');
    return await res.json();
}

async function loadProjectData() {
    try {
        const [entitiesRes, relationsRes] = await Promise.all([
            fetch(`${API_BASE}/projects/${currentProject.id}/entities`),
            fetch(`${API_BASE}/projects/${currentProject.id}/relations`)
        ]);
        
        if (entitiesRes.ok) {
            projectEntities = await entitiesRes.json();
        }
        if (relationsRes.ok) {
            projectRelations = await relationsRes.json();
        }
        
        // 预加载实体详情
        await preloadEntityDetails();
        
        currentData = {
            transcript_id: 'project_view',
            project_id: currentProject.id,
            segments: [],
            entities: projectEntities,
            full_text: '',
            created_at: new Date().toISOString()
        };
        
        renderGraph();
        renderEntityList();
        
    } catch (err) {
        console.error('Load project data failed:', err);
    }
}

async function preloadEntityDetails() {
    // 并行加载所有实体详情
    const promises = projectEntities.map(async (ent) => {
        try {
            const res = await fetch(`${API_BASE}/entities/${ent.id}/details`);
            if (res.ok) {
                entityDetailsCache[ent.id] = await res.json();
            }
        } catch (e) {
            console.error(`Failed to load entity ${ent.id} details:`, e);
        }
    });
    await Promise.all(promises);
}

// ==================== Agent Panel ====================

function initAgentPanel() {
    const chatInput = document.getElementById('chatInput');
    if (chatInput) {
        chatInput.addEventListener('keypress', (e) => {
            if (e.key === 'Enter' && !e.shiftKey) {
                e.preventDefault();
                sendAgentMessage();
            }
        });
    }
}

function toggleAgentPanel() {
    const panel = document.getElementById('agentPanel');
    const toggle = panel.querySelector('.agent-toggle');
    panel.classList.toggle('collapsed');
    toggle.textContent = panel.classList.contains('collapsed') ? '‹' : '›';
}

function addChatMessage(content, isUser = false, isTyping = false) {
    const container = document.getElementById('chatMessages');
    const msgDiv = document.createElement('div');
    msgDiv.className = `chat-message ${isUser ? 'user' : 'assistant'}`;
    
    if (isTyping) {
        msgDiv.innerHTML = `
            <div class="message-bubble typing-indicator" id="typingIndicator">
                <span></span><span></span><span></span>
            </div>
        `;
    } else {
        msgDiv.innerHTML = `<div class="message-bubble">${content}</div>`;
    }
    
    container.appendChild(msgDiv);
    container.scrollTop = container.scrollHeight;
    return msgDiv;
}

function removeTypingIndicator() {
    const indicator = document.getElementById('typingIndicator');
    if (indicator) {
        indicator.parentElement.remove();
    }
}

async function sendAgentMessage() {
    const input = document.getElementById('chatInput');
    const message = input.value.trim();
    if (!message) return;
    
    input.value = '';
    addChatMessage(message, true);
    addChatMessage('', false, true);
    
    try {
        // 判断是命令还是问答
        const isCommand = message.includes('合并') || message.includes('修改') || 
                          message.startsWith('把') || message.startsWith('将');
        
        if (isCommand) {
            // 执行命令
            const res = await fetch(`${API_BASE}/projects/${currentProject.id}/agent/command`, {
                method: 'POST',
                headers: { 'Content-Type': 'application/json' },
                body: JSON.stringify({ command: message })
            });
            
            removeTypingIndicator();
            
            if (res.ok) {
                const result = await res.json();
                let response = '';
                
                if (result.intent === 'merge_entities') {
                    if (result.success) {
                        response = `✅ 已合并 ${result.merged.length} 个实体到 "${result.target}"`;
                        await loadProjectData(); // 刷新数据
                    } else {
                        response = `❌ 合并失败：${result.error || '未找到匹配的实体'}`;
                    }
                } else if (result.intent === 'edit_entity') {
                    if (result.success) {
                        response = `✅ 已更新实体 "${result.entity?.name}"`;
                        await loadProjectData();
                    } else {
                        response = `❌ 编辑失败：${result.error || '未找到实体'}`;
                    }
                } else if (result.intent === 'answer_question') {
                    response = result.answer;
                } else {
                    response = result.message || result.explanation || '未识别的指令';
                }
                
                addChatMessage(response);
            } else {
                addChatMessage('❌ 请求失败，请重试');
            }
        } else {
            // RAG 问答
            const res = await fetch(`${API_BASE}/projects/${currentProject.id}/agent/query`, {
                method: 'POST',
                headers: { 'Content-Type': 'application/json' },
                body: JSON.stringify({ query: message, stream: false })
            });
            
            removeTypingIndicator();
            
            if (res.ok) {
                const result = await res.json();
                addChatMessage(result.answer);
            } else {
                addChatMessage('❌ 获取回答失败，请重试');
            }
        }
    } catch (err) {
        removeTypingIndicator();
        addChatMessage('❌ 网络错误，请检查连接');
        console.error('Agent error:', err);
    }
}

async function loadSuggestions() {
    addChatMessage('正在获取建议...', false, true);
    
    try {
        const res = await fetch(`${API_BASE}/projects/${currentProject.id}/agent/suggest`);
        removeTypingIndicator();
        
        if (res.ok) {
            const result = await res.json();
            const suggestions = result.suggestions || [];
            
            if (suggestions.length === 0) {
                addChatMessage('暂无建议，请先上传一些音频文件。');
                return;
            }
            
            let html = '<div style="margin-bottom:8px;">💡 基于项目数据的建议：</div>';
            suggestions.forEach((s, i) => {
                html += `
                    <div class="suggestion-card" onclick="applySuggestion(${i})">
                        <div class="suggestion-type">${s.type === 'action' ? '⚡ 操作' : '💡 洞察'}</div>
                        <div class="suggestion-title">${s.title}</div>
                        <div class="suggestion-desc">${s.description}</div>
                    </div>
                `;
            });
            
            const msgDiv = document.createElement('div');
            msgDiv.className = 'chat-message assistant';
            msgDiv.innerHTML = `<div class="message-bubble" style="padding:0;background:none;border:none;">${html}</div>`;
            document.getElementById('chatMessages').appendChild(msgDiv);
        }
    } catch (err) {
        removeTypingIndicator();
        addChatMessage('❌ 获取建议失败');
    }
}

function applySuggestion(index) {
    // 可以在这里实现建议的自动应用
    addChatMessage('建议功能开发中，敬请期待！');
}

// ==================== Transcript Rendering ====================

function renderTranscript() {
    const container = document.getElementById('transcriptContent');
    if (!container || !currentData || !currentData.segments) return;
    
    container.innerHTML = '';
    
    currentData.segments.forEach((seg, idx) => {
        const div = document.createElement('div');
        div.className = 'segment';
        div.dataset.index = idx;
        
        let text = seg.text;
        const entities = findEntitiesInText(seg.text);
        
        entities.sort((a, b) => b.start - a.start);
        
        entities.forEach(ent => {
            const before = text.slice(0, ent.start);
            const name = text.slice(ent.start, ent.end);
            const after = text.slice(ent.end);
            const details = entityDetailsCache[ent.id];
            const confidence = details?.mentions?.[0]?.confidence || 1.0;
            const lowConfClass = confidence < 0.7 ? 'low-confidence' : '';
            
            text = before + `<span class="entity ${lowConfClass}" data-id="${ent.id}" onclick="window.selectEntity('${ent.id}')" onmouseenter="showEntityCard(event, '${ent.id}')" onmouseleave="hideEntityCard()">${name}</span>` + after;
        });
        
        div.innerHTML = `
            <div class="speaker">${seg.speaker}</div>
            <div class="segment-text">${text}</div>
        `;
        
        container.appendChild(div);
    });
}

function findEntitiesInText(text) {
    if (!projectEntities || projectEntities.length === 0) return [];
    
    const found = [];
    projectEntities.forEach(ent => {
        const name = ent.name;
        let pos = 0;
        while ((pos = text.indexOf(name, pos)) !== -1) {
            found.push({
                id: ent.id,
                name: ent.name,
                start: pos,
                end: pos + name.length
            });
            pos += 1;
        }
        
        if (ent.aliases && ent.aliases.length > 0) {
            ent.aliases.forEach(alias => {
                let aliasPos = 0;
                while ((aliasPos = text.indexOf(alias, aliasPos)) !== -1) {
                    found.push({
                        id: ent.id,
                        name: alias,
                        start: aliasPos,
                        end: aliasPos + alias.length
                    });
                    aliasPos += 1;
                }
            });
        }
    });
    
    return found;
}

// ==================== Entity Card ====================

function initEntityCard() {
    const card = document.getElementById('entityCard');
    
    // 鼠标移出卡片时隐藏
    card.addEventListener('mouseleave', () => {
        card.classList.remove('show');
    });
}

function showEntityCard(event, entityId) {
    const card = document.getElementById('entityCard');
    const details = entityDetailsCache[entityId];
    const entity = projectEntities.find(e => e.id === entityId);
    
    if (!entity) return;
    
    // 更新卡片内容
    document.getElementById('cardName').textContent = entity.name;
    document.getElementById('cardBadge').textContent = entity.type;
    document.getElementById('cardBadge').className = `entity-type-badge type-${entity.type.toLowerCase()}`;
    document.getElementById('cardDefinition').textContent = entity.definition || '暂无定义';
    
    const mentionCount = details?.mentions?.length || 0;
    const relationCount = details?.relations?.length || 0;
    document.getElementById('cardMentions').textContent = `${mentionCount} 次提及`;
    document.getElementById('cardRelations').textContent = `${relationCount} 个关系`;
    
    // 定位卡片
    const rect = event.target.getBoundingClientRect();
    card.style.left = `${rect.left}px`;
    card.style.top = `${rect.bottom + 10}px`;
    
    // 确保不超出屏幕
    const cardRect = card.getBoundingClientRect();
    if (cardRect.right > window.innerWidth) {
        card.style.left = `${window.innerWidth - cardRect.width - 20}px`;
    }
    
    card.classList.add('show');
}

function hideEntityCard() {
    // 延迟隐藏，允许鼠标移到卡片上
    setTimeout(() => {
        const card = document.getElementById('entityCard');
        if (!card.matches(':hover')) {
            card.classList.remove('show');
        }
    }, 100);
}

// ==================== Graph Visualization ====================

function renderGraph() {
    const svg = d3.select('#graph-svg');
    svg.selectAll('*').remove();
    
    if (!projectEntities || projectEntities.length === 0) {
        svg.append('text')
            .attr('x', '50%')
            .attr('y', '50%')
            .attr('text-anchor', 'middle')
            .attr('fill', '#666')
            .text('暂无实体数据，请上传音频');
        return;
    }
    
    const container = svg.node().parentElement;
    const width = container.clientWidth;
    const height = container.clientHeight - 200;
    
    svg.attr('width', width).attr('height', height);
    
    const nodes = projectEntities.map(e => ({
        id: e.id,
        name: e.name,
        type: e.type,
        definition: e.definition,
        ...e
    }));
    
    const links = projectRelations.map(r => ({
        id: r.id,
        source: r.source_id,
        target: r.target_id,
        type: r.type,
        evidence: r.evidence
    })).filter(r => r.source && r.target);
    
    if (links.length === 0 && nodes.length > 1) {
        for (let i = 0; i < Math.min(nodes.length - 1, 5); i++) {
            links.push({ source: nodes[0].id, target: nodes[i + 1].id, type: 'related' });
        }
    }
    
    const colorMap = {
        'PROJECT': '#7b2cbf',
        'TECH': '#00d4ff',
        'PERSON': '#ff6b6b',
        'ORG': '#4ecdc4',
        'OTHER': '#666'
    };
    
    const simulation = d3.forceSimulation(nodes)
        .force('link', d3.forceLink(links).id(d => d.id).distance(120))
        .force('charge', d3.forceManyBody().strength(-400))
        .force('center', d3.forceCenter(width / 2, height / 2))
        .force('collision', d3.forceCollide().radius(50));
    
    // 关系连线
    const link = svg.append('g')
        .selectAll('line')
        .data(links)
        .enter().append('line')
        .attr('stroke', '#444')
        .attr('stroke-width', 1.5)
        .attr('stroke-opacity', 0.6)
        .style('cursor', 'pointer')
        .on('click', (e, d) => showProvenance(d));
    
    // 关系标签
    const linkLabel = svg.append('g')
        .selectAll('text')
        .data(links)
        .enter().append('text')
        .attr('font-size', '10px')
        .attr('fill', '#666')
        .attr('text-anchor', 'middle')
        .style('pointer-events', 'none')
        .text(d => d.type);
    
    // 节点组
    const node = svg.append('g')
        .selectAll('g')
        .data(nodes)
        .enter().append('g')
        .attr('class', 'node')
        .call(d3.drag()
            .on('start', dragstarted)
            .on('drag', dragged)
            .on('end', dragended))
        .on('click', (e, d) => window.selectEntity(d.id))
        .on('mouseenter', (e, d) => showEntityCard(e, d.id))
        .on('mouseleave', hideEntityCard);
    
    // 节点圆圈
    node.append('circle')
        .attr('r', 35)
        .attr('fill', d => colorMap[d.type] || '#666')
        .attr('stroke', '#fff')
        .attr('stroke-width', 2)
        .attr('class', 'node-circle');
    
    // 节点文字
    node.append('text')
        .text(d => d.name.length > 6 ? d.name.slice(0, 5) + '...' : d.name)
        .attr('text-anchor', 'middle')
        .attr('dy', 5)
        .attr('fill', '#fff')
        .attr('font-size', '11px')
        .attr('font-weight', '500')
        .style('pointer-events', 'none');
    
    // 节点类型图标
    node.append('text')
        .attr('dy', -45)
        .attr('text-anchor', 'middle')
        .attr('fill', d => colorMap[d.type] || '#666')
        .attr('font-size', '10px')
        .text(d => d.type)
        .style('pointer-events', 'none');
    
    simulation.on('tick', () => {
        link
            .attr('x1', d => d.source.x)
            .attr('y1', d => d.source.y)
            .attr('x2', d => d.target.x)
            .attr('y2', d => d.target.y);
        
        linkLabel
            .attr('x', d => (d.source.x + d.target.x) / 2)
            .attr('y', d => (d.source.y + d.target.y) / 2);
        
        node.attr('transform', d => `translate(${d.x},${d.y})`);
    });
    
    function dragstarted(e, d) {
        if (!e.active) simulation.alphaTarget(0.3).restart();
        d.fx = d.x;
        d.fy = d.y;
    }
    
    function dragged(e, d) {
        d.fx = e.x;
        d.fy = e.y;
    }
    
    function dragended(e, d) {
        if (!e.active) simulation.alphaTarget(0);
        d.fx = null;
        d.fy = null;
    }
}

// ==================== Provenance ====================

async function showProvenance(relation) {
    const modal = document.getElementById('provenanceModal');
    const body = document.getElementById('provenanceBody');
    
    modal.classList.add('show');
    body.innerHTML = '<p style="color:#666;">加载中...</p>';
    
    try {
        let content = '';
        
        if (relation.id) {
            // 从API获取溯源信息
            const res = await fetch(`${API_BASE}/relations/${relation.id}/provenance`);
            if (res.ok) {
                const data = await res.json();
                content = `
                    <div style="margin-bottom:20px;">
                        <div style="color:#888;font-size:0.85rem;margin-bottom:8px;">关系类型</div>
                        <div style="font-size:1.1rem;color:#00d4ff;">${data.source} → ${data.type} → ${data.target}</div>
                    </div>
                    
                    <div style="margin-bottom:20px;">
                        <div style="color:#888;font-size:0.85rem;margin-bottom:8px;">来源文档</div>
                        <div>${data.transcript?.filename || '未知文件'}</div>
                    </div>
                    
                    <div style="color:#888;font-size:0.85rem;margin-bottom:8px;">证据文本</div>
                    <div class="provenance-evidence">"${data.evidence || '无证据文本'}"</div>
                `;
            } else {
                content = '<p style="color:#ff6b6b;">获取溯源信息失败</p>';
            }
        } else {
            // 使用本地数据
            content = `
                <div style="margin-bottom:20px;">
                    <div style="color:#888;font-size:0.85rem;margin-bottom:8px;">关系类型</div>
                    <div style="font-size:1.1rem;color:#00d4ff;">${relation.source.name || relation.source} → ${relation.type} → ${relation.target.name || relation.target}</div>
                </div>
                
                <div style="color:#888;font-size:0.85rem;margin-bottom:8px;">证据文本</div>
                <div class="provenance-evidence">"${relation.evidence || '无证据文本'}"</div>
            `;
        }
        
        body.innerHTML = content;
    } catch (err) {
        body.innerHTML = '<p style="color:#ff6b6b;">加载失败</p>';
    }
}

function closeProvenance() {
    document.getElementById('provenanceModal').classList.remove('show');
}

// ==================== Entity List ====================

function renderEntityList() {
    const container = document.getElementById('entityList');
    if (!container) return;
    
    container.innerHTML = '<h3 style="margin-bottom:12px;color:#888;font-size:0.9rem;">项目实体</h3>';
    
    if (!projectEntities || projectEntities.length === 0) {
        container.innerHTML += '<p style="color:#666;font-size:0.85rem;">暂无实体，请上传音频文件</p>';
        return;
    }
    
    projectEntities.forEach(ent => {
        const div = document.createElement('div');
        div.className = 'entity-item';
        div.dataset.id = ent.id;
        div.onclick = () => window.selectEntity(ent.id);
        div.onmouseenter = (e) => showEntityCard(e, ent.id);
        div.onmouseleave = hideEntityCard;
        
        div.innerHTML = `
            <span class="entity-type-badge type-${ent.type.toLowerCase()}">${ent.type}</span>
            <div>
                <div style="font-weight:500;">${ent.name}</div>
                <div style="font-size:0.8rem;color:#666;">${ent.definition || '暂无定义'}</div>
            </div>
        `;
        
        container.appendChild(div);
    });
}

// ==================== Entity Selection ====================

window.selectEntity = function(entityId) {
    selectedEntity = entityId;
    const entity = projectEntities.find(e => e.id === entityId);
    if (!entity) return;
    
    // 高亮文本中的实体
    document.querySelectorAll('.entity').forEach(el => {
        if (el.dataset.id === entityId) {
            el.style.background = '#ff6b6b';
            el.style.color = '#fff';
        } else {
            el.style.background = '';
            el.style.color = '';
        }
    });
    
    // 高亮图谱中的节点
    d3.selectAll('.node-circle')
        .attr('stroke', d => d.id === entityId ? '#ff6b6b' : '#fff')
        .attr('stroke-width', d => d.id === entityId ? 4 : 2)
        .attr('r', d => d.id === entityId ? 40 : 35);
    
    // 高亮实体列表
    document.querySelectorAll('.entity-item').forEach(el => {
        if (el.dataset.id === entityId) {
            el.style.background = '#2a2a2a';
            el.style.borderLeft = '3px solid #ff6b6b';
        } else {
            el.style.background = '';
            el.style.borderLeft = '';
        }
    });
    
    console.log('Selected:', entity.name, entity.definition);
};

// ==================== Upload ====================

window.showUpload = function() {
    const el = document.getElementById('uploadOverlay');
    if (el) el.classList.add('show');
};

window.hideUpload = function() {
    const el = document.getElementById('uploadOverlay');
    if (el) el.classList.remove('show');
};

function initUpload() {
    const input = document.getElementById('fileInput');
    const overlay = document.getElementById('uploadOverlay');
    
    if (!input) return;
    
    input.addEventListener('change', async (e) => {
        if (!e.target.files.length) return;
        
        const file = e.target.files[0];
        if (overlay) {
            overlay.innerHTML = `
                <div style="text-align:center;">
                    <h2>正在分析...</h2>
                    <p style="color:#666;margin-top:10px;">${file.name}</p>
                    <p style="color:#888;margin-top:20px;font-size:0.9rem;">ASR转录 + 实体提取中</p>
                </div>
            `;
        }
        
        try {
            const result = await uploadAudio(file);
            
            currentData = result;
            
            await loadProjectData();
            
            if (result.segments && result.segments.length > 0) {
                renderTranscript();
            }
            
            if (overlay) overlay.classList.remove('show');
            
        } catch (err) {
            console.error('Upload failed:', err);
            if (overlay) {
                overlay.innerHTML = `
                    <div style="text-align:center;">
                        <h2 style="color:#ff6b6b;">分析失败</h2>
                        <p style="color:#666;margin-top:10px;">${err.message}</p>
                        <button class="btn" onclick="location.reload()" style="margin-top:20px;">重试</button>
                    </div>
                `;
            }
        }
    });
}