import { mkdtempSync, rmSync, writeFileSync } from 'node:fs'
import { createRequire } from 'node:module'
import { tmpdir } from 'node:os'
import { join } from 'node:path'
import { findKeys } from 'i18next-cli'
import { describe, expect, it } from 'vitest'

const require = createRequire(import.meta.url)
const commonJsFindKeys = require('i18next-cli').findKeys

const calls = {
  normal: "translate('probe.normal', 'Normal')",
  block: "translate /* comment */ ('probe.block', 'Block')",
  line: "translate\n// comment\n('probe.line', 'Line')",
  member: "i18n./* member */t /* call */ ('probe.member', 'Member')",
  optional: "translate?. /* optional */ ('probe.optional', 'Optional')",
  generic: "translate /* generic */ <string>('probe.generic', 'Generic')",
  dollar: "$api.t /* member */ ('probe.dollar', 'Dollar')",
  unicode: "transl\\u0061te('probe.unicode', 'Unicode')"
}

describe.each([
  ['ES module', findKeys],
  ['CommonJS', commonJsFindKeys]
])('localization extraction through %s', (_name, extractKeys) => {
  it('keeps keys in calls containing comments or escaped identifiers', async () => {
    const directory = mkdtempSync(join(tmpdir(), 'orca-extraction-syntax-'))
    try {
      for (const [name, source] of Object.entries(calls)) {
        writeFileSync(join(directory, `${name}.ts`), source)
      }
      const errors = []
      const result = await extractKeys(
        {
          locales: ['en'],
          extract: {
            input: [join(directory, '*.ts').replaceAll('\\', '/')],
            defaultNS: false,
            functions: ['t', '*.t', 'translate'],
            disablePlurals: true
          }
        },
        undefined,
        errors
      )
      expect(errors).toEqual([])
      expect([...result.allKeys.values()].map((entry) => entry.key).sort()).toEqual(
        Object.keys(calls)
          .map((name) => `probe.${name}`)
          .sort()
      )
    } finally {
      rmSync(directory, { recursive: true, force: true })
    }
  })

  it('preserves cross-file namespaces, source locations and fresh plugin transformations', async () => {
    const directory = mkdtempSync(join(tmpdir(), 'orca-extraction-passes-'))
    const file = join(directory, 'a-consumer.ts')
    const config = {
      locales: ['en'],
      extract: {
        input: [join(directory, '*.ts').replaceAll('\\', '/')],
        functions: ['t', '*.t'],
        disablePlurals: true
      }
    }
    try {
      writeFileSync(join(directory, 'z-constants.ts'), "export const NS = 'probe' as const\n")
      const source = [
        "import { NS } from './z-constants'",
        '/* 🙂 */ const { t } = useTranslation(NS);',
        "t('first', 'First')"
      ].join('\n')
      writeFileSync(file, source)
      const first = await extractKeys(config)
      expect([...first.allKeys.values()]).toMatchObject([
        { key: 'first', ns: 'probe', locations: [{ file, line: 3, column: 2 }] }
      ])

      writeFileSync(file, source.replace("'first'", "'updated'"))
      const updated = await extractKeys(config)
      expect([...updated.allKeys.values()].map((entry) => entry.key)).toEqual(['updated'])

      const transformed = await extractKeys({
        ...config,
        plugins: [
          {
            name: 'fixture-transform',
            onLoad: (code) => code.replace("'updated'", "'transformed'")
          }
        ]
      })
      expect([...transformed.allKeys.values()].map((entry) => entry.key)).toEqual(['transformed'])
    } finally {
      rmSync(directory, { recursive: true, force: true })
    }
  })
})