All files / features / auth SessionMonitor.tsx

83.51% Statements 76/91
66.66% Branches 28/42
86.95% Functions 20/23
86.07% Lines 68/79

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129        65x   65x 65x 65x 65x   492x 492x 1264x 1264x 1264x 1264x 1264x 1264x 492x 492x 492x 1x 1x 1x 1x   492x 137x       137x 131x 131x 131x 131x 1x 1x 1x   130x         130x 2x 2x 2x 2x   2x 2x   2x         130x 3x   127x     137x                   1x 1x 1x   1x 1x   1x 1x         492x 10x 10x 10x 10x                                                                      
import { useEffect, useRef, useState, useCallback } from 'react'
import { useNavigate } from 'react-router-dom'
import { useAuthStore } from './authStore'
import { Button } from '@/components/ui/button'
 
const WARNING_BEFORE_MS = 2 * 60 * 1000
const REFRESH_BEFORE_MS = 30 * 1000
const CHECK_INTERVAL_MS = 1000
 
export function SessionMonitor() {
  const navigate = useNavigate()
  const expiresAt = useAuthStore((s) => s.expiresAt)
  const idleExpiresAt = useAuthStore((s) => s.idleExpiresAt)
  const isAuthenticated = useAuthStore((s) => s.isAuthenticated)
  const setAuth = useAuthStore((s) => s.setAuth)
  const clearAuth = useAuthStore((s) => s.clearAuth)
  const user = useAuthStore((s) => s.user)
 
  const [remaining, setRemaining] = useState<number | null>(null)
  const hasRefreshed = useRef(false)
 
  const doClearAndRedirect = useCallback(() => {
    const path = window.location.hash.replace('#', '') || '/'
    sessionStorage.setItem('returnUrl', path)
    clearAuth()
    navigate('/login?expired=true', { replace: true })
  }, [clearAuth, navigate])
 
  IuseEffect(() => {
    if (!isAuthenticated || !idleExpiresAt || !expiresAt) {
      setRemaining(null)
      return
    }
 
    const interval = setInterval(() => {
      const now = Date.now()
      const idleDiff = new Date(idleExpiresAt).getTime() - now
      const absDiff = new Date(expiresAt).getTime() - now
 
      if (absDiff <= 0) {
        doClearAndRedirect()
   I     clearInterval(interval)
        return
      }

      if (idleDiff <= 0) {
        doClearAndRedirect()
        clearInterval(interval)
        return
     I }
 
      if (idleDiff <= REFRESH_BEFORE_MS && !hasRefreshed.current) {
     E   hasRefreshed.current = true
        fetch('/api/auth/refresh', { method: 'POST' })
          .then((res) => {
            if (!res.ok) throw new Error('Refresh failed')
            return res.json()
          })
          .then((data) => {
            if (user) {
              setAuth(user, data.expiresAt, data.idleExpiresAt)
            }
            hasRefreshed.current = false
          })
          .catch(() => {
            doClearAndRedirect()
          })
      }
 
      if (idleDiff <= WARNING_BEFORE_MS) {
        setRemaining(idleDiff)
      } else {
        setRemaining(null)
      }
    }, CHECK_INTERVAL_MS)
 
    return () => clearInterval(interval)
  }I, [isAuthenticated, idleExpiresAt, expiresAt, doClearAndRedirect, setAuth, user])
 
  function handleStayLoggedIn() {
   E fetch('/api/auth/refresh', { method: 'POST' })
      .then((res) => {
        if (!res.ok) throw new Error('Refresh failed')
        return res.json()
      })
      .then((data) => {
        if (user) {
          setAuth(user, data.expiresAt, data.idleExpiresAt)
        }
        setRemaining(null)
        hasRefreshed.current = false
      })
      .catch(() => {
        doClearAndRedirect()
      })
  }
 
  if (!isAuthenticated || remaining === null || remaining > WARNING_BEFORE_MS) return null
 
  const seconds = Math.ceil(remaining / 1000)
  const minutes = Math.floor(seconds / 60)
  const secs = seconds % 60
 
  return (
    <div
      role="dialog"
      className="fixed inset-0 z-50 flex items-center justify-center bg-black/50"
    >
      <div className="w-80 rounded-lg border border-border bg-card p-6 shadow-lg">
        <h2 className="mb-2 text-lg font-semibold">Session Timeout</h2>
        <p className="mb-4 text-sm text-muted-foreground">
          Your session will expire in {minutes}:{secs.toString().padStart(2, '0')}
        </p>
        <div className="mb-4 h-2 w-full overflow-hidden rounded-full bg-muted">
          <div
            className="h-full rounded-full bg-primary transition-all duration-1000"
            style={{
              width: `${(remaining / WARNING_BEFORE_MS) * 100}%`,
            }}
          />
        </div>
        <Button className="w-full" onClick={handleStayLoggedIn}>
          Stay logged in
        </Button>
      </div>
    </div>
  )
}