script.js

document.addEventListener('DOMContentLoaded', () => {
    const workspace = document.getElementById('workspace');
    
    document.querySelectorAll('.block-palette .block').forEach(block => {
        block.addEventListener('dragstart', e => {
            e.dataTransfer.setData('text/plain', block.dataset.type);
        });
    });

    workspace.addEventListener('dragover', e => {
        e.preventDefault();
    });
    
    workspace.addEventListener('drop', e => {
        e.preventDefault();
        const type = e.dataTransfer.getData('text/plain');
        
        const newBlock = document.createElement('div');
        newBlock.className = `block ${type}-block`;
        newBlock.innerHTML = `
            <span class="delete-btn">Remove</span>
            ${getBlockContent(type)}
        `;
        
        workspace.appendChild(newBlock);
        addDeleteListener(newBlock);
    });
});

function getBlockContent(type) {
    const templates = {
        straight: `Straight (
            <input type="number" placeholder="Distance" class="param">
        ) (
            <input type="number" placeholder="Power" class="param">
        )`,
        strafe: `Strafe (
            <input type="number" placeholder="Distance" class="param">
        ) (
            <input type="number" placeholder="Power" class="param">
        )`,
        turn: `Turn (
            <input type="number" placeholder="Degrees" class="param">
        )`
    };
    return templates[type] || '';
}

function addDeleteListener(block) {
    block.querySelector('.delete-btn').addEventListener('click', () => {
        block.remove();
    });
}

function convertBlocks() {
    const blocks = Array.from(document.querySelectorAll('#workspace .block'));
    let output = [];
    
    blocks.forEach((block, index) => {
        if (index === 0 && block.classList.contains('start-block')) return;
        
        const inputs = block.getElementsByTagName('input');
        if (block.classList.contains('straight-block')) {
            output.push(`addCommands(new Drivetrain_GyroStraight(${inputs[0].value || 0}, ${inputs[1].value || 0}));`);
        }
        else if (block.classList.contains('strafe-block')) {
            output.push(`addCommands(new Drivetrain_GyroStrafe(${inputs[0].value || 0}, ${inputs[1].value || 0}));`);
        }
        else if (block.classList.contains('turn-block')) {
            output.push(`addCommands(new Drivetrain_GyroTurn(${inputs[0].value || 0}));`);
        }
    });

    output.unshift("// Generated by Tobin's Auto Coder\n");
    
    showOutput(output.join('\n'));
}

function showOutput(code) {
    const modal = document.getElementById('outputModal');
    modal.style.display = 'block';
    
    document.getElementById('convertedCode').innerHTML = Prism.highlight(code, Prism.languages.javascript, 'javascript');
    document.getElementById('outputModal').style.display = 'block';

    document.querySelectorAll('.param').forEach(input => {
        input.addEventListener('dblclick', function() {
            this.focus();
            this.select();
        });
    });
    
    window.onclick = function(event) {
        const modal = document.getElementById('outputModal');
        if (event.target == modal) {
            closeModal();
        }
    }
    
    document.addEventListener('keydown', (e) => {
        if (e.key === 'Escape') closeModal();
    });
}

function copyCode() {
    const code = document.getElementById('convertedCode').textContent;
    navigator.clipboard.writeText(code).then(() => {
        alert('Code copied to clipboard!');
    });
}

function closeModal() {
    document.getElementById('outputModal').style.display = 'none';
}