fix: update prompt for auto update (#1262)

# Motivation

<!-- Why is this change necessary? -->

# Content

<!-- Please include a summary of the change -->

# Testing

<!-- How was the change tested? -->

# Please check the following before marking your PR as ready for review

- [ ] I have added tests for my changes
- [ ] I have updated the documentation or added new documentation as
needed

Author: rushilpatel0

src/codegen/cli/cli.py

@@ -61,12 +61,12 @@ def version_callback(value: bool):
 
 # Check for updates on startup (non-blocking)
 try:
-    from codegen.cli.commands.update import check_for_updates_on_startup
-
-    # Only check on actual command runs, not completions
+    # Only check when no arguments are passed (just "codegen" to launch TUI)
     import sys
 
-    if not any(arg in sys.argv for arg in ["--help", "-h", "completion", "--version"]):
+    from codegen.cli.commands.update import check_for_updates_on_startup
+
+    if len(sys.argv) == 1:
         check_for_updates_on_startup()
 except ImportError:
     pass  # Update check dependencies not available

src/codegen/cli/commands/update/main.py

@@ -9,7 +9,8 @@
 from rich.console import Console
 
 import codegen
-from .updater import UpdateManager, check_for_updates_on_startup
+
+from .updater import UpdateManager
 
 console = Console()
 

src/codegen/cli/commands/update/updater.py

@@ -23,7 +23,7 @@
 
 # Update configuration
 UPDATE_CHECK_FILE = Path.home() / ".codegen" / "update_check.json"
-UPDATE_CHECK_INTERVAL = timedelta(hours=24)  # Check for updates once per day
+UPDATE_CHECK_INTERVAL = timedelta(hours=12)  # Check for updates once per day
 
 
 class InstallMethod(Enum):
@@ -219,7 +219,7 @@ def _save_last_check_time(self) -> None:
         with open(UPDATE_CHECK_FILE, "w") as f:
             json.dump({"last_check": datetime.now().isoformat()}, f)
 
-    def perform_update(self, target_version: Optional[str] = None, dry_run: bool = False) -> bool:
+    def perform_update(self, target_version: Optional[str] = None, dry_run: bool = False, skip_confirmation: bool = False) -> bool:
         """Perform the update to a specific version or latest."""
         current_version = self._get_current_version()
 
@@ -248,8 +248,8 @@ def perform_update(self, target_version: Optional[str] = None, dry_run: bool = F
         if dry_run:
             return True
 
-        # Confirm update
-        if not self._confirm_update():
+        # Confirm update (skip if already confirmed)
+        if not skip_confirmation and not self._confirm_update():
             self.console.print("[yellow]Update cancelled[/yellow]")
             return False
 
@@ -382,15 +382,23 @@ def _update_via_homebrew(self, target: Version) -> bool:
 
 
 def check_for_updates_on_startup() -> None:
-    """Check for updates on CLI startup (non-blocking)."""
+    """Check for updates on CLI startup with blocking prompt."""
     try:
         # Only check if we haven't checked recently
         manager = UpdateManager()
-        result = manager.check_for_updates(force=False)
+        result = manager.check_for_updates(force=True)
 
         if result.update_available:
-            console.print(f"\n[cyan]ℹ️  A new version of Codegen CLI is available: {result.latest_version}[/cyan]")
-            console.print("[dim]Run 'codegen update' to upgrade[/dim]\n")
+            console.print(f"\n[cyan]ℹ️  A new version of Codegen CLI is available: {result.current_version} → {result.latest_version}[/cyan]")
+
+            if manager.perform_update():
+                console.print("\n[green]✓ Update completed successfully![/green]")
+                console.print("[yellow]Please restart your terminal or run a new codegen command to use the updated version.[/yellow]\n")
+                # Exit after successful update
+                sys.exit(0)
+            else:
+                console.print("\n[red]Update failed. Please try running 'codegen update' manually.[/red]\n")
+
     except Exception:
         # Silently ignore update check failures on startup
         pass