import { existsSync } from 'node:fs'
import { mkdir, rm, stat, writeFile } from 'node:fs/promises'
import { dirname, join, resolve } from 'node:path'
import { spawnSync } from 'node:child_process'
import { fileURLToPath } from 'node:url'
import {
  collectPackageEntrypoints,
  ecosystemPackageManifestEntry,
  loadEcosystemReleaseContext
} from './lib/ecosystem-packages.mjs'

const scriptDir = dirname(fileURLToPath(import.meta.url))
const sourceRoot = resolve(scriptDir, '..')
const args = process.argv.slice(2)

const readArg = (name, fallback) => {
  const index = args.indexOf(name)
  return index >= 0 ? args[index + 1] : fallback
}

const mode = args.includes('--publish')
  ? 'publish'
  : args.includes('--list')
    ? 'list'
    : 'pack'
const dryRun = args.includes('--dry-run')
const preflight = args.includes('--preflight')
const clean = args.includes('--clean')
const skipExisting = !args.includes('--no-skip-existing')
const npmRegistry = readArg(
  '--registry',
  process.env.FILE_VIEWER_NPM_REGISTRY ||
    process.env.NPM_CONFIG_REGISTRY ||
    process.env.npm_config_registry ||
    'https://registry.npmjs.org/'
)
const packDir = resolve(
  sourceRoot,
  readArg('--pack-dir', process.env.FILE_VIEWER_ECOSYSTEM_PACK_DIR || '.release/ecosystem')
)

const { rootPackage, entries } = await loadEcosystemReleaseContext(sourceRoot)

function run(command, commandArgs, cwd = sourceRoot) {
  console.log(`$ ${[command, ...commandArgs].join(' ')}`)
  const result = spawnSync(command, commandArgs, {
    cwd,
    encoding: 'utf8',
    stdio: 'inherit'
  })
  if (result.status !== 0) {
    throw new Error(`Command failed: ${command} ${commandArgs.join(' ')}`)
  }
}

function runPublish(command, commandArgs, cwd = sourceRoot) {
  console.log(`$ ${[command, ...commandArgs].join(' ')}`)
  const result = spawnSync(command, commandArgs, {
    cwd,
    stdio: 'inherit'
  })
  if (result.status === 0) {
    return 'published'
  }
  throw new Error(`Command failed: ${command} ${commandArgs.join(' ')}`)
}

function capture(command, commandArgs, cwd = sourceRoot) {
  const result = spawnSync(command, commandArgs, {
    cwd,
    encoding: 'utf8',
    stdio: 'pipe'
  })
  return {
    ok: result.status === 0,
    status: result.status,
    stdout: (result.stdout || '').trim(),
    stderr: (result.stderr || '').trim()
  }
}

function verifyNpmAuthentication() {
  const result = capture('npm', ['whoami', '--registry', npmRegistry])
  if (!result.ok || !result.stdout) {
    throw new Error(
      [
        'npm authentication is required before publishing ecosystem packages.',
        'Run `npm login` or `npm adduser` in an interactive terminal, complete MFA/passkey verification, then rerun `pnpm release:ecosystem:publish`.',
        result.stderr || result.stdout
      ]
        .filter(Boolean)
        .join('\n')
    )
  }
  console.log(`npm authenticated as ${result.stdout} on ${npmRegistry}`)
}

function isPackageVersionPublished(entry) {
  const result = capture(
    'npm',
    ['view', `${entry.packageName}@${entry.version}`, 'version', '--registry', npmRegistry]
  )
  return result.ok && result.stdout.trim() === entry.version
}

async function assertDirectory(path, label = path) {
  if (!existsSync(path)) {
    throw new Error(`Missing directory: ${label}`)
  }
  const info = await stat(path)
  if (!info.isDirectory()) {
    throw new Error(`Not a directory: ${label}`)
  }
}

async function assertFile(path, label = path) {
  if (!existsSync(path)) {
    throw new Error(`Missing file: ${label}`)
  }
  const info = await stat(path)
  if (!info.isFile()) {
    throw new Error(`Not a file: ${label}`)
  }
}

async function verifyPackageEntrypoints(entry, { requireFiles }) {
  for (const field of ['main', 'module', 'types']) {
    if (!entry.packageJson[field]) {
      throw new Error(`${entry.packageName} is missing package.json ${field}`)
    }
  }
  if (!entry.packageJson.exports?.['.']) {
    throw new Error(`${entry.packageName} is missing package.json exports["."]`)
  }

  if (!requireFiles) {
    return
  }

  for (const entrypoint of collectPackageEntrypoints(entry.packageJson)) {
    await assertFile(
      join(entry.absoluteDir, entrypoint),
      `${entry.packageName} entrypoint ${entrypoint}. Run pnpm release:ecosystem:build before packing or publishing`
    )
  }
}

async function verifyPackage(entry, options) {
  await assertDirectory(entry.absoluteDir, entry.packageDir)
  await assertFile(join(entry.absoluteDir, 'package.json'), `${entry.packageDir}/package.json`)
  await assertFile(join(entry.absoluteDir, 'README.md'), `${entry.packageDir}/README.md`)
  await assertFile(join(entry.absoluteDir, 'README.en.md'), `${entry.packageDir}/README.en.md`)
  if (entry.packageJson.private === true) {
    throw new Error(`${entry.packageName} is private and cannot be released`)
  }
  if (entry.version !== rootPackage.version) {
    throw new Error(`${entry.packageName} version ${entry.version} does not match root version ${rootPackage.version}`)
  }
  if (entry.packageJson.publishConfig?.access !== 'public') {
    throw new Error(`${entry.packageName} must publish with access=public`)
  }
  await verifyPackageEntrypoints(entry, options)
}

const names = new Set()
for (const entry of entries) {
  if (names.has(entry.packageName)) {
    throw new Error(`Duplicate release package: ${entry.packageName}`)
  }
  names.add(entry.packageName)
  await verifyPackage(entry, { requireFiles: mode !== 'list' && !preflight })
}

if (mode === 'list') {
  console.log(JSON.stringify({
    version: rootPackage.version,
    packages: entries.map(entry => ({
      ...ecosystemPackageManifestEntry(entry),
      tarballName: entry.tarballName
    }))
  }, null, 2))
  process.exit(0)
}

if (mode === 'pack') {
  if (clean) {
    await rm(packDir, { recursive: true, force: true })
  }
  await mkdir(packDir, { recursive: true })
  for (const entry of entries) {
    run('pnpm', ['-C', entry.packageDir, 'pack', '--pack-destination', packDir])
    await assertFile(join(packDir, entry.tarballName), `${entry.packageName} tarball`)
  }
  const manifest = {
    version: rootPackage.version,
    generatedAt: new Date().toISOString(),
    packageCount: entries.length,
    packages: entries.map(ecosystemPackageManifestEntry)
  }
  await writeFile(
    join(packDir, 'npm-release-manifest.json'),
    `${JSON.stringify(manifest, null, 2)}\n`,
    'utf8'
  )
  run('node', ['scripts/verify-ecosystem-tarballs.mjs', '--pack-dir', packDir])
  console.log(`Packed ${entries.length} ecosystem packages into ${packDir}`)
}

if (mode === 'publish') {
  if (!dryRun) {
    verifyNpmAuthentication()
  }
  if (preflight) {
    console.log(`Publish preflight passed for ${entries.length} ecosystem packages.`)
    process.exit(0)
  }
  let publishedCount = 0
  let skippedCount = 0
  for (const entry of entries) {
    if (!dryRun && skipExisting && isPackageVersionPublished(entry)) {
      console.log(`Skipping ${entry.packageName}@${entry.version}: already published on ${npmRegistry}`)
      skippedCount += 1
      continue
    }
    const publishArgs = [
      '-C',
      entry.packageDir,
      'publish',
      '--access',
      'public',
      '--no-git-checks',
      '--ignore-scripts',
      '--registry',
      npmRegistry
    ]
    if (dryRun) {
      publishArgs.push('--dry-run')
    }
    const publishResult = runPublish('pnpm', publishArgs)
    if (publishResult === 'already-published') {
      console.log(`Skipping ${entry.packageName}@${entry.version}: registry reports the version is already published.`)
      skippedCount += 1
    } else {
      publishedCount += 1
    }
  }
  console.log(
    dryRun
      ? `Dry-run published ${publishedCount} ecosystem packages.`
      : `Published ${publishedCount} ecosystem packages; skipped ${skippedCount} already published packages.`
  )
}