calebboyd
77 lines
2.2 KiB
TypeScript
77 lines
2.2 KiB
TypeScript
import { NexeCompiler, NexeError } from '../compiler'
|
|
import { resolve, relative } from 'path'
|
|
import { each } from '@calebboyd/semaphore'
|
|
import resolveFiles, { resolveFileNameSync } from 'resolve-dependencies'
|
|
import { dequote, STDIN_FLAG } from '../util'
|
|
import { Readable } from 'stream'
|
|
|
|
function getStdIn(stdin: Readable): Promise<string> {
|
|
let out = ''
|
|
return new Promise(resolve => {
|
|
stdin
|
|
.setEncoding('utf8')
|
|
.on('readable', () => {
|
|
let current
|
|
while ((current = stdin.read())) {
|
|
out += current
|
|
}
|
|
})
|
|
.on('end', () => resolve(out.trim()))
|
|
setTimeout(() => {
|
|
if (!out.trim()) {
|
|
resolve(out.trim())
|
|
}
|
|
}, 1000)
|
|
})
|
|
}
|
|
|
|
export default async function bundle(compiler: NexeCompiler, next: any) {
|
|
const { bundle, cwd, input: inputPath } = compiler.options
|
|
let input = inputPath
|
|
compiler.entrypoint = './' + relative(cwd, input)
|
|
compiler.startup = ';require("module").runMain();'
|
|
|
|
if (!bundle) {
|
|
await compiler.addResource(resolve(cwd, input))
|
|
return next()
|
|
}
|
|
|
|
let code = ''
|
|
if (typeof bundle === 'string') {
|
|
code = await require(bundle).createBundle(compiler.options)
|
|
}
|
|
|
|
if (input === STDIN_FLAG && (code = code || dequote(await getStdIn(process.stdin)))) {
|
|
compiler.stdinUsed = true
|
|
compiler.entrypoint = './__nexe_stdin.js'
|
|
await compiler.addResource(resolve(cwd, compiler.entrypoint), code)
|
|
return next()
|
|
}
|
|
|
|
if (input === STDIN_FLAG) {
|
|
const maybeInput = resolveFileNameSync(cwd, '.')
|
|
if (!maybeInput || !maybeInput.absPath) {
|
|
throw new NexeError('No valid input detected')
|
|
}
|
|
input = maybeInput.absPath
|
|
compiler.entrypoint = './' + relative(cwd, input)
|
|
}
|
|
|
|
const { files, warnings } = await resolveFiles(
|
|
input,
|
|
...Object.keys(compiler.bundle.index).filter(x => x.endsWith('.js')),
|
|
{ cwd, expand: true, loadContent: false }
|
|
)
|
|
|
|
if (
|
|
warnings.filter(x => x.startsWith('Error parsing file') && !x.includes('node_modules')).length
|
|
) {
|
|
throw new NexeError('Parsing Error:\n' + warnings.join('\n'))
|
|
}
|
|
|
|
await each(Object.keys(files), (filename: string) => compiler.addResource(filename), {
|
|
concurrency: 10
|
|
})
|
|
return next()
|
|
}
|