74 lines
2.0 KiB
TypeScript
74 lines
2.0 KiB
TypeScript
import path from "node:path"
|
|
|
|
import { red, yellow } from "kolorist"
|
|
import type { HmrContext, Plugin } from "vite"
|
|
|
|
function isNodeWithinCircularImports(
|
|
node: any,
|
|
nodeChain: any[],
|
|
currentChain: any[] = [node],
|
|
traversedModules = new Set<any>(),
|
|
): boolean {
|
|
if (traversedModules.has(node)) {
|
|
return false
|
|
}
|
|
traversedModules.add(node)
|
|
|
|
for (const importer of node.importers) {
|
|
if (importer === node) continue
|
|
|
|
const importerIndex = nodeChain.indexOf(importer)
|
|
if (importerIndex !== -1) {
|
|
const importChain = [
|
|
importer,
|
|
...[...currentChain].reverse(),
|
|
...nodeChain.slice(importerIndex, -1).reverse(),
|
|
].map((m) => path.relative(process.cwd(), m.file))
|
|
|
|
console.warn(yellow(`Circular imports detected: ${importChain.join("\n↳ ")}`))
|
|
return true
|
|
}
|
|
|
|
if (!currentChain.includes(importer)) {
|
|
const result = isNodeWithinCircularImports(
|
|
importer,
|
|
nodeChain,
|
|
currentChain.concat(importer),
|
|
traversedModules,
|
|
)
|
|
if (result) return result
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
export const circularImportRefreshPlugin = (): Plugin => ({
|
|
name: "circular-import-refresh",
|
|
configureServer(server) {
|
|
server.ws.on("message", (message) => {
|
|
console.info(message)
|
|
})
|
|
},
|
|
handleHotUpdate({ file, server }: HmrContext) {
|
|
const mod = server.moduleGraph.getModuleById(file)
|
|
|
|
// Check for circular imports
|
|
if (mod && isNodeWithinCircularImports(mod, [mod])) {
|
|
console.error(
|
|
red(
|
|
`Circular dependency detected in ${file} involving store files. Performing full page refresh.`,
|
|
),
|
|
)
|
|
|
|
server.ws.send({ type: "full-reload" })
|
|
return []
|
|
}
|
|
|
|
if (file.startsWith(path.resolve(process.cwd(), "src/store")) && file.endsWith(".ts")) {
|
|
console.warn(yellow(`[memory-hmr] Detected change in store file: ${file}. Reloading page.`))
|
|
server.ws.send({ type: "full-reload" })
|
|
return []
|
|
}
|
|
},
|
|
})
|