Electron应用内下载器实现

gy2022-02-152025-05-21

考虑到某些模块的正常运行离不开伪造http请求标头来欺骗服务器，以获取上面的某些敏感数据。

准备做一个应用内的定制版下载管理器，来承接各个模块可能需要触发的下载任务。

整个分支工程的框架大体如下：

一、外置下载引擎的可行性论证与选择

优秀的下载器motrix使用aria2作为其内核，支持断点续传和多进程下载。但因其技术文档难以寻找，我打算使用常见的request模块来下载文件。以后若有时间，我会把aria2也集成进去。

二、下载逻辑以及代码编写

1.作为一个独立对象储存在./insert/buildInDownloader

2.文件树目录结构：

3.下载原理：

创建写入流 fs.createWriteStream
get方法请求文件 request.get(options)
在request请求的回调函数中获取文件大小并设置length参数 length: response.headers[‘content-length’]
设置好length参数之后再重新request请求文件（否则无法获取length参数，进度信息会有误）
将request请求到的文件流pipe进progress-stream的实例（proStream）中去监听文件下载进度 request.get(options).pipe(proStream)及proStream.on(‘progress’, function(progress){…}）
文件下载完成，触发文件的finish事件

4.代码实现：

var request = require('request');
var progress = require('progress-stream');
var fs = require('fs');
var http = require("http");

var percentage = new Map();//外部变量，定期更新，以供外部读取
var progress_stream_instance = new Map();
var file_write_stream = new Map();

//当请求的非文本文件时，一定要设置options.encoding : null
/*options:
--task_id,
--save_path,
--referer,
--url,
--encoding

*/
function download(options) {
    //set map
    file_write_stream.set(options.task_id, fs.createWriteStream(options.save_path,{encoding:"binary"})); //文件的写入流

    var my_request = http.request({
        //只获取响应头数据
        method: 'HEAD',
        path: options.path,
        host: options.host
    }, function (res) {
        if (res.statusCode == 200) {
            console.log(file_write_stream);

            //编码
            //res.setEncoding(null);

            //创建progress_stream实例
            progress_stream_instance.set(options.task_id, progress({
                length: res.headers['content-length'],
                time: 500 /* ms */
            }));

            //绑定定期更新进度条事件
            progress_stream_instance.get(options.task_id).on('progress', function (progress) {
                //更新进度条
                percentage.set(options.task_id, Math.round(progress.percentage));
                /*
                更新UI里的数据
                */
            });

            //正式下载
            request.get("http://" + options.host + options.path, options).pipe(progress_stream_instance.get(options.task_id)).pipe(file_write_stream.get(options.task_id));

        } else {
            console.error('error');
        }
    });

    my_request.end();/*完成请求发送 如果任何一部分请求体没有被发送,它将被刷新到流 */
}