☰

Greasy Fork is available in English.

Netflix - subtitle downloader

Allows you to download subtitles from Netflix

La data de 25-11-2020. Vezi ultima versiune.

  1. // ==UserScript==
  2. // @name        Netflix - subtitle downloader
  3. // @description Allows you to download subtitles from Netflix
  4. // @license     MIT
  5. // @version     3.4.3
  6. // @namespace   tithen-firion.github.io
  7. // @include     https://www.netflix.com/*
  8. // @grant       unsafeWindow
  9. // @require     https://cdn.jsdelivr.net/gh/Stuk/jszip@579beb1d45c8d586d8be4411d5b2e48dea018c06/dist/jszip.min.js?version=3.1.5
  10. // @require     https://cdn.jsdelivr.net/gh/eligrey/FileSaver.js@283f438c31776b622670be002caf1986c40ce90c/dist/FileSaver.min.js?version=2018-12-29
  11. // ==/UserScript==
  12.  
  13. class ProgressBar { 
  14.   constructor(max) {
  15.     this.current = 0;
  16.     this.max = max;
  17.  
  18.     let container = document.querySelector('#userscript_progress_bars');
  19.     if(container === null) {
  20.       container = document.createElement('div');
  21.       container.id = 'userscript_progress_bars'
  22.       document.body.appendChild(container)
  23.       container.style
  24.       container.style.position = 'fixed';
  25.       container.style.top = 0;
  26.       container.style.left = 0;
  27.       container.style.width = '100%';
  28.       container.style.background = 'red';
  29.       container.style.zIndex = '99999999';
  30.     }
  31.  
  32.     this.progressElement = document.createElement('div');
  33.     this.progressElement.innerHTML = 'Click to stop';
  34.     this.progressElement.style.cursor = 'pointer';
  35.     this.progressElement.style.fontSize = '16px';
  36.     this.progressElement.style.textAlign = 'center';
  37.     this.progressElement.style.width = '100%';
  38.     this.progressElement.style.height = '20px';
  39.     this.progressElement.style.background = 'transparent';
  40.     this.stop = new Promise(resolve => {
  41.       this.progressElement.addEventListener('click', () => {resolve(STOP_THE_DOWNLOAD)});
  42.     });
  43.  
  44.     container.appendChild(this.progressElement);
  45.   }
  46.  
  47.   increment() {
  48.     this.current += 1;
  49.     if(this.current <= this.max) {
  50.       let p = this.current / this.max * 100;
  51.       this.progressElement.style.background = `linear-gradient(to right, green ${p}%, transparent ${p}%)`;
  52.     }
  53.   }
  54.  
  55.   destroy() {
  56.     this.progressElement.remove();
  57.   }
  58. }
  59.  
  60. const STOP_THE_DOWNLOAD = 'NETFLIX_SUBTITLE_DOWNLOADER_STOP_THE_DOWNLOAD';
  61. const MAIN_TITLE = '.player-status-main-title, .ellipsize-text>h4, .video-title>h4';
  62. const TRACK_MENU = '#player-menu-track-settings, .audio-subtitle-controller';
  63. const NEXT_EPISODE = '.player-next-episode:not(.player-hidden), .button-nfplayerNextEpisode';
  64.  
  65. const WEBVTT = 'webvtt-lssdh-ios8';
  66. const DFXP = 'dfxp-ls-sdh';
  67. const SIMPLE = 'simplesdh';
  68. const ALL_FORMATS = [WEBVTT, DFXP, SIMPLE];
  69.  
  70. const FORMAT_NAMES = {};
  71. FORMAT_NAMES[WEBVTT] = 'WebVTT';
  72. FORMAT_NAMES[DFXP] = 'DFXP/XML';
  73.  
  74. const EXTENSIONS = {};
  75. EXTENSIONS[WEBVTT] = 'vtt';
  76. EXTENSIONS[DFXP] = 'dfxp';
  77. EXTENSIONS[SIMPLE] = 'xml';
  78.  
  79. const DOWNLOAD_MENU = `<lh class="list-header">Netflix subtitle downloader</lh>
  80. <li class="list-header">Netflix subtitle downloader</li>
  81. <li class="track download">Download subs for this episode</li>
  82. <li class="track download-all">Download subs from this ep till last available</li>
  83. <li class="track force-all-lang">Force Netflix to show all languages: <span></span></li>
  84. <li class="track lang-setting">Languages to download: <span></span></li>
  85. <li class="track sub-format">Subtitle format: prefer <span></span></li>`;
  86.  
  87. const SCRIPT_CSS = `.player-timed-text-tracks, .track-list-subtitles{ border-right:1px solid #000 }
  88. .player-timed-text-tracks+.player-timed-text-tracks, .track-list-subtitles+.track-list-subtitles{ border-right:0 }
  89. .subtitle-downloader-menu { list-style:none }
  90. #player-menu-track-settings .subtitle-downloader-menu li.list-header,
  91. .audio-subtitle-controller .subtitle-downloader-menu lh.list-header{ display:none }`;
  92.  
  93. const SUB_TYPES = {
  94.   'subtitles': '',
  95.   'closedcaptions': '[cc]'
  96. };
  97.  
  98. let idOverrides = {};
  99. let zip;
  100. let subCache = {};
  101. let batch = false;
  102.  
  103. let forceSubs = localStorage.getItem('NSD_force-all-lang') !== 'false';
  104. let langs = localStorage.getItem('NSD_lang-setting') || '';
  105. let subFormat = localStorage.getItem('NSD_sub-format') || WEBVTT;
  106.  
  107. const setForceText = () => {
  108.   document.querySelector('.subtitle-downloader-menu > .force-all-lang > span').innerHTML = (forceSubs ? 'on' : 'off');
  109. };
  110. const setLangsText = () => {
  111.   document.querySelector('.subtitle-downloader-menu > .lang-setting > span').innerHTML = (langs === '' ? 'all' : langs);
  112. };
  113. const setFormatText = () => {
  114.   document.querySelector('.subtitle-downloader-menu > .sub-format > span').innerHTML = FORMAT_NAMES[subFormat];
  115. };
  116.  
  117. const toggleForceLang = () => {
  118.   forceSubs = !forceSubs;
  119.   if(forceSubs)
  120.     localStorage.removeItem('NSD_force-all-lang');
  121.   else
  122.     localStorage.setItem('NSD_force-all-lang', forceSubs);
  123.   document.location.reload();
  124. };
  125. const setLangToDownload = () => {
  126.   const result = prompt('Languages to download, comma separated. Leave empty to download all subtitles.\nExample: en,de,fr', langs);
  127.   if(result !== null) {
  128.     langs = result;
  129.     if(langs === '')
  130.       localStorage.removeItem('NSD_lang-setting');
  131.     else
  132.       localStorage.setItem('NSD_lang-setting', langs);
  133.     setLangsText();
  134.   }
  135. };
  136. const setSubFormat = () => {
  137.   if(subFormat === WEBVTT) {
  138.     localStorage.setItem('NSD_sub-format', DFXP);
  139.     subFormat = DFXP;
  140.   }
  141.   else {
  142.     localStorage.removeItem('NSD_sub-format');
  143.     subFormat = WEBVTT;
  144.   }
  145.   setFormatText();
  146. };
  147.  
  148. const asyncSleep = (seconds, value) => new Promise(resolve => {
  149.   window.setTimeout(resolve, seconds * 1000, value);
  150. });
  151.  
  152. const popRandomElement = arr => {
  153.   return arr.splice(arr.length * Math.random() << 0, 1)[0];
  154. };
  155.  
  156. // get show name or full name with episode number
  157. const __getTitle = full => {
  158.   if(typeof full === 'undefined')
  159.     full = true;
  160.   const titleElement = document.querySelector(MAIN_TITLE);
  161.   if(titleElement === null)
  162.     return null;
  163.   const title = [titleElement.textContent.replace(/[:*?"<>|\\\/]+/g, '_').replace(/ /g, '.')];
  164.   if(full) {
  165.     const episodeElement = titleElement.nextElementSibling;
  166.     if(episodeElement) {
  167.       const m = episodeElement.textContent.match(/^[^\d]*(\d+)[^\d]+(\d+)[^\d]*$/);
  168.       if(m && m.length == 3) {
  169.         title.push(`S${m[1].padStart(2, '0')}E${m[2].padStart(2, '0')}`);
  170.       }
  171.       else {
  172.         title.push(episodeElement.textContent.trim().replace(/[:*?"<>|\\\/]+/g, '_').replace(/ /g, '.'));
  173.       }
  174.     }
  175.     title.push('WEBRip.Netflix');
  176.   }
  177.   return title.join('.');
  178. };
  179. // helper function, periodically checking for the title and resolving promise if found
  180. const _getTitle = (full, resolve) => {
  181.   const title = __getTitle(full);
  182.   if(title === null)
  183.     window.setTimeout(_getTitle, 200, full, resolve);
  184.   else
  185.     resolve(title);
  186. };
  187. // promise of a title
  188. const getTitle = full => new Promise(resolve => {
  189.   _getTitle(full, resolve);
  190. });
  191.  
  192. const processSubInfo = async result => {
  193.   const tracks = result.timedtexttracks;
  194.   const titleP = getTitle();
  195.   const subs = {};
  196.   for(const track of tracks) {
  197.     if(track.isNoneTrack)
  198.       continue;
  199.  
  200.     let type = SUB_TYPES[track.rawTrackType];
  201.     if(typeof type === 'undefined')
  202.       type = `[${track.rawTrackType}]`;
  203.     const lang = track.language + type + (track.isForcedNarrative ? '-forced' : '');
  204.  
  205.     const formats = {};
  206.     for(let format of ALL_FORMATS) {
  207.       if(typeof track.ttDownloadables[format] !== 'undefined')
  208.         formats[format] = [Object.values(track.ttDownloadables[format].downloadUrls), EXTENSIONS[format]];
  209.     }
  210.  
  211.     if(Object.keys(formats).length > 0)
  212.       subs[lang] = formats;
  213.   }
  214.   subCache[result.movieId] = {titleP, subs};
  215.  
  216.   if(batch) {
  217.     downloadAll();
  218.   }
  219. };
  220.  
  221. const getSubsFromCache = () => {
  222.   const id = window.location.pathname.split('/').pop();
  223.   if(subCache.hasOwnProperty(id))
  224.     return subCache[id];
  225.  
  226.   let newID = undefined;
  227.   try {
  228.     newID = unsafeWindow.netflix.falcorCache.videos[id].current.value[1];
  229.   }
  230.   catch(ignore) {}
  231.   if(typeof newID !== 'undefined' && subCache.hasOwnProperty(newID))
  232.     return subCache[newID];
  233.  
  234.   newID = idOverrides[id];
  235.   if(typeof newID !== 'undefined' && subCache.hasOwnProperty(newID))
  236.     return subCache[newID];
  237.  
  238.   alert("Couldn't find subs, try refreshing the page.");
  239.   throw '';
  240. };
  241.  
  242. const pickFormat = formats => {
  243.   const preferred = ALL_FORMATS.slice();
  244.   if(subFormat === DFXP)
  245.     preferred.push(preferred.shift());
  246.  
  247.   for(let format of preferred) {
  248.     if(typeof formats[format] !== 'undefined')
  249.       return formats[format];
  250.   }
  251. };
  252.  
  253.  
  254. const _save = async (_zip, title) => {
  255.   const content = await _zip.generateAsync({type:'blob'});
  256.   saveAs(content, title + '.zip');
  257. };
  258.  
  259. const _download = async _zip => {
  260.   const showTitle = getTitle(false);
  261.   const {titleP, subs} = getSubsFromCache();
  262.   const downloaded = [];
  263.  
  264.   let filteredLangs;
  265.   if(langs === '')
  266.     filteredLangs = Object.keys(subs);
  267.   else {
  268.     const regularExpression = new RegExp(
  269.       '^(' + langs
  270.         .replace(/\[/g, '\\[')
  271.         .replace(/\]/g, '\\]')
  272.         .replace(/\-/g, '\\-')
  273.         .replace(/\s/g, '')
  274.         .replace(/,/g, '|')
  275.       + ')'
  276.     );
  277.     filteredLangs = [];
  278.     for(const lang of Object.keys(subs)) {
  279.       if(lang.match(regularExpression))
  280.         filteredLangs.push(lang);
  281.     }
  282.   }
  283.  
  284.   const progress = new ProgressBar(filteredLangs.length);
  285.   let stop = false;
  286.   for(const lang of filteredLangs) {
  287.     const [urls, extension] = pickFormat(subs[lang]);
  288.     while(urls.length > 0) {
  289.       let url = popRandomElement(urls);
  290.       const resultPromise = fetch(url, {mode: "cors"});
  291.       let result;
  292.       try {
  293.         // Promise.any isn't supported in all browsers, use Promise.race instead
  294.         result = await Promise.race([resultPromise, progress.stop, asyncSleep(30, STOP_THE_DOWNLOAD)]);
  295.       }
  296.       catch(e) {
  297.         // the only promise that can be rejected is the one from fetch
  298.         // if that happens we want to stop the download anyway
  299.         result = STOP_THE_DOWNLOAD;
  300.       }
  301.       if(result === STOP_THE_DOWNLOAD) {
  302.         stop = true;
  303.         break;
  304.       }
  305.       progress.increment();
  306.       const data = await result.text();
  307.       if(data.length > 0) {
  308.         downloaded.push({lang, data, extension});
  309.         break;
  310.       }
  311.     }
  312.     if(stop)
  313.       break;
  314.   }
  315.   const title = await titleP;
  316.  
  317.   downloaded.forEach(x => {
  318.     const {lang, data, extension} = x;
  319.     _zip.file(`${title}.${lang}.${extension}`, data);
  320.   });
  321.  
  322.   if(await Promise.race([progress.stop, {}]) === STOP_THE_DOWNLOAD)
  323.     stop = true;
  324.   progress.destroy();
  325.  
  326.   return [await showTitle, stop];
  327. };
  328.  
  329. const downloadThis = async () => {
  330.   const _zip = new JSZip();
  331.   const [showTitle, stop] = await _download(_zip);
  332.   _save(_zip, showTitle);
  333. };
  334.  
  335. const downloadAll = async () => {
  336.   zip = zip || new JSZip();
  337.   batch = true;
  338.   const [showTitle, stop] = await _download(zip);
  339.   const nextEp = document.querySelector(NEXT_EPISODE);
  340.   if(!stop && nextEp)
  341.     nextEp.click();
  342.   else {
  343.     await _save(zip, showTitle);
  344.     zip = undefined;
  345.     batch = false;
  346.   }
  347. };
  348.  
  349. const processMessage = e => {
  350.   const override = e.detail.id_override;
  351.   if(typeof override !== 'undefined')
  352.     idOverrides[override[0]] = override[1];
  353.   else
  354.     processSubInfo(e.detail);
  355. }
  356.  
  357. const injection = () => {
  358.   const WEBVTT = 'webvtt-lssdh-ios8';
  359.   const MANIFEST_URL = "manifest";
  360.   const forceSubs = localStorage.getItem('NSD_force-all-lang') !== 'false';
  361.  
  362.   // hijack JSON.parse and JSON.stringify functions
  363.   ((parse, stringify) => {
  364.     JSON.parse = function (text) {
  365.       const data = parse(text);
  366.       if (data && data.result && data.result.timedtexttracks && data.result.movieId) {
  367.         window.dispatchEvent(new CustomEvent('netflix_sub_downloader_data', {detail: data.result}));
  368.       }
  369.       return data;
  370.     };
  371.     JSON.stringify = function (data) {
  372.       if (data && typeof data.url === 'string' && data.url.indexOf(MANIFEST_URL) > -1) {
  373.         for (let v of Object.values(data)) {
  374.           try {
  375.             if (v.profiles)
  376.               v.profiles.unshift(WEBVTT);
  377.             if (v.showAllSubDubTracks != null && forceSubs)
  378.               v.showAllSubDubTracks = true;
  379.           }
  380.           catch (e) {
  381.             if (e instanceof TypeError)
  382.               continue;
  383.             else
  384.               throw e;
  385.           }
  386.         }
  387.       }
  388.       if(data && typeof data.movieId === 'number') {
  389.         try {
  390.           let videoId = data.params.sessionParams.uiplaycontext.video_id;
  391.           if(typeof videoId === 'number' && videoId !== data.movieId)
  392.             window.dispatchEvent(new CustomEvent('netflix_sub_downloader_data', {detail: {id_override: [videoId, data.movieId]}}));
  393.         }
  394.         catch(ignore) {}
  395.       }
  396.       return stringify(data);
  397.     };
  398.   })(JSON.parse, JSON.stringify);
  399. }
  400.  
  401. window.addEventListener('netflix_sub_downloader_data', processMessage, false);
  402.  
  403. // inject script
  404. const sc = document.createElement('script');
  405. sc.innerHTML = '(' + injection.toString() + ')()';
  406. document.head.appendChild(sc);
  407. document.head.removeChild(sc);
  408.  
  409. // add CSS style
  410. const s = document.createElement('style');
  411. s.innerHTML = SCRIPT_CSS;
  412. document.head.appendChild(s);
  413.  
  414. // add menu when it's not there
  415. const observer = new MutationObserver(function(mutations) {
  416.   mutations.forEach(function(mutation) {
  417.     mutation.addedNodes.forEach(function(node) {
  418.       if(node.nodeName.toUpperCase() == 'DIV') {
  419.         let trackMenu = (node.parentNode || node).querySelector(TRACK_MENU);
  420.         if(trackMenu !== null && trackMenu.querySelector('.subtitle-downloader-menu') === null) {
  421.           let ol = document.createElement('ol');
  422.           ol.setAttribute('class', 'subtitle-downloader-menu player-timed-text-tracks track-list track-list-subtitles');
  423.           ol.innerHTML = DOWNLOAD_MENU;
  424.           trackMenu.appendChild(ol);
  425.           ol.querySelector('.download').addEventListener('click', downloadThis);
  426.           ol.querySelector('.download-all').addEventListener('click', downloadAll);
  427.           ol.querySelector('.force-all-lang').addEventListener('click', toggleForceLang);
  428.           ol.querySelector('.lang-setting').addEventListener('click', setLangToDownload);
  429.           ol.querySelector('.sub-format').addEventListener('click', setSubFormat);
  430.           setForceText();
  431.           setLangsText();
  432.           setFormatText();
  433.         }
  434.       }
  435.     });
  436.   });
  437. });
  438. observer.observe(document.body, { childList: true, subtree: true });