-
Notifications
You must be signed in to change notification settings - Fork 1
/
vite.config.ts
93 lines (74 loc) · 2.39 KB
/
vite.config.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
import { Plugin, defineConfig } from 'vite';
import wasm from 'vite-plugin-wasm';
import path from 'path';
import { spawnSync } from 'child_process';
import fs from 'fs';
import chalk from 'chalk';
const log = console.log;
type WasmCompileArgs = {
flags: Iterable<string>;
};
function wasmCompile(args: WasmCompileArgs = { flags: [] }): Plugin {
const isProd = () => process.env.NODE_ENV === 'production';
function compileRustCrates() {
const cratesPath = path.resolve(__dirname, './crates');
const crates = fs.readdirSync(cratesPath).filter((it) => {
const item = path.resolve(cratesPath, it);
return fs.lstatSync(item).isDirectory();
});
log(chalk.yellow(`Found crates [${crates.join(', ')}]`));
for (const crate of crates) {
log();
const cratePath = path.resolve(cratesPath, crate);
const pkgPath = path.resolve(__dirname, `./pkg/${crate}`);
log(chalk.yellow(`Compiling ${crate}`));
const buildResult = spawnSync(
'wasm-pack',
[
'--log-level',
'warn',
'build',
cratePath,
isProd() ? '--release' : '--dev',
'--out-dir',
pkgPath,
'--',
...args.flags,
],
{ stdio: 'inherit' },
);
if (buildResult.error || buildResult.status !== 0) {
log(chalk.red(`Failed to compile ${crate}`));
}
}
log();
}
return {
name: 'wasm-compile',
enforce: 'pre',
configureServer(server) {
server.watcher.add(path.resolve(__dirname, './crates/**/src/*.rs'));
server.watcher.on('change', (file) => {
if (file.endsWith('.rs') || file.endsWith('Cargo.toml')) {
compileRustCrates();
}
});
},
buildStart(_params) {
compileRustCrates();
},
};
}
export default defineConfig({
resolve: {
alias: {
'@pkg': path.resolve(__dirname, './pkg'),
},
},
plugins: [wasmCompile({ flags: ['--features', 'wee_alloc'] }), wasm()],
build: {
target: 'esnext',
outDir: 'docs',
},
base: '/chip8-wasm/'
});