feat(admin): implement admin controls polling and restart prompt (#16627)

packages/cli/src/ui/components/AdminSettingsChangedDialog.test.tsx

@@ -0,0 +1,51 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { renderWithProviders } from '../../test-utils/render.js';
+import { describe, it, expect, vi, afterEach } from 'vitest';
+import { act } from 'react';
+import { AdminSettingsChangedDialog } from './AdminSettingsChangedDialog.js';
+
+const handleRestartMock = vi.fn();
+
+describe('AdminSettingsChangedDialog', () => {
+  afterEach(() => {
+    vi.restoreAllMocks();
+  });
+
+  it('renders correctly', () => {
+    const { lastFrame } = renderWithProviders(<AdminSettingsChangedDialog />);
+    expect(lastFrame()).toMatchSnapshot();
+  });
+
+  it('restarts on "r" key press', async () => {
+    const { stdin } = renderWithProviders(<AdminSettingsChangedDialog />, {
+      uiActions: {
+        handleRestart: handleRestartMock,
+      },
+    });
+
+    act(() => {
+      stdin.write('r');
+    });
+
+    expect(handleRestartMock).toHaveBeenCalled();
+  });
+
+  it.each(['r', 'R'])('restarts on "%s" key press', async (key) => {
+    const { stdin } = renderWithProviders(<AdminSettingsChangedDialog />, {
+      uiActions: {
+        handleRestart: handleRestartMock,
+      },
+    });
+
+    act(() => {
+      stdin.write(key);
+    });
+
+    expect(handleRestartMock).toHaveBeenCalled();
+  });
+});

packages/cli/src/ui/components/AdminSettingsChangedDialog.tsx

@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { Box, Text } from 'ink';
+import { theme } from '../semantic-colors.js';
+import { useKeypress } from '../hooks/useKeypress.js';
+import { useUIActions } from '../contexts/UIActionsContext.js';
+import { Command, keyMatchers } from '../keyMatchers.js';
+
+export const AdminSettingsChangedDialog = () => {
+  const { handleRestart } = useUIActions();
+
+  useKeypress(
+    (key) => {
+      if (keyMatchers[Command.RESTART_APP](key)) {
+        handleRestart();
+      }
+    },
+    { isActive: true },
+  );
+
+  const message =
+    'Admin settings have changed. Please restart the session to apply new settings.';
+
+  return (
+    <Box borderStyle="round" borderColor={theme.status.warning} paddingX={1}>
+      <Text color={theme.status.warning}>
+        {message} Press &apos;r&apos; to restart, or &apos;Ctrl+C&apos; twice to
+        exit.
+      </Text>
+    </Box>
+  );
+};

packages/cli/src/ui/components/DialogManager.tsx

@@ -30,6 +30,7 @@ import { useConfig } from '../contexts/ConfigContext.js';
 import { useSettings } from '../contexts/SettingsContext.js';
 import process from 'node:process';
 import { type UseHistoryManagerReturn } from '../hooks/useHistoryManager.js';
+import { AdminSettingsChangedDialog } from './AdminSettingsChangedDialog.js';
 import { IdeTrustChangeDialog } from './IdeTrustChangeDialog.js';
 
 interface DialogManagerProps {
@@ -50,6 +51,9 @@ export const DialogManager = ({
   const { constrainHeight, terminalHeight, staticExtraHeight, mainAreaWidth } =
     uiState;
 
+  if (uiState.adminSettingsChanged) {
+    return <AdminSettingsChangedDialog />;
+  }
   if (uiState.showIdeRestartPrompt) {
     return <IdeTrustChangeDialog reason={uiState.ideTrustRestartReason} />;
   }

packages/cli/src/ui/components/__snapshots__/AdminSettingsChangedDialog.test.tsx.snap

@@ -0,0 +1,8 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`AdminSettingsChangedDialog > renders correctly 1`] = `
+"╭──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ Admin settings have changed. Please restart the session to apply new settings. Press 'r' to restart, or 'Ctrl+C'     │
+│ twice to exit.                                                                                                       │
+╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯"
+`;