fix(extensions): address multiple reviewer comments

- ExtensionRegistry.update() now preserves original installed_at timestamp
- Add ExtensionRegistry.restore() for rollback (entry was removed)
- Clean up wrongly installed extension on ID mismatch before rollback
- Remove unused catalog_error parameter from _print_extension_info()

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: iamaeroplane

src/specify_cli/__init__.py

@@ -2472,7 +2472,7 @@ def extension_info(
 
     # Case 1: Found in catalog - show full catalog info
     if ext_info:
-        _print_extension_info(ext_info, manager, catalog_error)
+        _print_extension_info(ext_info, manager)
         return
 
     # Case 2: Installed locally but catalog lookup failed or not in catalog
@@ -2522,7 +2522,7 @@ def extension_info(
     raise typer.Exit(1)
 
 
-def _print_extension_info(ext_info: dict, manager, catalog_error: Optional[Exception] = None):
+def _print_extension_info(ext_info: dict, manager):
     """Print formatted extension info from catalog data."""
     # Header
     verified_badge = " [green]✓ Verified[/green]" if ext_info.get("verified") else ""
@@ -2783,6 +2783,11 @@ def extension_update(
 
                     # 7. Verify extension ID matches
                     if installed_manifest.id != extension_id:
+                        # Remove the wrongly installed extension before raising
+                        try:
+                            manager.remove(installed_manifest.id)
+                        except Exception:
+                            pass  # Best effort cleanup
                         raise ValueError(
                             f"Extension ID mismatch: expected '{extension_id}', got '{installed_manifest.id}'"
                         )
@@ -2840,9 +2845,9 @@ def extension_update(
                             config["hooks"][hook_name].extend(hooks)
                         hook_executor.save_project_config(config)
 
-                    # Restore registry entry
+                    # Restore registry entry (use restore() since entry was removed)
                     if backup_registry_entry:
-                        manager.registry.update(extension_id, backup_registry_entry)
+                        manager.registry.restore(extension_id, backup_registry_entry)
 
                     console.print(f"   [green]✓[/green] Rollback successful")
                 except Exception as rollback_error:

src/specify_cli/extensions.py

@@ -244,6 +244,25 @@ def update(self, extension_id: str, metadata: dict):
         """
         if extension_id not in self.data["extensions"]:
             raise KeyError(f"Extension '{extension_id}' is not installed")
+        # Preserve the original installed_at timestamp
+        existing = self.data["extensions"][extension_id]
+        original_installed_at = existing.get("installed_at")
+        self.data["extensions"][extension_id] = metadata
+        if original_installed_at and "installed_at" not in metadata:
+            self.data["extensions"][extension_id]["installed_at"] = original_installed_at
+        self._save()
+
+    def restore(self, extension_id: str, metadata: dict):
+        """Restore extension metadata to registry without modifying timestamps.
+
+        Use this method for rollback scenarios where you have a complete backup
+        of the registry entry (including installed_at) and want to restore it
+        exactly as it was.
+
+        Args:
+            extension_id: Extension ID
+            metadata: Complete extension metadata including installed_at
+        """
         self.data["extensions"][extension_id] = metadata
         self._save()