ui-controller.js

/**
 * TunnelVision UI Controller
 * Handles tree editor rendering, drag-and-drop, settings panel, and all user interactions.
 */

import { saveSettingsDebounced } from '../../../../script.js';
import { getContext } from '../../../st-context.js';
import { world_names, loadWorldInfo, saveWorldInfo } from '../../../world-info.js';
import { getAutoSummaryCount, resetAutoSummaryCount } from './auto-summary.js';
import { getActiveTunnelVisionBooks } from './tool-registry.js';
import {
    getTree,
    saveTree,
    deleteTree,
    isLorebookEnabled,
    setLorebookEnabled,
    createTreeNode,
    addEntryToNode,
    removeNode,
    removeEntryFromTree,
    getAllEntryUids,
    getSettings,
    getBookDescription,
    setBookDescription,
    getSelectedLorebook,
    setSelectedLorebook,
    isTrackerUid,
    isTrackerTitle,
    setTrackerUid,
    syncTrackerUidsForLorebook,
    getConnectionProfileId,
    setConnectionProfileId,
    listConnectionProfiles,
    getBookPermission,
    setBookPermission,
    SETTING_DEFAULTS,
} from './tree-store.js';
import { buildTreeFromMetadata, buildTreeWithLLM, generateSummariesForTree, ingestChatMessages } from './tree-builder.js';
import { registerTools, unregisterTools, getDefaultToolDescriptions, stripDynamicContent } from './tool-registry.js';
import { runDiagnostics } from './diagnostics.js';
import { applyRecurseLimit } from './index.js';
import { refreshHiddenToolCallMessages } from './activity-feed.js';
import { separateConditions, isEvaluableCondition, formatCondition, EVALUABLE_TYPES, CONDITION_LABELS, getKeywordProbability, setKeywordProbability } from './conditions.js';
import { callGenericPopup, POPUP_TYPE } from '../../../popup.js';


let currentLorebook = null;

function selectCurrentLorebook(bookName) {
    currentLorebook = bookName || null;
    setSelectedLorebook(currentLorebook);
}

function syncSelectedLorebook() {
    if (currentLorebook && world_names?.includes(currentLorebook)) {
        return;
    }

    const preferredLorebook = getSelectedLorebook();
    if (preferredLorebook && world_names?.includes(preferredLorebook)) {
        currentLorebook = preferredLorebook;
        return;
    }

    currentLorebook = null;
}

// ─── Event Bindings ──────────────────────────────────────────────

export function bindUIEvents() {
    // Main collapsible header
    $('#tv_header_toggle').on('click', function () {
        $(this).toggleClass('expanded');
        $(this).closest('.tv-container').find('.tv-settings-body').slideToggle(200);
    });

    $('#tv_global_enabled').on('change', onGlobalToggle);
    $('#tv_conditional_triggers_master').on('change', function () {
        const settings = getSettings();
        settings.conditionalTriggersEnabled = $(this).prop('checked');
        $('#tv_conditional_triggers').prop('checked', settings.conditionalTriggersEnabled);
        saveSettingsDebounced();
    });
    $('#tv_lorebook_select').on('change', onLorebookSelect);
    $('#tv_lorebook_enabled').on('change', onLorebookToggle);
    $('#tv_book_description').on('input', onBookDescriptionChange);
    $('#tv_build_metadata').on('click', onBuildFromMetadata);
    $('#tv_build_llm').on('click', onBuildWithLLM);
    $('#tv_open_tree_editor').on('click', onOpenTreeEditor);
    $('#tv_import_file').on('change', onImportTree);
    $('#tv_bulk_export').on('click', onBulkExport);
    $('#tv_bulk_import_file').on('change', onBulkImport);
    $('#tv_bulk_import').on('click', () => $('#tv_bulk_import_file').trigger('click'));

    $('#tv_run_diagnostics').on('click', onRunDiagnostics);

    // Lorebook filter
    $('#tv_lorebook_filter').on('input', onLorebookFilter);

    // Advanced Settings collapsible header
    $('#tv_advanced_header').on('click', function () {
        $(this).toggleClass('expanded');
        $(this).next('.tv-advanced-body').slideToggle(200);
    });

    // Auto-detect pattern
    $('#tv_auto_detect_pattern').on('input', function () {
        const settings = getSettings();
        settings.autoDetectPattern = $(this).val();
        saveSettingsDebounced();
    });

    // Per-tool toggles
    $(document).on('change', '.tv_tool_enabled', onToolToggle);

    // Per-tool confirmation toggles
    $('.tv_tool_confirm').on('change', onToolConfirmToggle);

    // Tool prompt overrides
    $('#tv_tool_prompt_overrides').on('input', '.tv-tool-prompt-textarea', onToolPromptChange);
    $('#tv_tool_prompt_overrides').on('click', '.tv-tool-prompt-reset', onToolPromptReset);

    // Search mode radio
    $('input[name="tv_search_mode"]').on('change', onSearchModeChange);

    // Collapsed tree depth
    $('#tv_collapsed_depth').on('change', onCollapsedDepthChange);

    // Selective retrieval
    $('#tv_selective_retrieval').on('change', onSelectiveRetrievalToggle);

    // Recurse limit
    $('#tv_recurse_limit').on('change', onRecurseLimitChange);

    // LLM build detail level
    $('#tv_llm_detail').on('change', onLlmDetailChange);

    // Tree granularity
    $('#tv_tree_granularity').on('change', onTreeGranularityChange);

    // LLM chunk size
    $('#tv_chunk_tokens').on('change', onChunkTokensChange);

    // Vector dedup toggle + threshold
    $('#tv_vector_dedup').on('change', onVectorDedupToggle);
    $('#tv_dedup_threshold').on('change', onDedupThresholdChange);

    // Chat ingest
    $('#tv_ingest_chat').on('click', onIngestChat);

    // Mandatory tool calls & prompt injection settings
    $('#tv_mandatory_tools').on('change', onMandatoryToolsToggle);
    $('#tv_mandatory_position').on('change', onPromptInjectionChange);
    $('#tv_mandatory_depth').on('change', onPromptInjectionChange);
    $('#tv_mandatory_role').on('change', onPromptInjectionChange);
    $('#tv_mandatory_prompt_text').on('change', onMandatoryPromptTextChange);
    $('#tv_mandatory_prompt_reset').on('click', onMandatoryPromptReset);
    $('#tv_notebook_position').on('change', onPromptInjectionChange);
    $('#tv_notebook_depth').on('change', onPromptInjectionChange);
    $('#tv_notebook_role').on('change', onPromptInjectionChange);
    $('#tv_stealth_mode').on('change', onStealthModeToggle);
    $('#tv_ephemeral_results').on('change', onEphemeralResultsToggle);
    $('.tv_ephemeral_tool').on('change', onEphemeralToolFilterChange);

    // Slash commands context setting
    $('#tv_command_context').on('change', onCommandContextChange);

    // Auto-summary settings
    $('#tv_auto_summary_enabled').on('change', onAutoSummaryToggle);
    $('#tv_auto_summary_interval').on('change', onAutoSummaryIntervalChange);
    $('#tv_auto_hide_summarized').on('change', onAutoHideSummarizedToggle);
    $('#tv_passthrough_constant').on('change', onPassthroughConstantToggle);
    $('#tv_allow_keyword_triggers').on('change', onAllowKeywordTriggersToggle);

    // Multi-book mode
    $('input[name="tv_multi_book_mode"]').on('change', onMultiBookModeChange);

    // Sidecar LLM (connection profile + sampler overrides)
    $('#tv_connection_profile').on('change', onConnectionProfileChange);
    $('#tv_sidecar_temperature').on('input', onSidecarTemperatureChange);
    $('#tv_sidecar_max_tokens').on('input', onSidecarMaxTokensChange);

    // Sidecar auto-retrieval
    $('#tv_sidecar_auto_retrieval').on('change', onSidecarAutoRetrievalToggle);
    $('#tv_sidecar_context_messages').on('input', onSidecarContextMessagesChange);
    $('#tv_sidecar_max_injection').on('input', onSidecarMaxInjectionChange);
    $('#tv_conditional_triggers').on('change', function () {
        const settings = getSettings();
        settings.conditionalTriggersEnabled = $(this).prop('checked');
        $('#tv_conditional_triggers_master').prop('checked', settings.conditionalTriggersEnabled);
        saveSettingsDebounced();
    });

    // Sidecar post-gen writer
    $('#tv_sidecar_post_gen_writer').on('change', onSidecarPostGenWriterToggle);
    $('#tv_sidecar_writer_context').on('input', onSidecarWriterContextChange);
    $('#tv_sidecar_writer_max_ops').on('input', onSidecarWriterMaxOpsChange);

    // Compact tool prompts
    $('#tv_compact_tool_prompts').on('change', onCompactToolPromptsToggle);

    // Per-lorebook permissions
    $('#tv_book_permission').on('change', onBookPermissionChange);

    // Backup & Restore collapsible header
    $('#tv_backup_header').on('click', function () {
        $(this).toggleClass('expanded');
        $(this).next('.tv-card-body').slideToggle(200);
    });

    // Diagnostics collapsible header
    $('#tv_diagnostics_header').on('click', function () {
        $(this).toggleClass('expanded');
        $(this).next('.tv-diagnostics-body').slideToggle(200);
    });

}

// ─── Refresh / Init ──────────────────────────────────────────────

export function refreshUI() {
    const settings = getSettings();
    const globalEnabled = settings.globalEnabled !== false;
    syncSelectedLorebook();

    $('#tv_global_enabled').prop('checked', globalEnabled);
    $('#tv_main_controls').toggle(globalEnabled);
    $('#tv_conditional_master_row').toggle(globalEnabled);
    $('#tv_conditional_triggers_master').prop('checked', settings.conditionalTriggersEnabled !== false);

    // Sync tool toggles from settings
    const disabledTools = settings.disabledTools || {};
    $('.tv_tool_enabled').each(function () {
        const toolName = $(this).data('tool');
        $(this).prop('checked', !disabledTools[toolName]);
    });

    // Sync tool confirmation toggles
    const confirmTools = settings.confirmTools || {};
    $('.tv_tool_confirm').each(function () {
        const toolName = $(this).data('tool');
        $(this).prop('checked', !!confirmTools[toolName]);
    });

    // Render tool prompt overrides
    renderToolPromptOverrides();

    // Sync search mode radio
    $(`input[name="tv_search_mode"][value="${settings.searchMode || 'traversal'}"]`).prop('checked', true);

    // Sync collapsed depth
    $('#tv_collapsed_depth').val(settings.collapsedDepth ?? 2);
    $('#tv_collapsed_depth_section').toggle((settings.searchMode || 'traversal') === 'collapsed');

    // Sync auto-detect pattern
    $('#tv_auto_detect_pattern').val(settings.autoDetectPattern || '');

    // Sync selective retrieval
    $('#tv_selective_retrieval').prop('checked', settings.selectiveRetrieval === true);

    // Sync recurse limit
    const recurseLimit = settings.recurseLimit ?? 5;
    $('#tv_recurse_limit').val(recurseLimit);
    $('#tv_recurse_warn').toggle(recurseLimit > 10);

    // Sync LLM detail level
    $('#tv_llm_detail').val(settings.llmBuildDetail || 'full');

    // Sync tree granularity
    $('#tv_tree_granularity').val(settings.treeGranularity ?? 0);

    // Sync LLM chunk size
    $('#tv_chunk_tokens').val(settings.llmChunkTokens ?? 30000);

    // Sync vector dedup
    const dedupEnabled = settings.enableVectorDedup === true;
    $('#tv_vector_dedup').prop('checked', dedupEnabled);
    $('#tv_dedup_threshold_row').toggle(dedupEnabled);
    $('#tv_dedup_threshold').val(settings.vectorDedupThreshold ?? 0.85);
    updateDedupStatus(dedupEnabled);

    // Sync mandatory tool calls & prompt injection
    $('#tv_mandatory_tools').prop('checked', settings.mandatoryTools === true);
    $('#tv_mandatory_prompt_options').toggle(settings.mandatoryTools === true);
    $('#tv_mandatory_position').val(settings.mandatoryPromptPosition || 'in_chat');
    $('#tv_mandatory_depth').val(settings.mandatoryPromptDepth ?? 1);
    $('#tv_mandatory_role').val(settings.mandatoryPromptRole || 'system');
    $('#tv_mandatory_prompt_text').val(settings.mandatoryPromptText || '');
    $('#tv_mandatory_depth_row').toggle((settings.mandatoryPromptPosition || 'in_chat') === 'in_chat');

    // Sync notebook injection settings
    $('#tv_notebook_position').val(settings.notebookPromptPosition || 'in_chat');
    $('#tv_notebook_depth').val(settings.notebookPromptDepth ?? 1);
    $('#tv_notebook_role').val(settings.notebookPromptRole || 'system');
    $('#tv_notebook_depth_row').toggle((settings.notebookPromptPosition || 'in_chat') === 'in_chat');

    $('#tv_stealth_mode').prop('checked', settings.stealthMode === true);
    $('#tv_ephemeral_results').prop('checked', settings.ephemeralResults === true);
    $('#tv_ephemeral_filter_options').toggle(settings.ephemeralResults === true);
    const filterList = settings.ephemeralToolFilter || [];
    $('.tv_ephemeral_tool').each(function () {
        $(this).prop('checked', filterList.includes($(this).val()));
    });

    // Sync slash command context setting
    $('#tv_command_context').val(settings.commandContextMessages ?? 50);

    // Sync auto-summary settings
    const autoEnabled = settings.autoSummaryEnabled === true;
    $('#tv_auto_summary_enabled').prop('checked', autoEnabled);
    $('#tv_auto_summary_options').toggle(autoEnabled);
    $('#tv_auto_summary_interval').val(settings.autoSummaryInterval ?? 20);
    $('#tv_auto_summary_count').text(getAutoSummaryCount());
    $('#tv_auto_hide_summarized').prop('checked', settings.autoHideSummarized !== false);
    $('#tv_passthrough_constant').prop('checked', settings.passthroughConstant !== false);
    $('#tv_allow_keyword_triggers').prop('checked', settings.allowKeywordTriggers === true);

    // Sync multi-book mode
    $(`input[name="tv_multi_book_mode"][value="${settings.multiBookMode || 'unified'}"]`).prop('checked', true);

    // Sync connection profile + sidecar sampler controls
    populateConnectionProfiles();

    populateLorebookDropdown();
    $('#tv_lorebook_controls').toggle(!!currentLorebook);

    if (currentLorebook) {
        loadLorebookUI(currentLorebook);
    }
}

function onLorebookFilter() {
    const query = $('#tv_lorebook_filter').val().toLowerCase().trim();
    $('#tv_lorebook_list .tv-lorebook-card').each(function () {
        const bookName = $(this).attr('data-book')?.toLowerCase() || '';
        $(this).toggle(!query || bookName.includes(query));
    });
}

function populateLorebookDropdown() {
    syncSelectedLorebook();
    const $list = $('#tv_lorebook_list');
    $list.empty();

    if (!world_names?.length) {
        $list.append('<div class="tv-help-text" style="text-align:center; padding: 12px;">No lorebooks found.</div>');
        return;
    }

    // Sort: TV-enabled first, then active in chat, then alphabetical
    const activeBooks = getActiveTunnelVisionBooks();
    const sorted = [...world_names].sort((a, b) => {
        const aTV = isLorebookEnabled(a) ? 1 : 0;
        const bTV = isLorebookEnabled(b) ? 1 : 0;
        if (aTV !== bTV) return bTV - aTV;
        const aActive = activeBooks.includes(a) ? 1 : 0;
        const bActive = activeBooks.includes(b) ? 1 : 0;
        if (aActive !== bActive) return bActive - aActive;
        return a.localeCompare(b);
    });

    for (const name of sorted) {
        const isActive = activeBooks.includes(name);
        const tvEnabled = isLorebookEnabled(name);
        const tree = getTree(name);
        const hasTree = !!tree?.root?.children?.length;

        const $card = $('<div class="tv-lorebook-card"></div>')
            .toggleClass('tv-lorebook-active', isActive)
            .toggleClass('tv-lorebook-selected', name === currentLorebook)
            .attr('data-book', name);

        const $info = $('<div class="tv-lorebook-card-info"></div>');
        const $name = $('<span class="tv-lorebook-card-name"></span>').text(name);
        $info.append($name);

        // Status badges
        const $badges = $('<div class="tv-lorebook-card-badges"></div>');
        if (!isActive) {
            $badges.append('<span class="tv-badge-inactive">inactive</span>');
        }
        if (tvEnabled) {
            $badges.append('<span class="tv-badge-tv-on"><i class="fa-solid fa-eye"></i> TV On</span>');
        }
        if (hasTree) {
            const count = (tree.root.children || []).length;
            $badges.append(`<span class="tv-badge-tree">${count} cat</span>`);
        }
        $info.append($badges);

        // Status indicator dot
        const dotClass = tvEnabled ? 'tv-dot-on' : (hasTree ? 'tv-dot-ready' : 'tv-dot-off');
        const $dot = $(`<span class="tv-lorebook-dot ${dotClass}"></span>`);

        $card.append($dot, $info);

        $card.on('click', () => {
            selectCurrentLorebook(name);
            $('.tv-lorebook-card').removeClass('tv-lorebook-selected');
            $card.addClass('tv-lorebook-selected');
            $('#tv_lorebook_controls').show();
            loadLorebookUI(name);
        });

        $list.append($card);
    }
}

// ─── Lorebook & Toggle Handlers ──────────────────────────────────

function onGlobalToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.globalEnabled = enabled;
    saveSettingsDebounced();
    $('#tv_main_controls').toggle(enabled);
    $('#tv_conditional_master_row').toggle(enabled);
    enabled ? registerTools() : unregisterTools();
}

function onLorebookSelect() {
    // Legacy handler for hidden select (kept for compatibility)
    const bookName = $(this).val();
    selectCurrentLorebook(bookName || null);
    $('#tv_lorebook_controls').toggle(!!bookName);
    if (bookName) loadLorebookUI(bookName);
}

async function loadLorebookUI(bookName) {
    const bookData = await loadWorldInfo(bookName);
    if (bookData?.entries) {
        await syncTrackerUidsForLorebook(bookName, bookData.entries);
    }
    $('#tv_lorebook_enabled').prop('checked', isLorebookEnabled(bookName));
    $('#tv_book_description').val(getBookDescription(bookName) || '');
    $('#tv_book_permission').val(getBookPermission(bookName));
    const tree = getTree(bookName);
    updateTreeStatus(bookName, tree);
    await renderTreeEditor(bookName, tree);
    await renderUnassignedEntries(bookName, tree, bookData);
    updateIngestUI();
}

function updateIngestUI() {
    const context = getContext();
    const hasChat = !!(context.chatId && context.chat?.length > 0);
    const hasBook = !!currentLorebook && isLorebookEnabled(currentLorebook);

    $('#tv_ingest_container').toggle(hasBook);

    if (hasChat) {
        const maxIdx = context.chat.length - 1;
        $('#tv_ingest_to').attr('max', maxIdx).val(maxIdx);
        $('#tv_ingest_from').attr('max', maxIdx);
        $('#tv_ingest_chat_info').text(`Chat has ${context.chat.length} messages (0-${maxIdx})`);
        $('#tv_ingest_chat').prop('disabled', false);
    } else {
        $('#tv_ingest_chat_info').text('No chat open. Open a chat to ingest messages.');
        $('#tv_ingest_chat').prop('disabled', true);
    }
}

function onLorebookToggle() {
    if (!currentLorebook) return;
    setLorebookEnabled(currentLorebook, $(this).prop('checked'));
    registerTools();
    populateLorebookDropdown(); // refresh badges
}

function onBookDescriptionChange() {
    if (!currentLorebook) return;
    const desc = $(this).val().trim();
    setBookDescription(currentLorebook, desc);
    saveSettingsDebounced();
}

function onToolToggle() {
    const toolName = $(this).data('tool');
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    const disabledTools = settings.disabledTools || {};
    if (enabled) {
        delete disabledTools[toolName];
    } else {
        disabledTools[toolName] = true;
    }
    settings.disabledTools = disabledTools;

    // Sync notebook injection setting with tool toggle
    if (toolName === 'TunnelVision_Notebook') {
        settings.notebookEnabled = enabled;
    }

    saveSettingsDebounced();
    registerTools();
}

// ─── Tool Confirmation Toggles ───────────────────────────────────

function onToolConfirmToggle() {
    const toolName = $(this).data('tool');
    const settings = getSettings();
    if (!settings.confirmTools) settings.confirmTools = {};
    settings.confirmTools[toolName] = $(this).prop('checked');
    saveSettingsDebounced();
    registerTools();
}

// ─── Tool Prompt Overrides ───────────────────────────────────────

function renderToolPromptOverrides() {
    const $container = $('#tv_tool_prompt_overrides');
    $container.empty();

    const settings = getSettings();
    const overrides = settings.toolPromptOverrides || {};
    const defaults = getDefaultToolDescriptions();

    for (const [toolName, defaultDesc] of Object.entries(defaults)) {
        const rawOverride = overrides[toolName] ? stripDynamicContent(overrides[toolName]) : null;
        const currentValue = rawOverride || defaultDesc;
        const isModified = !!rawOverride && rawOverride !== defaultDesc;
        const shortName = toolName.replace('TunnelVision_', '');

        const $block = $(`<div class="tv-tool-prompt-block ${isModified ? 'tv-tool-prompt-modified' : ''}"></div>`);
        const $header = $('<div class="tv-tool-prompt-header"></div>');
        $header.append(`<span class="tv-tool-prompt-label">${shortName}</span>`);
        $header.append(`<button class="tv-tool-prompt-reset" data-tool="${toolName}" title="Reset to default">Reset</button>`);
        $block.append($header);

        const $textarea = $(`<textarea class="tv-tool-prompt-textarea" data-tool="${toolName}" rows="4"></textarea>`);
        $textarea.val(currentValue);
        $block.append($textarea);

        $container.append($block);
    }
}

function onToolPromptChange() {
    const toolName = $(this).data('tool');
    const value = stripDynamicContent($(this).val());
    const settings = getSettings();
    if (!settings.toolPromptOverrides) settings.toolPromptOverrides = {};

    const defaults = getDefaultToolDescriptions();
    if (value === defaults[toolName]) {
        // Value matches default — remove override
        delete settings.toolPromptOverrides[toolName];
        $(this).closest('.tv-tool-prompt-block').removeClass('tv-tool-prompt-modified');
    } else {
        settings.toolPromptOverrides[toolName] = value;
        $(this).closest('.tv-tool-prompt-block').addClass('tv-tool-prompt-modified');
    }
    saveSettingsDebounced();
}

function onToolPromptReset() {
    const toolName = $(this).data('tool');
    const settings = getSettings();
    if (settings.toolPromptOverrides) {
        delete settings.toolPromptOverrides[toolName];
    }
    saveSettingsDebounced();

    const defaults = getDefaultToolDescriptions();
    const $block = $(this).closest('.tv-tool-prompt-block');
    $block.find('.tv-tool-prompt-textarea').val(defaults[toolName] || '');
    $block.removeClass('tv-tool-prompt-modified');
}

function onSearchModeChange() {
    const mode = $('input[name="tv_search_mode"]:checked').val();
    const settings = getSettings();
    settings.searchMode = mode;
    saveSettingsDebounced();
    $('#tv_collapsed_depth_section').toggle(mode === 'collapsed');
    // Re-register to rebuild tool description with new mode
    registerTools();
}

function onCollapsedDepthChange() {
    const raw = Number($('#tv_collapsed_depth').val());
    const clamped = Math.min(4, Math.max(1, Math.round(raw) || 2));
    $('#tv_collapsed_depth').val(clamped);
    const settings = getSettings();
    settings.collapsedDepth = clamped;
    saveSettingsDebounced();
    registerTools();
}

async function onSelectiveRetrievalToggle() {
    const settings = getSettings();
    settings.selectiveRetrieval = $(this).prop('checked');
    saveSettingsDebounced();
    await registerTools();
}

function onRecurseLimitChange() {
    const raw = Number($('#tv_recurse_limit').val());
    const clamped = Math.min(Math.max(Math.round(raw) || 5, 1), 50);
    $('#tv_recurse_limit').val(clamped);
    $('#tv_recurse_warn').toggle(clamped > 10);

    const settings = getSettings();
    settings.recurseLimit = clamped;
    saveSettingsDebounced();
    applyRecurseLimit(settings);
}

function onLlmDetailChange() {
    const settings = getSettings();
    settings.llmBuildDetail = $('#tv_llm_detail').val();
    saveSettingsDebounced();
}

function onTreeGranularityChange() {
    const settings = getSettings();
    settings.treeGranularity = Number($('#tv_tree_granularity').val()) || 0;
    saveSettingsDebounced();
}

function onChunkTokensChange() {
    const raw = Number($('#tv_chunk_tokens').val());
    const clamped = Math.min(Math.max(Math.round(raw / 1000) * 1000 || 30000, 5000), 500000);
    $('#tv_chunk_tokens').val(clamped);

    const settings = getSettings();
    settings.llmChunkTokens = clamped;
    saveSettingsDebounced();
}

function onVectorDedupToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.enableVectorDedup = enabled;
    saveSettingsDebounced();
    $('#tv_dedup_threshold_row').toggle(enabled);
    updateDedupStatus(enabled);
}

function onDedupThresholdChange() {
    const raw = Number($('#tv_dedup_threshold').val());
    const clamped = Math.min(Math.max(raw, 0.5), 0.99);
    $('#tv_dedup_threshold').val(clamped);

    const settings = getSettings();
    settings.vectorDedupThreshold = clamped;
    saveSettingsDebounced();
}

/**
 * Update the dedup status indicator.
 * @param {boolean} enabled
 */
function updateDedupStatus(enabled) {
    const $status = $('#tv_dedup_status');
    const $text = $('#tv_dedup_method_text');
    if (!enabled) {
        $status.hide();
        return;
    }
    $status.show();
    $text.text('Using trigram similarity — fast character n-gram matching that catches near-duplicates and morphological variants.');
}

// ─── Tree Building ───────────────────────────────────────────────

async function onBuildFromMetadata() {
    if (!currentLorebook) return;
    const $btn = $('#tv_build_metadata');
    try {
        $btn.prop('disabled', true).html('<span class="tv_loading"></span> Building...');
        const tree = await buildTreeFromMetadata(currentLorebook);
        toastr.success(`Built tree with ${(tree.root.children || []).length} categories`, 'TunnelVision');
        loadLorebookUI(currentLorebook);
        populateLorebookDropdown();
        registerTools();
    } catch (e) {
        toastr.error(e.message, 'TunnelVision');
        console.error('[TunnelVision]', e);
    } finally {
        $btn.prop('disabled', false).html('<i class="fa-solid fa-sitemap"></i> From Metadata');
    }
}

async function onBuildWithLLM() {
    if (!currentLorebook) return;
    const $btn = $('#tv_build_llm');
    const $progress = $('#tv_build_progress');
    const $progressText = $('#tv_build_progress_text');
    const $progressFill = $('#tv_build_progress_fill');
    const $progressDetail = $('#tv_build_progress_detail');

    try {
        $btn.prop('disabled', true).html('<span class="tv_loading"></span> Building...');
        $('#tv_build_metadata').prop('disabled', true);
        $progress.slideDown(200);
        $progressFill.css('width', '0%');
        $progressDetail.text('');

        const tree = await buildTreeWithLLM(currentLorebook, {
            onProgress: (msg, pct) => {
                $progressText.text(msg);
                if (typeof pct === 'number') {
                    $progressFill.css('width', `${Math.min(pct, 100)}%`);
                }
            },
            onDetail: (msg) => $progressDetail.text(msg),
        });

        $progressFill.css('width', '100%');
        $progressText.text('Done!');
        toastr.success(`LLM built tree with ${(tree.root.children || []).length} categories`, 'TunnelVision');
        loadLorebookUI(currentLorebook);
        populateLorebookDropdown();
        registerTools();
    } catch (e) {
        toastr.error(e.message, 'TunnelVision');
        console.error('[TunnelVision]', e);
    } finally {
        $btn.prop('disabled', false).html('<i class="fa-solid fa-brain"></i> With LLM');
        $('#tv_build_metadata').prop('disabled', false);
        setTimeout(() => $progress.slideUp(300), 2000);
    }
}

// ─── Chat Ingest ─────────────────────────────────────────────────

async function onIngestChat() {
    if (!currentLorebook) return;

    const context = getContext();
    if (!context.chatId || !context.chat?.length) {
        toastr.error('No chat is open. Open a chat first.', 'TunnelVision');
        return;
    }

    const from = parseInt($('#tv_ingest_from').val(), 10) || 0;
    const to = parseInt($('#tv_ingest_to').val(), 10) || 0;

    if (from > to) {
        toastr.warning('"From" must be less than or equal to "To".', 'TunnelVision');
        return;
    }

    const $btn = $('#tv_ingest_chat');
    const $progress = $('#tv_ingest_progress');
    const $progressText = $('#tv_ingest_progress_text');
    const $progressFill = $('#tv_ingest_progress_fill');
    const $progressDetail = $('#tv_ingest_progress_detail');

    try {
        $btn.prop('disabled', true).html('<span class="tv_loading"></span> Ingesting...');
        $progress.slideDown(200);
        $progressFill.css('width', '0%');
        $progressDetail.text('');

        const result = await ingestChatMessages(currentLorebook, {
            from,
            to,
            progress: (msg, pct) => {
                $progressText.text(msg);
                if (typeof pct === 'number') {
                    $progressFill.css('width', `${Math.min(pct, 100)}%`);
                }
            },
            detail: (msg) => $progressDetail.text(msg),
        });

        $progressFill.css('width', '100%');
        $progressText.text('Done!');
        toastr.success(`Created ${result.created} entries from chat (${result.errors} errors)`, 'TunnelVision');
        loadLorebookUI(currentLorebook);
        registerTools();
    } catch (e) {
        toastr.error(e.message, 'TunnelVision');
        console.error('[TunnelVision] Ingest error:', e);
    } finally {
        $btn.prop('disabled', false).html('<i class="fa-solid fa-download"></i> Ingest Messages');
        setTimeout(() => $progress.slideUp(300), 2000);
    }
}

function onMandatoryToolsToggle() {
    const settings = getSettings();
    settings.mandatoryTools = $(this).prop('checked');
    saveSettingsDebounced();
    $('#tv_mandatory_prompt_options').toggle(settings.mandatoryTools);
}

function onPromptInjectionChange() {
    const settings = getSettings();
    const $el = $(this);
    const id = $el.attr('id') || '';

    if (id.startsWith('tv_mandatory_')) {
        const field = id.replace('tv_mandatory_', '');
        if (field === 'position') {
            settings.mandatoryPromptPosition = $el.val();
            $('#tv_mandatory_depth_row').toggle($el.val() === 'in_chat');
        } else if (field === 'depth') {
            settings.mandatoryPromptDepth = Math.max(1, Math.round(Number($el.val()) || 1));
            $el.val(settings.mandatoryPromptDepth);
        } else if (field === 'role') {
            settings.mandatoryPromptRole = $el.val();
        }
    } else if (id.startsWith('tv_notebook_')) {
        const field = id.replace('tv_notebook_', '');
        if (field === 'position') {
            settings.notebookPromptPosition = $el.val();
            $('#tv_notebook_depth_row').toggle($el.val() === 'in_chat');
        } else if (field === 'depth') {
            settings.notebookPromptDepth = Math.max(1, Math.round(Number($el.val()) || 1));
            $el.val(settings.notebookPromptDepth);
        } else if (field === 'role') {
            settings.notebookPromptRole = $el.val();
        }
    }

    saveSettingsDebounced();
}

function onMandatoryPromptTextChange() {
    const settings = getSettings();
    settings.mandatoryPromptText = $(this).val() || '';
    saveSettingsDebounced();
}

function onMandatoryPromptReset() {
    const settings = getSettings();
    settings.mandatoryPromptText = SETTING_DEFAULTS.mandatoryPromptText;
    $('#tv_mandatory_prompt_text').val(settings.mandatoryPromptText);
    saveSettingsDebounced();
}

function onStealthModeToggle() {
    const settings = getSettings();
    settings.stealthMode = $(this).prop('checked');
    saveSettingsDebounced();
    void refreshHiddenToolCallMessages({ syncFlags: true });
}

function onEphemeralResultsToggle() {
    const settings = getSettings();
    settings.ephemeralResults = $(this).prop('checked');
    $('#tv_ephemeral_filter_options').toggle(settings.ephemeralResults);
    saveSettingsDebounced();
}

function onEphemeralToolFilterChange() {
    const settings = getSettings();
    const selected = [];
    $('.tv_ephemeral_tool:checked').each(function () {
        selected.push($(this).val());
    });
    settings.ephemeralToolFilter = selected;
    saveSettingsDebounced();
}

// ─── Slash Commands Settings ─────────────────────────────────────

function onCommandContextChange() {
    const raw = Number($('#tv_command_context').val());
    const clamped = Math.min(Math.max(Math.round(raw) || 50, 5), 500);
    $('#tv_command_context').val(clamped);
    const settings = getSettings();
    settings.commandContextMessages = clamped;
    saveSettingsDebounced();
}

// ─── Auto-Summary Settings ──────────────────────────────────────

function onAutoSummaryToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.autoSummaryEnabled = enabled;
    saveSettingsDebounced();
    $('#tv_auto_summary_options').toggle(enabled);
}

function onAutoSummaryIntervalChange() {
    const raw = Number($('#tv_auto_summary_interval').val());
    const clamped = Math.min(Math.max(Math.round(raw) || 20, 5), 200);
    $('#tv_auto_summary_interval').val(clamped);
    const settings = getSettings();
    settings.autoSummaryInterval = clamped;
    saveSettingsDebounced();
}

function onAutoHideSummarizedToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.autoHideSummarized = enabled;
    saveSettingsDebounced();
}

function onPassthroughConstantToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.passthroughConstant = enabled;
    saveSettingsDebounced();
}

function onAllowKeywordTriggersToggle() {
    const enabled = $(this).prop('checked');
    const settings = getSettings();
    settings.allowKeywordTriggers = enabled;
    saveSettingsDebounced();
}

// ─── Multi-Book Mode ─────────────────────────────────────────────

function onMultiBookModeChange() {
    const mode = $('input[name="tv_multi_book_mode"]:checked').val();
    const settings = getSettings();
    settings.multiBookMode = mode;
    saveSettingsDebounced();
    registerTools();
}

// ─── Sidecar LLM (Connection Profile + Sampler Overrides) ───────

function onConnectionProfileChange() {
    setConnectionProfileId($(this).val() || null);
    // Show/hide sampler controls when a profile is selected
    $('#tv_sidecar_sampler_fields').toggle(!!$(this).val());
}

function onSidecarTemperatureChange() {
    const val = parseFloat($(this).val());
    const settings = getSettings();
    settings.sidecarTemperature = val;
    $('#tv_sidecar_temp_val').text(val.toFixed(2));
    saveSettingsDebounced();
}

function onSidecarMaxTokensChange() {
    const settings = getSettings();
    settings.sidecarMaxTokens = Number($(this).val()) || 2048;
    saveSettingsDebounced();
}

function onSidecarAutoRetrievalToggle() {
    const settings = getSettings();
    settings.sidecarAutoRetrieval = $(this).prop('checked');
    $('#tv_sidecar_retrieval_fields').toggle(settings.sidecarAutoRetrieval);
    saveSettingsDebounced();
}

function onSidecarContextMessagesChange() {
    const settings = getSettings();
    settings.sidecarContextMessages = Number($(this).val()) || 10;
    saveSettingsDebounced();
}

function onSidecarMaxInjectionChange() {
    const settings = getSettings();
    settings.sidecarMaxInjectionTokens = Number($(this).val()) || 4000;
    saveSettingsDebounced();
}

function onSidecarPostGenWriterToggle() {
    const settings = getSettings();
    settings.sidecarPostGenWriter = $(this).prop('checked');
    $('#tv_sidecar_writer_fields').toggle(settings.sidecarPostGenWriter);
    saveSettingsDebounced();
}

function onCompactToolPromptsToggle() {
    const settings = getSettings();
    settings.compactToolPrompts = $(this).prop('checked');
    saveSettingsDebounced();
    registerTools();
}

function onSidecarWriterContextChange() {
    const settings = getSettings();
    settings.sidecarWriterContextMessages = Number($(this).val()) || 15;
    saveSettingsDebounced();
}

function onSidecarWriterMaxOpsChange() {
    const settings = getSettings();
    settings.sidecarWriterMaxOps = Number($(this).val()) || 5;
    saveSettingsDebounced();
}

// ─── Per-Lorebook Permissions ────────────────────────────────────

function onBookPermissionChange() {
    if (!currentLorebook) return;
    setBookPermission(currentLorebook, $(this).val() || 'read_write');
    registerTools();
}

function populateConnectionProfiles() {
    const $select = $('#tv_connection_profile');
    const currentVal = getConnectionProfileId() || '';

    // Keep the first option (default)
    $select.find('option:not(:first)').remove();

    for (const profile of listConnectionProfiles().sort((a, b) => (a.name || '').localeCompare(b.name || ''))) {
        if (!profile?.id || !profile?.name) continue;
        $select.append($('<option></option>').val(profile.id).text(profile.name));
    }

    $select.val(currentVal);

    // Sync sampler controls
    const settings = getSettings();
    $('#tv_sidecar_temperature').val(settings.sidecarTemperature ?? 0.2);
    $('#tv_sidecar_temp_val').text((settings.sidecarTemperature ?? 0.2).toFixed(2));
    $('#tv_sidecar_max_tokens').val(settings.sidecarMaxTokens || 2048);
    $('#tv_sidecar_sampler_fields').toggle(!!currentVal);

    // Sync auto-retrieval controls
    const autoRetrieval = settings.sidecarAutoRetrieval === true;
    $('#tv_sidecar_auto_retrieval').prop('checked', autoRetrieval);
    $('#tv_sidecar_retrieval_fields').toggle(autoRetrieval);
    $('#tv_sidecar_context_messages').val(settings.sidecarContextMessages ?? 10);
    $('#tv_sidecar_max_injection').val(settings.sidecarMaxInjectionTokens ?? 4000);

    // Sync conditional triggers toggle
    $('#tv_conditional_triggers').prop('checked', settings.conditionalTriggersEnabled !== false);

    // Sync compact tool prompts
    $('#tv_compact_tool_prompts').prop('checked', settings.compactToolPrompts === true);

    // Sync post-gen writer controls
    const postGenWriter = settings.sidecarPostGenWriter === true;
    $('#tv_sidecar_post_gen_writer').prop('checked', postGenWriter);
    $('#tv_sidecar_writer_fields').toggle(postGenWriter);
    $('#tv_sidecar_writer_context').val(settings.sidecarWriterContextMessages ?? 15);
    $('#tv_sidecar_writer_max_ops').val(settings.sidecarWriterMaxOps ?? 5);
}

// ─── Tree Management ─────────────────────────────────────────────

/**
 * Open the tree editor for a specific lorebook. Exported for use by the activity feed.
 * Sets currentLorebook so internal state stays consistent.
 * @param {string} bookName
 */
export async function openTreeEditorForBook(bookName) {
    selectCurrentLorebook(bookName);
    await onOpenTreeEditor();
}

async function onOpenTreeEditor() {
    if (!currentLorebook) return;
    const tree = getTree(currentLorebook);
    if (!tree || !tree.root) {
        toastr.warning('Build a tree first before opening the editor.', 'TunnelVision');
        return;
    }

    const bookData = await loadWorldInfo(currentLorebook);
    if (bookData?.entries) {
        await syncTrackerUidsForLorebook(currentLorebook, bookData.entries);
    }
    const entryLookup = buildEntryLookup(bookData);
    const bookName = currentLorebook;

    // State: which node is selected in the tree
    let selectedNode = tree.root;

    // Build the popup content
    const $popup = $('<div class="tv-popup-editor"></div>');

    // Toolbar
    const $toolbar = $(`<div class="tv-popup-toolbar">
        <div class="tv-popup-toolbar-left">
            <span class="tv-popup-title"><i class="fa-solid fa-folder-tree"></i> ${escapeHtml(bookName)}</span>
        </div>
        <div class="tv-popup-toolbar-right">
            <button class="tv-popup-btn" id="tv_popup_add_cat" title="Add category"><i class="fa-solid fa-folder-plus"></i> Add Category</button>
            <button class="tv-popup-btn" id="tv_popup_regen" title="Regenerate summaries"><i class="fa-solid fa-rotate"></i> Regen Summaries</button>
            <button class="tv-popup-btn" id="tv_popup_export" title="Export"><i class="fa-solid fa-file-export"></i></button>
            <button class="tv-popup-btn" id="tv_popup_import" title="Import"><i class="fa-solid fa-file-import"></i></button>
            <button class="tv-popup-btn tv-popup-btn-danger" id="tv_popup_delete" title="Delete tree"><i class="fa-solid fa-trash-can"></i></button>
        </div>
    </div>`);
    $popup.append($toolbar);

    // Search bar
    const $search = $(`<div class="tv-popup-search">
        <i class="fa-solid fa-magnifying-glass"></i>
        <input type="text" id="tv_popup_search" placeholder="Search categories and entries..." />
    </div>`);
    $popup.append($search);

    // Body: tree sidebar + main panel
    const $body = $('<div class="tv-popup-body"></div>');
    const $treeSidebar = $('<div class="tv-tree-sidebar"></div>');
    const $treeHeader = $('<div class="tv-tree-sidebar-header"><span>Tree</span></div>');
    const $treeScroll = $('<div class="tv-tree-sidebar-scroll"></div>');
    $treeSidebar.append($treeHeader, $treeScroll);

    const $mainPanel = $('<div class="tv-main-panel"></div>');

    $body.append($treeSidebar, $mainPanel);
    $popup.append($body);

    // --- Render functions ---

    function selectNode(node) {
        selectedNode = node;
        renderTreeNodes();
        renderMainPanel();
    }

    function isRootNode(node) {
        return !!node && node.id === tree.root.id;
    }

    function countActiveEntries(node) {
        return getAllEntryUids(node).filter(uid => !!entryLookup[uid] && !entryLookup[uid].disable).length;
    }

    function assignEntryToNode(uid, targetNode) {
        removeEntryFromTree(tree.root, uid);
        addEntryToNode(targetNode, uid);
        saveTree(bookName, tree);
    }

    function renderTreeNodes() {
        $treeScroll.empty();
        $treeScroll.append(buildTreeNode(tree.root, 0, { isRoot: true }));
        // Unassigned pseudo-node
        const unassigned = getUnassignedEntries(bookData, tree);
        if (unassigned.length > 0) {
            const $unRow = $('<div class="tv-tree-row tv-tree-row-unassigned"></div>');
            $unRow.append($('<span class="tv-tree-toggle"></span>'));
            $unRow.append($('<span class="tv-tree-dot" style="opacity:0.4"></span>'));
            $unRow.append($('<span class="tv-tree-label" style="color:var(--SmartThemeQuoteColor,#888)"></span>').text('Unassigned'));
            $unRow.append($(`<span class="tv-tree-count">${unassigned.length}</span>`));
            $unRow.on('click', () => {
                selectedNode = { id: '__unassigned__', label: 'Unassigned', entryUids: unassigned.map(e => e.uid), children: [] };
                renderTreeNodes();
                renderMainPanel();
            });
            if (selectedNode?.id === '__unassigned__') $unRow.addClass('active');
            $treeScroll.append($('<div style="margin-top:6px;padding-top:6px;border-top:1px solid var(--SmartThemeBorderColor,#444)"></div>').append($unRow));
        }
    }

    function buildTreeNode(node, depth, { isRoot = false } = {}) {
        const $wrapper = $('<div class="tv-tree-node"></div>');
        const hasChildren = (node.children || []).length > 0;
        const isActive = selectedNode?.id === node.id;
        const count = countActiveEntries(node);
        const label = isRoot ? 'Root' : (node.label || 'Unnamed');

        const $row = $(`<div class="tv-tree-row${isActive ? ' active' : ''}${isRoot ? ' tv-tree-row-root' : ''}"></div>`);
        const $toggle = $(`<span class="tv-tree-toggle">${hasChildren ? (node.collapsed ? '\u25B6' : '\u25BC') : ''}</span>`);
        const $dot = $('<span class="tv-tree-dot"></span>');
        const $label = $('<span class="tv-tree-label"></span>').text(label);
        const $count = $(`<span class="tv-tree-count">${count}</span>`);

        // Click toggle to expand/collapse
        $toggle.on('click', (e) => {
            e.stopPropagation();
            node.collapsed = !node.collapsed;
            saveTree(bookName, tree);
            renderTreeNodes();
        });

        // Click row to select
        $row.on('click', () => selectNode(node));

        // Drop target: drag entries onto tree nodes
        $row.on('dragover', (e) => { e.preventDefault(); $row.addClass('tv-tree-drop-target'); });
        $row.on('dragleave', () => $row.removeClass('tv-tree-drop-target'));
        $row.on('drop', (e) => {
            e.preventDefault();
            $row.removeClass('tv-tree-drop-target');
            const raw = e.originalEvent.dataTransfer.getData('text/plain');
            if (!raw || !/^\d+$/.test(raw)) return;
            const uid = Number(raw);
            assignEntryToNode(uid, node);
            selectNode(node);
            renderUnassignedEntries(bookName, tree, bookData);
            registerTools();
        });

        $row.append($toggle, $dot, $label, $count);
        $wrapper.append($row);

        // Children (recursive — no depth limit)
        if (hasChildren && !node.collapsed) {
            const $children = $('<div class="tv-tree-children"></div>');
            for (const child of node.children) {
                $children.append(buildTreeNode(child, depth + 1));
            }
            $wrapper.append($children);
        }

        return $wrapper;
    }

    function buildBreadcrumb(node) {
        if (node.id === '__unassigned__') {
            const $bc = $('<div class="tv-main-breadcrumb"></div>');
            const $rootCrumb = $('<span class="tv-bc-crumb"></span>').text('Root');
            $rootCrumb.on('click', () => selectNode(tree.root));
            $bc.append($rootCrumb);
            $bc.append($('<span class="tv-bc-sep">\u25B8</span>'));
            $bc.append($('<span class="tv-bc-current"></span>').text('Unassigned'));
            return $bc;
        }

        const path = [];
        const findPath = (current, target, trail) => {
            trail.push(current);
            if (current.id === target.id) return true;
            for (const child of (current.children || [])) {
                if (findPath(child, target, trail)) return true;
            }
            trail.pop();
            return false;
        };
        findPath(tree.root, node, path);

        const $bc = $('<div class="tv-main-breadcrumb"></div>');
        for (let i = 0; i < path.length; i++) {
            if (i > 0) $bc.append($('<span class="tv-bc-sep">\u25B8</span>'));
            const n = path[i];
            const label = n === tree.root ? 'Root' : (n.label || 'Unnamed');
            if (i < path.length - 1) {
                const $crumb = $('<span class="tv-bc-crumb"></span>').text(label);
                $crumb.on('click', () => selectNode(n));
                $bc.append($crumb);
            } else {
                $bc.append($('<span class="tv-bc-current"></span>').text(label));
            }
        }
        return $bc;
    }

    function renderMainPanel() {
        $mainPanel.empty();
        const node = selectedNode;
        if (!node) return;

        const isUnassigned = node.id === '__unassigned__';
        const isRoot = isRootNode(node);

        // Header
        const $header = $('<div class="tv-main-header"></div>');
        $header.append(buildBreadcrumb(node));

        const $titleRow = $('<div class="tv-main-title-row"></div>');
        if (!isUnassigned && !isRoot) {
            const $titleInput = $(`<input class="tv-main-title" type="text" />`).val(node.label || 'Unnamed');
            $titleInput.on('change', function () {
                node.label = $(this).val().trim() || 'Unnamed';
                saveTree(bookName, tree);
                renderTreeNodes();
                registerTools();
            });
            $titleRow.append($titleInput);

            const $actions = $('<div class="tv-main-title-actions"></div>');
            const $addSub = $('<button class="tv-popup-btn" title="Add sub-category"><i class="fa-solid fa-folder-plus"></i></button>');
            $addSub.on('click', () => {
                node.children = node.children || [];
                node.children.push(createTreeNode('New Sub-category'));
                node.collapsed = false;
                saveTree(bookName, tree);
                selectNode(node);
                registerTools();
            });
            const $regenSummary = $('<button class="tv-popup-btn" title="Regenerate summary for this node"><i class="fa-solid fa-arrows-rotate"></i></button>');
            $regenSummary.on('click', async (e) => {
                e.stopPropagation();
                const $icon = $regenSummary.find('i');
                try {
                    $regenSummary.prop('disabled', true);
                    $icon.addClass('fa-spin');
                    // Wrap in temp parent so the target node gets summarized (not skipped as root)
                    const tempWrapper = { children: [node], entryUids: [] };
                    await generateSummariesForTree(tempWrapper, bookName, true);
                    saveTree(bookName, tree);
                    renderTreeNodes();
                    renderMainPanel();
                    toastr.success(`Summary regenerated for "${node.label}".`, 'TunnelVision');
                } catch (err) {
                    toastr.error(err.message, 'TunnelVision');
                } finally {
                    $regenSummary.prop('disabled', false);
                    $icon.removeClass('fa-spin');
                }
            });
            const $delNode = $('<button class="tv-popup-btn tv-popup-btn-danger" title="Delete this node"><i class="fa-solid fa-trash-can"></i></button>');
            $delNode.on('click', () => {
                if (!confirm(`Delete "${node.label}" and unassign its entries?`)) return;
                removeNode(tree.root, node.id);
                saveTree(bookName, tree);
                selectedNode = tree.root;
                renderTreeNodes();
                renderMainPanel();
                renderUnassignedEntries(bookName, tree, bookData);
                registerTools();
            });
            $actions.append($addSub, $regenSummary, $delNode);
            $titleRow.append($actions);
        } else {
            $titleRow.append($('<div class="tv-main-title-static"></div>').text(isUnassigned ? 'Unassigned Entries' : 'Root'));
            if (isRoot) {
                const $actions = $('<div class="tv-main-title-actions"></div>');
                const $addSub = $('<button class="tv-popup-btn" title="Add category under root"><i class="fa-solid fa-folder-plus"></i></button>');
                $addSub.on('click', () => {
                    tree.root.children = tree.root.children || [];
                    tree.root.children.push(createTreeNode('New Category'));
                    tree.root.collapsed = false;
                    saveTree(bookName, tree);
                    selectNode(tree.root);
                    registerTools();
                });
                $actions.append($addSub);
                $titleRow.append($actions);
            }
        }
        $header.append($titleRow);
        $mainPanel.append($header);

        // Scrollable body
        const $body = $('<div class="tv-main-body"></div>');

        // Node summary
        if (node.summary && !isUnassigned && !isRoot) {
            $body.append($(`<div class="tv-node-summary">
                <div class="tv-node-summary-label">Node Summary</div>
                <div class="tv-node-summary-text"></div>
            </div>`).find('.tv-node-summary-text').text(node.summary).end());
        }

        // Direct entries
        const entryUids = node.entryUids || [];
        if (entryUids.length > 0) {
            const sectionLabel = isRoot ? 'Root Entries' : 'Direct Entries';
            $body.append($(`<div class="tv-entry-section-title">${sectionLabel} <span class="tv-entry-section-count">(${entryUids.length})</span></div>`));
            const $list = $('<div class="tv-entry-list-rows"></div>');
            for (const uid of entryUids) {
                const entry = entryLookup[uid];
                $list.append(buildEntryRow(uid, entry, node, bookName, tree, isUnassigned));
            }
            $body.append($list);
        }

        // Child nodes
        const children = node.children || [];
        if (children.length > 0) {
            $body.append($(`<div class="tv-entry-section-title">Sub-categories <span class="tv-entry-section-count">(${children.length})</span></div>`));
            const $cards = $('<div class="tv-child-cards"></div>');
            for (const child of children) {
                const childCount = countActiveEntries(child);
                const $card = $('<div class="tv-child-card"></div>');
                $card.append($('<span class="tv-tree-dot"></span>'));
                const $info = $('<div class="tv-child-card-info"></div>');
                $info.append($('<div class="tv-child-card-name"></div>').text(child.label || 'Unnamed'));
                if (child.summary) {
                    $info.append($('<div class="tv-child-card-summary"></div>').text(child.summary));
                }
                $card.append($info);
                $card.append($(`<span class="tv-child-card-count">${childCount}</span>`));
                $card.append($('<span class="tv-child-card-arrow">\u25B8</span>'));
                $card.on('click', () => {
                    child.collapsed = false;
                    saveTree(bookName, tree);
                    selectNode(child);
                });
                $cards.append($card);
            }
            $body.append($cards);
        }

        $mainPanel.append($body);
    }

    function buildEntryRow(uid, entry, node, bookName, tree, isUnassigned) {
        const label = entry ? (entry.comment || entry.key?.[0] || `#${uid}`) : `#${uid} (deleted)`;

        const $row = $(`<div class="tv-entry-row" draggable="true" data-uid="${uid}"></div>`);
        $row.append($('<span class="tv-entry-drag">\u22EE\u22EE</span>'));
        $row.append($('<span class="tv-entry-name"></span>').text(label));
        $row.append($(`<span class="tv-entry-uid">#${uid}</span>`));

        // Tracker toggle
        if (entry) {
            const tracked = isTrackerUid(bookName, uid);
            const $tracker = $(`<button class="tv-btn-icon tv-entry-tracker ${tracked ? 'is-on' : ''}" title="${tracked ? 'Tracked entry' : 'Track this entry'}"><i class="fa-solid ${tracked ? 'fa-location-crosshairs' : 'fa-location-dot'}"></i></button>`);
            $tracker.on('click', (e) => {
                e.stopPropagation();
                const nextTracked = !$tracker.hasClass('is-on');
                setTrackerUid(bookName, uid, nextTracked);
                $tracker.toggleClass('is-on', nextTracked);
                $tracker.attr('title', nextTracked ? 'Tracked entry' : 'Track this entry');
                $tracker.find('i').attr('class', `fa-solid ${nextTracked ? 'fa-location-crosshairs' : 'fa-location-dot'}`);
                registerTools();
            });
            $row.append($tracker);
        }

        // Enable/disable toggle
        if (entry) {
            const isDisabled = !!entry.disable;
            const $toggle = $(`<button class="tv-btn-icon tv-entry-toggle ${isDisabled ? 'is-off' : ''}" title="${isDisabled ? 'Enable entry' : 'Disable entry'}"><i class="fa-solid ${isDisabled ? 'fa-eye-slash' : 'fa-eye'}"></i></button>`);
            $toggle.on('click', async (e) => {
                e.stopPropagation();
                const wasTracked = isTrackerUid(bookName, uid);
                entry.disable = !entry.disable;
                await saveWorldInfo(bookName, bookData, true);
                if (entry.disable) {
                    setTrackerUid(bookName, uid, false);
                } else if (wasTracked || isTrackerTitle(entry.comment)) {
                    setTrackerUid(bookName, uid, true);
                }
                $toggle.toggleClass('is-off', !!entry.disable);
                $toggle.attr('title', entry.disable ? 'Enable entry' : 'Disable entry');
                $toggle.find('i').attr('class', `fa-solid ${entry.disable ? 'fa-eye-slash' : 'fa-eye'}`);
                $row.toggleClass('is-disabled', !!entry.disable);
                renderTreeNodes();
                renderMainPanel();
                await renderUnassignedEntries(bookName, tree, bookData);
                registerTools();
            });
            $row.append($toggle);
            if (isDisabled) $row.addClass('is-disabled');
        }

        if (!isUnassigned) {
            const $remove = $('<button class="tv-btn-icon tv-btn-danger-icon tv-entry-remove" title="Remove from node"><i class="fa-solid fa-xmark"></i></button>');
            $remove.on('click', async (e) => {
                e.stopPropagation();
                node.entryUids = (node.entryUids || []).filter(u => u !== uid);
                saveTree(bookName, tree);
                renderMainPanel();
                renderTreeNodes();
                await renderUnassignedEntries(bookName, tree, bookData);
                registerTools();
            });
            $row.append($remove);
        }

        // Drag
        $row.on('dragstart', (e) => {
            e.originalEvent.dataTransfer.setData('text/plain', String(uid));
            $row.addClass('dragging');
        });
        $row.on('dragend', () => $row.removeClass('dragging'));

        // Click to inline-expand entry detail
        if (entry) {
            $row.on('click', function () {
                const $existing = $row.next('.tv-entry-expand');
                if ($existing.length) {
                    $existing.slideUp(150, () => $existing.remove());
                    $row.removeClass('expanded');
                    return;
                }
                // Close any other expanded entries
                $row.closest('.tv-entry-list-rows').find('.tv-entry-expand').slideUp(150, function () { $(this).remove(); });
                $row.closest('.tv-entry-list-rows').find('.tv-entry-row').removeClass('expanded');

                $row.addClass('expanded');
                const $expand = $('<div class="tv-entry-expand" style="display:none"></div>');

                // Node summary context
                if (node.summary && !isUnassigned && !isRootNode(node)) {
                    $expand.append($(`<div class="tv-expand-node-box">
                        <div class="tv-expand-node-label">Parent node: ${escapeHtml(node.label || 'Unnamed')}</div>
                        <div class="tv-expand-node-text"></div>
                    </div>`).find('.tv-expand-node-text').text(node.summary).end());
                }

                // Keys (filter out condition tags — shown separately below)
                const allKeys = entry.key || [];
                const regularKeys = allKeys.filter(k => !isEvaluableCondition(k));
                if (regularKeys.length > 0) {
                    const $keys = $('<div class="tv-expand-keys"></div>');
                    $keys.append($('<span class="tv-expand-label">Keys</span>'));
                    const $tags = $('<div class="tv-expand-key-tags"></div>');
                    for (const k of regularKeys) {
                        $tags.append($('<span class="tv-expand-key-tag"></span>').text(k));
                    }
                    $keys.append($tags);
                    $expand.append($keys);
                }

                // Conditions editor — parsed from key[] and keysecondary[]
                {
                    const primaryParsed = separateConditions(entry.key || []);
                    const secondaryParsed = separateConditions(entry.keysecondary || []);
                    const hasPrimary = primaryParsed.conditions.length > 0;
                    const hasSecondary = secondaryParsed.conditions.length > 0;

                    const $condSection = $('<div class="tv-expand-conditions"></div>');
                    $condSection.append($('<span class="tv-expand-label">Conditions</span>'));

                    // Render existing condition tags
                    const $condTags = $('<div class="tv-condition-tags"></div>');

                    const renderCondTag = (cond, group) => {
                        const typeLabel = (CONDITION_LABELS[cond.type] || cond.type).toUpperCase();
                        const condStr = formatCondition(cond);
                        const negatedClass = cond.negated ? ' is-negated' : '';
                        const negTitle = cond.negated ? 'Negated — click to un-negate' : 'Not negated — click to negate';
                        const prob = getKeywordProbability(entry, condStr);
                        const probClass = prob < 100 ? ' tv-condition-prob-reduced' : '';
                        const $tag = $(`<span class="tv-condition-tag${negatedClass}" data-group="${group}" title="${group} condition">
                            <button class="tv-condition-neg-toggle tv-btn-icon" title="${negTitle}"><i class="fa-solid ${cond.negated ? 'fa-ban' : 'fa-check'}"></i></button>
                            <span class="tv-condition-type">${typeLabel}</span>:<span class="tv-condition-value">${escapeHtml(cond.value)}</span>
                            <span class="tv-condition-prob${probClass}" title="Click to change probability (0-100)">${prob}%</span>
                            <i class="fa-solid fa-xmark tv-condition-remove" title="Remove condition"></i>
                        </span>`);
                        $tag.find('.tv-condition-prob').on('click', (e) => {
                            e.stopPropagation();
                            const current = getKeywordProbability(entry, condStr);
                            const input = prompt(`Probability for ${condStr} (0-100):`, String(current));
                            if (input === null) return;
                            const val = parseInt(input, 10);
                            if (isNaN(val) || val < 0 || val > 100) return;
                            setKeywordProbability(entry, condStr, val);
                            saveWorldInfo(bookName, bookData, true);
                            $tag.find('.tv-condition-prob').text(`${val}%`).toggleClass('tv-condition-prob-reduced', val < 100);
                            toastr.info(`Set ${condStr} probability to ${val}%`, 'TunnelVision');
                        });
                        $tag.find('.tv-condition-neg-toggle').on('click', (e) => {
                            e.stopPropagation();
                            const oldStr = formatCondition(cond);
                            const targetArr = group === 'primary' ? entry.key : entry.keysecondary;
                            const idx = targetArr.indexOf(oldStr);
                            if (idx < 0) return;
                            // Migrate probability to new condition string
                            const oldProb = getKeywordProbability(entry, oldStr);
                            cond.negated = !cond.negated;
                            const newStr = formatCondition(cond);
                            targetArr[idx] = newStr;
                            if (oldProb < 100) {
                                setKeywordProbability(entry, newStr, oldProb);
                            }
                            if (entry.tvKeywordProbability?.[oldStr] !== undefined) {
                                delete entry.tvKeywordProbability[oldStr];
                            }
                            saveWorldInfo(bookName, bookData, true);
                            $tag.toggleClass('is-negated', !!cond.negated);
                            const $icon = $tag.find('.tv-condition-neg-toggle i');
                            $icon.attr('class', `fa-solid ${cond.negated ? 'fa-ban' : 'fa-check'}`);
                            $tag.find('.tv-condition-neg-toggle').attr('title', cond.negated ? 'Negated — click to un-negate' : 'Not negated — click to negate');
                            toastr.info(`${newStr} (${group})`, 'TunnelVision');
                        });
                        $tag.find('.tv-condition-remove').on('click', (e) => {
                            e.stopPropagation();
                            const targetArr = group === 'primary' ? entry.key : entry.keysecondary;
                            const idx = targetArr.indexOf(condStr);
                            if (idx >= 0) {
                                targetArr.splice(idx, 1);
                                // Clean up probability data
                                if (entry.tvKeywordProbability?.[condStr] !== undefined) {
                                    delete entry.tvKeywordProbability[condStr];
                                }
                                saveWorldInfo(bookName, bookData, true);
                                $tag.fadeOut(150, () => $tag.remove());
                                toastr.info(`Removed ${condStr}`, 'TunnelVision');
                            }
                        });
                        return $tag;
                    };

                    for (const c of primaryParsed.conditions) $condTags.append(renderCondTag(c, 'primary'));
                    for (const c of secondaryParsed.conditions) $condTags.append(renderCondTag(c, 'secondary'));

                    if (!hasPrimary && !hasSecondary) {
                        $condTags.append($('<span class="tv-condition-empty">No conditions set</span>'));
                    }
                    $condSection.append($condTags);

                    // Add condition controls
                    const typeOptions = [...EVALUABLE_TYPES].map(t => `<option value="${t}">${CONDITION_LABELS[t] || t}</option>`).join('');
                    const $addRow = $(`<div class="tv-condition-add-row">
                        <select class="tv-condition-type-select"><${typeOptions}</select>
                        <input type="text" class="tv-condition-value-input" placeholder="value (e.g. tense, forest)" />
                        <select class="tv-condition-group-select">
                            <option value="primary">Primary</option>
                            <option value="secondary">Secondary</option>
                        </select>
                        <button class="tv-popup-btn tv-popup-btn-sm tv-condition-add-btn" title="Add condition"><i class="fa-solid fa-plus"></i></button>
                    </div>`);

                    $addRow.find('.tv-condition-add-btn').on('click', (e) => {
                        e.stopPropagation();
                        const type = $addRow.find('.tv-condition-type-select').val();
                        const value = $addRow.find('.tv-condition-value-input').val().trim();
                        const group = $addRow.find('.tv-condition-group-select').val();
                        if (!value) {
                            toastr.warning('Enter a condition value.', 'TunnelVision');
                            return;
                        }
                        const condStr = `[${type}:${value}]`;
                        const targetArr = group === 'primary' ? (entry.key || (entry.key = [])) : (entry.keysecondary || (entry.keysecondary = []));
                        if (targetArr.includes(condStr)) {
                            toastr.warning('Condition already exists.', 'TunnelVision');
                            return;
                        }
                        targetArr.push(condStr);
                        saveWorldInfo(bookName, bookData, true);
                        // Add tag visually
                        $condTags.find('.tv-condition-empty').remove();
                        $condTags.append(renderCondTag({ type, value, negated: false }, group));
                        $addRow.find('.tv-condition-value-input').val('');
                        toastr.success(`Added ${condStr} (${group})`, 'TunnelVision');
                    });

                    // Allow Enter key to add
                    $addRow.find('.tv-condition-value-input').on('keydown', (e) => {
                        if (e.key === 'Enter') {
                            e.stopPropagation();
                            $addRow.find('.tv-condition-add-btn').trigger('click');
                        }
                    });

                    $condSection.append($addRow);
                    $expand.append($condSection);
                }

                // Content
                if (entry.content) {
                    $expand.append($('<div class="tv-expand-label">Content</div>'));
                    $expand.append($('<div class="tv-expand-content"></div>').text(entry.content));
                }

                // Edit button row
                const $editRow = $('<div class="tv-expand-edit-row"></div>');
                const $editBtn = $('<button class="tv-popup-btn tv-popup-btn-sm" title="Edit entry content"><i class="fa-solid fa-pen-to-square"></i> Edit</button>');
                $editBtn.on('click', (e) => {
                    e.stopPropagation();
                    // Toggle edit mode
                    const $contentEl = $expand.find('.tv-expand-content');
                    if ($contentEl.is('textarea')) {
                        // Save mode
                        entry.content = $contentEl.val();
                        saveWorldInfo(bookName, bookData, true);
                        const $newContent = $('<div class="tv-expand-content"></div>').text(entry.content);
                        $contentEl.replaceWith($newContent);
                        $editBtn.html('<i class="fa-solid fa-pen-to-square"></i> Edit');
                        toastr.success('Entry updated.', 'TunnelVision');
                    } else {
                        // Edit mode
                        const $textarea = $('<textarea class="tv-expand-content tv-expand-content-edit"></textarea>').val(entry.content || '');
                        if ($contentEl.length) {
                            $contentEl.replaceWith($textarea);
                        } else {
                            $expand.find('.tv-expand-label').last().after($textarea);
                        }
                        $textarea.focus();
                        $editBtn.html('<i class="fa-solid fa-floppy-disk"></i> Save');
                    }
                });
                $editRow.append($editBtn);
                $expand.append($editRow);

                $row.after($expand);
                $expand.slideDown(150);
            });
        }

        return $row;
    }

    // --- Initial render ---
    renderTreeNodes();
    renderMainPanel();

    // Wire toolbar buttons BEFORE showing popup (callGenericPopup awaits until close)
    $popup.find('#tv_popup_add_cat').on('click', () => {
        tree.root.children = tree.root.children || [];
        tree.root.children.push(createTreeNode('New Category'));
        tree.root.collapsed = false;
        saveTree(bookName, tree);
        renderTreeNodes();
        renderMainPanel();
        registerTools();
    });

    $popup.find('#tv_popup_regen').on('click', async () => {
        const $btn = $popup.find('#tv_popup_regen');
        try {
            $btn.prop('disabled', true).find('i').addClass('fa-spin');
            await generateSummariesForTree(tree.root, bookName);
            saveTree(bookName, tree);
            renderTreeNodes();
            renderMainPanel();
            registerTools();
            toastr.success('Summaries regenerated.', 'TunnelVision');
        } catch (e) {
            toastr.error(e.message, 'TunnelVision');
        } finally {
            $btn.prop('disabled', false).find('i').removeClass('fa-spin');
        }
    });

    $popup.find('#tv_popup_export').on('click', () => onExportTree());
    $popup.find('#tv_popup_import').on('click', () => $('#tv_import_file').trigger('click'));
    $popup.find('#tv_popup_delete').on('click', () => {
        if (!confirm(`Delete the entire tree for "${bookName}"?`)) return;
        deleteTree(bookName);
        toastr.info('Tree deleted.', 'TunnelVision');
        loadLorebookUI(bookName);
        populateLorebookDropdown();
        registerTools();
        $('.popup.active .popup-button-close, .popup:last-child [data-i18n="Close"]').trigger('click');
    });

    // Search filter
    $popup.find('#tv_popup_search').on('input', function () {
        const q = $(this).val().toLowerCase().trim();
        $treeScroll.find('.tv-tree-row').each(function () {
            if ($(this).hasClass('tv-tree-row-root')) {
                $(this).closest('.tv-tree-node').show();
                return;
            }
            const label = $(this).find('.tv-tree-label').text().toLowerCase();
            $(this).closest('.tv-tree-node').toggle(!q || label.includes(q));
        });
        $mainPanel.find('.tv-entry-row').each(function () {
            const name = $(this).find('.tv-entry-name').text().toLowerCase();
            $(this).toggle(!q || name.includes(q));
        });
    });

    // Show popup (blocks until user closes it)
    await callGenericPopup($popup, POPUP_TYPE.DISPLAY, '', {
        large: true,
        wide: true,
        allowVerticalScrolling: true,
        allowHorizontalScrolling: false,
    });

    // When popup closes, refresh sidebar UI
    loadLorebookUI(bookName);
    populateLorebookDropdown();
}

// ─── Tree Editor Helpers ─────────────────────────────────────────

function getUnassignedEntries(bookData, tree) {
    if (!bookData?.entries || !tree?.root) return [];
    const indexedUids = new Set(getAllEntryUids(tree.root));
    const unassigned = [];
    for (const key of Object.keys(bookData.entries)) {
        const entry = bookData.entries[key];
        if (entry.disable) continue;
        if (!indexedUids.has(entry.uid)) unassigned.push(entry);
    }
    return unassigned;
}

// ─── Import Sanitization ─────────────────────────────────────────

/**
 * Recursively sanitize an imported tree node.
 * Ensures all fields are the expected types, strips unexpected properties,
 * and prevents prototype pollution via __proto__ / constructor keys.
 * @param {Object} node
 */
function sanitizeImportedNode(node) {
    if (!node || typeof node !== 'object') return;

    // Enforce expected field types
    if (typeof node.id !== 'string' || !node.id) node.id = `tv_import_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
    if (typeof node.label !== 'string') node.label = 'Unnamed';
    if (typeof node.summary !== 'string') node.summary = '';
    if (!Array.isArray(node.entryUids)) node.entryUids = [];
    if (!Array.isArray(node.children)) node.children = [];

    // Sanitize entryUids — must be numbers
    node.entryUids = node.entryUids.filter(uid => typeof uid === 'number' && Number.isFinite(uid));

    // Strip any unexpected/dangerous keys (prototype pollution vectors)
    const allowed = new Set(['id', 'label', 'summary', 'entryUids', 'children', 'collapsed', 'isArc']);
    for (const key of Object.keys(node)) {
        if (!allowed.has(key)) delete node[key];
    }

    // Recurse children
    for (const child of node.children) {
        sanitizeImportedNode(child);
    }
}

// ─── Export / Import ─────────────────────────────────────────────

function onExportTree() {
    if (!currentLorebook) return;
    const tree = getTree(currentLorebook);
    if (!tree) {
        toastr.warning('No tree to export.', 'TunnelVision');
        return;
    }
    const blob = new Blob([JSON.stringify(tree, null, 2)], { type: 'application/json' });
    const url = URL.createObjectURL(blob);
    const a = document.createElement('a');
    a.href = url;
    a.download = `tunnelvision_${currentLorebook.replace(/[^a-z0-9]/gi, '_')}.json`;
    a.click();
    URL.revokeObjectURL(url);
    toastr.info('Tree exported.', 'TunnelVision');
}

function onImportTree(e) {
    if (!currentLorebook) return;
    const file = e.target.files?.[0];
    if (!file) return;

    const reader = new FileReader();
    reader.onload = (ev) => {
        try {
            const tree = JSON.parse(ev.target.result);
            if (!tree.root || !Array.isArray(tree.root.children)) {
                throw new Error('Invalid tree structure.');
            }
            // Sanitize imported tree to prevent injection of unexpected properties
            sanitizeImportedNode(tree.root);
            tree.lorebookName = currentLorebook;
            tree.lastBuilt = Date.now();
            // Strip any unexpected top-level keys
            const cleanTree = {
                lorebookName: tree.lorebookName,
                root: tree.root,
                version: Number(tree.version) || 1,
                lastBuilt: tree.lastBuilt,
            };
            saveTree(currentLorebook, cleanTree);
            toastr.success('Tree imported.', 'TunnelVision');
            loadLorebookUI(currentLorebook);
            registerTools();
        } catch (err) {
            toastr.error(`Import failed: ${err.message}`, 'TunnelVision');
        }
    };
    reader.readAsText(file);
    // Reset file input so same file can be re-imported
    $(e.target).val('');
}

function onBulkExport() {
    const settings = getSettings();
    const trees = settings.trees || {};
    const enabledLorebooks = settings.enabledLorebooks || {};
    const bookDescriptions = settings.bookDescriptions || {};
    const bookPermissions = settings.bookPermissions || {};

    // Build a snapshot of all trees + per-book settings
    const backup = {
        _tunnelvision_backup: true,
        version: 1,
        exportedAt: new Date().toISOString(),
        trees: JSON.parse(JSON.stringify(trees)),
        enabledLorebooks: JSON.parse(JSON.stringify(enabledLorebooks)),
        bookDescriptions: JSON.parse(JSON.stringify(bookDescriptions)),
        bookPermissions: JSON.parse(JSON.stringify(bookPermissions)),
    };

    const treeCount = Object.keys(trees).length;
    if (treeCount === 0) {
        toastr.warning('No trees to export.', 'TunnelVision');
        return;
    }

    const blob = new Blob([JSON.stringify(backup, null, 2)], { type: 'application/json' });
    const url = URL.createObjectURL(blob);
    const a = document.createElement('a');
    a.href = url;
    a.download = `tunnelvision_backup_${Date.now()}.json`;
    a.click();
    URL.revokeObjectURL(url);
    toastr.success(`Exported ${treeCount} tree(s) + settings.`, 'TunnelVision');
}

function onBulkImport(e) {
    const file = e.target.files?.[0];
    if (!file) return;

    const reader = new FileReader();
    reader.onload = (ev) => {
        try {
            const backup = JSON.parse(ev.target.result);
            if (!backup._tunnelvision_backup || !backup.trees) {
                throw new Error('Not a valid TunnelVision backup file.');
            }

            const settings = getSettings();
            const treeCount = Object.keys(backup.trees).length;

            // Confirm overwrite
            if (!confirm(`This will import ${treeCount} tree(s) and their settings. Existing trees with the same names will be overwritten. Continue?`)) {
                return;
            }

            // Import trees
            for (const [bookName, tree] of Object.entries(backup.trees)) {
                if (!tree?.root || !Array.isArray(tree.root.children)) {
                    console.warn(`[TunnelVision] Skipping invalid tree: ${bookName}`);
                    continue;
                }
                sanitizeImportedNode(tree.root);
                tree.lorebookName = bookName;
                saveTree(bookName, {
                    lorebookName: tree.lorebookName,
                    root: tree.root,
                    version: Number(tree.version) || 1,
                    lastBuilt: tree.lastBuilt || Date.now(),
                });
            }

            // Import per-book settings
            if (backup.enabledLorebooks) {
                for (const [bookName, enabled] of Object.entries(backup.enabledLorebooks)) {
                    setLorebookEnabled(bookName, enabled);
                }
            }
            if (backup.bookDescriptions) {
                for (const [bookName, desc] of Object.entries(backup.bookDescriptions)) {
                    setBookDescription(bookName, desc);
                }
            }
            if (backup.bookPermissions) {
                for (const [bookName, perm] of Object.entries(backup.bookPermissions)) {
                    setBookPermission(bookName, perm);
                }
            }

            toastr.success(`Imported ${treeCount} tree(s) + settings.`, 'TunnelVision');
            populateLorebookDropdown();
            registerTools();
        } catch (err) {
            toastr.error(`Bulk import failed: ${err.message}`, 'TunnelVision');
        }
    };
    reader.readAsText(file);
    $(e.target).val('');
}

// ─── Tree Status ─────────────────────────────────────────────────

function updateTreeStatus(bookName, tree) {
    const $info = $('#tv_tree_info');
    if (!tree) {
        $info.text('No tree built yet.');
        return;
    }
    const totalEntries = getAllEntryUids(tree.root).length;
    const categories = (tree.root.children || []).length;
    const date = new Date(tree.lastBuilt).toLocaleString();
    $info.text(`${categories} categories, ${totalEntries} indexed entries. Last built: ${date}`);
}

// ─── Tree Editor Rendering ───────────────────────────────────────

async function renderTreeEditor(bookName, tree) {
    const $container = $('#tv_tree_editor_container');

    if (!tree || !tree.root || ((tree.root.children || []).length === 0 && (tree.root.entryUids || []).length === 0)) {
        $container.hide();
        return;
    }

    $container.show();
    const totalEntries = getAllEntryUids(tree.root).length;
    const $count = $('#tv_tree_entry_count');
    if (totalEntries > 0) {
        $count.text(totalEntries).show();
    } else {
        $count.hide();
    }

    // Mini-kanban overview in sidebar
    const $overview = $('#tv_mini_kanban_overview');
    $overview.empty();
    const categories = [];
    if ((tree.root.entryUids || []).length > 0) {
        categories.push({
            label: 'Root',
            summary: 'Entries stored directly on the root node.',
            entryUids: tree.root.entryUids,
            children: [],
        });
    }
    categories.push(...(tree.root.children || []));
    const colors = ['#e84393', '#f0946c', '#6c5ce7', '#00b894', '#fdcb6e'];
    for (let i = 0; i < categories.length; i++) {
        const cat = categories[i];
        const count = getAllEntryUids(cat).length;
        const color = colors[i % colors.length];
        const $row = $(`<div class="tv-mini-cat">
            <div class="tv-mini-cat-stripe" style="background:${color}"></div>
            <div class="tv-mini-cat-info">
                <div class="tv-mini-cat-name"></div>
                <div class="tv-mini-cat-summary"></div>
            </div>
            <div class="tv-mini-cat-count">${count}</div>
        </div>`);
        $row.find('.tv-mini-cat-name').text(cat.label || 'Unnamed');
        $row.find('.tv-mini-cat-summary').text(cat.summary || '');
        $overview.append($row);
    }
}

// ─── Unassigned Entries ──────────────────────────────────────────

async function renderUnassignedEntries(bookName, tree, bookData = null) {
    const $container = $('#tv_unassigned_container');
    const $count = $('#tv_unassigned_count');
    const $list = $('#tv_unassigned_list');

    if (!tree || !tree.root) {
        $list.empty();
        $container.hide();
        return;
    }

    const resolvedBookData = bookData || await loadWorldInfo(bookName);
    if (!resolvedBookData || !resolvedBookData.entries) {
        $list.empty();
        $container.hide();
        return;
    }

    const unassigned = getUnassignedEntries(resolvedBookData, tree);
    $count.text(unassigned.length);
    $list.empty();

    for (const entry of unassigned) {
        const label = entry.comment || entry.key?.[0] || `#${entry.uid}`;
        const $chip = $('<button type="button" class="tv-unassigned-chip"></button>');
        $chip.append($('<span class="tv-unassigned-chip-label"></span>').text(label));
        $chip.append($(`<span class="tv-unassigned-chip-uid">#${entry.uid}</span>`));
        $chip.append($('<span class="tv-unassigned-chip-action"><i class="fa-solid fa-arrow-turn-down"></i> Root</span>'));
        $chip.on('click', async () => {
            addEntryToNode(tree.root, entry.uid);
            saveTree(bookName, tree);
            toastr.success(`Assigned "${label}" to Root.`, 'TunnelVision');
            await loadLorebookUI(bookName);
            populateLorebookDropdown();
            registerTools();
        });
        $list.append($chip);
    }

    if (unassigned.length === 0) {
        $container.hide();
    } else {
        $container.show();
    }
}

// ─── Diagnostics ─────────────────────────────────────────────────

async function onRunDiagnostics() {
    const $btn = $('#tv_run_diagnostics');
    const $output = $('#tv_diagnostics_output');

    $btn.prop('disabled', true).html('<span class="tv_loading"></span> Running...');
    $output.empty().show();

    try {
        const results = await runDiagnostics();
        for (const result of results) {
            const icon = result.status === 'pass' ? 'fa-check' : result.status === 'warn' ? 'fa-triangle-exclamation' : 'fa-xmark';
            const cssClass = `tv_diag_${result.status}`;
            const $item = $(`<div class="tv_diag_item ${cssClass}"><i class="fa-solid ${icon}"></i> ${escapeHtml(result.message)}</div>`);
            if (result.fix && typeof result.fix === 'function') {
                const $fixBtn = $(`<button class="tv-btn tv-btn-secondary" style="margin-left:8px;padding:2px 8px;font-size:0.85em;">${escapeHtml(result.fixLabel || 'Fix')}</button>`);
                $fixBtn.on('click', () => {
                    const msg = result.fix();
                    $fixBtn.replaceWith(`<span style="margin-left:8px;color:var(--tv-accent,#4fc3f7);">✓ ${escapeHtml(msg)}</span>`);
                });
                $item.append($fixBtn);
            }
            $output.append($item);
        }
    } catch (e) {
        $output.append(`<div class="tv_diag_item tv_diag_fail"><i class="fa-solid fa-xmark"></i> Diagnostics error: ${escapeHtml(e.message)}</div>`);
    } finally {
        $btn.prop('disabled', false).html('<i class="fa-solid fa-stethoscope"></i> Run Diagnostics');
    }
}

// ─── Utilities ───────────────────────────────────────────────────

function buildEntryLookup(bookData) {
    const lookup = {};
    if (!bookData || !bookData.entries) return lookup;
    for (const key of Object.keys(bookData.entries)) {
        const entry = bookData.entries[key];
        lookup[entry.uid] = entry;
    }
    return lookup;
}

function escapeHtml(str) {
    if (!str) return '';
    return String(str).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;');
}