import { resolve, join } from 'pathe'
import { defineNuxtModule, installModule, addPlugin, resolveModule } from '@nuxt/kit'
import { colors } from '@unocss/preset-uno'
import type { UnocssNuxtOptions } from '@unocss/nuxt'

export default defineNuxtModule({
  name: '@nuxthq/ui',
  configKey: 'ui',
  async setup (_options, nuxt) {
    const { prefix = 'u', colors: { primary = 'indigo', gray = 'zinc' } = {} } = _options || {}
    const { shortcuts = [], rules = [], variants = [], theme = {} } = _options?.unocss || {}

    const options: UnocssNuxtOptions = {
      theme: {
        colors: {
          gray: typeof gray === 'object' ? gray : (colors && colors[gray]),
          primary: typeof primary === 'object' ? primary : (colors && colors[primary])
        },
        ...theme
      },
      preflight: true,
      icons: {
        prefix: ''
      },
      shortcuts: {
        'bg-tw-white': 'bg-white dark:bg-black',
        'bg-tw-gray-50': 'bg-gray-50 dark:bg-gray-900',
        'bg-tw-gray-100': 'bg-gray-100 dark:bg-gray-800',
        'bg-tw-gray-200': 'bg-gray-200 dark:bg-gray-700',
        'bg-tw-gray-300': 'bg-gray-300 dark:bg-gray-600',
        'bg-tw-gray-400': 'bg-gray-400 dark:bg-gray-500',
        'bg-tw-gray-500': 'bg-gray-500 dark:bg-gray-400',
        'bg-tw-gray-600': 'bg-gray-600 dark:bg-gray-300',
        'bg-tw-gray-700': 'bg-gray-700 dark:bg-gray-200',
        'bg-tw-gray-800': 'bg-gray-800 dark:bg-gray-100',
        'bg-tw-gray-900': 'bg-gray-900 dark:bg-gray-50',
        'bg-tw-black': 'bg-black dark:bg-white',
        'text-tw-white': 'text-white dark:text-black',
        'text-tw-gray-50': 'text-gray-50 dark:text-gray-900',
        'text-tw-gray-100': 'text-gray-100 dark:text-gray-800',
        'text-tw-gray-200': 'text-gray-200 dark:text-gray-700',
        'text-tw-gray-300': 'text-gray-300 dark:text-gray-600',
        'text-tw-gray-400': 'text-gray-400 dark:text-gray-500',
        'text-tw-gray-500': 'text-gray-500 dark:text-gray-400',
        'text-tw-gray-600': 'text-gray-600 dark:text-gray-300',
        'text-tw-gray-700': 'text-gray-700 dark:text-gray-200',
        'text-tw-gray-800': 'text-gray-800 dark:text-gray-100',
        'text-tw-gray-900': 'text-gray-900 dark:text-gray-50',
        'text-tw-black': 'text-black dark:text-white',
        'border-tw-gray-100': 'border-gray-100 dark:border-gray-900',
        'border-tw-gray-200': 'border-gray-200 dark:border-gray-800',
        'border-tw-gray-300': 'border-gray-300 dark:border-gray-700',
        'border-tw-gray-400': 'border-gray-400 dark:border-gray-600',
        'border-tw-gray-500': 'border-gray-500 dark:border-gray-500',
        'border-tw-gray-600': 'border-gray-600 dark:border-gray-400',
        'border-tw-gray-700': 'border-gray-700 dark:border-gray-300',
        'border-tw-gray-800': 'border-gray-800 dark:border-gray-200',
        'border-tw-gray-900': 'border-gray-900 dark:border-gray-100',
        'divide-tw-gray-100': 'divide-gray-100 dark:divide-gray-900',
        'divide-tw-gray-200': 'divide-gray-200 dark:divide-gray-800',
        'divide-tw-gray-300': 'divide-gray-300 dark:divide-gray-700',
        'divide-tw-gray-400': 'divide-gray-400 dark:divide-gray-600',
        'divide-tw-gray-500': 'divide-gray-500 dark:divide-gray-500',
        'divide-tw-gray-600': 'divide-gray-600 dark:divide-gray-400',
        'divide-tw-gray-700': 'divide-gray-700 dark:divide-gray-300',
        'divide-tw-gray-800': 'divide-gray-800 dark:divide-gray-200',
        'divide-tw-gray-900': 'divide-gray-900 dark:divide-gray-100',
        'ring-tw-gray-100': 'ring-gray-100 dark:ring-gray-900',
        'ring-tw-gray-200': 'ring-gray-200 dark:ring-gray-800',
        'ring-tw-gray-300': 'ring-gray-300 dark:ring-gray-700',
        'ring-tw-gray-400': 'ring-gray-400 dark:ring-gray-600',
        'ring-tw-gray-500': 'ring-gray-500 dark:ring-gray-500',
        'ring-tw-gray-600': 'ring-gray-600 dark:ring-gray-400',
        'ring-tw-gray-700': 'ring-gray-700 dark:ring-gray-300',
        'ring-tw-gray-800': 'ring-gray-800 dark:ring-gray-200',
        'ring-tw-gray-900': 'ring-gray-900 dark:ring-gray-100',
        ...shortcuts
      },
      rules: [
        [/^shadow-?(.*)$/, ([, d], { theme }) => {
          // @ts-ignore
          const value = theme?.boxShadow?.[d || 'DEFAULT']
          if (value) {
            return {
              '--un-shadow-color': '0,0,0',
              '--un-shadow': value,
              'box-shadow': 'var(--un-shadow)'
            }
          }
        }],
        ...rules
      ],
      variants
    }

    await installModule(nuxt, { src: '@unocss/nuxt', options })

    const runtimeDir = resolve(__dirname, 'runtime')
    const componentsDir = resolve(__dirname, 'components')

    addPlugin(resolveModule('./plugin', { paths: runtimeDir }), { append: true })

    nuxt.hook('components:dirs', (dirs) => {
      dirs.push({
        path: join(componentsDir, 'elements'),
        prefix
      })
      dirs.push({
        path: join(componentsDir, 'feedback'),
        prefix
      })
      dirs.push({
        path: join(componentsDir, 'forms'),
        prefix
      })
      dirs.push({
        path: join(componentsDir, 'layout'),
        prefix
      })
      dirs.push({
        path: join(componentsDir, 'navigation'),
        prefix
      })
      dirs.push({
        path: join(componentsDir, 'overlays'),
        prefix
      })
    })

    nuxt.options.build.transpile.push('@popperjs/core')
  }
})