From 0fc36c865ba46cab42b67e52153cd6115d8087ab Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Mon, 17 May 2021 21:02:31 +0800
Subject: feat: Prepare for connection status badge.

---
 .../src/app/views/timeline-common/Timeline.tsx     | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

(limited to 'FrontEnd/src/app/views/timeline-common')

diff --git a/FrontEnd/src/app/views/timeline-common/Timeline.tsx b/FrontEnd/src/app/views/timeline-common/Timeline.tsx
index 8bdf4d3b..65378563 100644
--- a/FrontEnd/src/app/views/timeline-common/Timeline.tsx
+++ b/FrontEnd/src/app/views/timeline-common/Timeline.tsx
@@ -1,4 +1,5 @@
 import React from "react";
+import { HubConnectionState } from "@microsoft/signalr";
 
 import {
   HttpForbiddenError,
@@ -19,6 +20,7 @@ export interface TimelineProps {
   timelineName?: string;
   reloadKey: number;
   onReload: () => void;
+  onConnectionStateChanged?: (state: HubConnectionState) => void;
 }
 
 const Timeline: React.FC<TimelineProps> = (props) => {
@@ -35,17 +37,29 @@ const Timeline: React.FC<TimelineProps> = (props) => {
     setPosts([]);
   }, [timelineName]);
 
+  const onConnectionStateChanged =
+    React.useRef<((state: HubConnectionState) => void) | null>(null);
+
+  React.useEffect(() => {
+    onConnectionStateChanged.current = props.onConnectionStateChanged ?? null;
+  }, [props.onConnectionStateChanged]);
+
   React.useEffect(() => {
     if (timelineName != null && state === "loaded") {
       const timelinePostUpdate$ = getTimelinePostUpdate$(timelineName);
-      const subscription = timelinePostUpdate$.subscribe(() => {
-        onReload();
-      });
+      const subscription = timelinePostUpdate$.subscribe(
+        ({ update, state }) => {
+          if (update) {
+            onReload();
+          }
+          onConnectionStateChanged.current?.(state);
+        }
+      );
       return () => {
         subscription.unsubscribe();
       };
     }
-  }, [timelineName, state, onReload]);
+  }, [timelineName, state, onReload, onConnectionStateChanged]);
 
   React.useEffect(() => {
     if (timelineName != null) {
-- 
cgit v1.2.3