add recommended files support from vercel#63

stashpad · Jul 13, 2022 · 9ddf9e5 · 9ddf9e5
1 parent a06e6c5
commit 9ddf9e5
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/lib/cache.js b/lib/cache.js
@@ -134,6 +134,16 @@ module.exports = class Cache {
     for (const asset of release.assets) {
       const { name, browser_download_url, url, content_type, size } = asset
 
+      // Added new lines here
+      this.latest.files[name] = {
+        name,
+        api_url: url,
+        url: browser_download_url,
+        content_type,
+        size: Math.round(size / 1000000 * 10) / 10
+      }
+      // End new
+
       if (name === 'RELEASES') {
         try {
           if (!this.latest.files) {

diff --git a/lib/index.js b/lib/index.js
@@ -36,6 +36,7 @@ module.exports = config => {
   router.get('/download/:platform', routes.downloadPlatform)
   router.get('/update/:platform/:version', routes.update)
   router.get('/update/win32/:version/RELEASES', routes.releases)
+  router.get('/files/:filename', routes.files);
 
   return (req, res) => {
     router(req, res, finalhandler(req, res))

diff --git a/lib/routes.js b/lib/routes.js
@@ -196,6 +196,27 @@ module.exports = ({ cache, config }) => {
     res.end(content)
   }
 
+  // ADDED this route
+  exports.files = async (req, res) => {
+    const { filename } = req.params;
+    const latest = await loadCache();
+
+    if (!latest.files[filename]) {
+      send(res, 404, `can't load ${filename}`)
+      return
+    }
+
+    if (shouldProxyPrivateDownload) {
+      proxyPrivateDownload(latest.files[filename], req, res);
+      return
+    }
+
+    res.writeHead(302, {
+      Location: latest.files[filename].url
+    })
+    res.end()
+  }
+
   exports.overview = async (req, res) => {
     const latest = await loadCache()