injected.js

(function() {
    const originalFetch = window.fetch;
    const originalXHR = window.XMLHttpRequest;
    let mapInstance = null;
    let markers = [];

    function sendLog(type, message) {
        window.postMessage({ source: "GEOCOACH", type: "LOG", payload: { type, message } }, "*");
    }

    function sendToExtension(type, payload) {
        window.postMessage({ source: "GEOCOACH", type, payload }, "*");
    }

    // Generic recursive finder for [null, null, lat, lng] pattern
    function findCoords(obj, found = []) {
        if (!obj || typeof obj !== 'object') return found;
        
        if (Array.isArray(obj)) {
            // Check for [null, null, lat, lng]
            if (obj.length >= 4 && obj[0] === null && obj[1] === null && 
                typeof obj[2] === 'number' && typeof obj[3] === 'number') {
                // Sanity check coordinates
                if (Math.abs(obj[2]) <= 90 && Math.abs(obj[3]) <= 180) {
                     // Check if not already found to avoid dups in this pass
                     const exists = found.some(c => c.lat === obj[2] && c.lng === obj[3]);
                     if (!exists) {
                        found.push({ lat: obj[2], lng: obj[3] });
                     }
                }
            }
            for (let item of obj) {
                findCoords(item, found);
            }
        } else {
            for (let key in obj) {
                if (obj.hasOwnProperty(key)) {
                    findCoords(obj[key], found);
                }
            }
        }
        return found;
    }

    function processResponse(url, text) {
        if (url.includes("GetMetadata") || url.includes("GetViewportInfo")) {
            // sendLog("INFO", "Analyzing: " + url);
            try {
                // Clean prefix if exists
                let jsonText = text;
                if (text.startsWith(")]}'")) {
                    jsonText = text.substring(4);
                }
                
                const data = JSON.parse(jsonText);
                const coords = findCoords(data);
                
                if (coords.length > 0) {
                    sendLog("SUCCESS", `Found ${coords.length} coordinates in ${url}`);
                    sendToExtension("LOCATIONS_FOUND", coords);
                }
            } catch (e) {
                // sendLog("ERROR", "Parse error: " + e.message);
            }
        }
    }

    // Hook Fetch
    window.fetch = async function(...args) {
        const response = await originalFetch.apply(this, args);
        try {
            const url = args[0] ? args[0].toString() : "";
            const options = args[1] || {};
            const method = options.method ? options.method.toUpperCase() : "GET";

            // Detect guess submission (end of round/guess)
            if (url.includes("/guesses") && method === "POST") {
                sendLog("INFO", "Guess detected. Clearing coordinates.");
                sendToExtension("CLEAR_LOCATIONS", null);
            }

            if (url.includes("google") && (url.includes("GetMetadata") || url.includes("GetViewportInfo"))) {
                const clone = response.clone();
                clone.text().then(text => processResponse(url, text));
            }
        } catch (e) {
            console.error(e);
        }
        return response;
    };

    // Hook XHR
    const originalOpen = originalXHR.prototype.open;
    originalXHR.prototype.open = function(method, url) {
        this._url = url;
        originalOpen.apply(this, arguments);
    };

    const originalSend = originalXHR.prototype.send;
    originalXHR.prototype.send = function(body) {
        this.addEventListener('load', function() {
            if (this._url && (this._url.includes("GetMetadata") || this._url.includes("GetViewportInfo"))) {
                processResponse(this._url, this.responseText);
            }
        });
        originalSend.apply(this, arguments);
    };

    // Hook Google Maps
    function hookMap() {
        if (window.google && window.google.maps && window.google.maps.Map) {
            const originalMap = window.google.maps.Map;
            // Avoid double hooking
            if (originalMap._hooked) return;

            window.google.maps.Map = function(...args) {
                const map = new originalMap(...args);
                mapInstance = map;
                sendLog("INFO", "Map instance captured!");
                return map;
            };
            window.google.maps.Map.prototype = originalMap.prototype;
            // Copy statics
            for (let key in originalMap) {
                if (originalMap.hasOwnProperty(key)) {
                    window.google.maps.Map[key] = originalMap[key];
                }
            }
            window.google.maps.Map._hooked = true;
        } else {
            setTimeout(hookMap, 500);
        }
    }
    hookMap();

    // Listen for commands
    window.addEventListener("message", (event) => {
        if (event.data && event.data.target === "INJECTED") {
            if (event.data.type === "SHOW_PINS") {
                const locations = event.data.payload;
                if (mapInstance && window.google && window.google.maps) {
                    // Clear existing
                    markers.forEach(m => m.setMap(null));
                    markers = [];
                    
                    locations.forEach(loc => {
                        const marker = new window.google.maps.Marker({
                            position: { lat: loc.lat, lng: loc.lng },
                            map: mapInstance,
                            title: "Ground Truth",
                            icon: {
                                path: window.google.maps.SymbolPath.CIRCLE,
                                scale: 6,
                                fillColor: '#FF0000',
                                fillOpacity: 0.8,
                                strokeWeight: 1,
                                strokeColor: '#FFFFFF'
                            }
                        });
                        markers.push(marker);
                    });
                }
            }
        }
    });

})();