// File: web/AF_Enhanced_HTML_Note.js

import { app } from "../../scripts/app.js";

// Define inline CSS styles
const AF_HTML_NOTE_STYLES = `
    .af-html-note-container {
        width: 100%;
        height: 100%;
        min-height: 100px;
        background: var(--comfy-menu-bg, #2a2a2a);
        border: 1px solid var(--border-color, #555);
        padding: 8px; /* Reduced padding */
        box-sizing: border-box;
        overflow: hidden; /* Changed from auto to hidden */
        cursor: default;
        font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', sans-serif;
        line-height: 1.6;
        color: var(--input-text, #ffffff);
        display: flex;
        flex-direction: column;
    }

    .af-html-note-content {
        width: 100%;
        height: 100%;
        /* Allow double-clicks for editing */
        pointer-events: auto;
        overflow-y: auto; /* Move scroll to content instead of container */
        flex: 1;
    }

    .af-html-note-editor {
        width: 100%;
        height: 100%;
        background: var(--comfy-menu-bg, #2a2a2a);
        border: 1px solid var(--border-color, #555);
        padding: 8px; /* Reduced padding to match container */
        font-family: 'Monaco', 'Consolas', 'Courier New', monospace;
        line-height: 1.4;
        color: var(--input-text, #ffffff);
        box-sizing: border-box;
        resize: none;
        outline: none;
        font-size: 14px;
        display: none;
        pointer-events: auto;
        flex: 1;
    }

    /* Enable link interactions only when Ctrl is held */
    .af-html-note-content a {
        color: #58a6ff;
        text-decoration: none;
        /* Links are not clickable by default */
        pointer-events: none;
    }

    body.af-ctrl-active .af-html-note-content a {
        /* Links become clickable when Ctrl is held */
        pointer-events: auto;
        cursor: pointer;
    }

    body.af-ctrl-active .af-html-note-content a:hover {
        color: #79c0ff;
        text-decoration: underline;
    }

    .af-html-note-editor {
        width: 100%;
        height: 100%;
        background: var(--comfy-menu-bg, #2a2a2a);
        border: 1px solid var(--border-color, #555);
        padding: 16px;
        font-family: 'Monaco', 'Consolas', 'Courier New', monospace;
        line-height: 1.4;
        color: var(--input-text, #ffffff);
        box-sizing: border-box;
        resize: none;
        outline: none;
        font-size: 14px;
        display: none;
        pointer-events: auto;
    }

    .af-html-note-container.edit-mode .af-html-note-content {
        display: none;
    }

    .af-html-note-container.edit-mode .af-html-note-editor {
        display: block;
    }

    /* Rest of your CSS remains exactly the same... */
    .af-html-note-content h1, .af-html-note-content h2, .af-html-note-content h3, 
    .af-html-note-content h4, .af-html-note-content h5, .af-html-note-content h6 {
        margin-top: 0;
        margin-bottom: 12px;
        font-weight: 600;
    }

    .af-html-note-content h1 { font-size: 1.5em; }
    .af-html-note-content h2 { font-size: 1.3em; }
    .af-html-note-content h3 { font-size: 1.1em; }

    .af-html-note-content .af-section {
        padding: 12px 16px;
        margin: 8px 0;
        border-radius: 6px;
        border-left: 4px solid;
    }

    .af-html-note-content .af-section.af-positive {
        background: rgba(144, 238, 144, 0.1);
        border-left-color: #90EE90;
    }

    .af-html-note-content .af-section.af-negative {
        background: rgba(255, 107, 107, 0.1);
        border-left-color: #FF6B6B;
    }

    .af-html-note-content .af-section.af-neutral {
        background: rgba(224, 224, 224, 0.1);
        border-left-color: #E0E0E0;
    }

    .af-html-note-content .af-section.af-info {
        background: rgba(135, 206, 235, 0.1);
        border-left-color: #87CEEB;
    }

    .af-html-note-content .af-section.af-warning {
        background: rgba(255, 215, 0, 0.1);
        border-left-color: #FFD700;
    }

    .af-html-note-content .af-section.af-custom {
        background: rgba(186, 85, 211, 0.1);
        border-left-color: #BA55D3;
    }

    .af-html-note-content .af-spacer { height: 16px; margin: 8px 0; }
    .af-html-note-content .af-spacer-small { height: 8px; margin: 4px 0; }
    .af-html-note-content .af-spacer-large { height: 32px; margin: 16px 0; }
    .af-html-note-content .af-spacer-xl { height: 48px; margin: 24px 0; }

    .af-html-note-content code {
        background: rgba(0,0,0,0.3);
        padding: 2px 6px;
        border-radius: 4px;
        font-family: 'Monaco', 'Consolas', 'Courier New', monospace;
        font-size: 0.9em;
        color: #f0f6fc;
    }

    .af-html-note-content pre {
        background: rgba(0,0,0,0.4);
        border: 1px solid #444;
        border-radius: 6px;
        padding: 12px;
        margin: 16px 0;
        overflow-x: auto;
    }

    .af-html-note-content pre code {
        background: transparent;
        padding: 0;
    }

    .af-html-note-content ul, .af-html-note-content ol {
        padding-left: 20px;
        margin: 12px 0;
    }

    .af-html-note-content li { margin: 4px 0; }
    .af-html-note-content p { margin: 8px 0; }

    .af-html-note-content table {
        border-collapse: collapse;
        width: 100%;
        margin: 16px 0;
    }

    .af-html-note-content th, .af-html-note-content td {
        border: 1px solid var(--border-color, #555);
        padding: 8px 12px;
        text-align: left;
    }

    .af-html-note-content th {
        background: rgba(255,255,255,0.1);
        font-weight: 600;
    }

    .af-html-note-content hr {
        border: none;
        border-top: 1px solid var(--border-color, #555);
        margin: 16px 0;
        opacity: 0.6;
    }

    .af-html-note-container::-webkit-scrollbar { width: 8px; }
    .af-html-note-container::-webkit-scrollbar-track { background: rgba(0,0,0,0.1); }
    .af-html-note-container::-webkit-scrollbar-thumb { 
        background: var(--border-color, #555); 
        border-radius: 4px; 
    }
    .af-html-note-container::-webkit-scrollbar-thumb:hover { background: #777; }

    .af-html-note-editor::-webkit-scrollbar { width: 8px; }
    .af-html-note-editor::-webkit-scrollbar-track { background: rgba(0,0,0,0.1); }
    .af-html-note-editor::-webkit-scrollbar-thumb { 
        background: var(--border-color, #555); 
        border-radius: 4px; 
    }
    .af-html-note-editor::-webkit-scrollbar-thumb:hover { background: #777; }
`;

// Inject styles
function injectStyles() {
    const styleId = 'af-html-note-styles';
    if (!document.getElementById(styleId)) {
        const style = document.createElement('style');
        style.id = styleId;
        style.textContent = AF_HTML_NOTE_STYLES;
        document.head.appendChild(style);
    }
}

app.registerExtension({
    name: "AF.enhanced.html.note",
    
    async beforeRegisterNodeDef(nodeType, nodeData, app) {
        if (nodeData.name === "AF_Enhanced_HTML_Note") {
            injectStyles();
            
            // Store the original onNodeCreated
            const originalOnNodeCreated = nodeType.prototype.onNodeCreated;
            
            nodeType.prototype.onNodeCreated = function() {
                const result = originalOnNodeCreated?.apply(this, arguments);
                
                this.isEditMode = false;
                this.size = [400, 300];
                
                // Don't hide the original widget - instead, replace it with our custom display
                setTimeout(() => {
                    const htmlWidget = this.widgets?.find(w => w.name === "html_content");
                    if (htmlWidget && htmlWidget.element) {
                        // Store the original value
                        this.htmlContent = htmlWidget.value;
                        
                        // Create our custom container
                        this.createHTMLNoteDisplay(this.htmlContent);
                        
                        // Hide the original widget but keep its value
                        htmlWidget.computeSize = () => [0, 0];
                        htmlWidget.element.style.display = 'none';
                    }
                }, 0);
                
                return result;
            };

			nodeType.prototype.createHTMLNoteDisplay = function(content) {
				// Create container element
				this.containerElement = document.createElement("div");
				this.containerElement.className = "af-html-note-container";
				
				// Create HTML content display
				this.contentElement = document.createElement("div");
				this.contentElement.className = "af-html-note-content";
				this.contentElement.innerHTML = content;
				
				// Create editor element
				this.editorElement = document.createElement("textarea");
				this.editorElement.className = "af-html-note-editor";
				this.editorElement.value = content;
				
				// Append both elements to container
				this.containerElement.appendChild(this.contentElement);
				this.containerElement.appendChild(this.editorElement);
				
				// Add double-click to edit - works anytime (no Ctrl required)
				this.contentElement.addEventListener('dblclick', (e) => {
					if (!this.isEditMode) {
						this.enterEditMode();
						e.stopPropagation();
					}
				});

				// Handle links
				this.handleLinksInHTML();

				// Handle editor events
				this.editorElement.addEventListener('keydown', (e) => {
					if (e.key === 'Escape') {
						this.exitEditMode();
						e.preventDefault();
						e.stopPropagation();
					} else {
						e.stopPropagation();
					}
				});

				this.editorElement.addEventListener('blur', () => {
					setTimeout(() => {
						if (this.isEditMode && document.activeElement !== this.editorElement) {
							this.exitEditMode();
						}
					}, 10);
				});

				// Add as DOM widget
				this.addDOMWidget("html_note_display", "div", this.containerElement);
			};

            nodeType.prototype.enterEditMode = function() {
                if (this.isEditMode) return;
                
                this.isEditMode = true;
                this.containerElement.classList.add('edit-mode');
                this.editorElement.value = this.htmlContent;
                
                setTimeout(() => {
                    this.editorElement.focus();
                    this.editorElement.select();
                }, 10);
            };

            nodeType.prototype.exitEditMode = function() {
                if (!this.isEditMode) return;
                
                this.isEditMode = false;
                this.containerElement.classList.remove('edit-mode');
                this.htmlContent = this.editorElement.value;
                this.contentElement.innerHTML = this.htmlContent;
                
                // Update the original widget value
                const htmlWidget = this.widgets?.find(w => w.name === "html_content");
                if (htmlWidget) {
                    htmlWidget.value = this.htmlContent;
                    if (htmlWidget.callback) {
                        htmlWidget.callback(this.htmlContent);
                    }
                }
                
                // Re-attach link handlers
                this.handleLinksInHTML();
            };

			nodeType.prototype.handleLinksInHTML = function() {
				if (!this.contentElement) return;
				
				const links = this.contentElement.querySelectorAll('a');
				links.forEach(link => {
					// Remove any existing handlers
					link.onclick = null;
					
					// Add new handler that requires Ctrl key
					link.addEventListener('click', (e) => {
						// Only handle links when Ctrl is active
						if (!document.body.classList.contains('af-ctrl-active')) {
							e.preventDefault();
							e.stopPropagation();
							return false;
						}
						
						// Handle external links
						if (link.href && !link.href.includes('#') && !link.href.startsWith('javascript:')) {
							e.preventDefault();
							window.open(link.href, '_blank');
						}
						// For javascript: links and onclick handlers, let them execute naturally
						// Don't prevent default or stop propagation for these
						
						e.stopPropagation();
						return false;
					});
				});
			};

            // Global Ctrl key handling
            let ctrlKeyActive = false;

            document.addEventListener('keydown', (e) => {
                if (e.key === 'Control') {
                    ctrlKeyActive = true;
                    document.body.classList.add('af-ctrl-active');
                }
            });

            document.addEventListener('keyup', (e) => {
                if (e.key === 'Control') {
                    ctrlKeyActive = false;
                    document.body.classList.remove('af-ctrl-active');
                }
            });

            // Clean up when node is removed
            const originalOnRemoved = nodeType.prototype.onRemoved;
            nodeType.prototype.onRemoved = function() {
                document.body.classList.remove('af-ctrl-active');
                originalOnRemoved?.apply(this, arguments);
            };
        }
    }
});