1ec8174700
* Add js function to generate gpu and cuda detection * inference nitro manage via json file instead of bash and bat script * Add /usr/lib/x86_64-linux-gnu/ to linux check gpu * chore: add CPU - GPU toggle * correct file path * fix: exist file sync check * fix: get resources path * Fix error jan/engines create existed error * Seting sync to file * Fix error show notification for GPU * Set notify default to true --------- Co-authored-by: Hien To <tominhhien97@gmail.com> Co-authored-by: Louis <louis@jan.ai>
106 lines
3.7 KiB
TypeScript
106 lines
3.7 KiB
TypeScript
import { app, ipcMain } from 'electron'
|
|
import { resolve, join } from 'path'
|
|
import { WindowManager } from './../managers/window'
|
|
import request from 'request'
|
|
import { createWriteStream, renameSync } from 'fs'
|
|
import { DownloadEvent, DownloadRoute } from '@janhq/core'
|
|
const progress = require('request-progress')
|
|
import { DownloadManager } from '@janhq/core/node'
|
|
|
|
export function handleDownloaderIPCs() {
|
|
/**
|
|
* Handles the "pauseDownload" IPC message by pausing the download associated with the provided fileName.
|
|
* @param _event - The IPC event object.
|
|
* @param fileName - The name of the file being downloaded.
|
|
*/
|
|
ipcMain.handle(DownloadRoute.pauseDownload, async (_event, fileName) => {
|
|
DownloadManager.instance.networkRequests[fileName]?.pause()
|
|
})
|
|
|
|
/**
|
|
* Handles the "resumeDownload" IPC message by resuming the download associated with the provided fileName.
|
|
* @param _event - The IPC event object.
|
|
* @param fileName - The name of the file being downloaded.
|
|
*/
|
|
ipcMain.handle(DownloadRoute.resumeDownload, async (_event, fileName) => {
|
|
DownloadManager.instance.networkRequests[fileName]?.resume()
|
|
})
|
|
|
|
/**
|
|
* Handles the "abortDownload" IPC message by aborting the download associated with the provided fileName.
|
|
* The network request associated with the fileName is then removed from the networkRequests object.
|
|
* @param _event - The IPC event object.
|
|
* @param fileName - The name of the file being downloaded.
|
|
*/
|
|
ipcMain.handle(DownloadRoute.abortDownload, async (_event, fileName) => {
|
|
const rq = DownloadManager.instance.networkRequests[fileName]
|
|
DownloadManager.instance.networkRequests[fileName] = undefined
|
|
rq?.abort()
|
|
})
|
|
|
|
/**
|
|
* Downloads a file from a given URL.
|
|
* @param _event - The IPC event object.
|
|
* @param url - The URL to download the file from.
|
|
* @param fileName - The name to give the downloaded file.
|
|
*/
|
|
ipcMain.handle(DownloadRoute.downloadFile, async (_event, url, fileName) => {
|
|
const userDataPath = join(app.getPath('home'), 'jan')
|
|
if (
|
|
typeof fileName === 'string' &&
|
|
(fileName.includes('file:/') || fileName.includes('file:\\'))
|
|
) {
|
|
fileName = fileName.replace('file:/', '').replace('file:\\', '')
|
|
}
|
|
const destination = resolve(userDataPath, fileName)
|
|
const rq = request(url)
|
|
// downloading file to a temp file first
|
|
const downloadingTempFile = `${destination}.download`
|
|
|
|
progress(rq, {})
|
|
.on('progress', function (state: any) {
|
|
WindowManager?.instance.currentWindow?.webContents.send(
|
|
DownloadEvent.onFileDownloadUpdate,
|
|
{
|
|
...state,
|
|
fileName,
|
|
}
|
|
)
|
|
})
|
|
.on('error', function (err: Error) {
|
|
WindowManager?.instance.currentWindow?.webContents.send(
|
|
DownloadEvent.onFileDownloadError,
|
|
{
|
|
fileName,
|
|
err,
|
|
}
|
|
)
|
|
})
|
|
.on('end', function () {
|
|
if (DownloadManager.instance.networkRequests[fileName]) {
|
|
// Finished downloading, rename temp file to actual file
|
|
renameSync(downloadingTempFile, destination)
|
|
|
|
WindowManager?.instance.currentWindow?.webContents.send(
|
|
DownloadEvent.onFileDownloadSuccess,
|
|
{
|
|
fileName,
|
|
}
|
|
)
|
|
DownloadManager.instance.setRequest(fileName, undefined)
|
|
} else {
|
|
WindowManager?.instance.currentWindow?.webContents.send(
|
|
DownloadEvent.onFileDownloadError,
|
|
{
|
|
fileName,
|
|
err: 'Download cancelled',
|
|
}
|
|
)
|
|
}
|
|
})
|
|
.pipe(createWriteStream(downloadingTempFile))
|
|
|
|
DownloadManager.instance.setRequest(fileName, rq)
|
|
})
|
|
}
|