fix(youtube): YouTube changed it back, check for both cases

fent · Jul 12, 2019 · c32cbc9 · c32cbc9
1 parent 39f16e2
commit c32cbc9
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/background/sources/videos/youtube.js b/background/sources/videos/youtube.js
@@ -68,11 +68,15 @@ export default {
  data: { 'flow': 2 },
  responseType: 'text',
  });
- const key = 'window["ytInitialData"] = JSON.parse("';
- let response = body;
- response = response.slice(response.indexOf(key) + key.length);
- response = response.slice(0, response.indexOf('");\n'));
- response = response.replace(/\\([\\"])/g, '$1');
+ let match = /window\["ytInitialData"\] = (JSON\.parse\(")?(.+?)(?:"\))?;\n/
+ .exec(body);
+ if (!match) {
+ throw Error('Unable to find youtube data');
+ }
+ let response = match[2];
+ if (match[1]) {
+ response = response.replace(/\\([\\"])/g, '$1');
+ }
  try {
  response = JSON.parse(response);
  } catch (err) {