Refactor: Extract shared useConnectionStatus hook and fix test

🤖 Generated with Codebuff Co-Authored-By: Codebuff
<noreply@codebuff.com>

Author: brandonkachen

cli/src/components/status-indicator.tsx

@@ -5,15 +5,7 @@ import { getCodebuffClient } from '../utils/codebuff-client'
 
 import type { ChatTheme } from '../utils/theme-system'
 
-export const StatusIndicator = ({
-  isProcessing,
-  theme,
-  clipboardMessage,
-}: {
-  isProcessing: boolean
-  theme: ChatTheme
-  clipboardMessage?: string | null
-}) => {
+const useConnectionStatus = () => {
   const [isConnected, setIsConnected] = useState<boolean | null>(null)
 
   useEffect(() => {
@@ -39,6 +31,20 @@ export const StatusIndicator = ({
     return () => clearInterval(interval)
   }, [])
 
+  return isConnected
+}
+
+export const StatusIndicator = ({
+  isProcessing,
+  theme,
+  clipboardMessage,
+}: {
+  isProcessing: boolean
+  theme: ChatTheme
+  clipboardMessage?: string | null
+}) => {
+  const isConnected = useConnectionStatus()
+
   if (clipboardMessage) {
     return <span fg={theme.statusAccent}>{clipboardMessage}</span>
   }
@@ -70,30 +76,6 @@ export const useHasStatus = (
   isProcessing: boolean,
   clipboardMessage?: string | null,
 ): boolean => {
-  const [isConnected, setIsConnected] = useState<boolean | null>(null)
-
-  useEffect(() => {
-    const checkConnection = async () => {
-      const client = getCodebuffClient()
-      if (!client) {
-        setIsConnected(false)
-        return
-      }
-
-      try {
-        const connected = await client.checkConnection()
-        setIsConnected(connected)
-      } catch (error) {
-        setIsConnected(false)
-      }
-    }
-
-    checkConnection()
-
-    const interval = setInterval(checkConnection, 30000)
-
-    return () => clearInterval(interval)
-  }, [])
-
+  const isConnected = useConnectionStatus()
   return isConnected === false || isProcessing || !!clipboardMessage
 }