|
1 | 1 | 'use strict'
|
2 | 2 |
|
3 |
| -const getSocialVideoUrl = require('get-social-video-url') |
4 | 3 | const { isUrl } = require('@metascraper/helpers')
|
5 |
| -const createBrowserless = require('browserless') |
| 4 | +const youtubedl = require('youtube-dl') |
| 5 | +const { promisify } = require('util') |
| 6 | +const path = require('path') |
6 | 7 |
|
7 |
| -const PREFERRED_VIDEO_QUALITY = [ |
8 |
| - 'sd', |
9 |
| - 'hd', |
10 |
| - 'mobile' |
11 |
| -] |
| 8 | +const getVideoInfo = promisify(youtubedl.getInfo) |
12 | 9 |
|
13 |
| -const getVideoUrl = async ({ |
14 |
| - browserless, |
15 |
| - url, |
16 |
| - preferredVideoQuality = PREFERRED_VIDEO_QUALITY |
17 |
| -}) => { |
18 |
| - const videoQualities = await getSocialVideoUrl({ url, browserless }) |
19 |
| - const videoQuality = preferredVideoQuality.find(videoQuality => isUrl(videoQualities[videoQuality])) |
20 |
| - return videoQualities[videoQuality] |
21 |
| -} |
| 10 | +/** |
| 11 | + * Get a Video source quality not too high |
| 12 | + */ |
| 13 | +const getVideoUrl = ({formats}) => { |
| 14 | + const urls = formats |
| 15 | + .map(item => item.url) |
| 16 | + .filter(url => path.extname(url).startsWith('.mp4')) |
22 | 17 |
|
23 |
| -module.exports = (opts = {}) => { |
24 |
| - const { launchOpts = {}, preferredVideoQuality } = opts |
| 18 | + const index = Math.round(urls.length / 2) - 1 |
| 19 | + return urls[index] |
| 20 | +} |
25 | 21 |
|
26 |
| - const fn = async ({url}) => { |
27 |
| - const browserless = await createBrowserless(launchOpts) |
28 |
| - const result = await getVideoUrl({url, browserless, preferredVideoQuality}) |
29 |
| - const browserInstance = await browserless.browser |
30 |
| - await browserInstance.close() |
31 |
| - return result |
| 22 | +const getVideoProvider = async url => { |
| 23 | + try { |
| 24 | + const info = await getVideoInfo(url) |
| 25 | + const videoUrl = getVideoUrl(info) |
| 26 | + return isUrl(videoUrl) && videoUrl |
| 27 | + } catch (err) { |
| 28 | + return false |
32 | 29 | }
|
| 30 | +} |
33 | 31 |
|
34 |
| - return { video: fn } |
| 32 | +module.exports = () => { |
| 33 | + return { |
| 34 | + video: ({url}) => getVideoProvider(url) |
| 35 | + } |
35 | 36 | }
|
36 | 37 |
|
37 | 38 | module.exports.getVideoUrl = getVideoUrl
|
0 commit comments