/
cli.ts
120 lines (103 loc) · 2.91 KB
/
cli.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import minimist from 'minimist'
import { dim, red } from 'kolorist'
import { createServer } from 'vite'
import { ViteNodeServer } from './server'
import { ViteNodeRunner } from './client'
const argv = minimist(process.argv.slice(2), {
'alias': {
r: 'root',
c: 'config',
h: 'help',
w: 'watch',
s: 'silent',
i: 'inline-deps',
},
'--': true,
'string': ['root', 'config', 'inline-deps'],
'boolean': ['help', 'watch', 'silent'],
unknown(name: string) {
if (name[0] === '-') {
console.error(red(`Unknown argument: ${name}`))
help()
process.exit(1)
}
return true
},
})
if (argv.help) {
help()
process.exit(0)
}
if (!argv._.length) {
console.error(red('No files specified.'))
help()
process.exit(1)
}
// forward argv
process.argv = [...process.argv.slice(0, 2), ...(argv['--'] || [])]
run(argv)
function help() {
// eslint-disable-next-line no-console
console.log(`
Usage:
$ vite-node [options] [files]
Options:
-r, --root <path> ${dim('[string]')} use specified root directory
-c, --config <file> ${dim('[string]')} use specified config file
-w, --watch ${dim('[boolean]')} restart on file changes, similar to "nodemon"
-s, --silent ${dim('[boolean]')} do not emit errors and logs
--vue ${dim('[boolean]')} support for importing Vue component
-i, --inline-deps <deps> ${dim('[string]')} inline specified dependencies
`)
}
export interface CliOptions {
files?: string[]
_?: string[]
root?: string
config?: string
watch?: boolean
'inline-deps'?: string | string[]
}
async function run(options: CliOptions = {}) {
const files = options.files || options._ || []
const inlineDeps = options['inline-deps'] || []
const server = await createServer({
logLevel: 'error',
configFile: options.config,
root: options.root,
})
await server.pluginContainer.buildStart({})
const node = new ViteNodeServer(server, {
deps: {
inline: Array.isArray(inlineDeps) ? inlineDeps : [inlineDeps],
},
})
const runner = new ViteNodeRunner({
root: server.config.root,
base: server.config.base,
fetchModule(id) {
return node.fetchModule(id)
},
resolveId(id, importer) {
return node.resolveId(id, importer)
},
})
// provide the vite define variable in this context
await runner.executeId('/@vite/env')
for (const file of files)
await runner.executeFile(file)
if (!options.watch)
await server.close()
server.watcher.on('change', async (eventName, path) => {
// eslint-disable-next-line no-console
console.log(dim(`[${eventName}] ${path}`))
// invalidate module cache but not node_modules
Array.from(runner.moduleCache.keys())
.forEach((i) => {
if (!i.includes('node_modules'))
runner.moduleCache.delete(i)
})
for (const file of files)
await runner.executeFile(file)
})
}