90 lines
2.7 KiB
TypeScript
90 lines
2.7 KiB
TypeScript
import { get } from 'https';
|
|
import { rmSync, createWriteStream, chmod } from 'fs';
|
|
import { join } from 'path';
|
|
|
|
// endpoint to github API
|
|
const options = {
|
|
hostname: 'api.github.com',
|
|
path: '/repos/yt-dlp/yt-dlp/releases/latest',
|
|
headers: {
|
|
'User-Agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:88.0) Gecko/20100101 Firefox/88.0'
|
|
},
|
|
method: 'GET',
|
|
port: 443,
|
|
}
|
|
|
|
/**
|
|
* Build the binary url based on the release tag
|
|
* @param {string} release yt-dlp GitHub release tag
|
|
* @returns {*} the fetch options with the correct tag and headers
|
|
*/
|
|
function buildDonwloadOptions(release) {
|
|
return {
|
|
hostname: 'github.com',
|
|
path: `/yt-dlp/yt-dlp/releases/download/${release}/yt-dlp`,
|
|
headers: {
|
|
'User-Agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:88.0) Gecko/20100101 Firefox/88.0'
|
|
},
|
|
method: 'GET',
|
|
port: 443,
|
|
}
|
|
}
|
|
|
|
/**
|
|
* gets the yt-dlp latest binary URL from GitHub API
|
|
*/
|
|
async function update() {
|
|
// ensure that the binary has been removed
|
|
try {
|
|
rmSync(join(__dirname, '..', 'core', 'yt-dlp'))
|
|
}
|
|
catch (e) {
|
|
console.log('file not found!')
|
|
}
|
|
// body buffer
|
|
let chunks = []
|
|
get(options, res => {
|
|
// push the http packets chunks into the buffer
|
|
res.on('data', chunk => {
|
|
chunks.push(chunk)
|
|
});
|
|
// the connection has ended so build the body from the buffer
|
|
// parse it as a JSON and get the tag_name
|
|
res.on('end', () => {
|
|
const buffer = Buffer.concat(chunks)
|
|
const release = JSON.parse(buffer.toString())['tag_name']
|
|
console.log('The latest release is:', release)
|
|
// invoke the binary downloader
|
|
downloadBinary(buildDonwloadOptions(release))
|
|
})
|
|
})
|
|
}
|
|
/**
|
|
* Utility that Pipes the latest binary to a file
|
|
* @param {string} url yt-dlp GitHub release url
|
|
*/
|
|
function downloadBinary(url) {
|
|
get(url, res => {
|
|
// if it is a redirect follow the url
|
|
if (res.statusCode === 301 || res.statusCode === 302) {
|
|
return downloadBinary(res.headers.location)
|
|
}
|
|
let bin = createWriteStream(join(__dirname, '..', 'core', 'yt-dlp'))
|
|
res.pipe(bin)
|
|
// once the connection has ended make the file executable
|
|
res.on('end', () => {
|
|
chmod(join(__dirname, '..', 'core', 'yt-dlp'), 0o775, err => {
|
|
err ? console.error('failed updating!') : console.log('done!')
|
|
})
|
|
})
|
|
})
|
|
}
|
|
/**
|
|
* Invoke the yt-dlp update procedure
|
|
* @param {Socket} socket the current connection socket
|
|
*/
|
|
export function ytdlpUpdater(socket) {
|
|
update().then(() => {
|
|
socket.emit('updated')
|
|
})
|
|
} |