File: js/background.js

Recommend this page to a friend!
  Classes of Andras Toth  >  Subtitle Downloader  >  js/background.js  >  Download  
File: js/background.js
Role: Example script
Content type: text/plain
Description: Example script
Class: Subtitle Downloader
Chrome extension to download videos and subtitles
Author: By
Last change: Update of js/background.js
Date: 1 year ago
Size: 8,345 bytes
 

Contents

Class file image Download
(function() { 'use strict'; var videosTabId = null, state = 0, isCollectEnable = false, data = []; chrome.browserAction.onClicked.addListener(function(tab) { switch (++state) { case 1: isCollectEnable = true; setBadgeText('css/images/icon48.png', ''); videosTabId = null; break; case 2: isCollectEnable = true; setBadgeText('css/images/icon48-blue.png', ''); chrome.tabs.create({ url: 'videos.html' }, function(tab) { videosTabId = tab.id; updateVideoTab(); }); break; case 3: setBadgeText('css/images/icon48-gray.png', ''); chrome.tabs.remove(videosTabId, null); break; } }); function removeUrlParameters(url, parameters) { var arr = url.split('&'); return arr.filter(function(u) { return !parameters.find(function(p) { return u.indexOf(p) == 0 }); }).join('&'); } function updateUrlParameter(url, param, value) { var regex = new RegExp('(' + param + '=)[^\&]+'); return url.replace(regex, '$1' + value); } function getFilename(url) { try { var name = url.split(/(\\|\/|=|\?)/g).find(function(item) { return (/\.(mp4|srt|vtt|avi|webm|flv$|mkv|ogg|mp3|wav)/gi).test(item) }); return name ? name : 'unknown.mp4'; } catch (e) { return "unknown.mp4"; } } function updateVideoTab(dt) { setTimeout(function() { chrome.extension.sendMessage({ order: 'SendDataToVideoTab', data: dt ? [dt] : data }, null); if (!dt) { data = []; } }, 300); } function onMessage(request, sender, sendResponse) { if (request.order == 'setContextMenu') { chrome.contextMenus.removeAll(function() { chrome.contextMenus.create({ "title": "Download video only", "contexts": ["video"], "onclick": clickHandler.bind(null, -1) }); request.tracks.forEach(function(trk, index) { chrome.contextMenus.create({ "title": "Download video and " + (trk.language ? trk.language : trk.label) + " subtitle", "contexts": ["video"], "onclick": clickHandler.bind(null, index) }); }); }); } if (request.order == 'DownloadVideoAndSubResponse') { var lines = request.lines; chrome.downloads.download({ url: request.url, filename: getFilename(request.url), saveAs: true }, function(id) { chrome.downloads.onChanged.addListener(downloadStarted.bind(null, id, lines)); }); } } function startSubDownload(filename, lines) { if (lines && lines.length) { var blob = window.URL.createObjectURL(new Blob([lines.join('\n')], { type: 'text/plain;charset=utf-8;' })); var fileName = filename.split('\\').reverse()[0]; chrome.downloads.download({ url: blob, filename: fileName.split('.')[0].concat('.srt'), saveAs: true }); } } function downloadStarted(id, lines, downloadItem) { if (!downloadItem.error && id === downloadItem.id && downloadItem.filename && downloadItem.filename.current) { chrome.downloads.onChanged.removeListener(downloadStarted); startSubDownload(downloadItem.filename.current, lines); } } function clickHandler(index, info, tab) { var lines; chrome.tabs.sendMessage(tab.id, { order: "DownloadVideoAndSub", index: index }, null); } function setBadgeText(icon, txt) { chrome.browserAction.setIcon({ path: icon }); chrome.browserAction.setBadgeText({ text: txt.toString() }); } function arrayContains(array, item) { return array.length != 0 && array.find(function(elem) { return elem.url.indexOf(item) > -1 }); } function onBeforeRequest(details) { if (isCollectEnable || videosTabId) { if (!(/\.(js)/gi).test(details.url) && (/\.(mp4|srt|vtt|avi|webm|flv$|mkv)/gi).test(details.url)) { if ((/\.(vtt|srt)/gi).test(details.url)) { if (!arrayContains(data, details.url)) { data.push({ date: new Date().getTime(), type: 'subtitle', url: details.url }); if (videosTabId) { updateVideoTab(data[data.length - 1]); } setBadgeText('css/images/icon48-red.png', data.length); } } else { if (!arrayContains(data, details.url)) { data.push({ date: new Date().getTime(), type: 'video', url: details.url }); if (videosTabId) { updateVideoTab(data[data.length - 1]); } setBadgeText('css/images/icon48-red.png', data.length); } } } else if (!(/\.(js)/gi).test(details.url) && (/\.(mp3|ogg|wav)/gi).test(details.url)) { if (!arrayContains(data, details.url)) { data.push({ date: new Date().getTime(), type: 'audio', url: details.url }); if (videosTabId) { updateVideoTab(data[data.length - 1]); } setBadgeText('css/images/icon48-red.png', data.length); } } else if (!(/\.(js)/gi).test(details.url) && (/\.(googlevideo.com)/gi).test(details.url) && (/&rbuf/gi).test(details.url)) { var url = removeUrlParameters(details.url, ['range=', 'rn=', 'rbuf=', 'cpn=', 'c=', 'cver=']); if (!arrayContains(data, url)) { data.push({ date: new Date().getTime(), type: (/mime=audio/gi).test(url) ? 'audio' : 'video', url: url }); if (videosTabId) { updateVideoTab(data[data.length - 1]); } setBadgeText('css/images/icon48-red.png', data.length); } } } } chrome.extension.onMessage.addListener(onMessage); chrome.webRequest.onBeforeRequest.addListener(onBeforeRequest, { urls: ['<all_urls>'] }); chrome.tabs.onRemoved.addListener(function(tabid, removed) { if (tabid == videosTabId) { setBadgeText('css/images/icon48-gray.png', ''); isCollectEnable = false; state = 0; videosTabId = null; data = []; } }); chrome.webRequest.onHeadersReceived.addListener(function(info) { if (info.tabId == videosTabId) { var headers = info.responseHeaders; for (var i = headers.length - 1; i >= 0; --i) { var header = headers[i].name.toLowerCase(); if (header == 'x-frame-options' || header == 'frame-options') { headers.splice(i, 1); } } return { responseHeaders: headers }; } }, { urls: ['*://*/*'], types: ['sub_frame'] }, ['blocking', 'responseHeaders']); })();