From ac623b5f93db72ec5432b5ee556717629ff19a6d Mon Sep 17 00:00:00 2001
From: Rocka <i@rocka.me>
Date: Sat, 14 Dec 2024 14:39:54 +0800
Subject: [PATCH 1/2] feat(web): update ui after login or stream operation

---
 web/liveman/components/nodes-table.tsx      | 15 +++++++--
 web/liveman/liveman.tsx                     |  2 +-
 web/shared/api.ts                           |  8 ++---
 web/shared/authorization-middleware.ts      |  2 +-
 web/shared/components/dialog-new-stream.tsx |  6 ++--
 web/shared/components/page-layout.tsx       |  6 ++--
 web/shared/components/streams-table.tsx     | 34 ++++++++++++++-------
 7 files changed, 49 insertions(+), 24 deletions(-)

diff --git a/web/liveman/components/nodes-table.tsx b/web/liveman/components/nodes-table.tsx
index 1aeb949..ae6d9f5 100644
--- a/web/liveman/components/nodes-table.tsx
+++ b/web/liveman/components/nodes-table.tsx
@@ -1,6 +1,9 @@
-import { Button, Checkbox, Dropdown, Link, Table } from 'react-daisyui';
+import { useContext, useEffect } from 'preact/hooks';
+
+import { Badge, Button, Checkbox, Dropdown, Link, Table } from 'react-daisyui';
 import { ArrowPathIcon, EllipsisHorizontalIcon } from '@heroicons/react/24/outline';
 
+import { TokenContext } from '@/shared/context';
 import { useRefreshTimer } from '@/shared/hooks/use-refresh-timer';
 
 import { type Node, getNodes } from '../api';
@@ -12,11 +15,19 @@ async function getNodesSorted() {
 
 export function NodesTable() {
     const nodes = useRefreshTimer([], getNodesSorted);
+    const tokenContext = useContext(TokenContext);
+
+    useEffect(() => {
+        if (tokenContext.token.length > 0) {
+            nodes.updateData();
+        }
+    }, [tokenContext.token]);
 
     return (
         <>
             <div className="flex items-center gap-2 px-4 h-12">
-                <span className="font-bold text-lg mr-auto">Nodes (total: {nodes.data.length})</span>
+                <span className="font-bold text-lg">Nodes</span>
+                <Badge color="ghost" className="font-bold mr-auto">{nodes.data.length}</Badge>
                 <Button
                     size="sm"
                     color="ghost"
diff --git a/web/liveman/liveman.tsx b/web/liveman/liveman.tsx
index a58c98f..e9e7fb2 100644
--- a/web/liveman/liveman.tsx
+++ b/web/liveman/liveman.tsx
@@ -1,7 +1,7 @@
 import { useCallback, useRef, useState } from 'preact/hooks';
 import { Button } from 'react-daisyui';
 
-import type { Stream } from '@/shared/api';
+import { type Stream } from '@/shared/api';
 import { useNeedAuthorization } from '@/shared/hooks/use-need-authorization';
 import { StreamsTable } from '@/shared/components/streams-table';
 import { PageLayout } from '@/shared/components/page-layout';
diff --git a/web/shared/api.ts b/web/shared/api.ts
index 0df7837..df22401 100644
--- a/web/shared/api.ts
+++ b/web/shared/api.ts
@@ -11,15 +11,15 @@ export const addUnauthorizedCallback = authMiddleware.addUnauthorizedCallback;
 export const removeUnauthorizedCallback = authMiddleware.removeUnauthorizedCallback;
 
 export function deleteSession(streamId: string, clientId: string) {
-    return w.url(`/session/${streamId}/${clientId}`).delete();
+    return w.url(`/session/${streamId}/${clientId}`).delete().res();
 }
 
 export function createStream(streamId: string) {
-    return w.url(`/api/streams/${streamId}`).post();
+    return w.url(`/api/streams/${streamId}`).post().res();
 }
 
 export function deleteStream(streamId: string) {
-    return w.url(`/api/streams/${streamId}`).delete();
+    return w.url(`/api/streams/${streamId}`).delete().res();
 }
 
 type SessionConnectionState =
@@ -66,5 +66,5 @@ export function getStreams() {
 }
 
 export function cascade(streamId: string, params: Cascade) {
-    return w.url(`/api/cascade/${streamId}`).post(params);
+    return w.url(`/api/cascade/${streamId}`).post(params).res();
 }
diff --git a/web/shared/authorization-middleware.ts b/web/shared/authorization-middleware.ts
index 30ceb7b..1e71c6d 100644
--- a/web/shared/authorization-middleware.ts
+++ b/web/shared/authorization-middleware.ts
@@ -1,4 +1,4 @@
-import { ConfiguredMiddleware } from 'wretch';
+import { type ConfiguredMiddleware } from 'wretch';
 
 const Authorization = 'Authorization';
 
diff --git a/web/shared/components/dialog-new-stream.tsx b/web/shared/components/dialog-new-stream.tsx
index 0ab09b6..d0017c5 100644
--- a/web/shared/components/dialog-new-stream.tsx
+++ b/web/shared/components/dialog-new-stream.tsx
@@ -6,6 +6,7 @@ import { createStream } from '../api';
 
 interface Props {
     onNewStreamId(id: string): void
+    onStreamCreated(): void
 }
 
 export interface INewStreamDialog {
@@ -29,9 +30,10 @@ export const NewStreamDialog = forwardRef<INewStreamDialog, Props>((props, ref)
         setStreamId(e.currentTarget.value);
     };
 
-    const onConfirmNewStreamId = (_e: Event) => {
+    const onConfirmNewStreamId = async (_e: Event) => {
         props.onNewStreamId(streamId);
-        createStream(streamId);
+        await createStream(streamId);
+        props.onStreamCreated();
     };
 
     return (
diff --git a/web/shared/components/page-layout.tsx b/web/shared/components/page-layout.tsx
index 546e2ad..f47a975 100644
--- a/web/shared/components/page-layout.tsx
+++ b/web/shared/components/page-layout.tsx
@@ -1,9 +1,9 @@
-import { PropsWithChildren } from 'preact/compat';
+import { type PropsWithChildren } from 'preact/compat';
 import { PageHeader } from './page-header';
 
-import { ITokenContext, TokenContext } from '../context';
+import { type ITokenContext, TokenContext } from '../context';
 
-export type PageLayoutProps = ITokenContext & PropsWithChildren;
+export type PageLayoutProps = PropsWithChildren<ITokenContext>;
 
 export function PageLayout({ token, children }: PageLayoutProps) {
     return (
diff --git a/web/shared/components/streams-table.tsx b/web/shared/components/streams-table.tsx
index e5f5b94..d883d95 100644
--- a/web/shared/components/streams-table.tsx
+++ b/web/shared/components/streams-table.tsx
@@ -1,7 +1,7 @@
 import { useState, useRef, useEffect, useContext } from 'preact/hooks';
-import { ReactNode } from 'preact/compat';
+import { type ReactNode } from 'preact/compat';
 
-import { Button, Checkbox, Table } from 'react-daisyui';
+import { Badge, Button, Checkbox, Table } from 'react-daisyui';
 import { ArrowPathIcon, ArrowRightEndOnRectangleIcon, PlusIcon } from '@heroicons/react/24/outline';
 
 import { type Stream, getStreams, deleteStream } from '../api';
@@ -9,11 +9,11 @@ import { formatTime, nextSeqId } from '../utils';
 import { useRefreshTimer } from '../hooks/use-refresh-timer';
 import { TokenContext } from '../context';
 
-import { IClientsDialog, ClientsDialog } from './dialog-clients';
-import { ICascadeDialog, CascadePullDialog, CascadePushDialog } from './dialog-cascade';
-import { IPreviewDialog, PreviewDialog } from './dialog-preview';
-import { IWebStreamDialog, WebStreamDialog } from './dialog-web-stream';
-import { INewStreamDialog, NewStreamDialog } from './dialog-new-stream';
+import { type IClientsDialog, ClientsDialog } from './dialog-clients';
+import { type ICascadeDialog, CascadePullDialog, CascadePushDialog } from './dialog-cascade';
+import { type IPreviewDialog, PreviewDialog } from './dialog-preview';
+import { type IWebStreamDialog, WebStreamDialog } from './dialog-web-stream';
+import { type INewStreamDialog, NewStreamDialog } from './dialog-new-stream';
 
 async function getStreamsSorted() {
     const streams = await getStreams();
@@ -40,6 +40,12 @@ export function StreamsTable(props: StreamTableProps) {
     const refPreviewStreams = useRef<Map<string, IPreviewDialog>>(new Map());
     const tokenContext = useContext(TokenContext);
 
+    useEffect(() => {
+        if (tokenContext.token.length > 0) {
+            streams.updateData();
+        }
+    }, [tokenContext.token]);
+
     const handleViewClients = (id: string) => {
         setSelectedStreamId(id);
         refClients.current?.show();
@@ -114,10 +120,16 @@ export function StreamsTable(props: StreamTableProps) {
         window.open(url);
     };
 
+    const handleDestroyStream = async (id: string) => {
+        await deleteStream(id);
+        await streams.updateData();
+    };
+
     return (
         <>
             <div className="flex items-center gap-2 px-4 h-12">
-                <span className="font-bold text-lg mr-auto">Streams (total: {streams.data.length})</span>
+                <span className="font-bold text-lg">Streams</span>
+                <Badge color="ghost" className="font-bold mr-auto">{streams.data.length}</Badge>
                 {props.showCascade ? (
                     <Button
                         size="sm"
@@ -172,7 +184,7 @@ export function StreamsTable(props: StreamTableProps) {
                                 <Button size="sm" onClick={() => handleOpenPlayerPage(i.id)}>Player</Button>
                                 <Button size="sm" onClick={() => handleOpenDebuggerPage(i.id)}>Debugger</Button>
                                 {props.renderExtraActions?.(i)}
-                                <Button size="sm" color="error" onClick={() => deleteStream(i.id)}>Destroy</Button>
+                                <Button size="sm" color="error" onClick={() => handleDestroyStream(i.id)}>Destroy</Button>
                             </div>
                         </Table.Row>
                     ) : <tr><td colspan={6} className="text-center">N/A</td></tr>}
@@ -187,7 +199,7 @@ export function StreamsTable(props: StreamTableProps) {
                     onClick={handleNewStream}
                 >New Stream</Button>
                 {webStreams.map(s =>
-                    <Button size="sm" onClick={() => { handleOpenWebStream(s); }}>{s}</Button>
+                    <Button size="sm" onClick={() => handleOpenWebStream(s)}>{s}</Button>
                 )}
             </div>
 
@@ -218,7 +230,7 @@ export function StreamsTable(props: StreamTableProps) {
                 />
             )}
 
-            <NewStreamDialog ref={refNewStream} onNewStreamId={handleNewStreamId} />
+            <NewStreamDialog ref={refNewStream} onNewStreamId={handleNewStreamId} onStreamCreated={streams.updateData} />
 
             {webStreams.map(s =>
                 <WebStreamDialog

From 3c17e3e9ca2c97bed6e4b8cf263c51cd87217acd Mon Sep 17 00:00:00 2001
From: Rocka <i@rocka.me>
Date: Sat, 14 Dec 2024 16:06:23 +0800
Subject: [PATCH 2/2] feat(web): refresh streams table after kicking clients

---
 web/liveion/components/login.tsx         | 2 +-
 web/shared/components/dialog-clients.tsx | 8 +++++++-
 web/shared/components/streams-table.tsx  | 1 +
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/web/liveion/components/login.tsx b/web/liveion/components/login.tsx
index 4c343da..8add54a 100644
--- a/web/liveion/components/login.tsx
+++ b/web/liveion/components/login.tsx
@@ -1,6 +1,6 @@
 import { useEffect, useRef, useState } from 'preact/hooks';
 import { TargetedEvent } from 'preact/compat';
-import { Alert, Button, Modal } from 'react-daisyui';
+import { Alert, Button, Loading, Modal } from 'react-daisyui';
 import { WretchError } from 'wretch/resolver';
 
 import * as api from '@/shared/api';
diff --git a/web/shared/components/dialog-clients.tsx b/web/shared/components/dialog-clients.tsx
index 345f065..929c1dd 100644
--- a/web/shared/components/dialog-clients.tsx
+++ b/web/shared/components/dialog-clients.tsx
@@ -8,6 +8,7 @@ import { formatTime } from '../utils';
 interface Props {
     id: string
     sessions: Session[]
+    onClientKicked: () => void
 }
 
 export interface IClientsDialog {
@@ -25,6 +26,11 @@ export const ClientsDialog = forwardRef<IClientsDialog, Props>((props: Props, re
         };
     });
 
+    const handleKickClient = async (streamId: string, clientId: string) => {
+        await deleteSession(streamId, clientId);
+        props.onClientKicked();
+    };
+
     return (
         <Modal ref={refDialog} className="min-w-md max-w-[unset] w-[unset]">
             <Modal.Header className="mb-2">
@@ -44,7 +50,7 @@ export const ClientsDialog = forwardRef<IClientsDialog, Props>((props: Props, re
                                 <span>{c.id + (c.reforward ? '(reforward)' : '')}</span>
                                 <span>{c.state}</span>
                                 <span>{formatTime(c.createdAt)}</span>
-                                <Button size="sm" color="error" onClick={() => deleteSession(props.id, c.id)}>Kick</Button>
+                                <Button size="sm" color="error" onClick={() => handleKickClient(props.id, c.id)}>Kick</Button>
                             </Table.Row>
                         ): <tr><td colspan={4} className="text-center">N/A</td></tr>}
                     </Table.Body>
diff --git a/web/shared/components/streams-table.tsx b/web/shared/components/streams-table.tsx
index d883d95..8e6308e 100644
--- a/web/shared/components/streams-table.tsx
+++ b/web/shared/components/streams-table.tsx
@@ -207,6 +207,7 @@ export function StreamsTable(props: StreamTableProps) {
                 ref={refClients}
                 id={selectedStreamId}
                 sessions={streams.data.find(s => s.id == selectedStreamId)?.subscribe.sessions ?? []}
+                onClientKicked={streams.updateData}
             />
 
             {props.showCascade ? (