feat(cache): Cache tiles locally and reserve

Author: trevorgerhardt

bin/mastarm

@@ -18,6 +18,7 @@ commander
 commander
   .command('build [entries...]')
   .description('Bundle JavaScript & CSS')
+  .option('-F, --flyle', 'Cache and serve tiles.')
   .option('-p, --proxy <address>', 'Proxy calls through to target address.')
   .option('-s, --serve', 'Serve with budo. Auto-matically rebuilds on changes.')
   .option('-w, --watch', 'Rebuild on changes with watchify.')
@@ -30,6 +31,7 @@ commander
       const budo = require('../lib/budo')
       files.map(budo({
         config,
+        flyle: get('flyle'),
         proxy: get('proxy')
       }))
     } else {

lib/budo.js

@@ -5,6 +5,7 @@ const transform = require('./transform')
 
 module.exports = function ({
   config,
+  flyle,
   proxy
 }) {
   return function ([entry, outfile]) {
@@ -42,6 +43,16 @@ module.exports = function ({
         }
       })
     }
+    if (flyle) {
+      const serveTiles = require('./flyle')
+      budoOpts.middleware.push(function (req, res, next) {
+        if (req.url.indexOf('/tile') === 0) {
+          serveTiles(req, res)
+        } else {
+          next()
+        }
+      })
+    }
 
     budo
       .cli([entry + ':' + outfile], budoOpts)

lib/flyle.js

@@ -0,0 +1,58 @@
+const concat = require('concat-stream')
+const fs = require('fs')
+const http = require('http')
+const https = require('https')
+const mkdirp = require('mkdirp')
+const path = require('path')
+const parse = require('url').parse
+
+const DEFAULT_CACHE_DIRECTORY = `${process.env.HOME}/.flyle`
+const DEFAULT_PNG = path.resolve(__dirname, '../mastarm.png')
+
+module.exports = function (req, res) {
+  const url = parse(req.url, true).query.url
+  const isHttps = url.indexOf('https') !== -1
+  const get = isHttps ? https.get : http.get
+  const filePath = isHttps ? url.split('https://')[1] : url.split('http://')[1]
+  const fullPath = path.resolve(DEFAULT_CACHE_DIRECTORY, filePath)
+  fs.stat(fullPath, (err, stats) => {
+    if (!err && stats.isFile()) {
+      sendImg({path: fullPath, res})
+    } else {
+      get(url, (fileResponse) => {
+        mkdirp(path.dirname(fullPath), (err) => {
+          if (err) {
+            logAndSend({err, res})
+          } else {
+            fileResponse.pipe(concat((png) => {
+              fs.writeFile(fullPath, png, (err) => {
+                if (err) {
+                  logAndSend({err, res})
+                } else {
+                  sendImg({path: fullPath, res})
+                }
+              })
+            }))
+          }
+        })
+      }).on('error', (err) => {
+        logAndSend({err, res})
+      })
+    }
+  })
+}
+
+function logAndSend ({err, res}) {
+  console.error('flyle >> sending default image: ', err.message)
+  sendImg({
+    path: DEFAULT_PNG,
+    res
+  })
+}
+
+function sendImg ({path, res}) {
+  res.writeHead(200, {
+    'Content-Type': 'image/png'
+  })
+  fs.createReadStream(path).pipe(res)
+}

package.json

@@ -47,6 +47,7 @@
     "check-dependencies": "^0.12.0",
     "commander": "^2.9.0",
     "commitizen": "^2.8.2",
+    "concat-stream": "^1.5.1",
     "cz-conventional-changelog": "^1.1.6",
     "envify": "^3.4.1",
     "errorify": "^0.3.1",