Restores local prebuilds, previously removed in prebuild#81 / a069253

jchook · Dec 23, 2020 · 0167893 · 0167893
1 parent 9e62565
commit 0167893
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 38 deletions.
diff --git a/download.js b/download.js
@@ -15,58 +15,74 @@ var mkdirp = require('mkdirp-classic')
 
 function downloadPrebuild (downloadUrl, opts, cb) {
   var cachedPrebuild = util.cachedPrebuild(downloadUrl)
+  var localPrebuild = util.localPrebuild(downloadUrl)
   var tempFile = util.tempFile(cachedPrebuild)
   var log = opts.log || noop
 
-  ensureNpmCacheDir(function (err) {
-    if (err) return onerror(err)
+  if (opts.nolocal) return download()
 
-    log.info('looking for cached prebuild @', cachedPrebuild)
-    fs.access(cachedPrebuild, fs.R_OK | fs.W_OK, function (err) {
-      if (!(err && err.code === 'ENOENT')) {
-        log.info('found cached prebuild')
-        return unpack()
-      }
+  log.info('looking for local prebuild @', localPrebuild)
+  fs.access(localPrebuild, fs.R_OK | fs.W_OK, function (err) {
+    if (err && err.code === 'ENOENT') {
+      return download()
+    }
 
-      log.http('request', 'GET ' + downloadUrl)
-      var reqOpts = proxy({ url: downloadUrl }, opts)
+    log.info('found local prebuild')
+    cachedPrebuild = localPrebuild
+    unpack()
+  })
 
-      if (opts.token) {
-        reqOpts.url += '?access_token=' + opts.token
-        reqOpts.headers = {
-          'User-Agent': 'simple-get',
-          Accept: 'application/octet-stream'
+  function download () {
+    ensureNpmCacheDir(function (err) {
+      if (err) return onerror(err)
+
+      log.info('looking for cached prebuild @', cachedPrebuild)
+      fs.access(cachedPrebuild, fs.R_OK | fs.W_OK, function (err) {
+        if (!(err && err.code === 'ENOENT')) {
+          log.info('found cached prebuild')
+          return unpack()
         }
-      }
 
-      var req = get(reqOpts, function (err, res) {
-        if (err) return onerror(err)
-        log.http(res.statusCode, downloadUrl)
-        if (res.statusCode !== 200) return onerror()
-        mkdirp(util.prebuildCache(), function () {
-          log.info('downloading to @', tempFile)
-          pump(res, fs.createWriteStream(tempFile), function (err) {
-            if (err) return onerror(err)
-            fs.rename(tempFile, cachedPrebuild, function (err) {
-              if (err) return cb(err)
-              log.info('renaming to @', cachedPrebuild)
-              unpack()
+        log.http('request', 'GET ' + downloadUrl)
+        var reqOpts = proxy({ url: downloadUrl }, opts)
+
+        if (opts.token) {
+          reqOpts.url += '?access_token=' + opts.token
+          reqOpts.headers = {
+            'User-Agent': 'simple-get',
+            Accept: 'application/octet-stream'
+          }
+        }
+
+        var req = get(reqOpts, function (err, res) {
+          if (err) return onerror(err)
+          log.http(res.statusCode, downloadUrl)
+          if (res.statusCode !== 200) return onerror()
+          mkdirp(util.prebuildCache(), function () {
+            log.info('downloading to @', tempFile)
+            pump(res, fs.createWriteStream(tempFile), function (err) {
+              if (err) return onerror(err)
+              fs.rename(tempFile, cachedPrebuild, function (err) {
+                if (err) return cb(err)
+                log.info('renaming to @', cachedPrebuild)
+                unpack()
+              })
             })
           })
         })
-      })
 
-      req.setTimeout(30 * 1000, function () {
-        req.abort()
+        req.setTimeout(30 * 1000, function () {
+          req.abort()
+        })
       })
-    })
 
-    function onerror (err) {
-      fs.unlink(tempFile, function () {
-        cb(err || error.noPrebuilts(opts))
-      })
-    }
-  })
+      function onerror (err) {
+        fs.unlink(tempFile, function () {
+          cb(err || error.noPrebuilts(opts))
+        })
+      }
+    })
+  }
 
   function unpack () {
     var binaryName

diff --git a/util.js b/util.js
@@ -105,11 +105,16 @@ function packageOrigin (env, pkg) {
   }
 }
 
+function localPrebuild (url) {
+  return path.join('prebuilds', path.basename(url))
+}
+
 exports.getDownloadUrl = getDownloadUrl
 exports.getApiUrl = getApiUrl
 exports.getAssetUrl = getAssetUrl
 exports.urlTemplate = urlTemplate
 exports.cachedPrebuild = cachedPrebuild
+exports.localPrebuild = localPrebuild
 exports.prebuildCache = prebuildCache
 exports.npmCache = npmCache
 exports.tempFile = tempFile