dany/clapper
1
0
Fork 0
mirror of https://github.com/Rafostar/clapper.git synced 2025-08-30 16:02:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

YT: pass download info using object intead of array

Browse Source
This commit is contained in:
Rafał Dzięgiel
2021-03-16 10:20:01 +01:00
parent 4f46a7eaa8
commit e0a3ef78db
1 changed files with 34 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/youtube.js
View File

@@ -52,15 +52,16 @@ var YouTubeClient = GObject.registerClass({
debug(`obtaining YouTube video info: ${videoId}`); debug(`obtaining YouTube video info: ${videoId}`);
this.downloadingVideoId = videoId; this.downloadingVideoId = videoId;
let [info, isAborted] = await this._getInfoPromise(videoId).catch(debug); let result = await this._getInfoPromise(videoId).catch(debug);
if(!info) { if(!result || !result.data) {
if(isAborted) if(result && result.isAborted)
return reject(new Error('download aborted')); return reject(new Error('download aborted'));
debug(`failed, remaining tries: ${tries}`); debug(`failed, remaining tries: ${tries}`);
continue; continue;
} }
const info = result.data;
const invalidInfoMsg = ( const invalidInfoMsg = (
!info.playabilityStatus !info.playabilityStatus
@@ -96,15 +97,16 @@ var YouTubeClient = GObject.registerClass({
debug('using remembered decipher actions'); debug('using remembered decipher actions');
else { else {
const embedUri = `https://www.youtube.com/embed/${videoId}`; const embedUri = `https://www.youtube.com/embed/${videoId}`;
const [body, isAbortedBody] = result = await this._downloadDataPromise(embedUri).catch(debug);
await this._downloadDataPromise(embedUri).catch(debug);
if(isAbortedBody) if(result && result.isAborted)
break; break;
if(!body) else if(!result || !result.data) {
debug(new Error('could not download embed body'));
continue; continue;
}
const ytPath = body.match(/(?<=jsUrl\":\").*?(?=\")/gs)[0]; const ytPath = result.data.match(/(?<=jsUrl\":\").*?(?=\")/gs)[0];
if(!ytPath) { if(!ytPath) {
debug(new Error('could not find YouTube player URI')); debug(new Error('could not find YouTube player URI'));
break; break;
@@ -124,12 +126,15 @@ var YouTubeClient = GObject.registerClass({
actions = await this._getCacheFileActionsPromise(ytId).catch(debug); actions = await this._getCacheFileActionsPromise(ytId).catch(debug);
if(!actions) { if(!actions) {
const [pBody, isAbortedPlayer] = result = await this._downloadDataPromise(ytUri).catch(debug);
await this._downloadDataPromise(ytUri).catch(debug);
if(!pBody || isAbortedPlayer) { if(result && result.isAborted)
debug(new Error('could not download player body'));
break; break;
else if(!result || !result.data) {
debug(new Error('could not download player body'));
continue;
} }
actions = YTDL.sig.extractActions(pBody); actions = YTDL.sig.extractActions(pBody);
if(actions) { if(actions) {
debug('deciphered'); debug('deciphered');
@@ -207,7 +212,10 @@ var YouTubeClient = GObject.registerClass({
{ {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
const message = Soup.Message.new('GET', url); const message = Soup.Message.new('GET', url);
let data = ''; const result = {
data: '',
isAborted: false,
};
const chunkSignal = message.connect('got-chunk', (msg, chunk) => { const chunkSignal = message.connect('got-chunk', (msg, chunk) => {
debug(`got chunk of data, length: ${chunk.length}`); debug(`got chunk of data, length: ${chunk.length}`);
@@ -215,7 +223,7 @@ var YouTubeClient = GObject.registerClass({
const chunkData = chunk.get_data(); const chunkData = chunk.get_data();
if(!chunkData) return; if(!chunkData) return;
data += (chunkData instanceof Uint8Array) result.data += (chunkData instanceof Uint8Array)
? ByteArray.toString(chunkData) ? ByteArray.toString(chunkData)
: chunkData; : chunkData;
}); });
@@ -226,16 +234,16 @@ var YouTubeClient = GObject.registerClass({
debug('got message response'); debug('got message response');
const statusCode = msg.status_code; const statusCode = msg.status_code;
if(statusCode === 200)
return resolve(result);
/* Internal Soup codes mean download aborted /* Internal Soup codes mean download aborted
* or some other error that cannot be handled * or some other error that cannot be handled
* and we do not want to retry in such case */ * and we do not want to retry in such case */
if(statusCode < 10) if(statusCode < 10)
return resolve([null, true]); result.isAborted = true;
if(statusCode !== 200) return reject(new Error(`response code: ${statusCode}`));
return reject(new Error(`response code: ${statusCode}`));
resolve([data, false]);
}); });
}); });
} }
@@ -269,13 +277,13 @@ var YouTubeClient = GObject.registerClass({
].join('&'); ].join('&');
const url = `https://www.youtube.com/get_video_info?${query}`; const url = `https://www.youtube.com/get_video_info?${query}`;
this._downloadDataPromise(url).then(res => { this._downloadDataPromise(url).then(result => {
if(res[1]) if(result.isAborted)
return resolve([null, true]); return resolve(result);
debug('parsing video info JSON'); debug('parsing video info JSON');
const gstUri = Gst.Uri.from_string('?' + res[0]); const gstUri = Gst.Uri.from_string('?' + result.data);
if(!gstUri) if(!gstUri)
return reject(new Error('could not convert query to URI')); return reject(new Error('could not convert query to URI'));
@@ -294,7 +302,9 @@ var YouTubeClient = GObject.registerClass({
return reject(new Error('could not parse video info JSON')); return reject(new Error('could not parse video info JSON'));
debug('successfully parsed video info JSON'); debug('successfully parsed video info JSON');
resolve([info, false]); result.data = info;
resolve(result);
}) })
.catch(err => reject(err)); .catch(err => reject(err));
}); });