// ==UserScript==
// @name         WME EZRoad Mod
// @namespace    https://greasyfork.org/users/1087400
// @version      2.5.2
// @description  Easily update roads
// @author       https://github.com/michaelrosstarr, https://greasyfork.org/en/users/1087400-kid4rm90s
// @include 	 /^https:\/\/(www|beta)\.waze\.com\/(?!user\/)(.{2,6}\/)?editor.*$/
// @exclude      https://www.waze.com/user/*editor/*
// @exclude      https://www.waze.com/*/user/*editor/*
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_xmlhttpRequest
// @grant        GM_info
// @grant 		 unsafeWindow
// @icon         https://www.google.com/s2/favicons?sz=64&domain=waze.com
// @license      GNU GPL(v3)
// @connect      greasyfork.org
// @require      https://greasyfork.org/scripts/24851-wazewrap/code/WazeWrap.js
// @require      https://update.greasyfork.icu/scripts/509664/WME%20Utils%20-%20Bootstrap.js
// @downloadURL none
// ==/UserScript==

/*Script modified from WME EZRoad (https://greasyfork.org/en/scripts/518381-wme-ezsegments) original author: Michaelrosstarr and thanks to him*/

(function main() {
  "use strict";
   const updateMessage = `
<b>2.5.2 - 2025-05-31</b><br>
- Improved shortcut keybind logic and UI.<br>
- Prevented double alerts and unnecessary alerts when keybind is empty.<br>
- Minor bug fixes.`;
   const scriptName = GM_info.script.name;
   const scriptVersion = GM_info.script.version;
  const downloadUrl = 'https://greasyfork.org/scripts/528552-wme-ezroad-mod/code/wme-ezroad-mod.user.js';
   let wmeSDK;

const roadTypes = [
    { id: 1, name: 'Motorway', value: 3 },
	{ id: 2, name: 'Ramp', value: 4 },
	{ id: 3, name: 'Major Highway', value: 6 },
	{ id: 4, name: 'Minor Highway', value: 7 },
	{ id: 5, name: 'Primary Street', value: 2 },
    { id: 6, name: 'Street', value: 1 },
    { id: 7, name: 'Narrow Street', value: 22 },
	{ id: 8, name: 'Offroad', value: 8 },
	{ id: 9, name: 'Parking Road', value: 20 },
	{ id: 10, name: 'Private Road', value: 17 },
    { id: 11, name: 'Ferry', value: 15 },
	{ id: 12, name: 'Railroad', value: 18 },
	{ id: 13, name: 'Runway/Taxiway', value: 19 },
	{ id: 14, name: 'Foothpath', value: 5 },
	{ id: 15, name: 'Pedestrianised Area', value: 10 },
	{ id: 16, name: 'Stairway', value: 16 },
];
const defaultOptions = {
    roadType: 1,
    unpaved: false,
    setStreet: false,
    autosave: false,
    setSpeed: 40,
    setLock: false,
    updateSpeed: false,
    copySegmentName: false,
    locks: roadTypes.map(roadType => ({ id: roadType.id, lock: 1 })),
    speeds: roadTypes.map(roadType => ({ id: roadType.id, speed: 40 })),
    copySegmentAttributes: false,
    shortcutKey: "g",
};

const locks = [
    { id: 1, value: 1 },
    { id: 2, value: 2 },
    { id: 3, value: 3 },
    { id: 4, value: 4 },
    { id: 5, value: 5 },
    { id: 6, value: 6 },
	{ id: 'HRCS', value: 'HRCS' },
]

const log = (message) => {
    if (typeof message === 'string') {
        console.log('WME_EZRoads_Mod: ' + message);
    } else {
        console.log('WME_EZRoads_Mod: ', message);
    }
}

unsafeWindow.SDK_INITIALIZED.then(initScript);

function initScript() {
    wmeSDK = getWmeSdk({ scriptId: "wme-ez-roads-mod", scriptName: "EZ Roads Mod" });
    WME_EZRoads_Mod_bootstrap();
}

const getCurrentCountry = () => {
    return wmeSDK.DataModel.Countries.getTopCountry();
}

const getTopCity = () => {
    return wmeSDK.DataModel.Cities.getTopCity();
}

const getAllCities = () => {
    return wmeSDK.DataModel.Cities.getAll();
}

const saveOptions = (options) => {
    window.localStorage.setItem('WME_EZRoads_Mod_Options', JSON.stringify(options));
}

const getOptions = () => {
    const savedOptions = JSON.parse(window.localStorage.getItem('WME_EZRoads_Mod_Options')) || {};
	// Merge saved options with defaults to ensure all expected options exist
    return { ...defaultOptions, ...savedOptions };
	}

const WME_EZRoads_Mod_bootstrap = () => {
    if (
        !document.getElementById('edit-panel')
        || !wmeSDK.DataModel.Countries.getTopCountry()
    ) {
        setTimeout(WME_EZRoads_Mod_bootstrap, 250);
        return;
    }

    if (wmeSDK.State.isReady) {
        WME_EZRoads_Mod_init();
    } else {
        wmeSDK.Events.once({ eventName: 'wme-ready' }).then(WME_EZRoads_Mod_init());
    }
}

let openPanel;
let registeredShortcutKey = null;

const WME_EZRoads_Mod_init = () => {
    log("Initing");

    // Register shortcut using WME SDK
    const options = getOptions();
    registeredShortcutKey = options.shortcutKey || "g";
    registerShortcut(registeredShortcutKey);

    const roadObserver = new MutationObserver((mutations) => {
        mutations.forEach((mutation) => {
            for (let i = 0; i < mutation.addedNodes.length; i++) {
                const addedNode = mutation.addedNodes[i];

                if (addedNode.nodeType === Node.ELEMENT_NODE) {
                    let editSegment = addedNode.querySelector('#segment-edit-general');
                    if (editSegment) {
                        openPanel = editSegment;

                        // Check if THIS SPECIFIC panel already has the button
                        const parentElement = editSegment.parentNode;
                        if (!parentElement.querySelector('[data-ez-roadmod-button="true"]')) {
                            log("Creating Quick Set Road button for this panel");
                            const quickButton = document.createElement('wz-button');
                            quickButton.setAttribute('type', 'button');
                            quickButton.setAttribute('style', 'margin-bottom: 5px; width: 100%');
                            quickButton.setAttribute('disabled', 'false');
                            quickButton.setAttribute('data-ez-roadmod-button', 'true');
                            quickButton.setAttribute('id', 'ez-roadmod-quick-button-' + Date.now()); // Unique ID using timestamp
                            quickButton.classList.add('send-button', 'ez-comment-button');
                            quickButton.textContent = 'Quick Update Segment';
                            parentElement.insertBefore(quickButton, editSegment);
                            quickButton.addEventListener('mousedown', () => handleUpdate());
                            log("Button created for current panel");
                        } else {
                            log("This panel already has the button, skipping creation");
                        }
					}
				}
			}
        });
    });

    roadObserver.observe(document.getElementById('edit-panel'), { childList: true, subtree: true });

    constructSettings();

    // Remove the old keydown event, as shortcut is now handled by WME SDK

    log("Completed Init")
};

function registerShortcut(key) {
    // No need to remove previous shortcut; createShortcut with same shortcutId will update it
    if (wmeSDK.Shortcuts) {
        wmeSDK.Shortcuts.createShortcut({
            callback: handleUpdate,
            description: "Quick Update Segments.",
            shortcutId: "EZRoad_Mod_QuickUpdate",
            shortcutKeys: key,
        });
        registeredShortcutKey = key;
        console.log(`[EZRoads Mod] Shortcut '${key}' for Quick Update Segments enabled.`);
    }
}

const getEmptyCity = () => {
    return wmeSDK.DataModel.Cities.getCity({
        cityName: '',
        countryId: getCurrentCountry().id
    }) || wmeSDK.DataModel.Cities.addCity({
        cityName: '',
        countryId: getCurrentCountry().id
    });
}

const delayedUpdate = (updateFn, delay) => {
    return new Promise(resolve => {
        setTimeout(() => {
            updateFn();
            resolve();
        }, delay);
    });
};

function getHighestSegLock(segID) {
    const segObj = W.model.segments.getObjectById(segID);
    if (!segObj) {
        console.warn(`Segment object with ID ${segID} not found in W.model.segments.`);
        return 1; // Default lock level if segment not found
    }
    const segType = segObj.attributes.roadType;
    const checkedSegs = [];
    let forwardLock = null; // Initialize to L1
    let reverseLock = null; // Initialize to L1

    function processForNode(forwardID) {
        checkedSegs.push(forwardID);
        const forNode = W.model.segments.getObjectById(forwardID).getToNode();
        if (!forNode) return forwardLock; // defensive check

        const forNodeSegs = [...forNode.attributes.segIDs];

        for (let j = 0; j < forNodeSegs.length; j++) {
            if (forNodeSegs[j] === forwardID) { forNodeSegs.splice(j, 1); }
        }

        for (let i = 0; i < forNodeSegs.length; i++) {
            const conSegObj = W.model.segments.getObjectById(forNodeSegs[i]);
            if (!conSegObj) continue; // Defensive check - skip if segment object is not found
            const conSeg = conSegObj.attributes;

            if (conSeg.roadType !== segType) {
                forwardLock = Math.max(conSeg.lockRank + 0, forwardLock); // +1 to convert lockRank to lock level
            } else {
                for (let k = 0; k < forNodeSegs.length; k++) {
                    if (!checkedSegs.some(segNum => segNum === conSeg.id)) {
                        const tempRank = processForNode(conSeg.id);
                        forwardLock = Math.max(tempRank, forwardLock);
                    }
                }
            }
        }
        return forwardLock;
    }

    function processRevNode(reverseID) {
        checkedSegs.push(reverseID);
        const revNode = W.model.segments.getObjectById(reverseID).getFromNode();
        if (!revNode) return reverseLock; // defensive check, return current lock if node is not valid

        const revNodeSegs = [...revNode.attributes.segIDs];

        for (let j = 0; j < revNodeSegs.length; j++) {
            if (revNodeSegs[j] === reverseID) { revNodeSegs.splice(j, 1); }
        }

        for (let i = 0; i < revNodeSegs.length; i++) {
            const conSegObj = W.model.segments.getObjectById(revNodeSegs[i]);
             if (!conSegObj) continue; // Defensive check - skip if segment object is not found
            const conSeg = conSegObj.attributes;

            if (conSeg.roadType !== segType) {
                reverseLock = Math.max(conSeg.lockRank + 0, reverseLock); // +1 to convert lockRank to lock level
            } else {
                for (let k = 0; k < revNodeSegs.length; k++) {
                    if (!checkedSegs.some(segNum => segNum === conSeg.id)) {
                        const tempRank = processRevNode(conSeg.id);
                        reverseLock = Math.max(tempRank, reverseLock);
                    }
                }
            }
        }
        return reverseLock;
    }

    let calculatedLock = Math.max(processForNode(segID), processRevNode(segID));
    return Math.min(calculatedLock, 6); // Limit to L6
}

function pushCityNameAlert(cityId, alertMessageParts) {
    let cityName = '';
    if (cityId) {
        const city = wmeSDK.DataModel.Cities.getById({ cityId });
        cityName = city && city.name ? city.name : '';
    }
    alertMessageParts.push(`City Name: <b>${cityName || 'None'}</b>`);
}

const handleUpdate = () => {
    const selection = wmeSDK.Editing.getSelection();

    if (!selection || selection.objectType !== 'segment') return;

    log('Updating RoadType');
    const options = getOptions();
    let alertMessageParts = [];
    let updatedRoadType = false;
    let updatedLockLevel = false;
    let updatedSpeedLimit = false;
    let updatedPaved = false;
    let updatedCityName = false;
    let updatedSegmentName = false;
    const updatePromises = [];

    // If copySegmentAttributes is checked, copy all attributes from a connected segment
    if (options.copySegmentAttributes) {
        selection.ids.forEach(id => {
            updatePromises.push(delayedUpdate(() => {
                try {
                    const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
                    const fromNode = seg.fromNodeId;
                    const toNode = seg.toNodeId;
                    let connectedSegId = null;
                    const allSegs = wmeSDK.DataModel.Segments.getAll();
                    for (let s of allSegs) {
                        if (s.id !== id && (s.fromNodeId === fromNode || s.toNodeId === fromNode || s.fromNodeId === toNode || s.toNodeId === toNode)) {
                            connectedSegId = s.id;
                            break;
                        }
                    }
                    if (connectedSegId) {
                        const connectedSeg = wmeSDK.DataModel.Segments.getById({ segmentId: connectedSegId });
                        // Copy speed limits
                        wmeSDK.DataModel.Segments.updateSegment({
                            segmentId: id,
                            fwdSpeedLimit: connectedSeg.fwdSpeedLimit,
                            revSpeedLimit: connectedSeg.revSpeedLimit,
                            roadType: connectedSeg.roadType,
                            lockRank: connectedSeg.lockRank
                        });
                        // Copy address (primary, alt, city)
                        wmeSDK.DataModel.Segments.updateAddress({
                            segmentId: id,
                            primaryStreetId: connectedSeg.primaryStreetId,
                            alternateStreetIds: connectedSeg.alternateStreetIds || []
                        });
                        // Copy paved/unpaved
                        const isUnpaved = connectedSeg.flagAttributes && connectedSeg.flagAttributes.unpaved === true;
                        let toggled = false;
                        const segPanel = openPanel;
                        if (segPanel) {
                            const unpavedIcon = segPanel.querySelector('.w-icon-unpaved-fill');
                            if (unpavedIcon) {
                                const unpavedChip = unpavedIcon.closest('wz-checkable-chip');
                                if (unpavedChip) {
                                    if (isUnpaved !== (seg.flagAttributes && seg.flagAttributes.unpaved === true)) {
                                        unpavedChip.click();
                                        toggled = true;
                                    }
                                }
                            }
                        }
                        alertMessageParts.push(`Copied all attributes from connected segment.`);
                    } else {
                        alertMessageParts.push(`No connected segment found to copy attributes.`);
                    }
                } catch (error) {
                    console.error('Error copying all attributes:', error);
                }
            }, 100));
        });
        Promise.all(updatePromises).then(() => {
            if (alertMessageParts.length) {
                if (WazeWrap?.Alerts) {
                    WazeWrap.Alerts.info('EZRoads Mod', alertMessageParts.join('<br>'), false, false, 7000);
                } else {
                    alert('EZRoads Mod: ' + alertMessageParts.join('\n'));
                }
            }
        });
        return; // Skip normal update logic if copying all attributes
    }

    selection.ids.forEach(id => {

        // Road Type
        updatePromises.push(delayedUpdate(() => {
            if (options.roadType) {
                const seg = wmeSDK.DataModel.Segments.getById({segmentId: id});
				const selectedRoad = roadTypes.find(rt => rt.value === options.roadType);
                //alertMessageParts.push(`Road Type: <b>${selectedRoad.name}</b>`);
                //updatedRoadType = true;
				log(`Segment ID: ${id}, Current Road Type: ${seg.roadType}, Target Road Type: ${options.roadType}, Target Road Name : ${selectedRoad.name}`); // Log current and target road type
                if (seg.roadType === options.roadType) {
                    log(`Segment ID: ${id} already has the target road type: ${options.roadType}. Skipping update.`);
                    alertMessageParts.push(`Road Type: <b>${selectedRoad.name} exists. Skipping update.</b>`);
                    updatedRoadType = true;
                } else {
                    try {
                        wmeSDK.DataModel.Segments.updateSegment({ segmentId: id, roadType: options.roadType });
                        log('Road type updated successfully.');
                        alertMessageParts.push(`Road Type: <b>${selectedRoad.name}</b>`);
                        updatedRoadType = true;
                    } catch (error) {
                        console.error('Error updating road type:', error);
                    }
                }
            }
        }, 200)); // 200ms delay before road type update

        // Set lock if enabled
        updatePromises.push(delayedUpdate(() => {
            if (options.setLock) {
            const rank = wmeSDK.State.getUserInfo().rank;
            const selectedRoad = roadTypes.find(rt => rt.value === options.roadType);
            if (selectedRoad) {
                let lockSetting = options.locks.find(l => l.id === selectedRoad.id);
                if (lockSetting) {
                    let toLock = lockSetting.lock;
                    if (toLock === 'HRCS') {
                        toLock = getHighestSegLock(id);
                    } else {
                        toLock = parseInt(toLock, 10);
                        toLock = Math.max(toLock - 1, 0); // Adjust to 0-based rank, ensuring it does not go below 0
                    }

                if (rank < toLock) toLock = rank;

                log(toLock);

                try {
                    const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
                    let displayLockLevel = (toLock === 'HRCS' || isNaN(toLock)) ? 'HRCS' : `L${toLock + 1}`;
                    let currentDisplayLockLevel;
                    if (seg.lockRank === 'HRCS') { // Should not happen, but for safety
                        currentDisplayLockLevel = 'HRCS';
                    }
                    else {
                        currentDisplayLockLevel = `L${seg.lockRank + 1}`;
                    }
                    if (seg.lockRank === toLock || (lockSetting.lock === 'HRCS' && currentDisplayLockLevel === displayLockLevel)) { // Compare lock levels
                    log(`Segment ID: ${id} already has the target lock level: ${displayLockLevel}. Skipping update.`);
                    alertMessageParts.push(`Lock Level: <b>${displayLockLevel} exists. Skipping update.</b>`);
                    updatedLockLevel = true;
                    } else {
                    wmeSDK.DataModel.Segments.updateSegment({
                        segmentId: id,
                        lockRank: toLock
                    });
                    alertMessageParts.push(`Lock Level: <b>${displayLockLevel}</b>`);
                    updatedLockLevel = true;
                    }
                } catch (error) {
                    console.error('Error updating segment lock rank:', error);
                }
                }
            }
            }
        }, 300)); // 250ms delay before lock rank update

        // Speed Limit - use road-specific speed if updateSpeed is enabled
        updatePromises.push(delayedUpdate(() => {		
        if (options.updateSpeed) {
            const selectedRoad = roadTypes.find(rt => rt.value === options.roadType);
            if (selectedRoad) {
                const speedSetting = options.speeds.find(s => s.id === selectedRoad.id);
                log('Selected road for speed: ' + selectedRoad.name);
                log('Speed setting found: ' + (speedSetting ? 'yes' : 'no'));

                if (speedSetting) {
                    const speedValue = parseInt(speedSetting.speed, 10);
                    log('Speed value to set: ' + speedValue);

                    // Apply speed if it's a valid number (including 0)
                    if (!isNaN(speedValue) && speedValue >= 0) {
                        log('Applying speed: ' + speedValue);
                        const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
                        if (seg.fwdSpeedLimit !== speedValue || seg.revSpeedLimit !== speedValue) {
                            wmeSDK.DataModel.Segments.updateSegment({
                                segmentId: id,
                                fwdSpeedLimit: speedValue,
                                revSpeedLimit: speedValue
                            });
                            alertMessageParts.push(`Speed Limit: <b>${speedValue}</b>`);
                            updatedSpeedLimit = true;
                        } else {
                            log(`Segment ID: ${id} already has the target speed limit: ${speedValue}. Skipping update.`);
                            alertMessageParts.push(`Speed Limit: <b>${speedValue} exists. Skipping update.</b>`);
                            updatedSpeedLimit = true;
                        }
                        //alertMessageParts.push(`Speed Limit: <b>${speedValue}</b>`);
                        //updatedSpeedLimit = true;
                    } else {
                        log('Not applying speed - invalid value: ' + speedSetting.speed);
                        alertMessageParts.push(`Speed Limit: <b>Invalid value ${speedValue}</b>`);
                        updatedSpeedLimit = true;
                    }
                }
            }
        } else {
            log('Speed updates disabled');
        }
        }, 400)); // 300ms delay before lock rank update		

        // Handling the street
        if (options.setStreet) {
            let city;
            let street;
            city = getTopCity() || getEmptyCity();
            street = wmeSDK.DataModel.Streets.getStreet({
                cityId: city.id,
                streetName: '',
            });
            log(`City Name: ${city?.name}, City ID: ${city?.id}, Street ID: ${street?.id}`);
            if(!street) {
                street = wmeSDK.DataModel.Streets.addStreet({
                    streetName: '',
                    cityId: city.id
                });
                log(`Created new empty street. Street ID: ${street?.id}`);
            }
            try {
                wmeSDK.DataModel.Segments.updateAddress({
                    segmentId: id,
                    primaryStreetId: street.id
                });
                pushCityNameAlert(city.id, alertMessageParts);
                updatedCityName = true;
            } catch (error) {
                console.error('Error updating segment address:', error);
            }
        }

        log(options);

        // Updated unpaved handler with SegmentFlagAttributes and fallback
    updatePromises.push(delayedUpdate(() => {
        if (options.unpaved) {
            const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
            const isUnpaved = seg.flagAttributes && seg.flagAttributes.unpaved === true;
            let unpavedToggled = false;

            if (!isUnpaved) {
                // Only click if segment is not already unpaved
                const unpavedIcon = openPanel.querySelector('.w-icon-unpaved-fill');
                if (unpavedIcon) {
                    const unpavedChip = unpavedIcon.closest('wz-checkable-chip');
                    if (unpavedChip) {
                        unpavedChip.click();
                        log('Clicked unpaved chip (set to unpaved)');
                        unpavedToggled = true;
                    }
                }
                // If new method failed, try the old method as fallback for non-compact mode
                if (!unpavedToggled) {
                    try {
                        const wzCheckbox = openPanel.querySelector('wz-checkbox[name="unpaved"]');
                        if (wzCheckbox) {
                            const hiddenInput = wzCheckbox.querySelector('input[type="checkbox"][name="unpaved"]');
                            if (hiddenInput && !hiddenInput.checked) {
                                hiddenInput.click();
                                log('Clicked unpaved checkbox (set to unpaved, non-compact mode)');
                                unpavedToggled = true;
                            }
                        }
                    } catch (e) {
                        log('Fallback to non-compact mode unpaved toggle method failed: ' + e);
                    }
                }
                if (unpavedToggled) {
                    alertMessageParts.push(`Paved Status: <b>Unpaved</b>`);
                    updatedPaved = true;
                }
            } else {
                // Already unpaved, no action needed
                alertMessageParts.push(`Paved Status: <b>Unpaved (already set)</b>`);
                updatedPaved = true;
            }
        } else {
            // If option is not checked and segment is unpaved, set it as paved
            const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
            const isUnpaved = seg.flagAttributes && seg.flagAttributes.unpaved === true;
            let pavedToggled = false;

            if (isUnpaved) {
                // Click to set as paved
                const unpavedIcon = openPanel.querySelector('.w-icon-unpaved-fill');
                if (unpavedIcon) {
                    const unpavedChip = unpavedIcon.closest('wz-checkable-chip');
                    if (unpavedChip) {
                        unpavedChip.click();
                        log('Clicked unpaved chip (set to paved)');
                        pavedToggled = true;
                    }
                }
                // If new method failed, try the old method as fallback for non-compact mode
                if (!pavedToggled) {
                    try {
                        const wzCheckbox = openPanel.querySelector('wz-checkbox[name="unpaved"]');
                        if (wzCheckbox) {
                            const hiddenInput = wzCheckbox.querySelector('input[type="checkbox"][name="unpaved"]');
                            if (hiddenInput && hiddenInput.checked) {
                                hiddenInput.click();
                                log('Clicked unpaved checkbox (set to paved, non-compact mode)');
                                pavedToggled = true;
                            }
                        }
                    } catch (e) {
                        log('Fallback to non-compact mode paved toggle method failed: ' + e);
                    }
                }
                if (pavedToggled) {
                    alertMessageParts.push(`Paved Status: <b>Paved</b>`);
                    updatedPaved = true;
                }
            } else {
                // Already paved, no action needed
                alertMessageParts.push(`Paved Status: <b>Paved (already set)</b>`);
                updatedPaved = true;
            }
        }
    }, 500)); // 500ms delay for unpaved/paved toggle    

        // 3a. Copy segment name from connected segment if enabled
        updatePromises.push(delayedUpdate(() => {
            if (options.copySegmentName) {
                try {
                    const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
                    const fromNode = seg.fromNodeId;
                    const toNode = seg.toNodeId;
                    let connectedSegId = null;
                    const allSegs = wmeSDK.DataModel.Segments.getAll();
                    for (let s of allSegs) {
                        if (s.id !== id && (s.fromNodeId === fromNode || s.toNodeId === fromNode || s.fromNodeId === toNode || s.toNodeId === toNode)) {
                            connectedSegId = s.id;
                            break;
                        }
                    }
                    if (connectedSegId) {
                        const connectedSeg = wmeSDK.DataModel.Segments.getById({ segmentId: connectedSegId });
                        const streetId = connectedSeg.primaryStreetId;
                        const altStreetIds = connectedSeg.alternateStreetIds || [];
                        const street = wmeSDK.DataModel.Streets.getById({ streetId });
                        // Get alternate street names
                        let altNames = [];
                        altStreetIds.forEach(streetId => {
                            const altStreet = wmeSDK.DataModel.Streets.getById({ streetId });
                            if (altStreet && altStreet.name) altNames.push(altStreet.name);
                        });
                        if (street && (street.name || street.englishName || street.signText)) {
                            wmeSDK.DataModel.Segments.updateAddress({
                                segmentId: id,
                                primaryStreetId: streetId,
                                alternateStreetIds: altStreetIds
                            });
                            let aliasMsg = altNames.length ? ` (Alternatives: ${altNames.join(', ')})` : '';
                            alertMessageParts.push(`Copied Name: <b>${street.name || ''}</b>${aliasMsg}`);
                            updatedSegmentName = true;
                        } else {
                            alertMessageParts.push(`Copied Name: <b>None (connected segment has no name)</b>`);
                            updatedSegmentName = true;
                        }
                        // Always push city name as a separate alert
                        pushCityNameAlert(street.cityId, alertMessageParts);
                        updatedCityName = true;
                    } else {
                        alertMessageParts.push(`Copied Name: <b>None (no connected segment found)</b>`);
                        updatedSegmentName = true;
                    }
                } catch (error) {
                    console.error('Error copying segment name:', error);
                }
            }
        }, 100)); // Run early in the update chain
    })

    Promise.all(updatePromises).then(() => {
        // Always push city name alert if not already set by other actions
        selection.ids.forEach(id => {
            if (!alertMessageParts.some(part => part.startsWith("City Name"))) {
                const seg = wmeSDK.DataModel.Segments.getById({ segmentId: id });
                if (seg && seg.primaryStreetId) {
                    const street = wmeSDK.DataModel.Streets.getById({ streetId: seg.primaryStreetId });
                    if (street) {
                        pushCityNameAlert(street.cityId, alertMessageParts);
                        updatedCityName = true;
                    }
                }
            }
        });

        const showAlert = () => {
            const updatedFeatures = [];
            if(updatedCityName) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("City")));
            if(updatedSegmentName) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("Copied Name")));
            if(updatedRoadType) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("Road Type")));
            if(updatedLockLevel) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("Lock Level")));
            if(updatedSpeedLimit) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("Speed Limit")));
            if(updatedPaved) updatedFeatures.push(alertMessageParts.find(part => part.startsWith("Paved")));
            const message = updatedFeatures.filter(Boolean).join(', ');
            if (message) {
                if (WazeWrap?.Alerts) {
                    WazeWrap.Alerts.info('EZRoads Mod', `Segment updated with: ${message}`, false, false, 7000);
                } else {
                    alert('EZRoads Mod: Segment updated (WazeWrap Alerts not available)');
                }
            }
        }

        // Autosave - DELAYED AUTOSAVE
        if (options.autosave) {
            setTimeout(() => {
                log('Delayed Autosave starting...');
                wmeSDK.Editing.save().then(() => {
                    log('Delayed Autosave completed.');
                    showAlert();
                });
            }, 600); // 1000ms (1 second) delay before autosave
        } else {
            showAlert();
        }
    });
}

const constructSettings = () => {
    const localOptions = getOptions();
    let currentRoadType = localOptions.roadType;
    const update = (key, value) => {
        const options = getOptions();
        options[key] = value;
        localOptions[key] = value;
        saveOptions(options);
        // Only show alert if not updating shortcutKey
        if (key !== 'shortcutKey') {
            if (WazeWrap?.Alerts) {
                WazeWrap.Alerts.success('EZRoads Mod', 'Options saved!', false, false, 1500);
            } else {
                alert('EZRoads Mod: Options saved!');
            }
        }
    };

    // Update lock level for a specific road type
    const updateLockLevel = (roadTypeId, lockLevel) => {
        const options = getOptions();
        const lockIndex = options.locks.findIndex(l => l.id === roadTypeId);
        if (lockIndex !== -1) {
            options.locks[lockIndex].lock = lockLevel;
            localOptions.locks = options.locks;
            saveOptions(options);
            if (WazeWrap?.Alerts) {
                WazeWrap.Alerts.success('EZRoads Mod', 'Options saved!', false, false, 1500);
            } else {
                alert('EZRoads Mod: Options saved!');
            }
        }
    };

    // Update speed for a specific road type
    const updateSpeed = (roadTypeId, speed) => {
        const options = getOptions();
        const speedIndex = options.speeds.findIndex(s => s.id === roadTypeId);
        let speedValue = parseInt(speed, 10);
        if (isNaN(speedValue)) {
            speedValue = -1;
        }
        log(`Updating speed for road type ${roadTypeId} to ${speedValue}`);
        if (speedIndex !== -1) {
            options.speeds[speedIndex].speed = speedValue;
            localOptions.speeds = options.speeds;
            saveOptions(options);
            if (WazeWrap?.Alerts) {
                WazeWrap.Alerts.success('EZRoads Mod', 'Options saved!', false, false, 1500);
            } else {
                alert('EZRoads Mod: Options saved!');
            }
        }
    };

    // Reset all options to defaults
    const resetOptions = () => {
        saveOptions(defaultOptions);
        // Refresh the page to reload settings
        window.location.reload();
    };

    // Checkbox option definitions
    const checkboxOptions = [
        { id: 'setStreet', text: 'Set Street To None', key: 'setStreet' },
        { id: 'autosave', text: 'Autosave on Action', key: 'autosave' },
        { id: 'unpaved', text: 'Set as Unpaved (Uncheck for Paved)', key: 'unpaved' },
        { id: 'setLock', text: 'Set the lock level', key: 'setLock' },
        { id: 'updateSpeed', text: 'Update speed limits', key: 'updateSpeed' },
        { id: 'copySegmentName', text: 'Copy connected Segment Name', key: 'copySegmentName' },
        { id: 'copySegmentAttributes', text: 'Copy Connected Segment Attribute', key: 'copySegmentAttributes' } 
    ];

    // Helper function to create radio buttons
    const createRadioButton = (roadType) => {
        const id = `road-${roadType.id}`;
        const isChecked = localOptions.roadType === roadType.value;
        const lockSetting = localOptions.locks.find(l => l.id === roadType.id) || { id: roadType.id, lock: 1 };
        const speedSetting = localOptions.speeds.find(s => s.id === roadType.id) || { id: roadType.id, speed: 40 };

        const div = $(`<div class="ezroadsmod-option">
            <div class="ezroadsmod-radio-container">
                <input type="radio" id="${id}" name="defaultRoad" ${isChecked ? 'checked' : ''}>
                <label for="${id}">${roadType.name}</label>
                <select id="lock-level-${roadType.id}" class="road-lock-level" data-road-id="${roadType.id}" ${!localOptions.setLock ? 'disabled' : ''}>
                    ${locks.map(lock => `<option value="${lock.value}" ${lockSetting.lock === lock.value ? 'selected' : ''}>${lock.value === 'HRCS' ? 'HRCS' : 'L' + lock.value}</option>`).join('')}
                </select>
                <input type="number" id="speed-${roadType.id}" class="road-speed" data-road-id="${roadType.id}"
                       value="${speedSetting.speed}" min="-1" ${!localOptions.updateSpeed ? 'disabled' : ''}>
            </div>
        </div>`);

        div.find('input[type="radio"]').on('click', () => {
            update('roadType', roadType.value);
            currentRoadType = roadType.value;
        });

        div.find('select').on('change', function () {
            updateLockLevel(roadType.id, $(this).val());
        });

        div.find('input.road-speed').on('change', function () {
            // Get the value as a number
            const speedValue = parseInt($(this).val(), 10);
            // If it's not a number, reset to 0
            if (isNaN(speedValue)) {
                $(this).val(0);
                updateSpeed(roadType.id, 0);
            } else {
                updateSpeed(roadType.id, speedValue);
            }
        });

        return div;
    };

    // Helper function to create checkboxes
    const createCheckbox = (option) => {
    const isChecked = localOptions[option.key];
    const div = $(`<div class="ezroadsmod-option">
        <input type="checkbox" id="${option.id}" name="${option.id}" ${isChecked ? 'checked' : ''}>
        <label for="${option.id}">${option.text}</label>
    </div>`);
    div.on('click', () => {
        // Mutually exclusive logic for setStreet and copySegmentName
        if (option.key === 'setStreet' && $(`#${option.id}`).prop('checked')) {
            $('#copySegmentName').prop('checked', false);
            update('copySegmentName', false);
        }
        if (option.key === 'copySegmentName' && $(`#${option.id}`).prop('checked')) {
            $('#setStreet').prop('checked', false);
            update('setStreet', false);
        }
        // Mutually exclusive logic for copySegmentAttributes
        if (option.key === 'copySegmentAttributes' && $(`#${option.id}`).prop('checked')) {
            // Uncheck all except autosave and itself
            checkboxOptions.forEach(opt => {
                if (opt.key !== 'autosave' && opt.key !== 'copySegmentAttributes') {
                    $(`#${opt.id}`).prop('checked', false);
                    update(opt.key, false);
                }
            });
            update('copySegmentAttributes', true);
        } else if (option.key === 'copySegmentAttributes' && !$(`#${option.id}`).prop('checked')) {
            update('copySegmentAttributes', false);
        } else {
            update(option.key, $(`#${option.id}`).prop('checked'));
        }
    });
    return div;
};

    // -- Set up the tab for the script
    wmeSDK.Sidebar.registerScriptTab().then(({ tabLabel, tabPane }) => {
        tabLabel.innerText = 'EZRoads Mod';
        tabLabel.title = 'Easily Update Roads';

        // Setup base styles
        const styles = $(`<style>
            #ezroadsmod-settings h2, #ezroadsmod-settings h5 {
                margin-top: 0;
                margin-bottom: 10px;
            }
            .ezroadsmod-section {
                margin-bottom: 15px;
            }
            .ezroadsmod-option {
                margin-bottom: 8px;
            }
            .ezroadsmod-radio-container {
                display: flex;
                align-items: center;
            }
            .ezroadsmod-radio-container input[type="radio"] {
                margin-right: 5px;
            }
            .ezroadsmod-radio-container label {
                flex: 1;
                margin-right: 10px;
                text-align: left;
            }
            .ezroadsmod-radio-container select {
                width: 80px;
                margin-left: auto;
                margin-right: 5px;
            }
            .ezroadsmod-radio-container input.road-speed {
                width: 60px;
            }
            .ezroadsmod-reset-button {
                margin-top: 20px;
                padding: 8px 12px;
                background-color: #f44336;
                color: white;
                border: none;
                border-radius: 4px;
                cursor: pointer;
                font-weight: bold;
            }
            .ezroadsmod-reset-button:hover {
                background-color: #d32f2f;
            }
			.ezroadsmod-key {
				padding: 2px 4px;
				font-size: 90%;
				color: #fff;
				background-color: #333;
				border-radius: 3px;
				box-shadow: inset 0 -1px 0 rgba(0,0,0,.25);
				width:30px;
				text-align:center;
				border:none;
				outline:none;
				margin-left: 4px;
				margin-right: 4px;
			}
        </style>`);

        tabPane.innerHTML = '<div id="ezroadsmod-settings"></div>';
        const scriptContentPane = $('#ezroadsmod-settings');
		scriptContentPane.append(styles);

		// Header section
        const header = $(`<div class="ezroadsmod-section">
        <h2>EZRoads Mod</h2>
        <div>Current Version: <b>${scriptVersion}</b></div>
        <label for="ezroadsmod-shortcut-key">Update Keybind:</label>
        <input type="search" id="ezroadsmod-shortcut-key" class="ezroadsmod-key" maxlength="1" value="${localOptions.shortcutKey || 'g'}">
        <span style="font-size:smaller;">(e.g. a,b,1,2)</span>
        </div>
        </div>`);
        scriptContentPane.append(header);

        header.find('#ezroadsmod-shortcut-key').on('input', function () {
            let val = $(this).val().toLowerCase().replace(/[^a-z0-9]/g, '').slice(0, 1);
            $(this).val(val);
            update('shortcutKey', val || 'g');
            // Only show alert if a key was actually entered
            if (val) {
                if (WazeWrap?.Alerts) {
                    WazeWrap.Alerts.success('EZRoads Mod', 'Shortcut key updated! Please reload the page for it to take effect.', false, false, 3000);
                } else {
                    alert('Shortcut key updated! Please reload the page for it to take effect.');
                }
            }
        });

        // Road type and options header
        const roadTypeHeader = $(`<div class="ezroads-section">
            <div style="display: flex; align-items: center;">
                <div style="flex-grow: 1; text-align: center;">Road Type</div>
                <div style="width: 80px; text-align: center;">Lock</div>
                <div style="width: 60px; text-align: center;">Speed</div>
            </div>
        </div>`);
        scriptContentPane.append(roadTypeHeader);

        // Road type section with header
        const roadTypeSection = $(`<div class="ezroads-section">
            <div id="road-type-options"></div>
        </div>`);
        scriptContentPane.append(roadTypeSection);

        const roadTypeOptions = roadTypeSection.find('#road-type-options');
        roadTypes.forEach(roadType => {
            roadTypeOptions.append(createRadioButton(roadType));
        });
        // Additional options section
        const additionalSection = $(`<div class="ezroadsmod-section">
            <h5>Additional Options</h5>
            <div id="additional-options"></div>
        </div>`);
        scriptContentPane.append(additionalSection);

        const additionalOptions = additionalSection.find('#additional-options');
        checkboxOptions.forEach(option => {
            additionalOptions.append(createCheckbox(option));
        });

        // Update all lock dropdowns when setLock checkbox changes
        $(document).on('click', '#setLock', function () {
            const isChecked = $(this).prop('checked');
            $('.road-lock-level').prop('disabled', !isChecked);
        });

        // Update all speed inputs when updateSpeed checkbox changes
        $(document).on('click', '#updateSpeed', function () {
            const isChecked = $(this).prop('checked');
            $('.road-speed').prop('disabled', !isChecked);
            log('Speed update option changed to: ' + isChecked);
        });

        // Reset button section
        const resetButton = $(`<button class="ezroadsmod-reset-button">Reset All Options</button>`);
        resetButton.on('click', function () {
            if (confirm('Are you sure you want to reset all options to default values? It will reload the webpage!')) {
                resetOptions();
            }
        });
        scriptContentPane.append(resetButton);
    });
};
 function scriptupdatemonitor() {
        if (WazeWrap?.Ready) {
			bootstrap({ scriptUpdateMonitor: { downloadUrl } });
            WazeWrap.Interface.ShowScriptUpdate(scriptName, scriptVersion, updateMessage);
        } else {
            setTimeout(scriptupdatemonitor, 250);
        }
    }
    // Start the "scriptupdatemonitor"
    scriptupdatemonitor();
    console.log(`${scriptName} initialized.`);

    /*
Change Log

Version
2.5.2 - 2025-05-31
        - Improved shortcut keybind logic and UI.
        - Prevented double alerts and unnecessary alerts when keybind is empty.
        - Minor bug fixes.
2.5.1 - 2025-05-30
        - Added "Copy Connected Segment Attribute" option.
        - When enabled, all other options (except Autosave) are automatically unchecked.
        - Copies all major attributes from a connected segment: speed limit, primary name, alias name, city, paved/unpaved status, and lock level.
        - Ensures mutually exclusive logic for this option in the UI.

2.5.0 - 2025-05-29
        - Improved reliability of the unpaved toggle by adding a 500ms delay and fallback logic for both compact and non-compact UI modes.
        - Now ensures paved/unpaved status is always set correctly, regardless of UI mode.
        - Enhanced update logic: skips redundant updates for road type, lock, and speed if already set.
        - Improved city name and segment name copying logic, with more robust alerts and feedback.
        - UI improvements: disables lock/speed controls when their options are off; mutually exclusive logic for "Set Street To None" and "Copy connected Segment Name".
        - Improved logging for easier debugging and tracking.
        - Bug fixes and code cleanup for better reliability and maintainability.
2.4.9 - 2025-05-22
        - Improved reliability of unpaved toggle (500ms delay).
        - Minor bug fixes and code cleanup 
2.4.8 - 2025-05-21
        - Now pave/unpave works better. When unchecked, the segment will be paved!
2.4.7 - 2025-05-21
       Added support for copying alternative segment name from connected segment
       Thanks to -
        - MapOmatic, without him this would not be possible.
2.4.6 - 2025-05-20
       Added support for copying segment name from connected segment
       Limitation -
        - Currently Alt names wont be copied.
2.4.5 - 2025-05-11
       Added Support for HRCS
       Fixed -
        - Various other bugs        
		
*/

})();