Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refine postinstall script #85

Merged
merged 7 commits into from
Jul 9, 2017
Merged

Refine postinstall script #85

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
a4cda5e
split code
gaoqiankun Jul 8, 2017
d79dd38
Add show download progress
gaoqiankun Jul 8, 2017
371c0f1
Reuse downloaded local file
gaoqiankun Jul 8, 2017
c17bbbe
try to download to home dir and bin dir as a fallback
gaoqiankun Jul 8, 2017
3c4670b
download to temp dir and copy to local dir when success
gaoqiankun Jul 8, 2017
cb4d0c4
check cdn file in `install`
gaoqiankun Jul 8, 2017
06d1cc0
use base64 encode of cdn file name as local file name
gaoqiankun Jul 8, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
138 changes: 91 additions & 47 deletions postinstall.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
var os = require('os');
var fs = require('fs');
var path = require('path');
var readline = require('readline');
var Zip = require('decompress-zip');
var request = require('request');

Expand All @@ -21,67 +23,109 @@ var cdnFiles = {
var arch = process.env.NGROK_ARCH || (os.platform() + os.arch());
var cdnFile = cdnFiles[arch];

if (!cdnFile) {
console.error('ngrok - platform ' + arch + ' is not supported.');
process.exit(1);
var binPath = path.join(__dirname, 'bin');
var localPath;
try {
localPath = path.join(os.homedir(), '.ngrok');
fs.existsSync(localPath) || fs.mkdirSync(localPath);
} catch (err) {
localPath = binPath;
}
var localFileName = new Buffer(cdnFile).toString('base64');
var localFile = path.join(localPath, localFileName + '.zip');

var localPath = __dirname + '/bin/';
var localFile = localPath + 'ngrok.zip';
var attempts = 0;
var maxAttempts = 3;
install();

install(retry);
function install () {
if (fs.existsSync(localFile) && fs.statSync(localFile).size) {
extract(retry)
} else if (!cdnFile) {
console.error('ngrok - platform ' + arch + ' is not supported.');
process.exit(1);
} else {
download(retry);
}

var attempts = 0;
var maxAttempts = 3;

function retry(err) {
attempts++;
if (err && attempts === maxAttempts) {
console.error('ngrok - install failed', err);
return process.exit(1);
function retry(err) {
attempts++;
if (err && attempts === maxAttempts) {
console.error('ngrok - install failed', err);
return process.exit(1);
}
if (err) {
console.warn('ngrok - install failed, retrying');
return setTimeout(function() {
download(retry);
}, 500);
}
process.exit(0);
}
if (err) {
console.warn('ngrok - install failed, retrying');
return setTimeout(function() {
install(retry);
}, 500);
}

function extract(cb) {
console.log('ngrok - unpacking binary')
new Zip(localFile).extract({path: binPath})
.once('error', error)
.once('extract', function() {
var suffix = os.platform() === 'win32' ? '.exe' : '';
if (suffix === '.exe')
fs.writeFileSync(path.join(binPath, 'ngrok.cmd'), 'ngrok.exe');
var target = path.join(binPath, 'ngrok' + suffix);
fs.chmodSync(target, 0755);
if (!fs.existsSync(target) || fs.statSync(target).size <= 0)
return error(new Error('corrupted file ' + target));
console.log('ngrok - binary unpacked to ' + target);
cb(null);
});

function error(e) {
console.warn('ngrok - error unpacking binary', e);
cb(e);
}
process.exit(0);
}

function install(cb) {
function download(cb) {
console.log('ngrok - downloading binary ' + cdnFile);
var total = 0;
var downloaded = 0;
var shouldClearLine = false;

var showProgress = function () {
if (shouldClearLine) {
readline.clearLine(process.stdout);
readline.cursorTo(process.stdout, 0);
}

var progress = downloaded + (total ? ('/' + total) : '');
process.stdout.write('ngrok - downloading progress: ' + progress);
shouldClearLine= true;
}
var tempFile = path.join(os.tmpdir(), 'ngrok.zip');
var outputStream = fs.createWriteStream(tempFile);

request
.get(cdnFile)
.pipe(fs.createWriteStream(localFile))
.on('finish', function() {
console.log('ngrok - binary downloaded to ' + localFile);
extract();
.on('response', function (res) {
total = res.headers['content-length'];
total > 0 && showProgress();
})
.on('data', function (data) {
downloaded += data.length;
showProgress();
outputStream.write(data);
})
.on('complete', function () {
console.log('\nngrok - binary downloaded to ' + localFile);
var readStream = fs.createReadStream(tempFile);
readStream
.pipe(fs.createWriteStream(localFile))
.on('finish', () => extract(cb));
})
.on('error', function(e) {
console.warn('ngrok - error downloading binary', e);
cb(e);
});

function extract() {
console.log('ngrok - unpacking binary')
new Zip(localFile).extract({path: localPath})
.once('error', error)
.once('extract', function() {
var suffix = os.platform() === 'win32' ? '.exe' : '';
if (suffix === '.exe')
fs.writeFileSync(localPath + 'ngrok.cmd', 'ngrok.exe');
fs.unlinkSync(localFile);
var target = localPath + 'ngrok' + suffix;
fs.chmodSync(target, 0755);
if (!fs.existsSync(target) || fs.statSync(target).size <= 0)
return error(new Error('corrupted file ' + target));
console.log('ngrok - binary unpacked to ' + target);
cb(null);
});
}

function error(e) {
console.warn('ngrok - error unpacking binary', e);
cb(e);
}
}