monthly update: fixes
This commit is contained in:
18
lib/db.js
18
lib/db.js
@@ -27,13 +27,17 @@ async function get_db() {
|
||||
|
||||
async function insertDownload(url, title, thumbnail, size, PID) {
|
||||
const uid = uuid.v1()
|
||||
db
|
||||
.prepare(`
|
||||
INSERT INTO downloads
|
||||
(uid, url, title, thumbnail, size, process_pid)
|
||||
VALUES (?, ?, ?, ?, ?, ?)`
|
||||
)
|
||||
.run(uid, url, title, thumbnail, size, PID)
|
||||
try {
|
||||
db
|
||||
.prepare(`
|
||||
INSERT INTO downloads
|
||||
(uid, url, title, thumbnail, size, process_pid)
|
||||
VALUES (?, ?, ?, ?, ?, ?)`
|
||||
)
|
||||
.run(uid, url, title, thumbnail, size, PID)
|
||||
} catch (error) {
|
||||
logger('db', 'some error occourred')
|
||||
}
|
||||
|
||||
return uid
|
||||
}
|
||||
|
||||
@@ -62,11 +62,10 @@ async function download(socket, payload) {
|
||||
|
||||
async function retriveDownload(socket) {
|
||||
const downloads = await pruneDownloads();
|
||||
|
||||
if (downloads.length > 0) {
|
||||
for (const _download of downloads) {
|
||||
await killProcess(_download.process_pid);
|
||||
await download(socket, _download.url);
|
||||
await download(socket, _download);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -22,7 +22,7 @@ io.on('connection', socket => {
|
||||
logger('ws', `${socket.handshake.address} connected!`)
|
||||
// message listeners
|
||||
socket.on('send-url', args => {
|
||||
logger('ws', args)
|
||||
logger('ws', args?.url)
|
||||
download(socket, args)
|
||||
})
|
||||
socket.on('abort', () => {
|
||||
@@ -32,7 +32,7 @@ io.on('connection', socket => {
|
||||
ytdlpUpdater(socket)
|
||||
})
|
||||
socket.on('fetch-jobs', async () => {
|
||||
socket.emit('pending-jobs', await db.pruneDownloads())
|
||||
socket.emit('pending-jobs', await db.retrieveAll())
|
||||
})
|
||||
socket.on('retrieve-jobs', async () => {
|
||||
retriveDownload(socket)
|
||||
|
||||
Reference in New Issue
Block a user