content.js

let isClicked = false;
let gotMessage = (message, sender, sendResponse) => {
    replaceImg();
    isClicked = !isClicked;
    if(message.txt === "button" && isClicked) {
        colorText(document.body);
    } else {
        unColorText(document.body);
    }

    if(message.txt != "button") {
        let paragraphs = document.getElementsByTagName('p');
        iuText(paragraphs);
    }
}

let filenames = [
    "IU.jpg",
    "IU1.jpg",
    "IU2.jpg",
    "IU3.jpg",
    "IU4.jpg",
    "IU5.jpg"
];

let colorText = (element) => {
    if(element.hasChildNodes()) {
        element.childNodes.forEach(colorText);
        //text elements have no child nodes
    } else if (element.nodeType === Text.TEXT_NODE) {
        if(element.textContent.toLowerCase().includes('iu')) {
            let newElement = document.createElement('span');
            newElement.innerHTML = element.textContent.replace(/(iu)/gi, 
            '<span class="rainbow">$1</span>')
            element.replaceWith(newElement)
        }
    }
}

let replaceImg = () => {
    let imgs = document.getElementsByTagName('img');
    for (imgElt of imgs) {
        // let r = Math.floor(Math.random() * filenames.length);
        // let file = 'IUimg/' + filenames[r];
        // let url = chrome.extension.getURL(file);
        // imgElt.src = url;
        imgElt.src = 'https://iu.wonder.im/images/profile';
    }
    
}

let unColorText = (element) => {
    if(element.hasChildNodes()) {
        element.childNodes.forEach(unColorText);
    } 

    if(element.classList) {
        for(let i = 0; i < element.classList.length; i++) {
            if(element.classList.contains("rainbow")){
                element.classList.remove("rainbow");
            }
        }
    }
}

let iuText = (paragraphs) => {
    
    for( p of paragraphs) {
        // p.innerHTML = message.txt;
        let paragraph = p.textContent.split(" ");

        // console.log(paragraph)
        // console.log(paragraph.length);
        let newString = "";
        for(let i = 0; i < paragraph.length; i++) {
            newString = newString.concat("IU <3");
            if(i != paragraph.length - 1) {
                newString = newString.concat(" ");
            }
        }

        console.log(newString)

        p.innerHTML = newString;
    }
}

chrome.runtime.onMessage.addListener(gotMessage);