feat: support bun

package.json

@@ -30,6 +30,7 @@
   "homepage": "https://github.com/0x-jerry/onrm",
   "license": "MIT",
   "dependencies": {
+    "@iarna/toml": "^2.2.5",
     "chalk": "^5.3.0",
     "cli-highlight": "^2.1.11",
     "inquirer": "^9.2.14",
@@ -38,6 +39,7 @@
   },
   "devDependencies": {
     "@0x-jerry/x-release": "^0.4.0",
+    "@types/iarna__toml": "^2.0.5",
     "@types/inquirer": "^9.0.7",
     "@types/minimist": "^1.2.5",
     "@types/shelljs": "^0.8.15",

src/actions.ts

@@ -1,13 +1,14 @@
 import { getConfig, type ONRMConfig, ONRMRC, type RegistryConfig, saveConfig } from './config'
 import inquirer from 'inquirer'
-import { npm, RegistryManager, yarn } from './registryManager'
+import { bun, npm, RegistryManager, yarn } from './registryManager'
 import chalk from 'chalk'
 import { printTable } from './print'
 import highlight from 'cli-highlight'
 
 const managers: Record<string, RegistryManager> = {
   npm,
-  yarn
+  yarn,
+  bun
 }
 
 async function add(name: string, registry: string, homeUrl: string = '') {
@@ -88,14 +89,14 @@ function use(name: string, type?: 'npm' | 'yarn') {
   }
 
   if (type) {
-    managers[type].setConfig('registry', registryConf.registry)
+    managers[type].setRegistry(registryConf.registry)
     console.log(`Set registry(${chalk.yellow(name)}) for [${chalk.green(type)}] successful!`)
     return
   }
 
   for (const key in managers) {
     const manager = managers[key]
-    manager.setConfig('registry', registryConf.registry)
+    manager.setRegistry(registryConf.registry)
   }
 
   console.log(
@@ -113,7 +114,7 @@ function _printRegistry(registries: ONRMConfig['registries']) {
     const manager = managers[key]
     used.push({
       type: key,
-      registry: manager.getConfig('registry')
+      registry: manager.getRegistry()
     })
   }
 

src/registryManager/base.ts

@@ -3,8 +3,8 @@ import shelljs from 'shelljs'
 export abstract class RegistryManager {
   protected abstract checkIsExist(): boolean
 
-  abstract setConfig(key: string, value: string): boolean
-  abstract getConfig(key: string): string
+  abstract setRegistry(value: string): boolean
+  abstract getRegistry(): string
   abstract getVersion(): string
 
   private _isExist?: boolean

src/registryManager/bun.ts

@@ -0,0 +1,64 @@
+import { RegistryManager } from './base'
+import shelljs from 'shelljs'
+import fs from 'fs'
+import os from 'os'
+import path from 'path'
+import toml from '@iarna/toml'
+
+const CONFIG_FILE = '.bunfig.toml'
+
+class Bun extends RegistryManager {
+  configPath = `${path.join(os.homedir(), CONFIG_FILE)}`
+
+  protected checkIsExist(): boolean {
+    return !!shelljs.which('bun')
+  }
+
+  setRegistry(value: string): boolean {
+    if (!this.isExist()) {
+      return false
+    } else {
+      const conf = this.readBunConfig()
+
+      conf.install ||= {}
+      conf.install.registry = value
+
+      this.saveBunConfig(conf)
+      return true
+    }
+  }
+
+  getRegistry(): string {
+    if (!this.isExist()) {
+      return ''
+    }
+
+    const conf = this.readBunConfig()
+
+    return conf.install?.registry || 'https://registry.npmjs.org/'
+  }
+
+  getVersion(): string {
+    return this.isExist() ? this.exec('bun --version') : ''
+  }
+
+  readBunConfig(): any {
+    try {
+      if (!fs.existsSync(this.configPath)) {
+        return {}
+      }
+
+      const file = fs.readFileSync(this.configPath, 'utf-8')
+      return toml.parse(file)
+    } catch (error) {
+      console.log(`Parse ${this.configPath} failed`, error)
+      return {}
+    }
+  }
+
+  saveBunConfig(config: any) {
+    fs.writeFileSync(this.configPath, toml.stringify(config))
+  }
+}
+
+export const bun = new Bun()

src/registryManager/index.ts

@@ -1,3 +1,4 @@
 export * from './base'
 export * from './npm'
 export * from './yarn'
+export * from './bun'

src/registryManager/npm.ts

@@ -6,21 +6,21 @@ class Npm extends RegistryManager {
     return !!shelljs.which('npm')
   }
 
-  setConfig(key: string, value: string): boolean {
+  setRegistry(value: string): boolean {
     if (!this.isExist()) {
       return false
     } else {
-      this.exec(`npm config set ${key} ${value}`)
+      this.exec(`npm config set registry ${value}`)
       return true
     }
   }
 
-  getConfig(key: string): string {
+  getRegistry(): string {
     if (!this.isExist()) {
       return ''
     }
 
-    return this.exec(`npm config get ${key}`)
+    return this.exec(`npm config get registry`)
   }
 
   getVersion(): string {

src/registryManager/yarn.ts

@@ -10,21 +10,21 @@ class Yarn extends RegistryManager {
     return this.isExist() ? this.exec('yarn -v') : ''
   }
 
-  setConfig(key: string, value: string): boolean {
+  setRegistry(value: string): boolean {
     if (!this.isExist()) {
       return false
     } else {
-      this.exec(`yarn config set ${key} ${value}`)
+      this.exec(`yarn config set registry ${value}`)
       return true
     }
   }
 
-  getConfig(key: string): string {
+  getRegistry(): string {
     if (!this.isExist()) {
       return ''
     }
 
-    const stdout = this.exec(`yarn config get ${key}`)
+    const stdout = this.exec(`yarn config get registry`)
 
     return stdout.replace('warning package.json: No license field', '').trim()
   }