-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathserver.js
More file actions
69 lines (49 loc) · 1.66 KB
/
server.js
File metadata and controls
69 lines (49 loc) · 1.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
'use strict';
let http = require('http');
let https = require('https');
let fs = require('fs');
let express = require('express');
let cheerio = require('cheerio');
let youtubeStream = require('youtube-audio-stream');
let app = express();
app.server = http.createServer(app);
app.use(express.static('static'));
app.get('/', (req, res) => {
res.sendFile(`${__dirname}/static/index.html`);
});
app.get('/query', (req, res) => {
console.log(`Querying for '${req.query.term}'...`);
https.get('https://www.youtube.com/results?search_query=' + req.query.term, (resp) => {
let body = '';
resp.on('data', (chunk) => {
body += chunk;
});
resp.on('end', () => {
let $ = cheerio.load(body);
let videoLink = null;
let videoTitle = null;
$('a[rel="spf-prefetch"]').each((i, elem) => {
if ($(elem).attr('href').toString().indexOf('ad') === -1) {
videoLink = `https://www.youtube.com${$(elem).attr('href')}`;
videoTitle = $(elem).attr('title');
console.log(`Found match: '${videoTitle}'! Redirecting...`);
res.redirect('/download?url=' + videoLink + '&title=' + videoTitle);
return false;
}
});
});
});
});
app.get('/download', (req, res) => {
console.log(`Downloading '${req.query.title}'...`)
let stream = fs.createWriteStream('static/resources/music.mp3');
youtubeStream(req.query.url)
.pipe(stream);
stream.on('finish', () => {
console.log('Finished download!')
res.download(`static/resources/music.mp3`, req.query.title);
});
});
app.server.listen(process.env.PORT || 3000, () => {
console.log(`Server running @: http://127.0.0.1:${(process.env.PORT) ? process.env.port : 3000}`);
});