Auto-update notifications (#1110)

3 files changed, 67 insertions, 0 deletions

diff --git a/packages/cli/src/ui/App.tsx b/packages/cli/src/ui/App.tsx
index 9a4ecbd3..a5317b30 100644
--- a/packages/cli/src/ui/App.tsx
+++ b/packages/cli/src/ui/App.tsx
@@ -60,6 +60,8 @@ import {
 import { useGitBranchName } from './hooks/useGitBranchName.js';
 import { useTextBuffer } from './components/shared/text-buffer.js';
 import * as fs from 'fs';
+import { UpdateNotification } from './components/UpdateNotification.js';
+import { checkForUpdates } from './utils/updateCheck.js';
 
 const CTRL_EXIT_PROMPT_DURATION_MS = 1000;
 
@@ -76,6 +78,12 @@ export const AppWrapper = (props: AppProps) => (
 );
 
 const App = ({ config, settings, startupWarnings = [] }: AppProps) => {
+  const [updateMessage, setUpdateMessage] = useState<string | null>(null);
+
+  useEffect(() => {
+    checkForUpdates().then(setUpdateMessage);
+  }, []);
+
   const { history, addItem, clearItems, loadHistory } = useHistory();
   const {
     consoleMessages,
@@ -467,6 +475,7 @@ const App = ({ config, settings, startupWarnings = [] }: AppProps) => {
             <Box flexDirection="column" key="header">
               <Header terminalWidth={terminalWidth} />
               <Tips config={config} />
+              {updateMessage && <UpdateNotification message={updateMessage} />}
             </Box>,
             ...history.map((h) => (
               <HistoryItemDisplay
diff --git a/packages/cli/src/ui/components/UpdateNotification.tsx b/packages/cli/src/ui/components/UpdateNotification.tsx
new file mode 100644
index 00000000..b88c9bd5
--- /dev/null
+++ b/packages/cli/src/ui/components/UpdateNotification.tsx
@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { Box, Text } from 'ink';
+import { Colors } from '../colors.js';
+
+interface UpdateNotificationProps {
+  message: string;
+}
+
+export const UpdateNotification = ({ message }: UpdateNotificationProps) => (
+  <Box
+    borderStyle="round"
+    borderColor={Colors.AccentYellow}
+    paddingX={1}
+    marginY={1}
+  >
+    <Text color={Colors.AccentYellow}>{message}</Text>
+  </Box>
+);
diff --git a/packages/cli/src/ui/utils/updateCheck.ts b/packages/cli/src/ui/utils/updateCheck.ts
new file mode 100644
index 00000000..17d30378
--- /dev/null
+++ b/packages/cli/src/ui/utils/updateCheck.ts
@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import updateNotifier from 'update-notifier';
+import { readPackageUp } from 'read-package-up';
+import process from 'node:process';
+
+export async function checkForUpdates(): Promise<string | null> {
+  // read-package-up looks for the closest package.json from cwd
+  const pkgResult = await readPackageUp({ cwd: process.cwd() });
+  if (!pkgResult) {
+    return null;
+  }
+
+  const { packageJson } = pkgResult;
+  const notifier = updateNotifier({
+    pkg: {
+      name: packageJson.name,
+      version: packageJson.version,
+    },
+    // check every time
+    updateCheckInterval: 0,
+    // allow notifier to run in scripts
+    shouldNotifyInNpmScript: true,
+  });
+
+  if (notifier.update) {
+    return `Gemini CLI update available! ${notifier.update.current} → ${notifier.update.latest}\nRun npm install -g ${packageJson.name} to update`;
+  }
+
+  return null;
+}