options.js

document.addEventListener('DOMContentLoaded', () => {
    // --- HTML Elements ---
    const statusDiv = document.getElementById('status');
    const saveButton = document.getElementById('save-settings');
    const globalRegionSelect = document.getElementById('global-region');
    const authMethodRadios = document.querySelectorAll('input[name="authMethod"]');
    const directApiFields = document.getElementById('direct-api-fields');
    const awsGatewayFields = document.getElementById('aws-gateway-fields');
    const clientIdInput = document.getElementById('clientId');
    const clientSecretInput = document.getElementById('clientSecret');
    const gatewayUrlInput = document.getElementById('gatewayUrl');
    const newExclusionInput = document.getElementById('new-exclusion');
    const addExclusionButton = document.getElementById('add-exclusion');
    const exclusionListUl = document.getElementById('exclusion-list');

    // --- Static Data ---
    const apiUrls = {
        'us-1': 'https://api.crowdstrike.com',
        'us-2': 'https://api.us-2.crowdstrike.com',
        'eu-1': 'https://api.eu-1.crowdstrike.com',
        'us-gov-1': 'https://api.laggar.gcw.crowdstrike.com'
    };

    // --- UI Helper Functions ---
    function showStatus(message, isError = false) {
        statusDiv.textContent = message;
        statusDiv.style.color = isError ? '#ef5350' : '#66bb6a';
        setTimeout(() => { statusDiv.textContent = ''; }, 4000);
    }

    function toggleAuthFields(method) {
        directApiFields.style.display = (method === 'direct') ? 'block' : 'none';
        awsGatewayFields.style.display = (method === 'aws') ? 'block' : 'none';
    }

    // --- Core Save Function for API/Region/Method Settings ---
    async function saveSettings() {
        if (saveButton.disabled) return;
        saveButton.disabled = true;
        saveButton.textContent = 'Saving...';

        try {
            const settingsToSave = {
                authMethod: document.querySelector('input[name="authMethod"]:checked').value,
                globalRegion: globalRegionSelect.value,
                clientId: clientIdInput.value,
                clientSecret: clientSecretInput.value,
                gatewayUrl: gatewayUrlInput.value.trim()
            };

            let requiredOrigin = '';
            if (settingsToSave.authMethod === 'aws') {
                if (settingsToSave.gatewayUrl) {
                    requiredOrigin = new URL(settingsToSave.gatewayUrl).origin;
                }
            } else {
                requiredOrigin = apiUrls[settingsToSave.globalRegion];
            }
            
            if (requiredOrigin) {
                const permissionNeeded = `${requiredOrigin}/*`;
                const hasPermission = await chrome.permissions.contains({ origins: [permissionNeeded] });
                if (!hasPermission) {
                    const granted = await chrome.permissions.request({ origins: [permissionNeeded] });
                    if (!granted) {
                        showStatus('Permission for the new API endpoint was denied. Settings not saved.', true);
                        return; // Exit if user denies
                    }
                }
            }

            // Save the settings
            await chrome.storage.sync.set(settingsToSave);
            showStatus('Settings saved successfully!');

        } catch (error) {
            console.error("Error saving settings:", error);
            showStatus("An error occurred. Please check the console for details.", true);
        } finally {
            saveButton.disabled = false;
            saveButton.textContent = 'Save All Settings';
        }
    }

    // --- Functions to Restore UI State ---
    function restoreSettings() {
        chrome.storage.sync.get({
            authMethod: 'direct', 
            globalRegion: 'us-1', 
            clientId: '', 
            clientSecret: '', 
            gatewayUrl: '', 
            excludedSites: []
        }, (items) => {
            document.querySelector(`input[name="authMethod"][value="${items.authMethod}"]`).checked = true;
            toggleAuthFields(items.authMethod);
            globalRegionSelect.value = items.globalRegion;
            clientIdInput.value = items.clientId;
            clientSecretInput.value = items.clientSecret;
            gatewayUrlInput.value = items.gatewayUrl;
            renderExclusionList(items.excludedSites);
        });
    }

    // --- Exclusion List Functions ---
    function renderExclusionList(sites = []) {
        exclusionListUl.innerHTML = '';
        sites.forEach(site => {
            const li = document.createElement('li');
            li.textContent = site;
            const removeButton = document.createElement('button');
            removeButton.textContent = 'Remove';
            removeButton.dataset.site = site;
            removeButton.addEventListener('click', handleRemoveExclusion);
            li.appendChild(removeButton);
            exclusionListUl.appendChild(li);
        });
    }

    async function handleAddExclusion() {
        let newSite = newExclusionInput.value.trim();
        if (!newSite) return;
        try {
            if (!newSite.startsWith('http')) newSite = `https://${newSite}`;
            const normalizedSite = new URL(newSite).hostname.replace(/^www\./, '');
            const data = await chrome.storage.sync.get({ excludedSites: [] });
            const sites = data.excludedSites;
            if (!sites.includes(normalizedSite)) {
                sites.push(normalizedSite);
                await chrome.storage.sync.set({ excludedSites: sites });
                renderExclusionList(sites);
                newExclusionInput.value = '';
                showStatus('Site added to exclusion list.');
            } else { 
                showStatus('This site is already on the list.', true); 
            }
        } catch (e) { 
            showStatus('Invalid domain in exclusion list.', true); 
        }
    }

    async function handleRemoveExclusion(event) {
        const siteToRemove = event.target.dataset.site;
        const data = await chrome.storage.sync.get({ excludedSites: [] });
        const sites = data.excludedSites.filter(site => site !== siteToRemove);
        await chrome.storage.sync.set({ excludedSites: sites });
        renderExclusionList(sites);
        showStatus('Site removed from exclusion list.');
    }

    // --- Event Listeners ---
    saveButton.addEventListener('click', saveSettings);
    addExclusionButton.addEventListener('click', handleAddExclusion);
    newExclusionInput.addEventListener('keyup', e => {
        if (e.key === 'Enter') {
            handleAddExclusion();
        }
    });
    authMethodRadios.forEach(radio => radio.addEventListener('change', e => toggleAuthFields(e.target.value)));
    
    // Initial Load
    restoreSettings();
});