woking upgrade dependencie

Author: bjohansebas

commands/upgrade.ts

@@ -6,10 +6,11 @@ import prompts from 'prompts'
 import { coerce, compare } from 'semver'
 import { TRANSFORM_OPTIONS } from '../config'
 import { onCancel, promptSource } from '../utils/share'
+import { installDependency } from '../utils/packageManager'
 
 const transformerDirectory = join(__dirname, '../', 'transforms')
 
-export async function upgrade(source?: string) {
+export async function upgrade(source?: string, options?: Record<string, unknown>) {
   const sourceSelected = source || (await promptSource('Which directory should the codemods be applied to?'))
 
   if (!sourceSelected) {
@@ -88,6 +89,13 @@ export async function upgrade(source?: string) {
   console.log(`> ${results.skipped} codemods were skipped`)
   console.log(`> ${results.failed} codemods failed`)
   console.log(`> ${results.unmodified} codemods were skipped because they didn't change anything`)
+
+  console.log('\n> Installing latest express version')
+  const { upgradeDependency } = options ?? {}
+
+  if (upgradeDependency) {
+    installDependency(["express@latest"], sourceSelected)
+  }
 }
 
 function suggestCodemods(packageJson) {

index.ts

@@ -27,6 +27,8 @@ program
   .command('upgrade')
   .description('Upgrade your express server to the latest version.')
   .argument('[source]', 'Path to source files or directory to transform.')
+  .option('--no-upgrade-dependency', 'Do not upgrade express dependency')
+  .helpOption('-h, --help', 'Display this help message.')
   .action(upgrade)
 
 program.parse(process.argv)

package.json

@@ -4,10 +4,19 @@
   "version": "0.0.1",
   "description": "Codemods for updating express servers.",
   "main": "build/index.js",
-  "contributors": ["Sebastian Beltran <bjohansebas@gmail.com>", "Filip Kudla <filip.kudla.dev@gmail.com>"],
+  "contributors": [
+    "Sebastian Beltran <bjohansebas@gmail.com>",
+    "Filip Kudla <filip.kudla.dev@gmail.com>"
+  ],
   "license": "MIT",
   "bin": "build/index.js",
-  "files": ["build/transforms/*.js", "build/commands/*.js", "build/utils/*.js", "build/config.js", "build/index.js"],
+  "files": [
+    "build/transforms/*.js",
+    "build/commands/*.js",
+    "build/utils/*.js",
+    "build/config.js",
+    "build/index.js"
+  ],
   "scripts": {
     "dev": "tsc -d -w -p tsconfig.json",
     "build": "tsc -d -p tsconfig.json",
@@ -18,7 +27,9 @@
   },
   "dependencies": {
     "commander": "^12.1.0",
+    "execa": "4.1.0",
     "fast-glob": "^3.3.2",
+    "find-up": "^5.0.0",
     "jscodeshift": "^17.1.1",
     "picocolors": "^1.1.1",
     "prompts": "^2.4.2",

utils/packageManager.ts

@@ -0,0 +1,69 @@
+import findUp from 'find-up'
+import { basename } from 'node:path'
+import execa from 'execa'
+
+const ADD_CMD_FLAG = {
+    npm: 'install',
+    yarn: 'add',
+    pnpm: 'add',
+    bun: 'add',
+}  
+
+export function getPkgManager(root: string) {
+  try {
+    const lockFile = findUp.sync(
+      [
+        'package-lock.json',
+        'yarn.lock',
+        'pnpm-lock.yaml',
+        'bun.lock',
+        'bun.lockb',
+      ],
+      { cwd: root }
+    )
+
+    console.log(lockFile)
+
+    if (lockFile) {
+      switch (basename(lockFile)) {
+        case 'package-lock.json':
+          return 'npm'
+        case 'yarn.lock':
+          return 'yarn'
+        case 'pnpm-lock.yaml':
+          return 'pnpm'
+        case 'bun.lock':
+        case 'bun.lockb':
+          return 'bun'
+      }
+    }
+
+    return 'npm'
+  } catch {
+    return 'npm'
+  }
+}
+
+export function installDependency(
+    packageToInstall: string[],
+    root: string
+  ) {
+    if (packageToInstall.length === 0) return
+  
+    const packageManager = getPkgManager(root)
+  
+    if (!packageManager) throw new Error('Failed to find package manager')
+  
+    try {
+      execa.sync(packageManager, [ADD_CMD_FLAG[packageManager], ...packageToInstall], {
+        stdio: ['ignore', 'ignore', 'inherit'],
+        shell: true,
+        cwd: root,
+      })
+    } catch (error) {
+      throw new Error(
+        `Failed to install "${packageToInstall}". Please install it manually.`,
+        { cause: error }
+      )
+    }
+}