fix: only spawn node once (#111)

Author: achingbrain

src/core/commands/start-server.js

@@ -9,7 +9,7 @@ const favicon = require('ipfs-registry-mirror-common/handlers/favicon')
 const root = require('../handlers/root')
 const tarball = require('../handlers/tarball')
 const manifest = require('../handlers/manifest')
-const getIpfs = require('../middlewares/getIpfs')
+const getIpfs = require('../middleware/get-ipfs')
 
 const startServer = (config) => {
   const app = express()

src/core/handlers/manifest.js

@@ -30,7 +30,7 @@ module.exports = (config, app) => {
 
     log(`Loading manifest for ${moduleName}`)
 
-    const ipfs = response.locals.ipfs
+    const ipfs = await request.app.locals.ipfs
 
     try {
       const manifest = await loadManifest(config, ipfs.api, moduleName)

src/core/handlers/tarball.js

@@ -13,7 +13,7 @@ module.exports = (config, app) => {
 
     log(`Loading ${file}`)
 
-    const ipfs = response.locals.ipfs
+    const ipfs = await request.app.locals.ipfs
 
     try {
       const readStream = await loadTarball(config, ipfs.api, file)

src/core/middleware/get-ipfs.js

@@ -0,0 +1,71 @@
+'use strict'
+
+const startIpfs = require('../commands/start-ipfs')
+const request = require('ipfs-registry-mirror-common/utils/retry-request')
+const timeout = require('ipfs-registry-mirror-common/utils/timeout-promise')
+
+const cleanUpOps = []
+
+const cleanUp = () => {
+  Promise.all(
+    cleanUpOps.map(op => op())
+  )
+    .then(() => {
+      process.exit(0)
+    })
+}
+
+process.on('SIGTERM', cleanUp)
+process.on('SIGINT', cleanUp)
+
+module.exports = (options) => {
+  return async (req, res, next) => {
+    if (req.app.locals.ipfs) {
+      return next()
+    }
+
+    const createIpfs = async () => {
+      const ipfs = await startIpfs(options)
+
+      cleanUpOps.push(() => {
+        return new Promise((resolve) => {
+          if (options.ipfs.node !== 'proc') {
+            return resolve()
+          }
+
+          ipfs.stop(() => {
+            console.info('😈 IPFS node stopped') // eslint-disable-line no-console
+
+            resolve()
+          })
+        })
+      })
+
+      console.info('🗂️  Loading registry index from', options.registry) // eslint-disable-line no-console
+
+      try {
+        const mirror = await request(Object.assign({}, options.request, {
+          uri: options.registry,
+          json: true
+        }))
+
+        console.info('☎️  Dialling registry mirror', mirror.ipfs.addresses.join(',')) // eslint-disable-line no-console
+
+        await timeout(
+          ipfs.api.swarm.connect(mirror.ipfs.addresses[0]),
+          options.registryConnectTimeout
+        )
+
+        console.info('📱️ Connected to registry') // eslint-disable-line no-console
+      } catch (error) {
+        console.info('📴 Not connected to registry') // eslint-disable-line no-console
+      }
+
+      return ipfs
+    }
+
+    req.app.locals.ipfs = createIpfs()
+
+    next()
+  }
+}