import { createResolver, defineNuxtModule } from '@nuxt/kit'
import { getEnv, version } from '../config/env'
import type { BuildInfo } from '~/types'

const { resolve } = createResolver(import.meta.url)

export default defineNuxtModule({
  meta: {
    name: 'elk:build-env',
  },
  async setup(_options, nuxt) {
    const { env, commit, branch } = await getEnv()
    const buildInfo: BuildInfo = {
      version,
      time: +Date.now(),
      commit,
      branch,
      env,
    }

    nuxt.options.runtimeConfig.public.env = env
    nuxt.options.runtimeConfig.public.buildInfo = buildInfo

    nuxt.options.nitro.publicAssets ||= nuxt.options.nitro.publicAssets || []
    if (env === 'canary' || env === 'preview')
      nuxt.options.nitro.publicAssets.push({ dir: resolve('../public-staging') })
    else if (env === 'dev')
      nuxt.options.nitro.publicAssets.push({ dir: resolve('../public-dev') })
  },
})