codecane 1.0.403 → 1.0.405

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codecane",
-  "version": "1.0.403",
+  "version": "1.0.405",
   "description": "AI coding agent",
   "license": "MIT",
   "bin": {

package/scripts/codebuff-wrapper.js

@@ -30,6 +30,23 @@ const PLATFORM_TARGETS = {
   'win32-x64': 'codebuff-win32-x64.zip',
 }
 
+// Terminal utilities
+const term = {
+  clearLine: () => {
+    if (process.stderr.isTTY) {
+      process.stderr.write('\r\x1b[K')
+    }
+  },
+  write: (text) => {
+    term.clearLine()
+    process.stderr.write(text)
+  },
+  writeLine: (text) => {
+    term.clearLine()
+    process.stderr.write(text + '\n')
+  },
+}
+
 // Utility functions
 function httpGet(url, options = {}) {
   return new Promise((resolve, reject) => {
@@ -112,14 +129,22 @@ function compareVersions(v1, v2) {
   return 0
 }
 
-function showProgress(downloaded, total) {
-  if (total > 0) {
-    const percentage = Math.round((downloaded / total) * 100)
-    process.stderr.write(`\r${percentage}%`)
-  } else {
-    const downloadedMB = (downloaded / 1024 / 1024).toFixed(1)
-    process.stderr.write(`\r${downloadedMB} MB`)
-  }
+function formatBytes(bytes) {
+  if (bytes === 0) return '0 B'
+  const k = 1024
+  const sizes = ['B', 'KB', 'MB', 'GB']
+  const i = Math.floor(Math.log(bytes) / Math.log(k))
+  return parseFloat((bytes / Math.pow(k, i)).toFixed(1)) + ' ' + sizes[i]
+}
+
+function formatSpeed(bytesPerSecond) {
+  return formatBytes(bytesPerSecond) + '/s'
+}
+
+function createProgressBar(percentage, width = 30) {
+  const filled = Math.round((width * percentage) / 100)
+  const empty = width - filled
+  return '[' + '█'.repeat(filled) + '░'.repeat(empty) + ']'
 }
 
 async function downloadBinary(version) {
@@ -135,7 +160,7 @@ async function downloadBinary(version) {
   // Ensure config directory exists
   fs.mkdirSync(CONFIG.configDir, { recursive: true })
 
-  console.log(`Downloading codebuff v${version}...`)
+  term.write(`Downloading codebuff v${version}...`)
 
   const res = await httpGet(downloadUrl)
 
@@ -146,6 +171,7 @@ async function downloadBinary(version) {
   const totalSize = parseInt(res.headers['content-length'] || '0', 10)
   let downloadedSize = 0
   let lastProgressTime = Date.now()
+  let lastDownloadedSize = 0
 
   const chunks = []
 
@@ -155,13 +181,27 @@ async function downloadBinary(version) {
 
     const now = Date.now()
     if (now - lastProgressTime >= 100 || downloadedSize === totalSize) {
+      const elapsedSeconds = (now - lastProgressTime) / 1000
+      const bytesThisInterval = downloadedSize - lastDownloadedSize
+      const speed = bytesThisInterval / elapsedSeconds
+
       lastProgressTime = now
-      showProgress(downloadedSize, totalSize)
+      lastDownloadedSize = downloadedSize
+
+      if (totalSize > 0) {
+        const percentage = Math.round((downloadedSize / totalSize) * 100)
+        const progressBar = createProgressBar(percentage)
+        // const sizeInfo = `${formatBytes(downloadedSize)}/${formatBytes(totalSize)}`
+        const speedInfo = speed > 0 ? formatSpeed(speed) : ''
+
+        term.write(`Downloading... ${progressBar} ${percentage}% ${speedInfo}`)
+      } else {
+        term.write(`Downloading... ${formatBytes(downloadedSize)}`)
+      }
     }
   }
 
-  process.stderr.write('\n')
-  console.log('Extracting...')
+  term.write('Extracting...')
 
   const buffer = Buffer.concat(chunks)
 
@@ -200,6 +240,9 @@ async function downloadBinary(version) {
   } else {
     throw new Error(`Binary not found after extraction`)
   }
+
+  // Clear the line after successful download
+  term.clearLine()
 }
 
 async function ensureBinaryExists() {
@@ -214,6 +257,7 @@ async function ensureBinaryExists() {
     try {
       await downloadBinary(version)
     } catch (error) {
+      term.clearLine()
       console.error('❌ Failed to download codebuff:', error.message)
       console.error('Please try again later.')
       process.exit(1)
@@ -232,7 +276,7 @@ async function ensureBinaryExists() {
   }
 }
 
-async function checkForUpdates(runningProcess) {
+async function checkForUpdates(runningProcess, exitListener) {
   try {
     const currentVersion = getCurrentVersion()
     if (!currentVersion) return
@@ -241,25 +285,43 @@ async function checkForUpdates(runningProcess) {
     if (!latestVersion) return
 
     if (compareVersions(currentVersion, latestVersion) < 0) {
-      console.log(`Updating...\n`)
+      term.clearLine()
+      console.log(`\nUpdate available: ${currentVersion} → ${latestVersion}`)
+
+      // Remove the specific exit listener to prevent it from interfering with the update
+      runningProcess.removeListener('exit', exitListener)
 
       // Kill the running process
       runningProcess.kill('SIGTERM')
 
-      // Wait for graceful shutdown
-      await new Promise((resolve) => setTimeout(resolve, 500))
+      // Wait for the process to actually exit
+      await new Promise((resolve) => {
+        runningProcess.on('exit', resolve)
+        // Fallback timeout in case the process doesn't exit gracefully
+        setTimeout(() => {
+          if (!runningProcess.killed) {
+            runningProcess.kill('SIGKILL')
+          }
+          resolve()
+        }, 5000)
+      })
 
       await downloadBinary(latestVersion)
 
-      // Restart with new binary
+      // Restart with new binary - this replaces the current process
       const newChild = spawn(CONFIG.binaryPath, process.argv.slice(2), {
         stdio: 'inherit',
         cwd: process.cwd(),
+        detached: false,
       })
 
+      // Set up exit handler for the new process
       newChild.on('exit', (code) => {
         process.exit(code || 0)
       })
+
+      // Don't return - keep this function running to maintain the wrapper
+      return new Promise(() => {}) // Never resolves, keeps wrapper alive
     }
   } catch (error) {
     // Silently ignore update check errors
@@ -267,7 +329,6 @@ async function checkForUpdates(runningProcess) {
 }
 
 async function main() {
-  // Ensure binary exists
   await ensureBinaryExists()
 
   // Start codebuff
@@ -276,14 +337,17 @@ async function main() {
     cwd: process.cwd(),
   })
 
+  // Store reference to the exit listener so we can remove it during updates
+  const exitListener = (code) => {
+    process.exit(code || 0)
+  }
+
+  child.on('exit', exitListener)
+
   // Check for updates in background
   setTimeout(() => {
-    checkForUpdates(child)
+    checkForUpdates(child, exitListener)
   }, 100)
-
-  child.on('exit', (code) => {
-    process.exit(code || 0)
-  })
 }
 
 // Run the main function