import { execSync } from 'child_process'
import chalk from 'chalk'
import { platform } from 'os'
import { existsSync } from 'fs'

// 环境安装指南
const INSTALL_GUIDES = {
  'Node.js': 'https://nodejs.org/zh-cn/download/',
  pnpm: 'https://pnpm.io/zh/installation',
  Rust: 'https://www.rust-lang.org/tools/install',
  'WebView2 Runtime': 'https://developer.microsoft.com/microsoft-edge/webview2/'
}

// 更新指南
const UPDATE_GUIDES = {
  Rust: '请运行 `rustup update` 命令更新 Rust 版本'
}

// Windows 特定的检查路径
const WINDOWS_PATHS = {
  'WebView2 Runtime': [
    'C:\\Program Files (x86)\\Microsoft\\EdgeWebView\\Application',
    'C:\\Program Files\\Microsoft\\EdgeWebView\\Application',
    'C:\\Windows\\SystemApps\\Microsoft.Win32WebViewHost_cw5n1h2txyewy'
  ]
}

// 错误信息映射
const ERROR_MESSAGES = {
  ENOENT: '命令未找到',
  EPERM: '权限不足',
  EACCES: '访问被拒绝'
}

const checks = [
  {
    name: 'Node.js',
    command: 'node --version',
    versionExtractor: (output) => output.replace('v', ''),
    minVersion: '18.0.0',
    isRequired: true
  },
  {
    name: 'pnpm',
    command: 'pnpm --version',
    versionExtractor: (output) => output.trim(),
    minVersion: '9.0.0',
    isRequired: true
  },
  {
    name: 'Rust',
    command: 'rustc --version',
    versionExtractor: (output) => output.split(' ')[1],
    minVersion: '1.86.0',
    isRequired: true
  }
]

/**
 * 检查 WebView2 是否安装
 * @returns {boolean}
 */
const checkWebView2 = () => {
  try {
    // 检查注册表
    const regQuery =
      'reg query "HKLM\\SOFTWARE\\WOW6432Node\\Microsoft\\EdgeUpdate\\Clients\\{F3017226-FE2A-4295-8BDF-00C3A9A7E4C5}" /v pv'
    execSync(regQuery, { stdio: 'ignore' })
    return true
  } catch {
    // 如果注册表查询失败，检查文件路径
    return WINDOWS_PATHS['WebView2 Runtime'].some((path) => existsSync(path))
  }
}

// Windows 特定的检查
const windowsChecks = [
  {
    name: 'WebView2 Runtime',
    checkInstalled: checkWebView2,
    isRequired: true
  }
]

/**
 * 获取友好的错误提示
 * @param {Error} error 错误对象
 * @returns {string} 错误提示
 */
const getFriendlyErrorMessage = (error) => {
  const code = error.code || ''
  return ERROR_MESSAGES[code] || error.message || '未知错误'
}

/**
 * 比较版本号
 * @param {string} version1 当前版本
 * @param {string} version2 所需版本
 * @returns {number} 1: version1 大, -1: version2 大, 0: 相等
 */
const compareVersions = (version1, version2) => {
  const v1 = version1.replace(/[^0-9.]/g, '').split('.')
  const v2 = version2.replace(/[^0-9.]/g, '').split('.')

  for (let i = 0; i < Math.max(v1.length, v2.length); i++) {
    const num1 = parseInt(v1[i] || '0')
    const num2 = parseInt(v2[i] || '0')
    if (num1 > num2) return 1
    if (num1 < num2) return -1
  }
  return 0
}

function checkDependency(check) {
  try {
    const output = execSync(check.command).toString().trim()
    const version = check.versionExtractor(output)
    const isVersionValid = compareVersions(version, check.minVersion) >= 0

    if (isVersionValid) {
      console.log(chalk.green(`✓ ${check.name} 版本 ${output} 已安装\n`))
      return true
    } else {
      console.log(chalk.yellow(`⚠️ ${check.name} 版本过低`))
      console.log(chalk.yellow(`  当前版本: ${output}`))
      console.log(chalk.yellow(`  需要版本: >=${check.minVersion}`))

      // 对 Rust 进行特殊处理，提示使用 rustup update
      if (check.name === 'Rust') {
        console.log(chalk.yellow(`  ${UPDATE_GUIDES[check.name]}`))
      }

      console.log(chalk.gray(`  👉 升级指南: ${INSTALL_GUIDES[check.name]}`))
      return false
    }
  } catch (error) {
    const errorMessage = getFriendlyErrorMessage(error)
    console.log(chalk.red(`✗ ${check.name} 未安装`))
    console.log(chalk.red(`  原因: ${errorMessage}`))
    console.log(chalk.gray(`  👉 安装指南: ${INSTALL_GUIDES[check.name]}`))
    return false
  }
}

/**
 * 检查 Windows 特定的依赖
 * @param {Object} check 检查项
 * @returns {boolean} 是否通过检查
 */
function checkWindowsDependency(check) {
  try {
    const isInstalled = check.checkInstalled()
    if (isInstalled) {
      console.log(chalk.green(`✓ ${check.name} 已安装`))
      return true
    } else {
      console.log(chalk.red(`✗ ${check.name} 未安装`))
      console.log(chalk.gray(`  👉 安装指南: ${INSTALL_GUIDES[check.name]}`))
      return false
    }
  } catch (error) {
    const errorMessage = getFriendlyErrorMessage(error)
    console.log(chalk.red(`✗ ${check.name} 检查失败`))
    console.log(chalk.red(`  原因: ${errorMessage}`))
    return false
  }
}

function main() {
  const isWindows = platform() === 'win32'

  // 执行基本检查
  const results = checks.map(checkDependency)

  // 在 Windows 上执行额外检查
  if (isWindows) {
    console.log(chalk.blue(`\n[HuLa ${new Date().toLocaleTimeString()}] 正在检查 Windows 开发环境...\n`))
    const windowsResults = windowsChecks.map(checkWindowsDependency)
    results.push(...windowsResults)
  }

  if (results.every(Boolean)) {
    console.log(chalk.green('\n✓ 所有环境检查通过！'))
    process.exit(0)
  } else {
    console.log(chalk.red('\n✗ 环境依赖检查失败，请按照上述提示安装或更新依赖。'))
    process.exit(1)
  }
}

main()