//when a new tab is highlighted chrome.tabs.onActivated.addListener( function(activeInfo) { chrome.tabs.get(activeInfo.tabId, function(tab) { let id = getYouTubeVideoID(tab.url); //if this even is a YouTube tab if (id) { videoIDChange(id, activeInfo.tabId); } }) } ); //when a tab changes URLs chrome.tabs.onUpdated.addListener(function(tabId, changeInfo, tab) { if (changeInfo != undefined && changeInfo.url != undefined) { let id = getYouTubeVideoID(changeInfo.url); //if URL changed and is youtube video message contentScript the video id if (changeInfo.url && id) { videoIDChange(id, tabId); } } }); chrome.runtime.onMessage.addListener(function (request, sender, callback) { if (request.message == "submitTimes") { submitTimes(request.videoID, callback); //this allows the callback to be called later by the submitTimes function return true; } else if (request.message == "addSponsorTime") { addSponsorTime(request.time, request.videoID, callback); //this allows the callback to be called later return true; } else if (request.message == "getSponsorTimes") { getSponsorTimes(request.videoID, function(sponsorTimes) { callback({ sponsorTimes: sponsorTimes }) }); //this allows the callback to be called later return true; } else if (request.message == "submitVote") { submitVote(request.type, request.UUID, callback); //this allows the callback to be called later return true; } }); //add help page on install chrome.runtime.onInstalled.addListener(function (object) { chrome.storage.sync.get(["userID"], function(result) { const userID = result.userID; // If there is no userID, then it is the first install. if (!userID){ //open up the install page chrome.tabs.create({url: chrome.extension.getURL("/help/index.html")}); //generate a userID const newUserID = generateUUID(); //save this UUID chrome.storage.sync.set({ "userID": newUserID, "serverAddress": serverAddress, //the last video id loaded, to make sure it is a video id change "sponsorVideoID": null, "previousVideoID": null }); } }); }); //gets the sponsor times from memory function getSponsorTimes(videoID, callback) { let sponsorTimes = []; let sponsorTimeKey = "sponsorTimes" + videoID; chrome.storage.sync.get([sponsorTimeKey], function(result) { let sponsorTimesStorage = result[sponsorTimeKey]; if (sponsorTimesStorage != undefined && sponsorTimesStorage.length > 0) { sponsorTimes = sponsorTimesStorage; } callback(sponsorTimes) }); } function addSponsorTime(time, videoID, callback) { getSponsorTimes(videoID, function(sponsorTimes) { //add to sponsorTimes if (sponsorTimes.length > 0 && sponsorTimes[sponsorTimes.length - 1].length < 2) { //it is an end time sponsorTimes[sponsorTimes.length - 1][1] = time; } else { //it is a start time let sponsorTimesIndex = sponsorTimes.length; sponsorTimes[sponsorTimesIndex] = []; sponsorTimes[sponsorTimesIndex][0] = time; } //save this info let sponsorTimeKey = "sponsorTimes" + videoID; chrome.storage.sync.set({[sponsorTimeKey]: sponsorTimes}, callback); }); } function submitVote(type, UUID, callback) { chrome.storage.sync.get(["serverAddress", "userID"], function(result) { const serverAddress = result.serverAddress; const userID = result.userID; //publish this vote sendRequestToServer("GET", serverAddress + "/api/voteOnSponsorTime?UUID=" + UUID + "&userID=" + userID + "&type=" + type, function(xmlhttp, error) { if (xmlhttp.readyState == 4 && xmlhttp.status == 200) { callback({ successType: 1 }); } else if (xmlhttp.readyState == 4 && xmlhttp.status == 405) { //duplicate vote callback({ successType: 0, statusCode: xmlhttp.status }); } else if (error) { //error while connect callback({ successType: -1, statusCode: xmlhttp.status }); } }) }) } function submitTimes(videoID, callback) { //get the video times from storage let sponsorTimeKey = 'sponsorTimes' + videoID; chrome.storage.sync.get([sponsorTimeKey, "serverAddress"], function(result) { let sponsorTimes = result[sponsorTimeKey]; const serverAddress = result.serverAddress; const userID = result.userID; if (sponsorTimes != undefined && sponsorTimes.length > 0) { //submit these times for (let i = 0; i < sponsorTimes.length; i++) { //submit the sponsorTime sendRequestToServer("GET", serverAddress + "/api/postVideoSponsorTimes?videoID=" + videoID + "&startTime=" + sponsorTimes[i][0] + "&endTime=" + sponsorTimes[i][1] + "&userID=" + userID, function(xmlhttp, error) { if (xmlhttp.readyState == 4 && !error) { callback({ statusCode: xmlhttp.status }); if (xmlhttp.status == 200) { //add these to the storage log chrome.storage.sync.get(["sponsorTimesContributed"], function(result) { let currentContributionAmount = 0; if (result.sponsorTimesContributed != undefined) { //current contribution amount is known currentContributionAmount = result.sponsorTimesContributed; } //save the amount contributed chrome.storage.sync.set({"sponsorTimesContributed": currentContributionAmount + sponsorTimes.length}); }); } } else if (error) { callback({ statusCode: -1 }); } }); } } }); } function videoIDChange(currentVideoID, tabId) { //send a message to the content script chrome.tabs.sendMessage(tabId, { message: 'ytvideoid', id: currentVideoID }); chrome.storage.sync.get(["sponsorVideoID", "previousVideoID"], function(result) { const sponsorVideoID = result.sponsorVideoID; const previousVideoID = result.previousVideoID; //not a url change if (sponsorVideoID == currentVideoID){ return; } chrome.storage.sync.set({ "sponsorVideoID": currentVideoID }); //warn them if they had unsubmitted times if (previousVideoID != null) { //get the sponsor times from storage let sponsorTimeKey = 'sponsorTimes' + previousVideoID; chrome.storage.sync.get([sponsorTimeKey], function(result) { let sponsorTimes = result[sponsorTimeKey]; if (sponsorTimes != undefined && sponsorTimes.length > 0) { //warn them that they have unsubmitted sponsor times chrome.notifications.create("stillThere" + Math.random(), { type: "basic", title: "Do you want to submit the sponsor times for watch?v=" + previousVideoID + "?", message: "You seem to have left some sponsor times unsubmitted. Go back to that page to submit them (they are not deleted).", iconUrl: "./icons/LogoSponsorBlocker256px.png" }); } //set the previous video id to the currentID chrome.storage.sync.set({ "previousVideoID": currentVideoID }); }); } else { chrome.storage.sync.set({ "previousVideoID": currentVideoID }); } }); } function sendRequestToServer(type, address, callback) { let xmlhttp = new XMLHttpRequest(); xmlhttp.open(type, address, true); if (callback != undefined) { xmlhttp.onreadystatechange = function () { callback(xmlhttp, false); }; xmlhttp.onerror = function(ev) { callback(xmlhttp, true); }; } //submit this request xmlhttp.send(); } function getYouTubeVideoID(url) { // Return video id or false var regExp = /^.*((youtu.be\/)|(v\/)|(\/u\/\w\/)|(embed\/)|(watch\?))\??v?=?([^#\&\?]*).*/; var match = url.match(regExp); return (match && match[7].length == 11) ? match[7] : false; } //uuid generator function from https://gist.github.com/jed/982883 function generateUUID(a){return a?(a^Math.random()*16>>a/4).toString(16):([1e7]+-1e3+-4e3+-8e3+-1e11).replace(/[018]/g,generateUUID)}