coding-agent-search 0.5.2

//! JavaScript generation for HTML export.
//!
//! Generates inline JavaScript for:
//! - Search functionality (text search with highlighting)
//! - Theme toggle (light/dark mode)
//! - Tool call expand/collapse
//! - Encryption/decryption (Web Crypto API)

use super::template::ExportOptions;
use tracing::debug;

/// Bundle of JavaScript for the template.
pub struct ScriptBundle {
    /// Inline JavaScript to include in the document
    pub inline_js: String,
}

/// Generate all JavaScript for the template.
pub fn generate_scripts(options: &ExportOptions) -> ScriptBundle {
    let mut scripts = Vec::new();

    // Core utilities
    scripts.push(generate_core_utils());

    // Search functionality
    if options.include_search {
        scripts.push(generate_search_js());
    }

    // Theme toggle
    if options.include_theme_toggle {
        scripts.push(generate_theme_js());
    }

    // Tool call toggle
    if options.show_tool_calls {
        scripts.push(generate_tool_toggle_js());
    }

    // Encryption/decryption
    if options.encrypt {
        scripts.push(generate_decryption_js());
    }

    // World-class UI/UX enhancements (always included)
    scripts.push(generate_world_class_js());

    // Initialize on load
    scripts.push(generate_init_js(options));

    let inline_js = scripts.join("\n\n");
    debug!(
        component = "scripts",
        operation = "generate",
        include_search = options.include_search,
        include_theme_toggle = options.include_theme_toggle,
        show_tool_calls = options.show_tool_calls,
        encrypt = options.encrypt,
        inline_bytes = inline_js.len(),
        "Generated inline scripts"
    );

    ScriptBundle { inline_js }
}

fn generate_core_utils() -> String {
    r#"// Core utilities
const $ = (sel) => document.querySelector(sel);
const $$ = (sel) => document.querySelectorAll(sel);

// Toast notifications
const Toast = {
    container: null,

    init() {
        this.container = document.createElement('div');
        this.container.id = 'toast-container';
        this.container.style.cssText = 'position:fixed;bottom:1rem;right:1rem;z-index:9999;display:flex;flex-direction:column;gap:0.5rem;';
        document.body.appendChild(this.container);
    },

    show(message, type = 'info') {
        if (!this.container) this.init();
        const toast = document.createElement('div');
        toast.className = 'toast toast-' + type;
        toast.style.cssText = 'padding:0.75rem 1rem;background:var(--bg-surface);border:1px solid var(--border);border-radius:6px;color:var(--text-primary);box-shadow:0 4px 12px rgba(0,0,0,0.3);transform:translateX(100%);transition:transform 0.3s ease;';
        toast.textContent = message;
        this.container.appendChild(toast);
        requestAnimationFrame(() => toast.style.transform = 'translateX(0)');
        setTimeout(() => {
            toast.style.transform = 'translateX(100%)';
            setTimeout(() => toast.remove(), 300);
        }, 3000);
    }
};

// Copy to clipboard
async function copyToClipboard(text) {
    try {
        await navigator.clipboard.writeText(text);
        Toast.show('Copied to clipboard', 'success');
        return true;
    } catch (e) {
        // Fallback for older browsers
        const textarea = document.createElement('textarea');
        textarea.value = text;
        textarea.style.position = 'fixed';
        textarea.style.opacity = '0';
        document.body.appendChild(textarea);
        textarea.select();
        let ok = false;
        try {
            ok = document.execCommand('copy');
        } catch (e2) {
            // execCommand threw — ok stays false
        }
        textarea.remove();
        if (ok) {
            Toast.show('Copied to clipboard', 'success');
            return true;
        }
        Toast.show('Copy failed', 'error');
    }
    return false;
}

// Copy code block
async function copyCodeBlock(btn) {
    const pre = btn.closest('pre');
    const code = pre.querySelector('code');
    const ok = await copyToClipboard(code ? code.textContent : pre.textContent);
    if (ok) {
        btn.classList.add('copied');
        setTimeout(() => btn.classList.remove('copied'), 1500);
    }
}

// Print handler
function printConversation() {
    // Expand all collapsed sections before print
    $$('details, .tool-call').forEach(el => {
        if (el.tagName === 'DETAILS') el.open = true;
        else el.classList.add('expanded');
    });
    window.print();
}"#
        .to_string()
}

fn generate_search_js() -> String {
    r#"// Search functionality
const Search = {
    input: null,
    countEl: null,
    matches: [],
    currentIndex: -1,
    _initialized: false,

    init() {
        this.input = $('#search-input');
        this.countEl = $('#search-count');
        if (!this.input) return;

        if (!this.countEl && this.input.parentNode) {
            const count = document.createElement('span');
            count.id = 'search-count';
            count.className = 'search-count';
            count.hidden = true;
            this.input.parentNode.appendChild(count);
            this.countEl = count;
        }
        if (!this.countEl) return;

        if (!this._initialized) {
            this.input.addEventListener('input', () => this.search());
            this.input.addEventListener('keydown', (e) => {
                if (e.key === 'Enter') {
                    e.preventDefault();
                    if (e.shiftKey) {
                        this.prev();
                    } else {
                        this.next();
                    }
                } else if (e.key === 'Escape') {
                    this.clear();
                    this.input.blur();
                }
            });

            // Keyboard shortcut: Ctrl/Cmd + F for search
            document.addEventListener('keydown', (e) => {
                if ((e.ctrlKey || e.metaKey) && e.key === 'f') {
                    e.preventDefault();
                    this.input.focus();
                    this.input.select();
                }
            });
            this._initialized = true;
        }
    },

    search() {
        this.clearHighlights();
        $$('.message.search-hit').forEach((el) => el.classList.remove('search-hit'));
        const query = this.input.value.trim().toLowerCase();
        if (!query) {
            this.countEl.hidden = true;
            return;
        }

        this.matches = [];
        const hitMessages = new Set();
        let searchRoots = $$('.message');
        if (!searchRoots || searchRoots.length === 0) {
            searchRoots = $$('.message-content');
        }
        searchRoots.forEach((el) => {
            const messageEl = el.classList && el.classList.contains('message') ? el : el.closest('.message');
            const walker = document.createTreeWalker(el, NodeFilter.SHOW_TEXT);
            let node;
            while ((node = walker.nextNode())) {
                const text = node.textContent.toLowerCase();
                let index = text.indexOf(query);
                while (index !== -1) {
                    this.matches.push({ node, index, length: query.length });
                    if (messageEl) hitMessages.add(messageEl);
                    index = text.indexOf(query, index + 1);
                }
            }
        });

        hitMessages.forEach((el) => el.classList.add('search-hit'));
        this.highlightAll();
        this.updateCount();

        if (this.matches.length > 0) {
            this.currentIndex = 0;
            this.scrollToCurrent();
        }
    },

    highlightAll() {
        // Process in reverse to preserve indices
        for (let i = this.matches.length - 1; i >= 0; i--) {
            const match = this.matches[i];
            const range = document.createRange();
            try {
                range.setStart(match.node, match.index);
                range.setEnd(match.node, match.index + match.length);
                const span = document.createElement('span');
                span.className = 'search-highlight';
                span.dataset.matchIndex = i;
                range.surroundContents(span);
            } catch (e) {
                // Skip invalid ranges
            }
        }
    },

    clearHighlights() {
        const parents = new Set();
        $$('.search-highlight').forEach((el) => {
            const parent = el.parentNode;
            while (el.firstChild) {
                parent.insertBefore(el.firstChild, el);
            }
            parent.removeChild(el);
            parents.add(parent);
        });
        // Merge adjacent text nodes so subsequent searches work correctly
        parents.forEach((p) => p.normalize());
        this.matches = [];
        this.currentIndex = -1;
    },

    updateCount() {
        if (this.matches.length > 0) {
            this.countEl.textContent = `${this.currentIndex + 1}/${this.matches.length}`;
            this.countEl.hidden = false;
        } else {
            this.countEl.textContent = 'No results';
            this.countEl.hidden = false;
        }
    },

    scrollToCurrent() {
        $$('.search-current').forEach((el) => el.classList.remove('search-current'));
        if (this.currentIndex >= 0 && this.currentIndex < this.matches.length) {
            const highlight = $(`[data-match-index="${this.currentIndex}"]`);
            if (highlight) {
                highlight.classList.add('search-current');
                highlight.scrollIntoView({ behavior: 'smooth', block: 'center' });
            }
        }
        this.updateCount();
    },

    next() {
        if (this.matches.length === 0) return;
        this.currentIndex = (this.currentIndex + 1) % this.matches.length;
        this.scrollToCurrent();
    },

    prev() {
        if (this.matches.length === 0) return;
        this.currentIndex = (this.currentIndex - 1 + this.matches.length) % this.matches.length;
        this.scrollToCurrent();
    },

    clear() {
        this.input.value = '';
        this.clearHighlights();
        this.countEl.hidden = true;
    }
};"#
    .to_string()
}

fn generate_theme_js() -> String {
    r#"// Theme toggle
const Theme = {
    toggle: null,

    init() {
        this.toggle = $('#theme-toggle');
        if (!this.toggle) return;

        // Load saved preference or system preference
        const saved = localStorage.getItem('cass-theme');
        const prefersDark = window.matchMedia('(prefers-color-scheme: dark)').matches;
        const theme = saved || (prefersDark ? 'dark' : 'light');
        document.documentElement.setAttribute('data-theme', theme);

        this.toggle.addEventListener('click', () => this.toggleTheme());

        // Listen for system theme changes
        window.matchMedia('(prefers-color-scheme: dark)').addEventListener('change', (e) => {
            if (!localStorage.getItem('cass-theme')) {
                document.documentElement.setAttribute('data-theme', e.matches ? 'dark' : 'light');
            }
        });
    },

    toggleTheme() {
        const current = document.documentElement.getAttribute('data-theme');
        const next = current === 'dark' ? 'light' : 'dark';
        document.documentElement.setAttribute('data-theme', next);
        localStorage.setItem('cass-theme', next);
    }
};"#
    .to_string()
}

fn generate_tool_toggle_js() -> String {
    r#"// Tool call expand/collapse
const ToolCalls = {
    init() {
        $$('.tool-call-header').forEach((header) => {
            if (header.dataset.toolToggleBound === 'true') return;
            header.dataset.toolToggleBound = 'true';
            header.addEventListener('click', () => {
                const toolCall = header.closest('.tool-call');
                toolCall.classList.toggle('expanded');
            });
        });
    }
};

// Tool badge popover controller
const ToolPopovers = {
    activePopover: null,
    activeBadge: null,
    _outsideClickBound: false,

    init() {
        this.initBadges();
        this.initOverflowBadges();
        this.initOutsideClick();
    },

    initBadges() {
        $$('.tool-badge:not(.tool-overflow)').forEach(badge => {
            if (badge.dataset.popoverBound === 'true') return;
            badge.dataset.popoverBound = 'true';
            // Helper to get popover - looks up fresh each time since popover may be built dynamically
            const getPopover = () => badge.querySelector('.tool-popover');

            // Show on hover (desktop)
            badge.addEventListener('mouseenter', () => this.show(badge, getPopover()));
            badge.addEventListener('mouseleave', () => this.hide(badge, getPopover()));

            // Show on focus (keyboard accessibility)
            badge.addEventListener('focus', () => this.show(badge, getPopover()));
            badge.addEventListener('blur', (e) => {
                // Don't hide if focus moves within the popover
                const popover = getPopover();
                if (!popover || !popover.contains(e.relatedTarget)) {
                    this.hide(badge, popover);
                }
            });

            // Toggle on click (mobile support)
            badge.addEventListener('click', (e) => {
                e.preventDefault();
                e.stopPropagation();
                this.toggle(badge, getPopover());
            });

            // Keyboard support
            badge.addEventListener('keydown', (e) => {
                if (e.key === 'Enter' || e.key === ' ') {
                    e.preventDefault();
                    this.toggle(badge, getPopover());
                } else if (e.key === 'Escape') {
                    this.hide(badge, getPopover());
                    badge.focus();
                }
            });
        });
    },

    initOverflowBadges() {
        $$('.tool-overflow').forEach(btn => {
            if (btn.dataset.overflowBound === 'true') return;
            btn.dataset.overflowBound = 'true';
            // Store original text
            btn.dataset.originalText = btn.textContent.trim();

            btn.addEventListener('click', (e) => {
                e.preventDefault();
                e.stopPropagation();
                const container = btn.closest('.message-header-right');
                if (!container) return;

                const isExpanded = container.classList.toggle('expanded');
                btn.textContent = isExpanded ? 'Less' : btn.dataset.originalText;
                btn.setAttribute('aria-expanded', isExpanded);
            });
        });
    },

    initOutsideClick() {
        if (this._outsideClickBound) return;
        this._outsideClickBound = true;
        document.addEventListener('click', (e) => {
            if (!e.target.closest('.tool-badge')) {
                this.hideAll();
            }
        });
    },

    show(badge, popover) {
        if (!popover) {
            // Build popover from data attributes if not present
            popover = this.buildPopover(badge);
            if (!popover) return;
        }

        // Hide any other active popover first
        if (this.activeBadge && this.activeBadge !== badge) {
            this.hide(this.activeBadge, this.activePopover);
        }

        popover.classList.add('visible');
        badge.setAttribute('aria-expanded', 'true');
        this.position(badge, popover);

        this.activePopover = popover;
        this.activeBadge = badge;
    },

    hide(badge, popover) {
        if (popover) {
            popover.classList.remove('visible');
            popover.style.position = '';
            popover.style.top = '';
            popover.style.left = '';
        }
        if (badge) {
            badge.setAttribute('aria-expanded', 'false');
        }
        if (this.activeBadge === badge) {
            this.activePopover = null;
            this.activeBadge = null;
        }
    },

    hideAll() {
        $$('.tool-popover.visible').forEach(p => {
            p.classList.remove('visible');
        });
        $$('.tool-badge[aria-expanded="true"]').forEach(b => {
            b.setAttribute('aria-expanded', 'false');
        });
        this.activePopover = null;
        this.activeBadge = null;
    },

    toggle(badge, popover) {
        const isVisible = popover && popover.classList.contains('visible');
        if (isVisible) {
            this.hide(badge, popover);
        } else {
            this.show(badge, popover);
        }
    },

    buildPopover(badge) {
        // Build a popover from data attributes if no inline popover exists
        const name = badge.dataset.toolName;
        const input = badge.dataset.toolInput;
        const output = badge.dataset.toolOutput;

        if (!name) return null;

        const popover = document.createElement('div');
        popover.className = 'tool-popover';
        popover.setAttribute('role', 'tooltip');

        let html = '<div class="tool-popover-header"><strong>' + this.escapeHtml(name) + '</strong></div>';

        if (input && input.trim()) {
            html += '<div class="tool-popover-section"><span class="tool-popover-label">Input</span><pre><code>' + this.escapeHtml(input) + '</code></pre></div>';
        }

        if (output && output.trim()) {
            html += '<div class="tool-popover-section"><span class="tool-popover-label">Output</span><pre><code>' + this.escapeHtml(output) + '</code></pre></div>';
        }

        popover.innerHTML = html;
        badge.appendChild(popover);
        return popover;
    },

    escapeHtml(text) {
        const div = document.createElement('div');
        div.textContent = text;
        return div.innerHTML;
    },

    position(badge, popover) {
        // Skip positioning on mobile - CSS handles bottom sheet style
        if (window.innerWidth < 768) {
            return;
        }

        popover.style.position = 'fixed';

        // Use fixed positioning relative to viewport
        const badgeRect = badge.getBoundingClientRect();
        const viewportWidth = window.innerWidth;
        const viewportHeight = window.innerHeight;
        const margin = 8;

        // Measure popover dimensions (temporarily make visible for measurement)
        popover.style.visibility = 'hidden';
        popover.style.display = 'block';
        const popoverRect = popover.getBoundingClientRect();
        popover.style.display = '';
        popover.style.visibility = '';

        // Default: position below and align left edge with badge
        let top = badgeRect.bottom + margin;
        let left = badgeRect.left;

        // Flip up if would overflow bottom
        if (top + popoverRect.height > viewportHeight - margin) {
            top = badgeRect.top - popoverRect.height - margin;
            popover.classList.add('popover-above');
        } else {
            popover.classList.remove('popover-above');
        }

        // Flip to align right edge if would overflow right
        if (left + popoverRect.width > viewportWidth - margin) {
            left = Math.max(margin, badgeRect.right - popoverRect.width);
        }

        // Ensure not off left edge
        left = Math.max(margin, left);

        // Ensure not off top edge
        top = Math.max(margin, top);

        popover.style.top = top + 'px';
        popover.style.left = left + 'px';
    }
};"#
    .to_string()
}

fn generate_world_class_js() -> String {
    r#"// World-class UI/UX enhancements
const WorldClass = {
    scrollProgress: null,
    floatingNav: null,
    gradientMesh: null,
    lastScrollY: 0,
    ticking: false,
    currentMessageIndex: -1,
    messages: [],
    _initialized: false,

    init() {
        this.messages = Array.from($$('.message'));
        this.scrollProgress = $('#scroll-progress');
        this.floatingNav = $('#floating-nav');
        this.initFloatingNav();
        this.initIntersectionObserver();
        this.initMessageLinks();
        // Bind document/window-level handlers only once to avoid duplicates
        // after decryption re-init (these targets survive innerHTML replacement)
        if (!this._initialized) {
            this.initKeyboardNav();
            this.initScrollHandler();
            this.initShareButton();
            this._initialized = true;
        }
    },

    initFloatingNav() {
        if (!this.floatingNav) return;

        const scrollTopBtn = $('#scroll-top');
        if (scrollTopBtn) {
            scrollTopBtn.onclick = () => {
                window.scrollTo({ top: 0, behavior: 'smooth' });
            };
        }
    },

    initScrollHandler() {
        const toolbar = $('.toolbar');
        let lastScrollY = window.scrollY;
        let scrollDirection = 'up';

        const updateScroll = () => {
            const scrollY = window.scrollY;
            const scrollHeight = document.documentElement.scrollHeight - window.innerHeight;
            const progress = scrollHeight > 0 ? (scrollY / scrollHeight) * 100 : 0;

            // Update progress bar
            if (this.scrollProgress) {
                this.scrollProgress.style.width = `${progress}%`;
            }

            // Show/hide floating nav
            if (this.floatingNav) {
                if (scrollY > 300) {
                    this.floatingNav.classList.add('visible');
                } else {
                    this.floatingNav.classList.remove('visible');
                }
            }

            // Mobile: hide toolbar on scroll down (only if wide enough scroll)
            if (toolbar && window.innerWidth < 768) {
                scrollDirection = scrollY > lastScrollY ? 'down' : 'up';
                if (scrollDirection === 'down' && scrollY > 200) {
                    toolbar.classList.add('toolbar-hidden');
                } else {
                    toolbar.classList.remove('toolbar-hidden');
                }
            }

            lastScrollY = scrollY;
            this.ticking = false;
        };

        window.addEventListener('scroll', () => {
            if (!this.ticking) {
                requestAnimationFrame(updateScroll);
                this.ticking = true;
            }
        }, { passive: true });
    },

    initIntersectionObserver() {
        if (!('IntersectionObserver' in window)) return;

        const reduceMotion = window.matchMedia('(prefers-reduced-motion: reduce)').matches;
        if (reduceMotion) {
            this.messages.forEach((msg) => {
                msg.style.opacity = '1';
                msg.style.transform = 'none';
                msg.classList.add('in-view');
            });
            return;
        }

        const observer = new IntersectionObserver((entries) => {
            entries.forEach(entry => {
                if (entry.isIntersecting) {
                    entry.target.classList.add('in-view');
                    observer.unobserve(entry.target);
                }
            });
        }, {
            threshold: 0.1,
            rootMargin: '0px 0px -50px 0px'
        });

        // Initially hide messages for animation
        // Must match CSS @keyframes messageReveal 'from' state exactly
        this.messages.forEach((msg, i) => {
            msg.style.opacity = '0';
            msg.style.transform = 'translateY(24px) scale(0.97)';
            setTimeout(() => observer.observe(msg), i * 30);
        });
    },

    initKeyboardNav() {
        document.addEventListener('keydown', (e) => {
            // Ignore if in input/textarea
            if (e.target.matches('input, textarea')) return;

            switch(e.key) {
                case 'j':
                    e.preventDefault();
                    this.navigateMessage(1);
                    break;
                case 'k':
                    e.preventDefault();
                    this.navigateMessage(-1);
                    break;
                case 'g':
                    e.preventDefault();
                    this.navigateToMessage(0);
                    break;
                case 'G':
                    e.preventDefault();
                    this.navigateToMessage(this.messages.length - 1);
                    break;
                case '/':
                    if (!e.ctrlKey && !e.metaKey) {
                        e.preventDefault();
                        const searchInput = $('#search-input');
                        if (searchInput) {
                            searchInput.focus();
                            searchInput.select();
                        }
                    }
                    break;
                case '?':
                    e.preventDefault();
                    this.showShortcutsHint();
                    break;
            }
        });
    },

    navigateMessage(direction) {
        const newIndex = Math.max(0, Math.min(this.messages.length - 1, this.currentMessageIndex + direction));
        this.navigateToMessage(newIndex);
    },

    navigateToMessage(index) {
        // Remove focus from current
        if (this.currentMessageIndex >= 0 && this.messages[this.currentMessageIndex]) {
            this.messages[this.currentMessageIndex].classList.remove('keyboard-focus');
        }

        this.currentMessageIndex = index;
        const msg = this.messages[index];
        if (msg) {
            msg.classList.add('keyboard-focus');
            msg.scrollIntoView({ behavior: 'smooth', block: 'center' });
        }
    },

    showShortcutsHint() {
        let hint = $('.shortcuts-hint');
        if (!hint) {
            hint = document.createElement('div');
            hint.className = 'shortcuts-hint';
            hint.innerHTML = '<kbd>j</kbd>/<kbd>k</kbd> navigate • <kbd>g</kbd> first • <kbd>G</kbd> last • <kbd>/</kbd> search • <kbd>?</kbd> help';
            document.body.appendChild(hint);
        }
        hint.classList.add('visible');
        setTimeout(() => hint.classList.remove('visible'), 3000);
    },

    initMessageLinks() {
        this.messages.forEach((msg, i) => {
            if (msg.querySelector('.message-link')) return;
            const btn = document.createElement('button');
            btn.className = 'message-link';
            btn.title = 'Copy link to message';
            btn.setAttribute('aria-label', 'Copy link to message');
            btn.innerHTML = '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><path d="M10 13a5 5 0 007.54.54l3-3a5 5 0 00-7.07-7.07l-1.72 1.71"/><path d="M14 11a5 5 0 00-7.54-.54l-3 3a5 5 0 007.07 7.07l1.71-1.71"/></svg>';
            btn.onclick = (e) => {
                e.stopPropagation();
                const id = msg.id || `msg-${i}`;
                if (!msg.id) msg.id = id;
                const url = `${window.location.href.split('#')[0]}#${id}`;
                copyToClipboard(url);
                btn.classList.add('copied');
                setTimeout(() => btn.classList.remove('copied'), 1500);
            };
            msg.appendChild(btn);
        });
    },

    initShareButton() {
        if (!navigator.share) return;

        const toolbar = $('.toolbar');
        if (!toolbar) return;

        const shareBtn = document.createElement('button');
        shareBtn.className = 'toolbar-btn';
        shareBtn.title = 'Share';
        shareBtn.setAttribute('aria-label', 'Share');
        shareBtn.innerHTML = '<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><path d="M4 12v8a2 2 0 002 2h12a2 2 0 002-2v-8"/><polyline points="16,6 12,2 8,6"/><line x1="12" y1="2" x2="12" y2="15"/></svg>';
        shareBtn.onclick = async () => {
            try {
                await navigator.share({
                    title: document.title,
                    url: window.location.href
                });
            } catch (e) {
                if (e.name !== 'AbortError') {
                    Toast.show('Share failed', 'error');
                }
            }
        };
        toolbar.appendChild(shareBtn);
    }
};

// Touch ripple effect for mobile
function createRipple(event) {
    const button = event.currentTarget;
    const rect = button.getBoundingClientRect();
    const ripple = document.createElement('span');
    const size = Math.max(rect.width, rect.height);
    ripple.style.width = ripple.style.height = `${size}px`;
    ripple.style.left = `${event.clientX - rect.left - size/2}px`;
    ripple.style.top = `${event.clientY - rect.top - size/2}px`;
    ripple.className = 'ripple';
    button.appendChild(ripple);
    setTimeout(() => ripple.remove(), 600);
}

// Add ripple to touch devices
if ('ontouchstart' in window) {
    document.addEventListener('DOMContentLoaded', () => {
        $$('.toolbar button, .floating-btn').forEach(btn => {
            btn.addEventListener('touchstart', createRipple);
        });
    });
}"#
    .to_string()
}

fn generate_decryption_js() -> String {
    r#"// Decryption using Web Crypto API
const Crypto = {
    modal: null,
    form: null,
    errorEl: null,

    init() {
        this.modal = $('#password-modal');
        this.form = $('#password-form');
        this.errorEl = $('#decrypt-error');

        if (!this.modal || !this.form) return;

        this.form.addEventListener('submit', (e) => {
            e.preventDefault();
            this.decrypt();
        });
    },

    async decrypt() {
        const passphrase = $('#password-input').value;
        if (!passphrase) return;

        try {
            this.errorEl.hidden = true;

            // Get encrypted content
            const encryptedEl = $('#encrypted-content');
            if (!encryptedEl) throw new Error('No encrypted content found');

            const encryptedData = JSON.parse(encryptedEl.textContent);
            const { salt, iv, ciphertext, iterations } = encryptedData;
            if (!salt || !iv || !ciphertext || !Number.isInteger(iterations) || iterations <= 0) {
                throw new Error('Invalid encryption parameters');
            }

            // Derive key from password
            const enc = new TextEncoder();
            const keyMaterial = await crypto.subtle.importKey(
                'raw',
                enc.encode(passphrase),
                'PBKDF2',
                false,
                ['deriveBits', 'deriveKey']
            );

            const key = await crypto.subtle.deriveKey(
                {
                    name: 'PBKDF2',
                    salt: this.base64ToBytes(salt),
                    iterations: iterations,
                    hash: 'SHA-256'
                },
                keyMaterial,
                { name: 'AES-GCM', length: 256 },
                false,
                ['decrypt']
            );

            // Decrypt
            const decrypted = await crypto.subtle.decrypt(
                {
                    name: 'AES-GCM',
                    iv: this.base64ToBytes(iv)
                },
                key,
                this.base64ToBytes(ciphertext)
            );

            // Replace content
            const dec = new TextDecoder();
            const plaintext = dec.decode(decrypted);
            const conversation = $('#conversation');
            conversation.innerHTML = plaintext;

            // Hide modal
            this.modal.hidden = true;
            this.form.reset();

            // Re-initialize tool calls and popovers
            if (typeof ToolCalls !== 'undefined') {
                ToolCalls.init();
            }
            if (typeof ToolPopovers !== 'undefined') {
                ToolPopovers.init();
            }
            if (typeof Search !== 'undefined') {
                Search.init();
            }
            if (typeof WorldClass !== 'undefined') {
                WorldClass.init();
            }
            if (typeof __cassAttachCodeCopyButtons === 'function') {
                __cassAttachCodeCopyButtons();
            }

        } catch (e) {
            this.errorEl.textContent = 'Decryption failed. Wrong password?';
            this.errorEl.hidden = false;
        }
    },

    base64ToBytes(base64) {
        const binary = atob(base64);
        const bytes = new Uint8Array(binary.length);
        for (let i = 0; i < binary.length; i++) {
            bytes[i] = binary.charCodeAt(i);
        }
        return bytes;
    }
};"#
    .to_string()
}

fn generate_init_js(options: &ExportOptions) -> String {
    let mut inits = Vec::new();

    if options.include_search {
        inits.push("try { Search.init(); } catch (e) { console.error('Search init failed', e); }");
    }

    if options.include_theme_toggle {
        inits.push("try { Theme.init(); } catch (e) { console.error('Theme init failed', e); }");
    }

    if options.show_tool_calls {
        inits.push(
            "try { ToolCalls.init(); } catch (e) { console.error('ToolCalls init failed', e); }",
        );
        inits.push("try { ToolPopovers.init(); } catch (e) { console.error('ToolPopovers init failed', e); }");
    }

    if options.encrypt {
        inits.push("try { Crypto.init(); } catch (e) { console.error('Crypto init failed', e); }");
    }

    // World-class UI/UX enhancements (always init)
    inits.push(
        "try { WorldClass.init(); } catch (e) { console.error('WorldClass init failed', e); }",
    );

    // Always add code block copy buttons and print button handler.
    inits.push(
        "try { __cassAttachCodeCopyButtons(); } catch (e) { console.error('Code copy init failed', e); }",
    );

    let copy_button_helpers = r#"// Add copy buttons to code blocks
// Idempotent so encrypted exports can re-run this after decrypting content.
const __cassAttachCodeCopyButtons = () => {
    $$('pre code').forEach((code) => {
        const pre = code.parentNode;
        if (!pre || pre.querySelector('.copy-code-btn')) return;
        const btn = document.createElement('button');
        btn.className = 'copy-code-btn';
        btn.innerHTML = '<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><rect x="9" y="9" width="13" height="13" rx="2"/><path d="M5 15H4a2 2 0 01-2-2V4a2 2 0 012-2h9a2 2 0 012 2v1"/></svg>';
        btn.title = 'Copy code';
        btn.setAttribute('aria-label', 'Copy code');
        btn.onclick = () => copyCodeBlock(btn);
        btn.style.cssText = 'position:absolute;top:0.5rem;right:0.5rem;padding:0.25rem;background:var(--bg-surface);border:1px solid var(--border);border-radius:4px;color:var(--text-muted);cursor:pointer;transition:opacity 0.2s;';
        pre.style.position = 'relative';
        pre.appendChild(btn);
    });
};"#;

    inits.push(
        r#"
    // Print button handler
    const printBtn = $('#print-btn');
    if (printBtn) printBtn.addEventListener('click', printConversation);

    // Global keyboard shortcut: Ctrl/Cmd + P for print
    document.addEventListener('keydown', (e) => {
        if ((e.ctrlKey || e.metaKey) && e.key === 'p') {
            e.preventDefault();
            printConversation();
        }
    });"#,
    );

    format!(
        r#"{}

// Initialize after DOM is ready (or immediately if already ready)
const __cassInitAll = () => {{
    {}
}};

if (document.readyState === 'loading') {{
    document.addEventListener('DOMContentLoaded', __cassInitAll);
}} else {{
    __cassInitAll();
}}"#,
        copy_button_helpers,
        inits.join("\n    ")
    )
}

#[cfg(test)]
mod tests {
    use super::*;

    macro_rules! assert_inline_js_contains {
        ($bundle:expr, $needle:literal) => {
            assert!($bundle.inline_js.contains($needle));
        };
    }

    #[test]
    fn test_generate_scripts_includes_search() {
        let opts = ExportOptions {
            include_search: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "const Search");
        assert_inline_js_contains!(bundle, "Search.init()");
    }

    #[test]
    fn test_search_init_is_idempotent_for_decryption_reinit() {
        let opts = ExportOptions {
            encrypt: true,
            include_search: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "currentIndex: -1,\n    _initialized: false");
        assert_inline_js_contains!(
            bundle,
            "if (!this._initialized) {\n            this.input.addEventListener('input'"
        );
        assert_inline_js_contains!(bundle, "this._initialized = true;");
        assert_inline_js_contains!(bundle, "Search.init()");
    }

    #[test]
    fn test_generate_scripts_excludes_search_when_disabled() {
        let opts = ExportOptions {
            include_search: false,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        assert!(!bundle.inline_js.contains("const Search"));
    }

    #[test]
    fn test_generate_scripts_includes_theme_toggle() {
        let opts = ExportOptions {
            include_theme_toggle: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "const Theme");
        assert_inline_js_contains!(bundle, "localStorage.getItem");
    }

    #[test]
    fn test_generate_scripts_includes_encryption() {
        let opts = ExportOptions {
            encrypt: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "const Crypto");
        assert_inline_js_contains!(bundle, "crypto.subtle");
    }

    #[test]
    fn test_generate_scripts_includes_toast_and_copy() {
        let opts = ExportOptions::default();
        let bundle = generate_scripts(&opts);

        // Toast notifications
        assert_inline_js_contains!(bundle, "const Toast");
        assert_inline_js_contains!(bundle, "Toast.show");

        // Copy to clipboard
        assert_inline_js_contains!(bundle, "copyToClipboard");
        assert_inline_js_contains!(bundle, "navigator.clipboard");

        // Fallback for older browsers
        assert_inline_js_contains!(bundle, "execCommand");
    }

    #[test]
    fn test_generate_scripts_includes_print_handler() {
        let opts = ExportOptions::default();
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "printConversation");
        assert_inline_js_contains!(bundle, "window.print");
    }

    #[test]
    fn test_generate_scripts_includes_keyboard_shortcuts() {
        let opts = ExportOptions {
            include_search: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        // Ctrl+F for search
        assert_inline_js_contains!(bundle, "e.key === 'f'");
        // Ctrl+P for print
        assert_inline_js_contains!(bundle, "e.key === 'p'");
        // Escape to clear
        assert_inline_js_contains!(bundle, "'Escape'");
    }

    #[test]
    fn test_generate_scripts_includes_copy_code_buttons() {
        let opts = ExportOptions::default();
        let bundle = generate_scripts(&opts);

        assert_inline_js_contains!(bundle, "copy-code-btn");
        assert_inline_js_contains!(bundle, "copyCodeBlock");
    }

    #[test]
    fn test_generate_scripts_includes_world_class_enhancements() {
        let opts = ExportOptions::default();
        let bundle = generate_scripts(&opts);

        // WorldClass object and initialization
        assert_inline_js_contains!(bundle, "const WorldClass");
        assert_inline_js_contains!(bundle, "WorldClass.init()");

        // Scroll progress indicator
        assert_inline_js_contains!(bundle, "scroll-progress");

        // Floating navigation
        assert_inline_js_contains!(bundle, "initFloatingNav");
        assert_inline_js_contains!(bundle, "scroll-top");

        // Keyboard navigation (vim-style j/k)
        assert_inline_js_contains!(bundle, "initKeyboardNav");
        assert_inline_js_contains!(bundle, "case 'j':");
        assert_inline_js_contains!(bundle, "case 'k':");

        // Message link copying
        assert_inline_js_contains!(bundle, "initMessageLinks");
        assert_inline_js_contains!(bundle, "message-link");
        assert_inline_js_contains!(bundle, "msg.querySelector('.message-link')");

        // Intersection observer for animations
        assert_inline_js_contains!(bundle, "IntersectionObserver");
        assert_inline_js_contains!(bundle, "in-view");

        // Native share API support
        assert_inline_js_contains!(bundle, "navigator.share");

        // Touch ripple effect
        assert_inline_js_contains!(bundle, "createRipple");
    }

    #[test]
    fn test_world_class_keyboard_shortcuts() {
        let opts = ExportOptions::default();
        let bundle = generate_scripts(&opts);

        // Vim-style navigation
        assert_inline_js_contains!(bundle, "navigateMessage(1)"); // j - next
        assert_inline_js_contains!(bundle, "navigateMessage(-1)"); // k - previous

        // Jump to first/last (g/G)
        assert_inline_js_contains!(bundle, "case 'g':");

        // Search shortcut (/)
        assert_inline_js_contains!(bundle, "case '/':");

        // Help shortcut (?)
        assert_inline_js_contains!(bundle, "case '?':");
        assert_inline_js_contains!(bundle, "showShortcutsHint");
    }

    #[test]
    fn test_tool_popovers_functionality() {
        let opts = ExportOptions {
            show_tool_calls: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        // ToolPopovers object exists
        assert_inline_js_contains!(bundle, "const ToolPopovers");
        assert_inline_js_contains!(bundle, "ToolPopovers.init()");

        // Hover support (desktop)
        assert_inline_js_contains!(bundle, "mouseenter");
        assert_inline_js_contains!(bundle, "mouseleave");

        // Focus support (keyboard accessibility)
        assert_inline_js_contains!(bundle, "addEventListener('focus'");
        assert_inline_js_contains!(bundle, "addEventListener('blur'");

        // Click support (mobile/touch)
        assert!(
            bundle
                .inline_js
                .contains("this.toggle(badge, getPopover())")
        );

        // Escape key support
        assert_inline_js_contains!(bundle, "e.key === 'Escape'");

        // aria-expanded updates
        assert_inline_js_contains!(bundle, "setAttribute('aria-expanded'");

        // Viewport positioning
        assert_inline_js_contains!(bundle, "getBoundingClientRect");
        assert_inline_js_contains!(bundle, "viewportWidth");
        assert_inline_js_contains!(bundle, "viewportHeight");

        // Overflow badge expansion
        assert_inline_js_contains!(bundle, "initOverflowBadges");
        assert_inline_js_contains!(bundle, "tool-overflow");
        assert_inline_js_contains!(bundle, "btn.dataset.overflowBound");

        // Outside click to close
        assert_inline_js_contains!(bundle, "initOutsideClick");
        assert_inline_js_contains!(bundle, "hideAll");
        assert_inline_js_contains!(bundle, "_outsideClickBound");

        // Re-init guards
        assert_inline_js_contains!(bundle, "header.dataset.toolToggleBound");
    }

    #[test]
    fn test_tool_popovers_reinit_after_decryption() {
        let opts = ExportOptions {
            encrypt: true,
            show_tool_calls: true,
            ..Default::default()
        };
        let bundle = generate_scripts(&opts);

        // After decryption, both ToolCalls and ToolPopovers should be reinitialized
        assert_inline_js_contains!(bundle, "ToolCalls.init()");
        assert_inline_js_contains!(bundle, "ToolPopovers.init()");
        assert_inline_js_contains!(bundle, "__cassAttachCodeCopyButtons();");
        assert_inline_js_contains!(bundle, "const __cassAttachCodeCopyButtons");
        assert_inline_js_contains!(bundle, "pre.querySelector('.copy-code-btn')");
    }
}