-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.mjs
158 lines (142 loc) · 4.71 KB
/
index.mjs
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
import { toESModule } from "elm-esm";
import compiler from "node-elm-compiler";
import { normalize, relative } from "path";
import injectAssets from "./assetsInjector.mjs";
import injectHMR from "./injectHMR.mjs";
import acquireLock from "./mutex.mjs";
const trimDebugMessage = (code) =>
code.replace(/(console\.warn\('Compiled in DEBUG mode)/, "// $1");
const viteProjectPath = (dependency) =>
`/${relative(process.cwd(), dependency)}`;
const parseImportId = (id) => {
const parsedId = new URL(id, "file://");
const fileName = parsedId.pathname.split("/").filter(Boolean).pop();
const pathname = parsedId.pathname;
const valid = fileName && pathname.endsWith(".elm");
const withParams = parsedId.searchParams.getAll("with");
return {
fileName,
valid,
pathname,
withParams,
};
};
export const plugin = (opts) => {
const compilableFiles = new Map();
const debug = opts?.debug;
const optimize = opts?.optimize;
return {
name: "vite-plugin-elm",
enforce: "pre",
handleHotUpdate({ file, server, modules }) {
const { valid } = parseImportId(file);
if (!valid) return;
const modulesToCompile = [];
compilableFiles.forEach((dependencies, compilableFile) => {
if (dependencies.has(normalize(file))) {
const module = server.moduleGraph.getModuleById(compilableFile);
if (module) modulesToCompile.push(module);
}
});
if (modulesToCompile.length > 0) {
server.ws.send({
type: "custom",
event: "hot-update-dependents",
data: modulesToCompile.map(({ url }) => url),
});
return modulesToCompile;
} else {
return modules;
}
},
configureServer(server) {
server.middlewares.use((req, res, next) => {
if (req.url.endsWith(".elm")) {
req.url += "?import";
res.setHeader("Content-Type", "application/javascript");
}
next();
});
},
async load(id) {
const { valid, fileName, pathname, withParams } = parseImportId(id);
if (!valid) return;
const accompanies = await (() => {
if (withParams.length > 0) {
const importTree = this.getModuleIds();
let importer = "";
for (const moduleId of importTree) {
if (moduleId === id) break;
importer = moduleId;
}
const resolveAcoompany = async (accompany) =>
(await this.resolve(accompany, importer))?.id ?? "";
return Promise.all(withParams.map(resolveAcoompany));
} else {
return Promise.resolve([]);
}
})();
const targets = [pathname, ...accompanies].filter(
(target) => target !== ""
);
compilableFiles.delete(id);
const dependencies = (
await Promise.all(
targets.map((target) => compiler.findAllDependencies(target))
)
).flat();
compilableFiles.set(id, new Set([...accompanies, ...dependencies]));
const releaseLock = await acquireLock();
try {
const isBuild = process.env.NODE_ENV === "production";
let compiled = await compiler.compileToString(targets, {
output: ".js",
optimize:
typeof optimize === "boolean" ? optimize : !debug && isBuild,
verbose: isBuild,
debug: debug ?? !isBuild,
});
const moduleName = fileName.replace(".elm", "");
compiled = compiled.replace(
"'init':",
`'vNode': (args) => {
var result = A2(
_Json_run,
$elm$json$Json$Decode$value,
_Json_wrap(args)
);
var initPair = $author$project$${moduleName}$init(result.a);
var model = initPair.a;
return $author$project$${moduleName}$view(model);
}, 'init':
`
);
const esm = injectAssets(toESModule(compiled));
// Apparently `addWatchFile` may not exist: https://github.com/hmsk/vite-plugin-elm/pull/36
if (this.addWatchFile) {
dependencies.forEach(this.addWatchFile.bind(this));
}
return {
code: isBuild
? esm
: trimDebugMessage(
injectHMR(esm, dependencies.map(viteProjectPath))
),
map: null,
};
} catch (e) {
if (e instanceof Error && e.message.includes("-- NO MAIN")) {
const message = `${viteProjectPath(
pathname
)}: NO MAIN .elm file is requested to transform by vite. Probably, this file is just a depending module`;
throw message;
} else {
throw e;
}
} finally {
releaseLock();
}
},
};
};
export default plugin;