diff --git a/src/components/Header.tsx b/src/components/Header.tsx
index e3387aa..f2305d9 100644
--- a/src/components/Header.tsx
+++ b/src/components/Header.tsx
@@ -1,7 +1,9 @@
import { ModeToggle } from "@/components/ThemeSwitcher"
import { Separator } from "@/components/ui/separator"
import { Skeleton } from "@/components/ui/skeleton"
+import { useWebSocketContext } from "@/hooks/use-websocket-context"
import { fetchLoginUser, fetchSetting } from "@/lib/nezha-api"
+import { cn } from "@/lib/utils"
import { useQuery } from "@tanstack/react-query"
import { DateTime } from "luxon"
import { useEffect, useRef, useState } from "react"
@@ -9,6 +11,7 @@ import { useTranslation } from "react-i18next"
import { useNavigate } from "react-router-dom"
import { LanguageSwitcher } from "./LanguageSwitcher"
+import { Button } from "./ui/button"
function Header() {
const { t } = useTranslation()
@@ -21,6 +24,10 @@ function Header() {
refetchOnWindowFocus: true,
})
+ const { lastMessage, connected } = useWebSocketContext()
+
+ const onlineCount = connected ? (lastMessage ? JSON.parse(lastMessage.data).online || 0 : 0) : "..."
+
const siteName = settingData?.data?.site_name
// @ts-expect-error CustomLogo is a global variable
@@ -68,6 +75,15 @@ function Header() {