63 lines
1.8 KiB
TypeScript
63 lines
1.8 KiB
TypeScript
import * as https from 'https'
|
||
import * as fs from 'fs'
|
||
import { BrowserWindow } from 'electron'
|
||
import * as http from 'http'
|
||
|
||
let mainWindow: BrowserWindow | null = null
|
||
|
||
export function setMainWindow(window: BrowserWindow) {
|
||
mainWindow = window
|
||
}
|
||
|
||
export function downloadFile(url: string, outputPath: string): Promise<void> {
|
||
return new Promise((resolve, reject) => {
|
||
console.log(`开始下载文件: ${url}`)
|
||
console.log(`保存路径: ${outputPath}`)
|
||
|
||
const file = fs.createWriteStream(outputPath)
|
||
// 创建HTTP客户端，兼容https和http
|
||
const client = url.startsWith('https') ? https : http
|
||
|
||
client
|
||
.get(url, response => {
|
||
const totalSize = parseInt(response.headers['content-length'] || '0', 10)
|
||
let downloadedSize = 0
|
||
|
||
console.log(`文件大小: ${totalSize} bytes`)
|
||
|
||
response.pipe(file)
|
||
|
||
response.on('data', chunk => {
|
||
downloadedSize += chunk.length
|
||
const progress = totalSize ? Math.round((downloadedSize / totalSize) * 100) : 0
|
||
|
||
console.log(`下载进度: ${progress}% (${downloadedSize}/${totalSize})`)
|
||
|
||
if (mainWindow) {
|
||
mainWindow.webContents.send('download-progress', {
|
||
progress,
|
||
status: 'downloading',
|
||
message: `下载中... ${progress}%`,
|
||
})
|
||
}
|
||
})
|
||
|
||
file.on('finish', () => {
|
||
file.close()
|
||
console.log(`文件下载完成: ${outputPath}`)
|
||
resolve()
|
||
})
|
||
|
||
file.on('error', err => {
|
||
console.error(`文件写入错误: ${err.message}`)
|
||
fs.unlink(outputPath, () => {}) // 删除不完整的文件
|
||
reject(err)
|
||
})
|
||
})
|
||
.on('error', err => {
|
||
console.error(`下载错误: ${err.message}`)
|
||
reject(err)
|
||
})
|
||
})
|
||
}
|