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 []
    }
  },
})