isolated-function 0.1.6 → 0.1.7

package/README.md

@@ -30,6 +30,7 @@
       - [memory](#memory)
       - [throwError](#throwerror)
       - [timeout](#timeout)
+      - [tmpdir](#tmpdir)
   - [=\> (fn(\[...args\]), teardown())](#-fnargs-teardown)
     - [fn](#fn)
     - [teardown](#teardown)
@@ -232,6 +233,13 @@ Default: `Infinity`
 
 Timeout after a specified amount of time, in milliseconds.
 
+##### tmpdir
+
+Type: `function`<br>
+Default: `fs.mkdtemp(path.join(require('os').tmpdir(), 'compile-'))`
+
+The temporal folder to use for installing code dependencies.
+
 ### => (fn([...args]), teardown())
 
 #### fn

package/package.json

@@ -2,7 +2,7 @@
   "name": "isolated-function",
   "description": "Runs untrusted code in a Node.js v8 sandbox.",
   "homepage": "https://github.com/Kikobeats/isolated-function",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "main": "src/index.js",
   "exports": {
     ".": "./src/index.js"

package/src/compile/index.js

@@ -3,7 +3,6 @@
 const { execSync } = require('child_process')
 const esbuild = require('esbuild')
 const fs = require('fs/promises')
-const { tmpdir } = require('os')
 const $ = require('tinyspawn')
 const path = require('path')
 
@@ -22,8 +21,10 @@ const packageManager = (() => {
   }
 })()
 
-const getTmp = async content => {
-  const cwd = await fs.mkdtemp(path.join(tmpdir(), 'compile-'))
+const tmpdirDefault = () => fs.mkdtemp(path.join(require('os').tmpdir(), 'compile-'))
+
+const getTmp = async (content, tmpdir) => {
+  const cwd = await tmpdir()
   await fs.mkdir(cwd, { recursive: true })
 
   const filepath = path.join(cwd, 'index.js')
@@ -33,10 +34,10 @@ const getTmp = async content => {
   return { filepath, cwd, content, cleanup }
 }
 
-module.exports = async snippet => {
+module.exports = async (snippet, tmpdir = tmpdirDefault) => {
   const compiledTemplate = generateTemplate(snippet)
   const dependencies = detectDependencies(compiledTemplate)
-  const tmp = await getTmp(transformDependencies(compiledTemplate))
+  const tmp = await getTmp(transformDependencies(compiledTemplate), tmpdir)
 
   await $(packageManager.init, { cwd: tmp.cwd })
   await $(`${packageManager.install} ${dependencies.join(' ')}`, {
@@ -52,7 +53,7 @@ module.exports = async snippet => {
   })
 
   await tmp.cleanup()
-  return getTmp(result.outputFiles[0].text)
+  return getTmp(result.outputFiles[0].text, tmpdir)
 }
 
 module.exports.detectDependencies = detectDependencies

package/src/index.js

@@ -20,9 +20,9 @@ const flags = ({ filename, memory }) => {
   return flags.join(' ')
 }
 
-module.exports = (snippet, { timeout, memory, throwError = true } = {}) => {
+module.exports = (snippet, { tmpdir, timeout, memory, throwError = true } = {}) => {
   if (!['function', 'string'].includes(typeof snippet)) throw new TypeError('Expected a function')
-  const compilePromise = compile(snippet)
+  const compilePromise = compile(snippet, tmpdir)
 
   const fn = async (...args) => {
     let duration