JoyLau's Blog

import $ from 'jquery';
import semver from 'semver';
import request from 'request';
import progress from 'request-progress';

//global.fs = require('fs');
//global.cp = require('child_process');
const fs = window.fs;
const cp = window.cp;
const electron = window.electron;
const {app, shell} = electron.remote;

state = {
    check: true,
    latest: {},
    // wait,download,install,error
    update: 'wait',
    downloadState: {}
};

// 检查更新
$.ajax({
    url: appConfig.updateCheckURL,
    timeout: 10000,
    type: 'GET',
    cache:false,
    success: function (data) {
        let latest = data[0];
        if(semver.satisfies(latest.version, '>' + app.getVersion())){
            if (latest.force) {
                that.updateVersion();
            }
        }
    },
    complete: function (XMLHttpRequest, status) {
        that.setState({
            check: false
        })
    }
});


updateVersion(){
    let that = this;
    const platform = osInfo.platform();
    try {
        const downloadUrl = platform === 'darwin' ? this.state.latest.download.dmg : platform === 'win32' ? this.state.latest.download.winSetup : '';
        if (downloadUrl === '') return;

        const downloadUrlArr = downloadUrl.split("/");

        const filename = downloadUrlArr[downloadUrlArr.length-1];

        const savePath = osInfo.tmpdir() + '/' + filename;

        const _request = request(downloadUrl);
        progress(_request, {
            // throttle: 2000,                    // Throttle the progress event to 2000ms, defaults to 1000ms
            // delay: 1000,                       // Only start to emit after 1000ms delay, defaults to 0ms
            // lengthHeader: 'x-transfer-length'  // Length header to use, defaults to content-length
        })
            .on('progress', function (state) {
                // The state is an object that looks like this:
                // {
                //     percent: 0.5,               // Overall percent (between 0 to 1)
                //     speed: 554732,              // The download speed in bytes/sec
                //     size: {
                //         total: 90044871,        // The total payload size in bytes
                //         transferred: 27610959   // The transferred payload size in bytes
                //     },
                //     time: {
                //         elapsed: 36.235,        // The total elapsed seconds since the start (3 decimals)
                //         remaining: 81.403       // The remaining seconds to finish (3 decimals)
                //     }
                // }
                that.setState({downloadState: state})
            })
            .on('error', function (err) {
                that.setState({
                    downloadState:{
                        error: true
                    }
                })
            })
            .on('end', function () {
                if (that.state.update === 'error') return;
                that.setState({
                    update: 'install',
                });

                setTimeout(function () {
                    if (platform === 'darwin'){
                        const appName = pjson.build.productName;
                        const appVersion = app.getVersion();
                        console.info(appName,appVersion);
                        // 挂载
                        cp.execSync(`hdiutil attach '${savePath}' -nobrowse`, {
                            stdio: ['ignore', 'ignore', 'ignore']
                        });

                        // 覆盖原 app
                        cp.execSync(`rm -rf '/Applications/${appName}.app' && cp -R '/Volumes/${appName} ${appVersion}/${appName}.app' '/Applications/${appName}.app'`);

                        // 卸载挂载的 dmg
                        cp.execSync(`hdiutil eject '/Volumes/${appName} ${appVersion}'`, {
                            stdio: ['ignore', 'ignore', 'ignore']
                        });

                        // 重启
                        app.relaunch();
                        app.quit();
                    }

                    if (platform === 'win32') {
                        shell.openItem(savePath);
                        setTimeout(function () {
                            app.quit();
                        },1500)
                    }
                },2000)
            })
            .pipe(fs.createWriteStream(savePath));

        that.setState({update:'download'});
    } catch (e) {
        console.info(e);
        that.setState({
            update: 'error',
        });
    }
}