background.js

function isEmpty(obj) {
    return Object.keys(obj).length === 0;
}
function checkArrayForString(arr, string) {
    for(let i = 0; i < arr.length; i++) {
        if(arr[i].indexOf(string) > -1) {
            return true;
        }
    };
    return false;
}

function startJustRead(tab) {
    var tabId = tab ? tab.id : null; // Defaults to the current tab
    chrome.tabs.executeScript(tabId, {
        file: "content_script.js", // Script to inject into page and run in sandbox
        allFrames: false // This injects script into iframes in the page and doesn't work before 4.0.266.0.
    });

    // Add a badge to signify the extension is in use
    chrome.browserAction.setBadgeBackgroundColor({color:[242, 38, 19, 230]});
    chrome.browserAction.setBadgeText({text:"on"});

    // Check if we need to add the site to JR's autorun list
    chrome.storage.sync.get("alwaysAddAR", function(result) {
        if(result && result["alwaysAddAR"]) {
            addSiteToAutorunList(null, tab);
        }
    });

    setTimeout(function() {
        chrome.browserAction.setBadgeText({text:""});
    }, 2000);
}

function startSelectText() {
    chrome.tabs.executeScript(null, {
        code: 'var useText = true;' // Ghetto way of signaling to select text instead of
    }, function() {                 // using Chrome messages
        startJustRead();
    });
}

function createPageCM() {
    // Create a right click menu option
    pageCMId = chrome.contextMenus.create({
         title: "View this page using Just Read",
         id: "pageCM",
         contexts: ["page"],
         onclick: startJustRead
    });
}
function createHighlightCM() {
    // Create an entry to allow user to use currently selected text
    highlightCMId = chrome.contextMenus.create({
        title: "View this selection in Just Read",
        id: "highlightCM",
        contexts:["selection"],
        onclick: function(info, tab) {
            chrome.tabs.executeScript(null, {
                code: 'var textToRead = true'
            }, function() {
                startJustRead();
            });
        }
    });
}
function createLinkCM() {
    // Create an entry to allow user to open a given link using Just read
    linkCMId = chrome.contextMenus.create({
        title: "View the linked page using Just Read",
        id: "linkCM",
        contexts:["link"],
        onclick: function(info, tab) {
            chrome.tabs.create(
                { url: info.linkUrl, active: false },
                function(newTab) {
                    chrome.tabs.executeScript(newTab.id, {
                        code: 'var runOnLoad = true'
                    }, function() {
                        startJustRead(newTab);
                    });
                }
            );
            
        }
    });
}
function createAutorunCM() {
    // Create an entry to allow user to open a given link using Just read
    autorunCMId = chrome.contextMenus.create({
        title: "Add this site to Just Read' auto-run list",
        id: "autorunCM",
        contexts:["page"],
        onclick: addSiteToAutorunList
    });
}
function addSiteToAutorunList(info, tab) {
    chrome.storage.sync.get('auto-enable-site-list', function(result) {
        let url = new URL((info != null && info.pageUrl) || tab.url);
        let entry;
        if(url.pathname !== "/"
        && url.pathname !== "") {
            entry = url.hostname + "/.+";
        } else {
            entry = url.hostname;
        }

        let currentDomains = result['auto-enable-site-list'];

        if(!isEmpty(currentDomains)) {
            if(!currentDomains.includes(entry)) {
                chrome.storage.sync.set({
                    'auto-enable-site-list': [...currentDomains, entry],
                }, function() {
                    if(checkArrayForString(currentDomains, url.hostname)) {
                        console.log("Auto-run entry added.\n\nWarning: An auto-run entry with the same hostname has already been added. Be careful to not add two duplicates.");
                    } else {
                        console.log('Auto-run entry added.');
                    }
                });
            } else {
                console.log("Entry already exists inside of Just Read's auto-run list. Not adding new entry.")
            }
        } else {
            chrome.storage.sync.set({ 'auto-enable-site-list': [entry] });
        }
    });
}


var pageCMId = highlightCMId = linkCMId = undefined;
function updateCMs() {
    chrome.storage.sync.get(["enable-pageCM", "enable-highlightCM", "enable-linkCM"], function (result) {
        var size = 0;
        
        for(var key in result) {
            size++;
            
            if(key === "enable-pageCM") {
                if(result[key]) {
                    if(typeof pageCMId == "undefined")
                        createPageCM();
                } else {
                    if(typeof pageCMId != "undefined") {
                        chrome.contextMenus.remove("pageCM");
                        pageCMId = undefined;
                    }
                }
            } else if(key === "enable-highlightCM") {
                if(result[key]) {
                    if(typeof highlightCMId == "undefined")
                        createHighlightCM();
                } else {
                    if(typeof highlightCMId != "undefined") {
                        chrome.contextMenus.remove("highlightCM");
                        highlightCMId = undefined;
                    }
                }
            } else if(key === "enable-linkCM") {
                if(result[key]) {
                    if(typeof linkCMId == "undefined")
                        createLinkCM();
                } else {
                    if(typeof linkCMId != "undefined") {
                        chrome.contextMenus.remove("linkCM");
                        linkCMId = undefined;
                    }
                }
            } else if(key === "enable-autorunCM") {
                if(result[key]) {
                    if(typeof autorunCMId == "undefined")
                        createAutorunCM();
                } else {
                    if(typeof autorunCMId != "undefined") {
                        chrome.contextMenus.remove("autorunCM");
                        autorunCMId = undefined;
                    }
                }
            } 
        }
        
        if(size === 0) {
            createPageCM();
            createHighlightCM();
            createLinkCM();
            createAutorunCM();
        }
    });
}

// Listen for the extension's click
chrome.browserAction.onClicked.addListener(startJustRead);

// Add our context menus
updateCMs();

// Listen for the keyboard shortcut
chrome.commands.onCommand.addListener(function(command) {
    if(command == "open-just-read")
        startJustRead();
    if(command == "select-text")
        startSelectText();
});

// Listen for messages
chrome.runtime.onMessage.addListener(function(request, sender, sendResponse) {
    if(request === "Open options") {
        chrome.runtime.openOptionsPage();
    } else if(request.updateCMs === "true") {
        updateCMs();
    } else if(request.savedVersion) { 
        let tempObj = {};
        tempObj.content = request.savedVersion;
        tempObj.url = sender.url;
        localStorage.setItem('JRSavedPage', JSON.stringify(tempObj));
    } else if(request.hasSavedVersion === "true") {
        let lastSavedPage = JSON.parse(localStorage.getItem('JRSavedPage'));
        if(lastSavedPage
        && sender.url === lastSavedPage.url) {
            sendResponse({ content: lastSavedPage.content });
        }
    }
});

// Create an entry to allow user to select an element to read from
chrome.contextMenus.create({
    title: "Select content to read",
    contexts: ["browser_action"],
    onclick: function() {
        startSelectText();
    }
});

// Create an entry to give information about the premium version
chrome.contextMenus.create({
    title: "Get Just Read Premium",
    contexts: ["browser_action"],
    onclick: function() {
        window.open('https://justread.link', '_blank');
    }
});

chrome.tabs.onUpdated.addListener( function (tabId, changeInfo, tab) {
    if (changeInfo.status == 'loading') {
        // Auto enable on sites specified
        chrome.storage.sync.get('auto-enable-site-list', function (siteListObj) {
            var siteList;   
            if(siteListObj) {
                siteList = siteListObj['auto-enable-site-list'];
                var url = tab.url;
                
                if(typeof siteList != "undefined") {
                    for(var i = 0; i < siteList.length; i++) {
                        var regex = new RegExp(siteList[i], "i");
        
                        if( url.match( regex ) ) {
                            chrome.tabs.executeScript(tabId, {
                                code: 'var runOnLoad = true;' // Ghetto way of signaling to run on load
                            }, function() {                   // instead of using Chrome messages
                                startJustRead(tab);
                            });
                        }
                    }
                }
            }
        });
    }
});