|
| 1 | +import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigger } from '@/components/ui/dropdown-menu'; |
| 2 | +import { Button } from '@/components/ui/button'; |
| 3 | +import { ChevronDownIcon, RefreshCwIcon } from 'lucide-react'; |
| 4 | +import { useEffect, useState } from 'react'; |
| 5 | +import { router } from '@inertiajs/react'; |
| 6 | + |
| 7 | +export default function Refresh() { |
| 8 | + const [poll, setPoll] = useState<{ |
| 9 | + stop: VoidFunction; |
| 10 | + start: VoidFunction; |
| 11 | + }>(); |
| 12 | + const [polling, setPolling] = useState(false); |
| 13 | + const storedInterval = (localStorage.getItem('refresh_interval') as '5' | '10' | '30' | '60' | '0') || '10'; |
| 14 | + const [refreshInterval, setRefreshInterval] = useState<5 | 10 | 30 | 60 | 0>( |
| 15 | + storedInterval === '0' ? 0 : (parseInt(storedInterval) as 5 | 10 | 30 | 60), |
| 16 | + ); |
| 17 | + const intervalLabels = { |
| 18 | + 5: '5s', |
| 19 | + 10: '10s', |
| 20 | + 30: '30s', |
| 21 | + 60: '1m', |
| 22 | + 0: 'OFF', |
| 23 | + }; |
| 24 | + |
| 25 | + const refresh = () => { |
| 26 | + router.reload({ |
| 27 | + onStart: () => { |
| 28 | + setPolling(true); |
| 29 | + }, |
| 30 | + onFinish: () => { |
| 31 | + setPolling(false); |
| 32 | + }, |
| 33 | + }); |
| 34 | + }; |
| 35 | + |
| 36 | + useEffect(() => { |
| 37 | + poll?.stop(); |
| 38 | + if (refreshInterval > 0) { |
| 39 | + setPoll(router.poll(refreshInterval * 1000)); |
| 40 | + } else { |
| 41 | + poll?.stop(); |
| 42 | + setPoll(undefined); |
| 43 | + } |
| 44 | + localStorage.setItem('refresh_interval', refreshInterval.toString()); |
| 45 | + // eslint-disable-next-line react-hooks/exhaustive-deps |
| 46 | + }, [refreshInterval]); |
| 47 | + |
| 48 | + return ( |
| 49 | + <div className="flex items-center"> |
| 50 | + <Button variant="outline" size="sm" className="md:rounded-r-none" onClick={refresh} disabled={polling}> |
| 51 | + {polling ? <RefreshCwIcon className="animate-spin" /> : <RefreshCwIcon className="lg:hidden" />} |
| 52 | + <span className="hidden md:block">Refresh</span> |
| 53 | + </Button> |
| 54 | + <DropdownMenu> |
| 55 | + <DropdownMenuTrigger asChild> |
| 56 | + <Button variant="outline" size="sm" className="hidden rounded-l-none border-l-0 md:flex"> |
| 57 | + {intervalLabels[refreshInterval] || 'Unknown'} |
| 58 | + <ChevronDownIcon /> |
| 59 | + </Button> |
| 60 | + </DropdownMenuTrigger> |
| 61 | + <DropdownMenuContent align="end"> |
| 62 | + <DropdownMenuItem onSelect={() => setRefreshInterval(5)}>5s</DropdownMenuItem> |
| 63 | + <DropdownMenuItem onSelect={() => setRefreshInterval(10)}>10s</DropdownMenuItem> |
| 64 | + <DropdownMenuItem onSelect={() => setRefreshInterval(30)}>30s</DropdownMenuItem> |
| 65 | + <DropdownMenuItem onSelect={() => setRefreshInterval(60)}>1m</DropdownMenuItem> |
| 66 | + <DropdownMenuItem onSelect={() => setRefreshInterval(0)}>OFF</DropdownMenuItem> |
| 67 | + </DropdownMenuContent> |
| 68 | + </DropdownMenu> |
| 69 | + </div> |
| 70 | + ); |
| 71 | +} |
0 commit comments