Bazsalanszky/destreamer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added a simulation flag to only extract video tile and urls without any file downloaded

Browse source 
Removede an outdated argv check because all arguments are now parsed with yargs
This commit is contained in:
Luca 2020-03-26 16:44:28 +01:00
parent 31fb4284ed
commit 047e164a86
1 changed files with 59 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

92
destreamer.ts
View file

@ -12,20 +12,36 @@ import sanitize = require('sanitize-filename')
const args: string[] = process.argv.slice(2); // TODO: Remove this const args: string[] = process.argv.slice(2); // TODO: Remove this
const argv = yargs.options({ const argv = yargs.options({
videoUrls: { type: 'array', demandOption: true }, videoUrls: { type: 'array', demandOption: true },
username: { type: 'string', demandOption: true }, username: { type: 'string', demandOption: true },
outputDirectory: { type: 'string', default: 'videos' }, outputDirectory: { type: 'string', default: 'videos' },
format: { alias:"f", format: {
describe: 'Expose youtube-dl --format option, for details see\n https://github.com/ytdl-org/youtube-dl/blob/master/README.md#format-selection', alias:"f",
type:'string', describe: 'Expose youtube-dl --format option, for details see\n https://github.com/ytdl-org/youtube-dl/blob/master/README.md#format-selection',
demandOption: false type:'string',
} demandOption: false
},
simulate: {
alias: "s",
describe: "If this is set to true no video will be downloaded and the script will log the video info (default: false)",
type: "boolean",
default: false,
demandOption: false
}
}).argv; }).argv;
console.info('Video URLs: %s', argv.videoUrls); if (argv.simulate){
console.info('Username: %s', argv.username); console.info('Video URLs: %s', argv.videoUrls);
console.info('Output Directory: %s', argv.outputDirectory); console.info('Username: %s', argv.username);
console.info('Video/Audio Quality: %s', argv.format); term.blue("There will be no video downloaded, it's only a simulation \n")
} else {
console.info('Video URLs: %s', argv.videoUrls);
console.info('Username: %s', argv.username);
console.info('Output Directory: %s', argv.outputDirectory);
console.info('Video/Audio Quality: %s', argv.format);
}
function sanityChecks() { function sanityChecks() {
try { try {
@ -52,11 +68,8 @@ function sanityChecks() {
fs.mkdirSync(argv.outputDirectory); fs.mkdirSync(argv.outputDirectory);
} }
if (args[0] == null || args[0].length < 10) { /* Removed check on the first argoumenti not being null or
console.error('Pass in video URL as first argument:\n' + longer than 10 since we have yargs now */
'Example: npm start https://www.microsoftstream.com/video/6f1a382b-e20c-44c0-98fc-5608286e48bc\n');
process.exit(-1);
}
} }
async function rentVideoForLater(videoUrls: string[], username: string, outputDirectory: string) { async function rentVideoForLater(videoUrls: string[], username: string, outputDirectory: string) {
@ -92,20 +105,24 @@ async function rentVideoForLater(videoUrls: string[], username: string, outputDi
await sleep(4000); await sleep(4000);
console.log('Looking up AMS stream locator...'); console.log('Looking up AMS stream locator...');
// let amp: any;
let document: any; let document: any;
const amsUrl = await page.evaluate( const amsUrl = await page.evaluate(
// maybe there should be some check in case the url fetch fails // maybe there should be some check in case the url fetch fails
() => { return document?.querySelector(".azuremediaplayer")?.player?.cache_?.src; } () => {
return document?.querySelector(".azuremediaplayer")?.player?.cache_?.src;
}
); );
// console.log(`Video url is: ${amsUrl}`); // console.log(`Video url is: ${amsUrl}`);
console.log('Fetching title'); console.log('Fetching title...');
let title = await page.evaluate( let title = await page.evaluate(
// Using optional chaining to return handle null case, generating default name // Using optional chaining to return handle null case, generating default name
() => { return document?.querySelector(".title")?.textContent?.trim() ?? () => {
`Video${videoUrls.indexOf(videoUrl)}`; } return document?.querySelector(".title")?.textContent?.trim() ??
`Video${videoUrls.indexOf(videoUrl)}`;
}
); );
// Implemented sanitize-filename as suggested in issue #11 // Implemented sanitize-filename as suggested in issue #11
@ -114,22 +131,31 @@ async function rentVideoForLater(videoUrls: string[], username: string, outputDi
if (title == "") if (title == "")
title = `Video${videoUrls.indexOf(videoUrl)}` title = `Video${videoUrls.indexOf(videoUrl)}`
console.log(`Video title is: ${title}`); //console.log(`Video title is: ${title}`);
console.log('Constructing HLS URL...'); console.log('Constructing HLS URL...');
const hlsUrl = amsUrl.substring(0, amsUrl.lastIndexOf('/')) + '/manifest(format=m3u8-aapl)'; const hlsUrl = amsUrl.substring(0, amsUrl.lastIndexOf('/')) + '/manifest(format=m3u8-aapl)';
console.log('Spawning youtube-dl with cookie and HLS URL...'); // If the simulate flag is true skip the download
let format = '' if (!argv.simulate) {
if (argv.format) { console.log('Spawning youtube-dl with cookie and HLS URL...');
format = `-f "${argv.format}"` let format = ''
if (argv.format) {
format = `-f "${argv.format}"`
}
const youtubedlCmd = 'youtube-dl --no-call-home --no-warnings ' + format +
` --output "${outputDirectory}/${title}.mp4" --add-header Cookie:"${cookie}" "${hlsUrl}"`
// console.log(`\n\n[DEBUG] Invoking youtube-dl: ${youtubedlCmd}\n\n`);
var result = execSync(youtubedlCmd, { stdio: 'inherit' });
} else {
// Logging the video info
term.blue("Video title is: ")
console.log(`${title}`)
term.blue("Video url is: ")
console.log(`${hlsUrl}`)
} }
const youtubedlCmd = 'youtube-dl --no-call-home --no-warnings ' + format +
` --output "${outputDirectory}/${title}.mp4" --add-header Cookie:"${cookie}" "${hlsUrl}"`
// console.log(`\n\n[DEBUG] Invoking youtube-dl: ${youtubedlCmd}\n\n`);
var result = execSync(youtubedlCmd, { stdio: 'inherit' });
} }
console.log("At this point Chrome's job is done, shutting it down..."); console.log("At this point Chrome's job is done, shutting it down...");