refactor(cli): code review cleanup fix for tab+tab (#18967)

packages/cli/src/ui/hooks/useRepeatedKeyPress.ts

@@ -0,0 +1,69 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { useRef, useCallback, useEffect, useState } from 'react';
+
+export interface UseRepeatedKeyPressOptions {
+  onRepeat?: (count: number) => void;
+  onReset?: () => void;
+  windowMs: number;
+}
+
+export function useRepeatedKeyPress(options: UseRepeatedKeyPressOptions) {
+  const [pressCount, setPressCount] = useState(0);
+  const pressCountRef = useRef(0);
+  const timerRef = useRef<NodeJS.Timeout | null>(null);
+
+  // To avoid stale closures
+  const optionsRef = useRef(options);
+  useEffect(() => {
+    optionsRef.current = options;
+  }, [options]);
+
+  const resetCount = useCallback(() => {
+    if (timerRef.current) {
+      clearTimeout(timerRef.current);
+      timerRef.current = null;
+    }
+    if (pressCountRef.current > 0) {
+      pressCountRef.current = 0;
+      setPressCount(0);
+      optionsRef.current.onReset?.();
+    }
+  }, []);
+
+  const handlePress = useCallback((): number => {
+    const newCount = pressCountRef.current + 1;
+    pressCountRef.current = newCount;
+    setPressCount(newCount);
+
+    if (timerRef.current) {
+      clearTimeout(timerRef.current);
+    }
+
+    timerRef.current = setTimeout(() => {
+      pressCountRef.current = 0;
+      setPressCount(0);
+      timerRef.current = null;
+      optionsRef.current.onReset?.();
+    }, optionsRef.current.windowMs);
+
+    optionsRef.current.onRepeat?.(newCount);
+
+    return newCount;
+  }, []);
+
+  useEffect(
+    () => () => {
+      if (timerRef.current) {
+        clearTimeout(timerRef.current);
+      }
+    },
+    [],
+  );
+
+  return { pressCount, handlePress, resetCount };
+}