dev.mts

// pnpm i only-allow esno prompts cross-spawn kolorist magicast del -D -w

import type { ASTNode } from 'magicast'
import fs from 'node:fs'
import spawn from 'cross-spawn'
import { cyan } from 'kolorist'
import { loadFile, writeFile } from 'magicast'
import { addVitePlugin } from 'magicast/helpers'
import prompts from 'prompts'

type VueVersion = '3' | '2.7' | '2.6'

const vueVersion: VueVersion[] = ['3', '2.7', '2.6']

const toVitePlugin: Record<VueVersion, string> = {
  3: '@vitejs/plugin-vue',
  2.7: '@vitejs/plugin-vue2',
  2.6: 'vite-plugin-vue2',
}

const toPackageOptions: Record<VueVersion, Record<string, Record<string, any>>> = {
  3: {
    devDependencies: {
      '@vitejs/plugin-vue': 'latest',
      '@vue/compiler-sfc': 'latest',
      '@vue/test-utils': 'latest',
      'vue': 'latest',
    },
  },
  2.7: {
    devDependencies: {
      '@vitejs/plugin-vue2': 'latest',
      '@vue/test-utils': 'legacy',
      'vue': '~2.7',
      'vue-template-compiler': '~2.7',
    },
  },
  2.6: {
    devDependencies: {
      '@vue/composition-api': 'latest',
      '@vue/test-utils': 'legacy',
      'vite-plugin-vue2': 'latest',
      'unplugin-vue2-script-setup': 'latest',
      'vue': '~2.6',
      'vue-template-compiler': '~2.6',
    },
  },
}

async function dev() {
  const { targetVersion }: { targetVersion: VueVersion } = await prompts({
    type: 'select',
    name: 'targetVersion',
    message: 'Select Vue version',
    choices: Array.from(vueVersion, value => ({ title: value, value })),
  })

  if (!targetVersion) {
    return
  }

  console.info(cyan('Fetching origin...'))
  spawn('git', ['pull'], { stdio: 'inherit' })

  console.info(cyan(`Switching to Vue ${targetVersion}...`))
  const mod = await loadFile('./vite.config.ts')

  // imported 表示命名导入的值，默认导入是 default
  // k 和 mod.imports[k].local 和 constructor 三者一致，表示导入取的别名

  // 删掉 vue 相关引入
  const existedVuePlugins: Record<string, boolean> = {}
  for (const k in mod.imports) {
    for (const vueVersion in toVitePlugin) {
      if (mod.imports[k] && [toVitePlugin[vueVersion as VueVersion], 'unplugin-vue2-script-setup/vite'].includes(mod.imports[k].from)) {
        delete mod.imports[k]
        existedVuePlugins[k] = true
      }
    }
  }

  // 删掉 vue 相关插件
  const options = mod.exports.default.$type === 'function-call'
    ? mod.exports.default.$args[0]
    : mod.exports.default
  if (Object.keys(existedVuePlugins).length && options.plugins?.length) {
    for (let i = options.plugins.length - 1; i > 0; i--) {
      const p = options.plugins[i]
      if (p?.$type === 'function-call' && existedVuePlugins[p.$callee]) {
        options.plugins.splice(i, 1)
      }
    }
  }

  // 添加 vue 相关插件
  addVitePlugin(mod, {
    from: toVitePlugin[targetVersion],
    imported: targetVersion === '2.6' ? 'createVuePlugin' : 'default',
    constructor: 'vue',
  })
  if (targetVersion === '2.6') {
    addVitePlugin(mod, {
      from: 'unplugin-vue2-script-setup/vite',
      imported: 'default',
      constructor: 'ScriptSetup',
    })
  }

  await writeFile(mod as unknown as ASTNode, './vite.config.ts')
  spawn('npx', ['eslint', './vite.config.ts', '--fix'], { stdio: 'inherit' })

  let isDepsChanged = false

  const pkg = JSON.parse(fs.readFileSync('./package.json', 'utf-8'))

  // 删除非目标版本的依赖
  for (const ver of vueVersion) {
    if (ver !== targetVersion) {
      for (const option in toPackageOptions[ver]) {
        for (const dep in toPackageOptions[ver][option]) {
          if (pkg[option][dep] && !toPackageOptions[targetVersion][option][dep]) {
            delete pkg[option][dep]
            isDepsChanged = true
          }
        }
      }
    }
  }

  // 添加目标版本的依赖
  for (const option in toPackageOptions[targetVersion]) {
    for (const dep in toPackageOptions[targetVersion][option]) {
      const depVer = toPackageOptions[targetVersion][option][dep]
      if (pkg[option][dep] !== depVer) {
        pkg[option][dep] = depVer
        isDepsChanged = true
      }
    }
  }

  if (isDepsChanged) {
    fs.writeFileSync('./package.json', JSON.stringify(pkg, null, 2))
    console.info(cyan('Linting package.json...'))
    spawn('npx', ['eslint', './package.json', '--fix'], { stdio: 'inherit' })
    await installDependencies()
  }

  spawn.sync('npx', ['vite', '--open', '--force'], { stdio: 'inherit' })

  async function installDependencies() {
    /* if (['darwin', 'linux'].includes(process.platform)) {
      console.info(cyan('Checking pnpm version...'))
      const latestPNPMVersion = spawn.sync('npm', ['view', 'pnpm', 'version']).stdout.toString().trim()
      const currentPNPMVersion = spawn.sync('pnpm', ['-v']).stdout.toString().trim()
      // Mac 自带 curl，Linux 不一定，Windows 不支持指定 pnpm 版本
      if (latestPNPMVersion !== currentPNPMVersion) {
        console.info(cyan('Upgrading pnpm...'))
        try {
          console.info(execSync(`curl -fsSL https://get.pnpm.io/install.sh | env PNPM_VERSION=${latestPNPMVersion} sh -`).toString())
          // const curlProcess = spawn.sync('curl', ['-fsSL', 'https://get.pnpm.io/install.sh'], {
          //   env: { PNPM_VERSION: latestPNPMVersion },
          //   stdio: ['pipe', 'pipe', 'pipe'], // Redirect stdin, stdout, and stderr
          // })
          // if (curlProcess.status === 0) {
          //   // If curl was successful, execute the shell command
          //   const shCommand = 'sh'
          //   const shArgs = ['-']
          //
          //   const shProcess = spawn.sync(shCommand, shArgs, {
          //     input: curlProcess.stdout, // Pass the stdout of curl as input to sh
          //     stdio: ['pipe', 'inherit', 'inherit'], // Redirect stdin, inherit stdout and stderr
          //   })
          //
          //   if (shProcess.status === 0) {
          //     console.info('pnpm installation successful.')
          //   } else {
          //     console.error('pnpm installation failed.')
          //   }
          // } else {
          //   console.error('curl command failed.')
          // }
          console.info(cyan('Setting registry...'))
          spawn.sync('pnpm', ['config', 'set', 'registry', 'https://registry.npmmirror.com'], { stdio: 'inherit' })
          // console.info(cyan('Installing node lts...'))
          // spawn.sync('pnpm', ['env', 'use', '-g', 'lts'], { stdio: 'inherit' })
          console.info(cyan('Installing global packages...'))
          spawn('pnpm', ['add', 'cnpm', '@antfu/ni', '-g'], { stdio: 'inherit' })
          console.info(cyan('Deleting ./node_modules...'))
          await deleteAsync(['./node_modules'])
        } catch (e) {

        }
      }
    } */
    console.info(cyan('Installing dependencies...'))
    spawn.sync('pnpm', ['i'], { stdio: 'inherit' })
    spawn.sync('npx', ['vue-demi-switch', targetVersion === '2.6' ? '2' : targetVersion], { stdio: 'inherit' })
  }
}

try {
  dev()
}
catch (e) {
  console.error(e)
}