Spaces:
Sleeping
Sleeping
File size: 2,605 Bytes
f871fed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
'use client'
import { useEffect, useState, useCallback } from 'react'
import { ConnectionError } from '@/lib/types/config'
import { ConnectionErrorOverlay } from '@/components/errors/ConnectionErrorOverlay'
import { getConfig, resetConfig } from '@/lib/config'
interface ConnectionGuardProps {
children: React.ReactNode
}
export function ConnectionGuard({ children }: ConnectionGuardProps) {
const [error, setError] = useState<ConnectionError | null>(null)
const [isChecking, setIsChecking] = useState(true)
const checkConnection = useCallback(async () => {
setIsChecking(true)
setError(null)
// Reset config cache to force a fresh fetch
resetConfig()
try {
const config = await getConfig()
// Check if database is offline
if (config.dbStatus === 'offline') {
setError({
type: 'database-offline',
details: {
message: 'The API server is running, but the database is not accessible',
attemptedUrl: config.apiUrl,
},
})
setIsChecking(false)
return
}
// If we got here, connection is good
setError(null)
setIsChecking(false)
} catch (err) {
// API is unreachable
const errorMessage =
err instanceof Error ? err.message : 'Unknown error occurred'
const attemptedUrl =
typeof window !== 'undefined'
? `${window.location.origin}/api/config`
: undefined
setError({
type: 'api-unreachable',
details: {
message: 'The Open Notebook API server could not be reached',
technicalMessage: errorMessage,
stack: err instanceof Error ? err.stack : undefined,
attemptedUrl,
},
})
setIsChecking(false)
}
}, [])
// Check connection on mount
useEffect(() => {
checkConnection()
}, [checkConnection])
// Add keyboard shortcut for retry (R key)
useEffect(() => {
const handleKeyPress = (e: KeyboardEvent) => {
if (error && (e.key === 'r' || e.key === 'R')) {
e.preventDefault()
checkConnection()
}
}
window.addEventListener('keydown', handleKeyPress)
return () => window.removeEventListener('keydown', handleKeyPress)
}, [error, checkConnection])
// Show overlay if there's an error
if (error) {
return <ConnectionErrorOverlay error={error} onRetry={checkConnection} />
}
// Show nothing while checking (prevents flash of content)
if (isChecking) {
return null
}
// Render children if connection is good
return <>{children}</>
}
|