// Adapted from https://github.com/chakra-ui/panda-vscode/blob/2042dc4af60072e499c8725ac06c5baa1a36d0c8/packages/language-server/src/builder-resolver.ts#L1

import { Builder, PandaContext } from '@pandacss/node'
import { PandaError } from '@pandacss/shared'
import { findConfig } from '@pandacss/config'
import path from 'node:path'
import { isMatch } from 'micromatch'

/**
 * - Calls builder.setup() for the closest config file to the given filepath
 * - Keep track of all the config files and their builders
 * - Share the setup promise so it can be awaited anywhere
 */
export class BuilderResolver {
  private builderByConfigDirpath = new Map<string, Builder>()
  private synchronizingByConfigDirpath = new Map<string, Promise<PandaContext | undefined> | false>()

  private configDirpathList = new Set<string>()
  private configDirPathByFilepath = new Map<string, string>()
  private configPathByDirpath = new Map<string, string>()

  findConfigDirpath<T>(filepath: string, onFound: (configDirPath: string, configPath: string) => T) {
    const cachedDir = this.configDirPathByFilepath.get(filepath)
    if (cachedDir) {
      return onFound(cachedDir, this.configPathByDirpath.get(cachedDir)!)
    }

    const dirpath = path.dirname(filepath)

    for (const configDirpath of this.configDirpathList) {
      if (dirpath.startsWith(configDirpath)) {
        this.configDirPathByFilepath.set(filepath, configDirpath)
        return onFound(configDirpath, this.configPathByDirpath.get(configDirpath)!)
      }
    }
  }

  get(filepath: string) {
    return this.findConfigDirpath(filepath, (configDirpath) => this.builderByConfigDirpath.get(configDirpath))
  }

  // isContextSynchronizing(filepath: string) {
  //   return this.findConfigDirpath(filepath, (configDirpath) => this.synchronizingByConfigDirpath.get(configDirpath))
  // }

  create(configPath: string) {
    const builder = new Builder()

    const dirpath = path.dirname(configPath)
    this.configDirPathByFilepath.set(configPath, dirpath)
    this.builderByConfigDirpath.set(dirpath, builder)
    this.configDirpathList.add(dirpath)
    this.configPathByDirpath.set(dirpath, configPath)

    return this
  }

  async setup(filepath: string) {
    return this.findConfigDirpath(filepath, async (configDirpath, configPath) => {
      const builder = this.builderByConfigDirpath.get(configDirpath)
      if (!builder) {
        return
      }

      const current = this.synchronizingByConfigDirpath.get(configDirpath)
      if (current) {
        return current
      }

      const synchronizing = builder.setup({ configPath, cwd: configDirpath })
      this.synchronizingByConfigDirpath.set(configDirpath, synchronizing)

      synchronizing.finally(() => {
        this.synchronizingByConfigDirpath.set(configDirpath, false)
      })

      return synchronizing
    })
  }

  async getOrCreate(filepath: string) {
    const context = await this.findConfigDirpath(filepath, (configDirpath, configPath) => {
      const builder = this.builderByConfigDirpath.get(configDirpath)
      if (builder) return builder.context

      return this.create(configPath).setup(filepath)
    })

    if (context) return context

    try {
      const dirpath = path.dirname(filepath)
      const configPath = findConfig({ cwd: dirpath })
      return this.create(configPath).setup(filepath)
    } catch (err) {
      if (err instanceof PandaError) {
        return
      }

      throw err
    }
  }

  isIncluded(filepath: string) {
    return this.findConfigDirpath(filepath, (configDirpath) => {
      const builder = this.builderByConfigDirpath.get(configDirpath)
      if (!builder) return false

      const ctx = builder.context
      if (!ctx) return false

      return isMatch(path.relative(configDirpath, filepath), ctx.config.include)
    })
  }
}