control-panel.js

// Control panel logic for IP Connection Analysis
// This file contains all logic for the control panel UI and its event handlers
import { getFlowColors, getInvalidLabels, getInvalidReason, getFlowColor } from './legends.js';
import { MAX_FLOW_LIST_ITEMS, FLOW_LIST_RENDER_BATCH } from './config.js';

export function initControlPanel(options) {
    // options: { onResetView, ... }
    const panel = document.getElementById('control-panel');
    if (!panel) return;

    const dragHandle = document.getElementById('control-panel-drag-handle');

    // --- Drag-to-move + click-to-collapse (like legend panel in timearcs) ---
    if (dragHandle) {
        let dragState = null;

        const onDrag = (e) => {
            if (!dragState) return;
            const dist = Math.hypot(e.clientX - dragState.startX, e.clientY - dragState.startY);
            if (dist > 5) {
                dragState.hasMoved = true;
                panel.classList.add('control-panel-dragging');
                const newLeft = Math.max(0, Math.min(window.innerWidth - 60, e.clientX - dragState.offsetX));
                // Leave room for the zoom bar attached above the panel
                const zoomBar = document.getElementById('zoomControlsSection');
                const zoomBarH = zoomBar ? zoomBar.offsetHeight : 0;
                const newTop = Math.max(zoomBarH, Math.min(window.innerHeight - 40, e.clientY - dragState.offsetY));
                panel.style.left = `${newLeft}px`;
                panel.style.top = `${newTop}px`;
                panel.style.right = 'auto';
            }
        };
        const onDragEnd = () => {
            if (dragState && !dragState.hasMoved) {
                // Click — toggle collapse
                panel.classList.toggle('control-panel-collapsed');
            }
            if (dragState) {
                panel.classList.remove('control-panel-dragging');
                dragState = null;
            }
            document.removeEventListener('mousemove', onDrag);
            document.removeEventListener('mouseup', onDragEnd);
        };

        dragHandle.addEventListener('mousedown', (e) => {
            if (e.button !== 0) return;
            const rect = panel.getBoundingClientRect();
            dragState = {
                offsetX: e.clientX - rect.left,
                offsetY: e.clientY - rect.top,
                startX: e.clientX,
                startY: e.clientY,
                hasMoved: false
            };
            document.addEventListener('mousemove', onDrag);
            document.addEventListener('mouseup', onDragEnd);
            e.preventDefault();
        });
    }

    // Reset View button — keep in normal flow inside control panel
    const resetBtn = document.getElementById('resetView');
    if (resetBtn) {
        resetBtn.style.position = '';
        resetBtn.style.width = '100%';
        resetBtn.style.margin = '0';
        resetBtn.style.padding = '10px 0';
        resetBtn.style.borderTop = '1px solid #eee';
        resetBtn.style.borderRadius = '0 0 6px 6px';
        if (options && typeof options.onResetView === 'function') {
            resetBtn.onclick = options.onResetView;
        }
    }

    // Wire up collapsible control-group sections
    panel.querySelectorAll('.control-group.collapsible > .collapsible-header').forEach(header => {
        header.addEventListener('click', () => {
            header.parentElement.classList.toggle('collapsed');
        });
    });
}

// Control panel render and update helpers (moved from main file)
export function createIPCheckboxes(uniqueIPs, onChange) {
    const container = document.getElementById('ipCheckboxes');
    if (!container) return;
    container.innerHTML = '';

    // Persist collapsed state across re-renders
    // Default to collapsed (show only selected) if coming from TimeArcs with pre-selected IPs
    if (container._collapsed === undefined) {
        const hasPrefilterIPs = window.brushSelectionPrefilterIPs && window.brushSelectionPrefilterIPs.length > 0;
        container._collapsed = hasPrefilterIPs;
    }

    // Build checkboxes (identical to original)
    uniqueIPs.forEach(ip => {
        const div = document.createElement('div');
        div.style.marginBottom = '5px';
        div.className = 'ip-item';
        div.dataset.ip = ip;
        const checkbox = document.createElement('input');
        checkbox.type = 'checkbox';
        checkbox.id = `ip-${ip.replace(/\./g, '-')}`;
        checkbox.value = ip;
        checkbox.checked = false;
        if (typeof onChange === 'function') {
            checkbox.addEventListener('change', onChange);
        }
        const label = document.createElement('label');
        label.htmlFor = checkbox.id;
        label.textContent = ip;
        label.style.marginLeft = '5px';
        label.style.fontSize = '12px';
        label.style.cursor = 'pointer';
        div.appendChild(checkbox);
        div.appendChild(label);
        container.appendChild(div);
    });

    // --- Collapse/expand toggle (placed after #ipSelector, outside scroll) ---
    const selector = document.getElementById('ipSelector');
    let toggleDiv = document.getElementById('ipCollapseToggle');
    if (!toggleDiv && selector) {
        toggleDiv = document.createElement('div');
        toggleDiv.id = 'ipCollapseToggle';
        toggleDiv.style.cssText = 'margin-top:6px;';
        selector.parentNode.insertBefore(toggleDiv, selector.nextSibling);
    }
    if (toggleDiv) {
        toggleDiv.innerHTML = '';
        const toggleLink = document.createElement('a');
        toggleLink.href = '#';
        toggleLink.style.cssText = 'font-size:11px; color:#007bff; text-decoration:none; cursor:pointer;';
        toggleLink.addEventListener('click', (e) => {
            e.preventDefault();
            container._collapsed = !container._collapsed;
            applyCollapseState(container, toggleLink);
        });
        toggleDiv.appendChild(toggleLink);
        applyCollapseState(container, toggleLink);
    }
}

// Separate function so it can't interfere with checkbox event flow
function applyCollapseState(container, toggleLink) {
    const items = container.querySelectorAll('.ip-item');
    if (container._collapsed) {
        let checkedCount = 0;
        items.forEach(item => {
            const cb = item.querySelector('input[type="checkbox"]');
            if (cb && cb.checked) {
                item.style.display = '';
                checkedCount++;
            } else {
                item.style.display = 'none';
            }
        });
        const unchecked = items.length - checkedCount;
        toggleLink.textContent = checkedCount === 0
            ? `Show all ${items.length} IPs`
            : `Show ${unchecked} more IP${unchecked !== 1 ? 's' : ''}`;
    } else {
        items.forEach(item => { item.style.display = ''; });
        toggleLink.textContent = 'Show only selected';
    }
}

export function filterIPList(searchTerm) {
    document.querySelectorAll('.ip-item').forEach(item => {
        const ip = item.dataset.ip || '';
        const matches = ip.toLowerCase().includes((searchTerm || '').toLowerCase());
        item.style.display = matches ? 'block' : 'none';
    });
}

// Re-apply the current collapse state (useful after checkboxes are programmatically checked)
export function refreshIPCollapseState() {
    const container = document.getElementById('ipCheckboxes');
    const toggleDiv = document.getElementById('ipCollapseToggle');
    if (!container || !toggleDiv) return;

    const toggleLink = toggleDiv.querySelector('a');
    if (toggleLink && container._collapsed !== undefined) {
        applyCollapseState(container, toggleLink);
    }
}

export function filterFlowList(searchTerm) {
    const items = document.querySelectorAll('#flowList .flow-item');
    const term = (searchTerm || '').toLowerCase();
    items.forEach(item => {
        const text = (item.innerText || item.textContent || '').toLowerCase();
        item.style.display = text.includes(term) ? '' : 'none';
    });
}

// Progress UI for flow processing/list rendering
export function showFlowProgress(label = 'Working…', percent = 0) {
    const box = document.getElementById('flowProgress');
    const bar = document.getElementById('flowProgressBar');
    const lbl = document.getElementById('flowProgressLabel');
    if (!box || !bar || !lbl) return;
    box.style.display = 'block';
    lbl.textContent = label;
    const pct = Math.max(0, Math.min(1, Number(percent) || 0));
    bar.style.width = `${Math.round(pct * 100)}%`;
}

export function updateFlowProgress(percent, label) {
    const bar = document.getElementById('flowProgressBar');
    if (!bar) return;
    if (label) {
        const lbl = document.getElementById('flowProgressLabel');
        if (lbl) lbl.textContent = label;
    }
    const pct = Math.max(0, Math.min(1, Number(percent) || 0));
    bar.style.width = `${Math.round(pct * 100)}%`;
}

export function hideFlowProgress() {
    const box = document.getElementById('flowProgress');
    if (!box) return;
    box.style.display = 'none';
}

// CSV loading progress UI
export function showCsvProgress(label = 'Loading CSV file...', percent = 0) {
    const box = document.getElementById('csvProgress');
    const bar = document.getElementById('csvProgressBar');
    const lbl = document.getElementById('csvProgressLabel');
    if (!box || !bar || !lbl) return;
    box.style.display = 'block';
    lbl.textContent = label;
    const pct = Math.max(0, Math.min(1, Number(percent) || 0));
    bar.style.width = `${Math.round(pct * 100)}%`;
}

export function updateCsvProgress(percent, label) {
    const bar = document.getElementById('csvProgressBar');
    if (!bar) return;
    if (label) {
        const lbl = document.getElementById('csvProgressLabel');
        if (lbl) lbl.textContent = label;
    }
    const pct = Math.max(0, Math.min(1, Number(percent) || 0));
    bar.style.width = `${Math.round(pct * 100)}%`;
}

export function hideCsvProgress() {
    const box = document.getElementById('csvProgress');
    if (!box) return;
    box.style.display = 'none';
}

// Paginated flow list rendering
// hasPacketData: if false, disables "Export CSV" and "View Packets" buttons (flow_list.json mode)
export function createFlowListCapped(flows, selectedFlowIds, formatBytes, formatTimestamp, exportFlowToCSV, zoomToFlow, updateTcpFlowPacketsGlobal, flowColors = {}, enterFlowDetailMode = null, hasPacketData = true) {
    const container = document.getElementById('flowListModalList') || document.getElementById('flowList');
    if (!container) return;
    if (!flows || flows.length === 0) {
        container.innerHTML = '<div style="color:#666; text-align:center; padding:20px;">No flows to display</div>';
        return;
    }

    // Sort flows in-place (avoids creating a copy - safe since callers pass new arrays)
    // If already sorted (e.g., from filterByIPs), this is O(n)
    flows.sort((a, b) => a.startTime - b.startTime);
    const total = flows.length;
    const flowsPerPage = MAX_FLOW_LIST_ITEMS || 500;
    const totalPages = Math.ceil(total / flowsPerPage);

    // Store pagination state on container
    if (!container._paginationState) {
        container._paginationState = {
            currentPage: 1,
            totalPages: totalPages,
            flows: flows,
            flowsPerPage: flowsPerPage
        };
    } else {
        // Update state with new flows
        container._paginationState.flows = flows;
        container._paginationState.totalPages = totalPages;
        container._paginationState.flowsPerPage = flowsPerPage;
        // Reset to page 1 if current page is beyond new total
        if (container._paginationState.currentPage > totalPages) {
            container._paginationState.currentPage = 1;
        }
    }

    const state = container._paginationState;
    const startIndex = (state.currentPage - 1) * flowsPerPage;
    const endIndex = Math.min(startIndex + flowsPerPage, total);
    const currentPageFlows = flows.slice(startIndex, endIndex);

    // Clear container and create pagination UI
    container.innerHTML = '';
    
    // Create pagination header
    const paginationHeader = document.createElement('div');
    paginationHeader.style.cssText = 'display:flex; justify-content:space-between; align-items:center; margin-bottom:10px; padding:8px; background:#f8f9fa; border-radius:4px; font-size:12px;';
    
    const pageInfo = document.createElement('div');
    pageInfo.style.cssText = 'color:#555;';
    pageInfo.innerHTML = `Page ${state.currentPage} of ${totalPages} • Showing ${(startIndex + 1).toLocaleString()}-${endIndex.toLocaleString()} of ${total.toLocaleString()} flows`;
    
    const paginationControls = document.createElement('div');
    paginationControls.style.cssText = 'display:flex; gap:8px; align-items:center;';
    
    // Previous button
    const prevBtn = document.createElement('button');
    prevBtn.textContent = '← Previous';
    prevBtn.disabled = state.currentPage === 1;
    prevBtn.style.cssText = 'padding:4px 8px; border:1px solid #ced4da; border-radius:3px; background:#fff; cursor:pointer; font-size:11px;';
    if (prevBtn.disabled) prevBtn.style.opacity = '0.5';
    
    // Page selector
    const pageSelect = document.createElement('select');
    pageSelect.style.cssText = 'padding:4px; border:1px solid #ced4da; border-radius:3px; font-size:11px;';
    for (let i = 1; i <= totalPages; i++) {
        const option = document.createElement('option');
        option.value = i;
        option.textContent = i;
        if (i === state.currentPage) option.selected = true;
        pageSelect.appendChild(option);
    }
    
    // Next button
    const nextBtn = document.createElement('button');
    nextBtn.textContent = 'Next →';
    nextBtn.disabled = state.currentPage === totalPages;
    nextBtn.style.cssText = 'padding:4px 8px; border:1px solid #ced4da; border-radius:3px; background:#fff; cursor:pointer; font-size:11px;';
    if (nextBtn.disabled) nextBtn.style.opacity = '0.5';
    
    // Event handlers
    const renderPage = (page) => {
        state.currentPage = page;
        createFlowListCapped(flows, selectedFlowIds, formatBytes, formatTimestamp, exportFlowToCSV, zoomToFlow, updateTcpFlowPacketsGlobal, flowColors, enterFlowDetailMode, hasPacketData);
    };
    
    prevBtn.addEventListener('click', () => {
        if (state.currentPage > 1) renderPage(state.currentPage - 1);
    });
    
    nextBtn.addEventListener('click', () => {
        if (state.currentPage < totalPages) renderPage(state.currentPage + 1);
    });
    
    pageSelect.addEventListener('change', (e) => {
        renderPage(parseInt(e.target.value));
    });
    
    paginationControls.appendChild(prevBtn);
    paginationControls.appendChild(pageSelect);
    paginationControls.appendChild(nextBtn);
    
    paginationHeader.appendChild(pageInfo);
    paginationHeader.appendChild(paginationControls);
    container.appendChild(paginationHeader);

    const invalidLabels = getInvalidLabels();

    // Render current page flows
    const BATCH = Math.max(50, Number(FLOW_LIST_RENDER_BATCH) || 200);
    const shouldShowProgress = currentPageFlows.length >= BATCH * 2;
    if (shouldShowProgress) {
        showFlowProgress('Rendering flow list…', 0);
    }

    let index = 0;
    function renderBatch() {
        const end = Math.min(currentPageFlows.length, index + BATCH);
        for (let i = index; i < end; i++) {
            const flow = currentPageFlows[i];
            const duration = Math.max(0, Math.round((flow.endTime - flow.startTime) / 1000000));
            const { utcTime: startTime } = formatTimestamp(flow.startTime);
            const { utcTime: endTime } = formatTimestamp(flow.endTime);
            const reason = getInvalidReason(flow);
            const color = getFlowColor(flow, flowColors);

            let closeTypeText = '';
            if (reason) {
                const label = invalidLabels[reason] || 'Invalid';
                closeTypeText = `
                    <span style="display:inline-flex; align-items:center; gap:6px;">
                        <span style=\"display:inline-block; width:10px; height:10px; border-radius:2px; background:${color}; border:1px solid #fff; box-shadow:0 0 0 1px rgba(0,0,0,0.08);\"></span>
                        <span style=\"color:#333;\">${label}</span>
                    </span>`;
            } else if (flow.closeType === 'graceful' || flow.closeType === 'abortive') {
                const label = flow.closeType === 'graceful' ? 'Graceful close' : 'Abortive close';
                closeTypeText = `
                    <span style="display:inline-flex; align-items:center; gap:6px;">
                        <span style=\"display:inline-block; width:10px; height:10px; border-radius:2px; background:${color}; border:1px solid #fff; box-shadow:0 0 0 1px rgba(0,0,0,0.08);\"></span>
                        <span style=\"color:#333;\">${label}</span>
                    </span>`;
            } else if (flow.establishmentComplete) {
                closeTypeText = '• Still open';
            } else {
                closeTypeText = '• Incomplete';
            }

            const item = document.createElement('div');
            item.className = 'flow-item';
            item.dataset.flowId = String(flow.id);
            item.style.borderLeft = `4px solid ${color}`;

            // Check if this specific flow has packet data (embedded or global)
            const flowHasPacketData = hasPacketData || flow._hasEmbeddedPackets;

            // Build button HTML based on whether packet data is available for this flow
            const viewBtnHTML = flowHasPacketData
                ? `<button class=\"flow-view-btn\" data-flow-id=\"${flow.id}\" style=\"padding:2px 8px; font-size:10px; border:1px solid #2196F3; border-radius:3px; background:#2196F3; color:white; cursor:pointer; font-weight:bold;\" title=\"View packets with arcs\">📊 View Packets</button>`
                : `<button class=\"flow-view-btn\" data-flow-id=\"${flow.id}\" style=\"padding:2px 8px; font-size:10px; border:1px solid #ccc; border-radius:3px; background:#eee; color:#999; cursor:not-allowed; font-weight:bold;\" title=\"Packet data not available (summary mode)\" disabled>📊 View Packets</button>`;

            const exportBtnHTML = flowHasPacketData
                ? `<button class=\"flow-export-btn\" data-flow-id=\"${flow.id}\" style=\"margin-left:auto; padding:2px 6px; font-size:10px; border:1px solid #ced4da; border-radius:3px; background:#fff; cursor:pointer;\">Export CSV</button>`
                : `<button class=\"flow-export-btn\" data-flow-id=\"${flow.id}\" style=\"margin-left:auto; padding:2px 6px; font-size:10px; border:1px solid #ccc; border-radius:3px; background:#eee; color:#999; cursor:not-allowed;\" title=\"Packet data not available (summary mode)\" disabled>Export CSV</button>`;

            item.innerHTML = `
                <input type=\"checkbox\" class=\"flow-checkbox\" id=\"flow-${flow.id}\" ${selectedFlowIds.has(String(flow.id)) ? 'checked' : ''}>
                <div class=\"flow-info\">
                    <div class=\"flow-connection\">${flow.initiator}:${flow.initiatorPort} ↔ ${flow.responder}:${flow.responderPort}</div>
                    <div class=\"flow-details\">
                        <span class=\"flow-status ${flow.state}\">${String(flow.state || '').replace('_',' ')}</span>
                        <span>${flow.totalPackets} packets</span>
                        <span>${formatBytes(flow.totalBytes)}</span>
                        <span>${duration}s duration</span>
                        <span>${closeTypeText}</span>
                        ${viewBtnHTML}
                        ${exportBtnHTML}
                    </div>
                    <div style=\"font-size:10px; color:#999; margin-top:3px;\">Start: ${startTime} • End: ${endTime}</div>
                </div>`;

            const cb = item.querySelector('.flow-checkbox');
            cb.addEventListener('change', (e) => {
                const flowId = String(flow.id);
                if (cb.checked) { selectedFlowIds.add(flowId); item.classList.add('selected'); }
                else { selectedFlowIds.delete(flowId); item.classList.remove('selected'); }
                if (typeof updateTcpFlowPacketsGlobal === 'function') updateTcpFlowPacketsGlobal();
            });
            item.addEventListener('click', (e) => {
                if (e.target && e.target.type !== 'checkbox') cb.click();
            });
            const exportBtn = item.querySelector('.flow-export-btn');
            exportBtn.addEventListener('click', (e) => {
                e.stopPropagation();
                if (typeof exportFlowToCSV === 'function') exportFlowToCSV(flow);
            });

            // View Packets button - enters flow detail mode
            const viewBtn = item.querySelector('.flow-view-btn');
            if (viewBtn) {
                viewBtn.addEventListener('click', (e) => {
                    e.stopPropagation();
                    if (typeof enterFlowDetailMode === 'function') {
                        // Close the modal before entering flow detail mode
                        const modalOverlay = document.getElementById('flowListModalOverlay');
                        if (modalOverlay) modalOverlay.style.display = 'none';
                        enterFlowDetailMode(flow);
                    } else {
                        console.warn('Flow detail mode not available');
                    }
                });
            }

            if (selectedFlowIds.has(String(flow.id))) item.classList.add('selected');
            container.appendChild(item);
        }

        index = end;
        if (shouldShowProgress) {
            updateFlowProgress(end / currentPageFlows.length);
        }
        if (index < currentPageFlows.length) {
            requestAnimationFrame(renderBatch);
        } else if (shouldShowProgress) {
            hideFlowProgress();
        }
    }

    requestAnimationFrame(renderBatch);

    // If rendering in modal, update header count
    const countEl = document.getElementById('flowListModalCount');
    if (countEl) countEl.textContent = `${total.toLocaleString()} flow(s)`;
}

export function wireControlPanelControls(opts) {
    const on = (id, type, handler) => { const el = document.getElementById(id); if (el && handler) el.addEventListener(type, handler); };
    on('ipSearch', 'input', (e) => { if (opts.onIpSearch) opts.onIpSearch(e.target.value); });
    on('selectAllIPs', 'click', () => { if (opts.onSelectAllIPs) opts.onSelectAllIPs(); });
    on('clearAllIPs', 'click', () => { if (opts.onClearAllIPs) opts.onClearAllIPs(); });
    on('collapseAllRows', 'click', () => { if (opts.onCollapseAllRows) opts.onCollapseAllRows(); });
    on('expandAllRows', 'click', () => { if (opts.onExpandAllRows) opts.onExpandAllRows(); });

    on('showTcpFlows', 'change', (e) => { if (opts.onToggleShowTcpFlows) opts.onToggleShowTcpFlows(e.target.checked); });
    on('showEstablishment', 'change', (e) => { if (opts.onToggleEstablishment) opts.onToggleEstablishment(e.target.checked); });
    on('showDataTransfer', 'change', (e) => { if (opts.onToggleDataTransfer) opts.onToggleDataTransfer(e.target.checked); });
    on('showClosing', 'change', (e) => { if (opts.onToggleClosing) opts.onToggleClosing(e.target.checked); });
    on('showGroundTruth', 'change', (e) => { if (opts.onToggleGroundTruth) opts.onToggleGroundTruth(e.target.checked); });

    // Render mode radios
    const modeCircles = document.getElementById('renderModeCircles');
    const modeBars = document.getElementById('renderModeBars');
    if (modeCircles && modeBars && opts.onToggleRenderMode) {
        const handler = () => {
            const mode = modeBars.checked ? 'bars' : 'circles';
            opts.onToggleRenderMode(mode);
        };
        modeCircles.addEventListener('change', handler);
        modeBars.addEventListener('change', handler);
    }
}

// Inline SVG arc icon matching the flag color legend in the packet view
function flagArcIcon(color) {
    // Semi-circle arc curving right, matching legends.js drawFlagLegend arc shape
    return `<svg width="14" height="14" viewBox="0 0 14 14" style="flex-shrink:0; margin-right:4px;"><path d="M 7 1 A 6 6 0 0 1 7 13" fill="none" stroke="${color}" stroke-width="4" stroke-linecap="round"/></svg>`;
}

// Order flags by TCP lifecycle: handshake → transfer → closing, unknowns at end
const FLAG_PHASE_ORDER = [
    'SYN', 'SYN+ACK',          // Handshake
    'ACK', 'PSH', 'PSH+ACK',   // Data transfer
    'FIN', 'FIN+ACK',           // Graceful close
    'RST', 'RST+ACK',          // Abortive close
    'OTHER'                     // Catch-all
];

function sortFlagsByTcpPhase(entries) {
    const order = new Map(FLAG_PHASE_ORDER.map((f, i) => [f, i]));
    return entries.sort(([a], [b]) => (order.get(a) ?? 99) - (order.get(b) ?? 99));
}

// Build 2-column grid HTML from sorted [flag, count] entries
// Column-first order: fill left column top-to-bottom, then right column (max 6 rows)
function buildFlagStatsGrid(sortedFlags, flagColors) {
    const maxRows = 6;
    // Fill first column up to maxRows before starting second column
    const col1Count = Math.min(maxRows, sortedFlags.length);
    const col1 = sortedFlags.slice(0, col1Count);
    const col2 = sortedFlags.slice(col1Count);

    const renderItem = ([flag, count]) => {
        const color = flagColors[flag] || '#95a5a6';
        return `<div style="display:flex; align-items:center; cursor:pointer; min-width:0;" data-flag="${flag}">${flagArcIcon(color)}<span style="white-space:nowrap; overflow:hidden; text-overflow:ellipsis;">${flag}: ${count.toLocaleString()}</span></div>`;
    };

    let html = '<div style="display:flex; gap:8px;">';
    html += `<div style="display:flex; flex-direction:column; gap:3px; flex:1; min-width:0;">${col1.map(renderItem).join('')}</div>`;
    if (col2.length > 0) {
        html += `<div style="display:flex; flex-direction:column; gap:3px; flex:1; min-width:0;">${col2.map(renderItem).join('')}</div>`;
    }
    html += '</div>';
    return html;
}

export function updateFlagStats(packets, classifyFlags, flagColors) {
    const container = document.getElementById('flagStats');
    if (!container) return;
    if (!packets || packets.length === 0) {
        container.innerHTML = '<div style="color: #666;">No data to display</div>';
        return;
    }
    const flagCounts = {};
    packets.forEach(packet => {
        // Use pre-classified flag_type/flagType for binned data, fall back to classifyFlags for raw packets
        const ft = packet.flagType || packet.flag_type || classifyFlags(packet.flags);
        const count = packet.count || 1;
        flagCounts[ft] = (flagCounts[ft] || 0) + count;
    });
    const sortedFlags = sortFlagsByTcpPhase(Object.entries(flagCounts));
    container.innerHTML = sortedFlags.length > 0 ? buildFlagStatsGrid(sortedFlags, flagColors) : '<div style="color:#666;">No TCP packets found</div>';
}

export function updateSizeLegend(globalMaxBinCount, radiusMin, radiusMax) {
    const container = document.getElementById('sizeLegend');
    if (!container) return;
    const maxCount = Math.max(1, globalMaxBinCount);
    if (maxCount <= 1) {
        container.innerHTML = '<div style="color:#666;">No data loaded</div>';
        return;
    }
    const midCount = Math.max(1, Math.round(maxCount / 2));
    const values = [1, midCount, maxCount];
    // sqrtScale matching ip_bar_diagram.js rScale
    const rScale = (v) => radiusMin + (radiusMax - radiusMin) * Math.sqrt((v - 1) / Math.max(1, maxCount - 1));
    const radii = values.map(v => Math.max(radiusMin, rScale(v)));
    const maxR = Math.max(...radii);

    // Horizontal layout: circles side by side, bottom-aligned, with labels below
    const gap = 12;
    const pad = 4;
    let items = '';
    for (let i = 0; i < values.length; i++) {
        const r = radii[i];
        const d = r * 2;
        const topPad = (maxR - r) * 2; // push smaller circles down to bottom-align
        items += `<div style="display:flex; flex-direction:column; align-items:center; gap:2px;">` +
            `<svg width="${d + 2}" height="${d + 2}" style="margin-top:${topPad}px;"><circle cx="${r + 1}" cy="${r + 1}" r="${r}" fill="none" stroke="#555" stroke-width="1"/></svg>` +
            `<span style="font-size:10px; color:#333; white-space:nowrap;">${values[i].toLocaleString()}</span>` +
            `</div>`;
    }

    container.innerHTML = `<div style="display:flex; align-items:flex-end; gap:${gap}px; padding:${pad}px 0;">${items}</div>`;
}

export function updateIPStats(packets, flagColors, formatBytes) {
    const container = document.getElementById('ipStats');
    if (!container) return;
    if (!packets || packets.length === 0) {
        container.innerHTML = '<div style="color: #666;">Select IPs to view statistics</div>';
        return;
    }
    const selectedIPs = Array.from(document.querySelectorAll('#ipCheckboxes input[type="checkbox"]:checked')).map(cb => cb.value);
    if (selectedIPs.length === 0) {
        container.innerHTML = '<div style="color: #666;">Select IPs to view statistics</div>';
        return;
    }
    const ipStats = {};
    selectedIPs.forEach(ip => {
        ipStats[ip] = { sent:0, received:0, total:0, bytes_sent:0, bytes_received:0, total_bytes:0, connections:new Set(), flags_sent:{}, flags_received:{} };
        Object.keys(flagColors).forEach(flag => { ipStats[ip].flags_sent[flag]=0; ipStats[ip].flags_received[flag]=0; });
    });
    packets.forEach(p => {
        const fStr = p.flagType || p.flag_type || (typeof p.flags === 'string' ? p.flags : '') || '';
        const count = p.count || 1;
        const size = (p.total_length || p.length || 0);
        if (ipStats[p.src_ip]) { const s=ipStats[p.src_ip]; s.sent+=count; s.total+=count; s.bytes_sent+=size; s.total_bytes+=size; s.connections.add(p.dst_ip); if (s.flags_sent[fStr]!==undefined) s.flags_sent[fStr]+=count; }
        if (ipStats[p.dst_ip]) { const s=ipStats[p.dst_ip]; s.received+=count; s.total+=count; s.bytes_received+=size; s.total_bytes+=size; s.connections.add(p.src_ip); if (s.flags_received[fStr]!==undefined) s.flags_received[fStr]+=count; }
    });
    const flagsToHtml = (m) => {
        const arr = Object.entries(m).filter(([,c])=>c>0).sort(([,a],[,b])=>b-a);
        if (!arr.length) return '<span style="color:#999; font-style:italic;">None</span>';
        return arr.map(([flag,count])=>`<span style="display:inline-flex; align-items:center; gap:4px; padding:2px 4px; border:1px solid #e9ecef; border-radius:3px; background:#fff;"><span style="width:10px; height:10px; background:${flagColors[flag]||'#bdc3c7'}; border-radius:2px;"></span><span style="font-size:10px; color:#555;">${flag} (${count})</span></span>`).join(' ');
    };
    let html = '<div style="font-weight:bold; margin-bottom:10px; border-bottom:1px solid #eee; padding-bottom:5px;">Selected IP Statistics</div>';
    selectedIPs.forEach(ip=>{
        const s = ipStats[ip];
        const connectionCount = s.connections.size;
        html += `
          <div style="margin-bottom:15px; padding:8px; border:1px solid #e9ecef; border-radius:3px;">
            <div style="font-weight:bold; color:#495057; margin-bottom:8px;">${ip}</div>
            <div style="display:grid; grid-template-columns:1fr 1fr; gap:5px; font-size:11px; margin-bottom:8px;">
              <div>Sent: ${s.sent.toLocaleString()}</div>
              <div>Received: ${s.received.toLocaleString()}</div>
              <div>Total: ${s.total.toLocaleString()}</div>
              <div>Unique Peers: ${connectionCount}</div>
              <div>Bytes Sent: ${formatBytes(s.bytes_sent)}</div>
              <div>Bytes Recv: ${formatBytes(s.bytes_received)}</div>
            </div>
            <div style="margin-bottom:5px;"><div style="font-size:10px; color:#666; margin-bottom:3px;">Flags Sent:</div><div style="display:flex; flex-wrap:wrap; gap:3px;">${flagsToHtml(s.flags_sent)}</div></div>
            <div><div style="font-size:10px; color:#666; margin-bottom:3px;">Flags Received:</div><div style="display:flex; flex-wrap:wrap; gap:3px;">${flagsToHtml(s.flags_received)}</div></div>
          </div>`;
    });
    container.innerHTML = html;
}

export function createFlowList(flows, selectedFlowIds, formatBytes, formatTimestamp, exportFlowToCSV, zoomToFlow, updateTcpFlowPacketsGlobal, flowColors = {}) {
    const container = document.getElementById('flowListModalList') || document.getElementById('flowList');
    if (!container) return;
    if (!flows || flows.length === 0) {
        container.innerHTML = '<div style="color:#666; text-align:center; padding:20px;">No flows to display</div>';
        return;
    }
    const sorted = [...flows].sort((a,b)=>a.startTime - b.startTime);

    // Use flow legend helpers from legends.js
    const closeColors = getFlowColors(flowColors);
    const invalidLabels = getInvalidLabels();
    let html = '';
    sorted.forEach(flow => {
        const duration = Math.round((flow.endTime - flow.startTime) / 1000000);
        const { utcTime: startTime } = formatTimestamp(flow.startTime);
        const { utcTime: endTime } = formatTimestamp(flow.endTime);
        const reason = getInvalidReason(flow);
        const color = getFlowColor(flow, flowColors);
        let closeTypeText = '';
        if (reason) {
            const label = invalidLabels[reason] || 'Invalid';
            closeTypeText = `
                <span style="display:inline-flex; align-items:center; gap:6px;">
                    <span style="display:inline-block; width:10px; height:10px; border-radius:2px; background:${color}; border:1px solid #fff; box-shadow:0 0 0 1px rgba(0,0,0,0.08);"></span>
                    <span style="color:#333;">${label}</span>
                </span>`;
        } else if (flow.closeType === 'graceful' || flow.closeType === 'abortive') {
            const label = flow.closeType === 'graceful' ? 'Graceful close' : 'Abortive close';
            closeTypeText = `
                <span style="display:inline-flex; align-items:center; gap:6px;">
                    <span style="display:inline-block; width:10px; height:10px; border-radius:2px; background:${color}; border:1px solid #fff; box-shadow:0 0 0 1px rgba(0,0,0,0.08);"></span>
                    <span style="color:#333;">${label}</span>
                </span>`;
        } else if (flow.establishmentComplete) {
            closeTypeText = '• Still open';
        } else {
            closeTypeText = '• Incomplete';
        }
        html += `
          <div class="flow-item" data-flow-id="${flow.id}" style="border-left: 4px solid ${color};">
            <input type="checkbox" class="flow-checkbox" id="flow-${flow.id}" ${selectedFlowIds.has(String(flow.id)) ? 'checked' : ''}>
            <div class="flow-info">
              <div class="flow-connection">${flow.initiator}:${flow.initiatorPort} ↔ ${flow.responder}:${flow.responderPort}</div>
              <div class="flow-details">
                <span class="flow-status ${flow.state}">${flow.state.replace('_',' ')}</span>
                <span>${flow.totalPackets} packets</span>
                <span>${formatBytes(flow.totalBytes)}</span>
                <span>${duration}s duration</span>
                <span>${closeTypeText}</span>
                <button class="flow-export-btn" data-flow-id="${flow.id}" style="margin-left:auto; padding:2px 6px; font-size:10px; border:1px solid #ced4da; border-radius:3px; background:#fff; cursor:pointer;">Export CSV</button>
              </div>
              <div style="font-size:10px; color:#999; margin-top:3px;">Start: ${startTime} • End: ${endTime}</div>
            </div>
          </div>`;
    });
    container.innerHTML = html;
    container.querySelectorAll('.flow-export-btn').forEach(btn => btn.addEventListener('click', (e) => {
        e.stopPropagation();
        const flowId = e.currentTarget.dataset.flowId;
        const f = flows.find(x => String(x.id) === String(flowId));
        if (f) exportFlowToCSV(f);
    }));
    container.querySelectorAll('.flow-checkbox').forEach(cb => cb.addEventListener('change', (e) => {
        const flowId = e.target.id.replace('flow-','');
        const flowItem = e.target.closest('.flow-item');
        if (e.target.checked) { selectedFlowIds.add(flowId); flowItem.classList.add('selected'); }
        else { selectedFlowIds.delete(flowId); flowItem.classList.remove('selected'); }
        if (typeof updateTcpFlowPacketsGlobal === 'function') updateTcpFlowPacketsGlobal();
    }));
    container.querySelectorAll('.flow-item').forEach(item => item.addEventListener('click', (e) => {
        if (e.target && e.target.type !== 'checkbox') {
            const cb = item.querySelector('.flow-checkbox');
            if (cb) cb.click();
        }
    }));
    selectedFlowIds.forEach(id => { const el = container.querySelector(`[data-flow-id="${id}"]`); if (el) el.classList.add('selected'); });

    // If rendering in modal, update header count
    const countEl = document.getElementById('flowListModalCount');
    if (countEl) countEl.textContent = `${flows.length.toLocaleString()} flow(s)`;
}

// Modal helpers for the Flow List popup
export function showFlowListModal() {
    const overlay = document.getElementById('flowListModalOverlay');
    if (!overlay) return;
    overlay.style.display = 'flex';
    // Center modal on first open or keep previous position if moved
    try {
        const modal = document.getElementById('flowListModal');
        if (!modal) return;
        const hasPosition = modal.dataset.positioned === 'true';
        if (!hasPosition) {
            // Center within viewport
            const vw = window.innerWidth, vh = window.innerHeight;
            const rect = modal.getBoundingClientRect();
            const left = Math.max(8, (vw - rect.width) / 2);
            const top = Math.max(8, (vh - rect.height) / 2);
            modal.style.left = `${left}px`;
            modal.style.top = `${top}px`;
            modal.dataset.positioned = 'true';
        }
    } catch (_) {}
}

export function hideFlowListModal() {
    const overlay = document.getElementById('flowListModalOverlay');
    if (overlay) overlay.style.display = 'none';
}

let flowListModalWired = false;
export function wireFlowListModalControls({ onSelectAll, onClearAll, onSearch } = {}) {
    if (flowListModalWired) return;
    flowListModalWired = true;
    const overlay = document.getElementById('flowListModalOverlay');
    const closeBtn = document.getElementById('flowListModalClose');
    const selectAllBtn = document.getElementById('flowListModalSelectAll');
    const clearAllBtn = document.getElementById('flowListModalClearAll');
    const searchInput = document.getElementById('flowListModalSearch');
    const modal = document.getElementById('flowListModal');
    const header = modal ? modal.querySelector('.modal-header') : null;

    // Overlay is non-blocking and click-through; close via button only
    if (closeBtn) closeBtn.addEventListener('click', hideFlowListModal);
    if (selectAllBtn && typeof onSelectAll === 'function') selectAllBtn.addEventListener('click', onSelectAll);
    if (clearAllBtn && typeof onClearAll === 'function') clearAllBtn.addEventListener('click', onClearAll);
    if (searchInput && typeof onSearch === 'function') searchInput.addEventListener('input', (e) => onSearch(e.target.value));

    // Draggable modal via header
    if (modal && header) {
        let isDragging = false;
        let startX = 0, startY = 0, origLeft = 0, origTop = 0;
        const onMouseMove = (e) => {
            if (!isDragging) return;
            const dx = e.clientX - startX;
            const dy = e.clientY - startY;
            const newLeft = Math.max(8, Math.min(window.innerWidth - modal.offsetWidth - 8, origLeft + dx));
            const newTop = Math.max(8, Math.min(window.innerHeight - modal.offsetHeight - 8, origTop + dy));
            modal.style.left = `${newLeft}px`;
            modal.style.top = `${newTop}px`;
            modal.dataset.positioned = 'true';
        };
        const onMouseUp = () => {
            if (!isDragging) return;
            isDragging = false;
            document.removeEventListener('mousemove', onMouseMove);
            document.removeEventListener('mouseup', onMouseUp);
        };
        header.addEventListener('mousedown', (e) => {
            // Only start dragging with primary button
            if (e.button !== 0) return;
            isDragging = true;
            startX = e.clientX;
            startY = e.clientY;
            const rect = modal.getBoundingClientRect();
            origLeft = rect.left;
            origTop = rect.top;
            document.addEventListener('mousemove', onMouseMove);
            document.addEventListener('mouseup', onMouseUp);
        });
    }
}

export function updateTcpFlowStats(flows, selectedFlowIds, formatBytes) {
    const container = document.getElementById('tcpFlowStats');
    if (!container) return;
    if (!flows || flows.length === 0) {
        container.innerHTML = 'Select 2 or more IP addresses to view TCP flow statistics';
        container.style.color = '#666';
        return;
    }
    const totalStats = {
        total: flows.length,
        established: flows.filter(f => f.establishmentComplete === true || f.state === 'established' || f.state === 'closed').length,
        withData: flows.filter(f => f.dataTransferStarted === true).length,
        gracefulClose: flows.filter(f => f.closeType === 'graceful').length,
        abortiveClose: flows.filter(f => f.closeType === 'abortive').length,
        invalid: flows.filter(f => f.closeType === 'invalid' || f.state === 'invalid').length,
        totalPackets: flows.reduce((sum, f) => sum + (parseInt(f.totalPackets) || 0), 0),
        totalBytes: flows.reduce((sum, f) => sum + (parseInt(f.totalBytes) || 0), 0)
    };
    const selectedFlows = flows.filter(f => selectedFlowIds.has(String(f.id)));
    const selectedStats = {
        selected: selectedFlows.length,
        established: selectedFlows.filter(f => f.establishmentComplete === true || f.state === 'established' || f.state === 'closed').length,
        withData: selectedFlows.filter(f => f.dataTransferStarted === true).length,
        gracefulClose: selectedFlows.filter(f => f.closeType === 'graceful').length,
        abortiveClose: selectedFlows.filter(f => f.closeType === 'abortive').length,
        invalid: selectedFlows.filter(f => f.closeType === 'invalid' || f.state === 'invalid').length,
        totalPackets: selectedFlows.reduce((sum, f) => sum + (parseInt(f.totalPackets) || 0), 0),
        totalBytes: selectedFlows.reduce((sum, f) => sum + (parseInt(f.totalBytes) || 0), 0)
    };
    let statsHTML = `<strong>${totalStats.total} TCP flow(s) for selected IPs (${selectedStats.selected} checked)</strong><br>`;
    if (totalStats.total > 0) {
        statsHTML += `<div style="margin-top:8px;">`;
        statsHTML += `• Fully established: ${totalStats.established}<br>`;
        statsHTML += `• With data transfer: ${totalStats.withData}<br>`;
        statsHTML += `• Graceful close: ${totalStats.gracefulClose}<br>`;
        statsHTML += `• Abortive close: ${totalStats.abortiveClose}<br>`;
        if (totalStats.invalid > 0) statsHTML += `• <span style="color:#e74c3c;">Invalid connections: ${totalStats.invalid}</span><br>`;
        statsHTML += `• Total packets: ${totalStats.totalPackets.toLocaleString()}<br>`;
        statsHTML += `• Total bytes: ${formatBytes(totalStats.totalBytes)}`;
        statsHTML += `</div>`;
        if (selectedStats.selected > 0) {
            statsHTML += `<div style=\"margin-top:8px; padding-top:8px; border-top:1px solid #eee; color:#007bff;\">`;
            statsHTML += `<strong>Checked flows (${selectedStats.selected}):</strong><br>`;
            statsHTML += `• Packets: ${selectedStats.totalPackets.toLocaleString()}, Bytes: ${formatBytes(selectedStats.totalBytes)}`;
            statsHTML += `</div>`;
        }
    } else {
        statsHTML += `<div style="color:#999; font-style:italic; margin-top:5px;">No flows match selected IP addresses</div>`;
    }
    container.innerHTML = statsHTML;
    container.style.color = '#27ae60';
}

export function updateGroundTruthStatsUI(html, ok=true) {
    const container = document.getElementById('groundTruthStats');
    if (!container) return;
    container.innerHTML = html;
    container.style.color = ok ? '#27ae60' : '#e74c3c';
}