// ==UserScript==
// @name         Google Search Correction Fix
// @namespace    http://tampermonkey.net/
// @version      1.0
// @description  Fix Google search bar reverting to incorrect spelling after correction
// @author       Satoko
// @license MIT
// @match        https://www.google.com/search?*
// @match        https://www.google.ru/search?*
// @match        https://www.google.com.ua/search?*
// @match        https://www.google.ac/search?*
// @match        https://www.google.ad/search?*
// @match        https://www.google.ae/search?*
// @match        https://www.google.com.af/search?*
// @match        https://www.google.com.ag/search?*
// @match        https://www.google.com.ai/search?*
// @match        https://www.google.al/search?*
// @match        https://www.google.am/search?*
// @match        https://www.google.co.ao/search?*
// @match        https://www.google.com.ar/search?*
// @match        https://www.google.as/search?*
// @match        https://www.google.at/search?*
// @match        https://www.google.com.au/search?*
// @match        https://www.google.az/search?*
// @match        https://www.google.ba/search?*
// @match        https://www.google.com.bd/search?*
// @match        https://www.google.be/search?*
// @match        https://www.google.bf/search?*
// @match        https://www.google.bg/search?*
// @match        https://www.google.com.bh/search?*
// @match        https://www.google.bi/search?*
// @match        https://www.google.bj/search?*
// @match        https://www.google.com.bn/search?*
// @match        https://www.google.com.bo/search?*
// @match        https://www.google.com.br/search?*
// @match        https://www.google.bs/search?*
// @match        https://www.google.bt/search?*
// @match        https://www.google.co.bw/search?*
// @match        https://www.google.by/search?*
// @match        https://www.google.com.bz/search?*
// @match        https://www.google.ca/search?*
// @match        https://www.google.cd/search?*
// @match        https://www.google.cf/search?*
// @match        https://www.google.cg/search?*
// @match        https://www.google.ch/search?*
// @match        https://www.google.ci/search?*
// @match        https://www.google.co.ck/search?*
// @match        https://www.google.cl/search?*
// @match        https://www.google.cm/search?*
// @match        https://www.google.cn/search?*
// @match        https://www.google.com.co/search?*
// @match        https://www.google.co.cr/search?*
// @match        https://www.google.com.cu/search?*
// @match        https://www.google.cv/search?*
// @match        https://www.google.com.cy/search?*
// @match        https://www.google.cz/search?*
// @match        https://www.google.de/search?*
// @match        https://www.google.dj/search?*
// @match        https://www.google.dk/search?*
// @match        https://www.google.dm/search?*
// @match        https://www.google.com.do/search?*
// @match        https://www.google.dz/search?*
// @match        https://www.google.com.ec/search?*
// @match        https://www.google.ee/search?*
// @match        https://www.google.com.eg/search?*
// @match        https://www.google.es/search?*
// @match        https://www.google.com.et/search?*
// @match        https://www.google.fi/search?*
// @match        https://www.google.com.fj/search?*
// @match        https://www.google.fm/search?*
// @match        https://www.google.fr/search?*
// @match        https://www.google.ga/search?*
// @match        https://www.google.ge/search?*
// @match        https://www.google.gg/search?*
// @match        https://www.google.com.gh/search?*
// @match        https://www.google.com.gi/search?*
// @match        https://www.google.gl/search?*
// @match        https://www.google.gm/search?*
// @match        https://www.google.gr/search?*
// @match        https://www.google.com.gt/search?*
// @match        https://www.google.gy/search?*
// @match        https://www.google.com.hk/search?*
// @match        https://www.google.hn/search?*
// @match        https://www.google.hr/search?*
// @match        https://www.google.ht/search?*
// @match        https://www.google.hu/search?*
// @match        https://www.google.co.id/search?*
// @match        https://www.google.ie/search?*
// @match        https://www.google.co.il/search?*
// @match        https://www.google.im/search?*
// @match        https://www.google.co.in/search?*
// @match        https://www.google.iq/search?*
// @match        https://www.google.is/search?*
// @match        https://www.google.it/search?*
// @match        https://www.google.je/search?*
// @match        https://www.google.com.jm/search?*
// @match        https://www.google.jo/search?*
// @match        https://www.google.co.jp/search?*
// @match        https://www.google.co.ke/search?*
// @match        https://www.google.com.kh/search?*
// @match        https://www.google.ki/search?*
// @match        https://www.google.kg/search?*
// @match        https://www.google.co.kr/search?*
// @match        https://www.google.com.kw/search?*
// @match        https://www.google.kz/search?*
// @match        https://www.google.la/search?*
// @match        https://www.google.com.lb/search?*
// @match        https://www.google.li/search?*
// @match        https://www.google.lk/search?*
// @match        https://www.google.co.ls/search?*
// @match        https://www.google.lt/search?*
// @match        https://www.google.lu/search?*
// @match        https://www.google.lv/search?*
// @match        https://www.google.com.ly/search?*
// @match        https://www.google.co.ma/search?*
// @match        https://www.google.md/search?*
// @match        https://www.google.me/search?*
// @match        https://www.google.mg/search?*
// @match        https://www.google.mk/search?*
// @match        https://www.google.ml/search?*
// @match        https://www.google.com.mm/search?*
// @match        https://www.google.mn/search?*
// @match        https://www.google.ms/search?*
// @match        https://www.google.com.mt/search?*
// @match        https://www.google.mu/search?*
// @match        https://www.google.mv/search?*
// @match        https://www.google.mw/search?*
// @match        https://www.google.com.mx/search?*
// @match        https://www.google.com.my/search?*
// @match        https://www.google.co.mz/search?*
// @match        https://www.google.com.na/search?*
// @match        https://www.google.com.nf/search?*
// @match        https://www.google.com.ng/search?*
// @match        https://www.google.com.ni/search?*
// @match        https://www.google.ne/search?*
// @match        https://www.google.nl/search?*
// @match        https://www.google.no/search?*
// @match        https://www.google.com.np/search?*
// @match        https://www.google.nr/search?*
// @match        https://www.google.nu/search?*
// @match        https://www.google.co.nz/search?*
// @match        https://www.google.com.om/search?*
// @match        https://www.google.com.pa/search?*
// @match        https://www.google.com.pe/search?*
// @match        https://www.google.com.pg/search?*
// @match        https://www.google.com.ph/search?*
// @match        https://www.google.com.pk/search?*
// @match        https://www.google.pl/search?*
// @match        https://www.google.pn/search?*
// @match        https://www.google.com.pr/search?*
// @match        https://www.google.ps/search?*
// @match        https://www.google.pt/search?*
// @match        https://www.google.com.py/search?*
// @match        https://www.google.com.qa/search?*
// @match        https://www.google.ro/search?*
// @match        https://www.google.ru/search?*
// @match        https://www.google.rw/search?*
// @match        https://www.google.com.sa/search?*
// @match        https://www.google.com.sb/search?*
// @match        https://www.google.sc/search?*
// @match        https://www.google.se/search?*
// @match        https://www.google.com.sg/search?*
// @match        https://www.google.sh/search?*
// @match        https://www.google.si/search?*
// @match        https://www.google.sk/search?*
// @match        https://www.google.com.sl/search?*
// @match        https://www.google.sn/search?*
// @match        https://www.google.so/search?*
// @match        https://www.google.sm/search?*
// @match        https://www.google.sr/search?*
// @match        https://www.google.st/search?*
// @match        https://www.google.com.sv/search?*
// @match        https://www.google.td/search?*
// @match        https://www.google.tg/search?*
// @match        https://www.google.co.th/search?*
// @match        https://www.google.com.tj/search?*
// @match        https://www.google.tk/search?*
// @match        https://www.google.tl/search?*
// @match        https://www.google.tm/search?*
// @match        https://www.google.tn/search?*
// @match        https://www.google.to/search?*
// @match        https://www.google.com.tr/search?*
// @match        https://www.google.tt/search?*
// @match        https://www.google.com.tw/search?*
// @match        https://www.google.co.tz/search?*
// @match        https://www.google.com.ua/search?*
// @match        https://www.google.co.ug/search?*
// @match        https://www.google.co.uk/search?*
// @match        https://www.google.com.uy/search?*
// @match        https://www.google.co.uz/search?*
// @match        https://www.google.com.vc/search?*
// @match        https://www.google.co.ve/search?*
// @match        https://www.google.vg/search?*
// @match        https://www.google.co.vi/search?*
// @match        https://www.google.com.vn/search?*
// @match        https://www.google.vu/search?*
// @match        https://www.google.ws/search?*
// @match        https://www.google.rs/search?*
// @match        https://www.google.co.za/search?*
// @match        https://www.google.co.zm/search?*
// @match        https://www.google.co.zw/search?*
// @grant        none
// @downloadURL none
// ==/UserScript==

(function() {
    'use strict';

    console.log('Google Search Correction Fix loaded');

    function fixCorrectionLinks() {
        const correctionLinks = document.querySelectorAll('a[href*="spell=1"], a[href*="correction"], .gL9Hy, .gNG8Bc, a[jsname*="spell"]');

        correctionLinks.forEach(link => {
            if (link.hasAttribute('data-fixed')) return;

            link.setAttribute('data-fixed', 'true');

            const originalHref = link.href;

            link.addEventListener('click', function(e) {
                e.preventDefault();
                e.stopPropagation();

                console.log('Intercepted correction click, redirecting to:', originalHref);

                window.location.href = originalHref;
            });
        });
    }

    function fixDidYouMean() {
        const didYouMeanElements = document.querySelectorAll('#fprs a, .aCOpRe em, .MUxGbd, .gqFSUc');

        didYouMeanElements.forEach(element => {
            if (element.hasAttribute('data-fixed')) return;

            const link = element.closest('a') || element;
            if (link.href && link.href.includes('google.com/search')) {
                link.setAttribute('data-fixed', 'true');

                link.addEventListener('click', function(e) {
                    if (this.href && this.href !== window.location.href) {
                        e.preventDefault();
                        e.stopPropagation();
                        console.log('Intercepted "Did you mean" click');
                        window.location.href = this.href;
                    }
                });
            }
        });
    }

    function fixSearchBox() {
        const searchBox = document.querySelector('input[name="q"]');
        if (searchBox) {
            const urlParams = new URLSearchParams(window.location.search);
            const urlQuery = urlParams.get('q');

            if (urlQuery && searchBox.value !== urlQuery) {
                if (!searchBox.value.includes(urlQuery) && urlQuery.includes(searchBox.value)) {
                    const newUrl = new URL(window.location.href);
                    newUrl.searchParams.set('q', searchBox.value);
                    window.history.replaceState(null, '', newUrl.toString());
                    console.log('Fixed URL to match search box:', searchBox.value);
                }
            }
        }
    }

    function runAllFixes() {
        fixCorrectionLinks();
        fixDidYouMean();
        fixSearchBox();
    }

    runAllFixes();

    const observer = new MutationObserver(function(mutations) {
        let shouldRun = false;
        mutations.forEach(function(mutation) {
            if (mutation.addedNodes.length) {
                shouldRun = true;
            }
        });
        if (shouldRun) {
            setTimeout(runAllFixes, 100);
        }
    });

    observer.observe(document.body, {
        childList: true,
        subtree: true
    });

    setInterval(runAllFixes, 2000);

})();