From 3a7e4970d4628fb208dfe89fd45314e1236bc197 Mon Sep 17 00:00:00 2001
From: CrescentLeaf <crescentleaf@noreply.codeberg.org>
Date: Mon, 8 Sep 2025 03:09:05 +0800
Subject: [PATCH] ui: login & register

---
 client/ui/App.tsx                   | 51 +++++++++++-----------
 client/ui/dialog/LoginDialog.tsx    | 49 +++++++++++++++------
 client/ui/dialog/RegisterDialog.tsx | 67 +++++++++++++++++++++++++++++
 3 files changed, 127 insertions(+), 40 deletions(-)
 create mode 100644 client/ui/dialog/RegisterDialog.tsx

diff --git a/client/ui/App.tsx b/client/ui/App.tsx
index a95c445..5aaed24 100644
--- a/client/ui/App.tsx
+++ b/client/ui/App.tsx
@@ -9,11 +9,11 @@ import User from "../api/client_data/User.ts"
 import RecentChat from "../api/client_data/RecentChat.ts"
 
 import * as React from 'react'
-import * as CryptoES from 'crypto-es'
 import { Button, Dialog, NavigationRail, TextField } from "mdui"
 import Split from 'split.js'
 import 'mdui/jsx.zh-cn.d.ts'
 import { checkApiSuccessOrSncakbar } from "./snackbar.ts";
+import RegisterDialog from "./dialog/RegisterDialog.tsx";
 
 declare global {
     namespace React {
@@ -27,7 +27,7 @@ declare global {
 
 export default function App() {
     const [recentsList, setRecentsList] = React.useState([
-         {
+        {
             id: '0',
             avatar: "https://www.court-records.net/mugshot/aa6-004-maya.png",
             title: "麻油衣酱",
@@ -38,11 +38,11 @@ export default function App() {
             avatar: "https://www.court-records.net/mugshot/aa6-004-maya.png",
             title: "Maya Fey",
             content: "我是绫里真宵, 是一名灵媒师~"
-        }, 
+        },
     ] as RecentChat[])
     const [contactsMap, setContactsMap] = React.useState({
         所有: [
-             {
+            {
                 id: '0',
                 avatar: "https://www.court-records.net/mugshot/aa6-004-maya.png",
                 nickname: "麻油衣酱",
@@ -51,7 +51,7 @@ export default function App() {
                 id: '0',
                 avatar: "https://www.court-records.net/mugshot/aa6-004-maya.png",
                 nickname: "Maya Fey",
-            }, 
+            },
         ],
     } as unknown as { [key: string]: User[] })
     const [navigationItemSelected, setNavigationItemSelected] = React.useState('Recents')
@@ -62,25 +62,16 @@ export default function App() {
     })
 
     const loginDialogRef: React.MutableRefObject<Dialog | null> = React.useRef(null)
-    const inputAccountRef: React.MutableRefObject<TextField | null> = React.useRef(null)
-    const inputPasswordRef: React.MutableRefObject<TextField | null> = React.useRef(null)
-    const registerButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
-    const loginButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
+    const loginInputAccountRef: React.MutableRefObject<TextField | null> = React.useRef(null)
+    const loginInputPasswordRef: React.MutableRefObject<TextField | null> = React.useRef(null)
 
-    useEventListener(loginButtonRef as React.MutableRefObject<Button>, 'click', async () => {
-        const account = inputAccountRef.current!.value
-        const password = inputPasswordRef.current!.value
-
-        const re = await Client.invoke("User.login", {
-            account: account,
-            password: CryptoES.SHA256(password),
-        })
-        
-        if (checkApiSuccessOrSncakbar(re, "登錄失敗")) return
-    })
+    const registerDialogRef: React.MutableRefObject<Dialog | null> = React.useRef(null)
+    const registerInputUserNameRef: React.MutableRefObject<TextField | null> = React.useRef(null)
+    const registerInputNickNameRef: React.MutableRefObject<TextField | null> = React.useRef(null)
+    const registerInputPasswordRef: React.MutableRefObject<TextField | null> = React.useRef(null)
 
     React.useEffect(() => {
-        ;(async () => {
+        ; (async () => {
             Split(['#SideBar', '#ChatFragment'], {
                 sizes: [25, 75],
                 minSize: [200, 400],
@@ -89,7 +80,7 @@ export default function App() {
 
             Client.connect()
             const re = await Client.invoke("User.auth", {
-                access_token: data.access_token,
+                access_token: data.access_token || '',
             })
             if (re.code == 401)
                 loginDialogRef.current!.open = true
@@ -107,10 +98,18 @@ export default function App() {
         }}>
             <LoginDialog
                 loginDialogRef={loginDialogRef}
-                inputAccountRef={inputAccountRef}
-                inputPasswordRef={inputPasswordRef}
-                registerButtonRef={registerButtonRef}
-                loginButtonRef={loginButtonRef} />
+                loginInputAccountRef={loginInputAccountRef}
+                loginInputPasswordRef={loginInputPasswordRef}
+                registerDialogRef={registerDialogRef} />
+
+            <RegisterDialog
+                registerDialogRef={registerDialogRef}
+                registerInputUserNameRef={registerInputUserNameRef}
+                registerInputNickNameRef={registerInputNickNameRef}
+                registerInputPasswordRef={registerInputPasswordRef}
+                loginInputAccountRef={loginInputAccountRef}
+                loginInputPasswordRef={loginInputPasswordRef} />
+
             <mdui-navigation-rail contained value="Recents" ref={navigationRailRef}>
                 <mdui-button-icon icon="menu" slot="top"></mdui-button-icon>
 
diff --git a/client/ui/dialog/LoginDialog.tsx b/client/ui/dialog/LoginDialog.tsx
index d5f2e67..b8705bc 100644
--- a/client/ui/dialog/LoginDialog.tsx
+++ b/client/ui/dialog/LoginDialog.tsx
@@ -1,29 +1,50 @@
 import * as React from 'react'
-import { Button, Dialog, TextField } from "mdui";
+import { Button, Dialog, TextField } from "mdui"
+import useEventListener from "../useEventListener.ts"
+import { checkApiSuccessOrSncakbar } from "../snackbar.ts"
+import Client from "../../api/Client.ts"
+
+import * as CryptoES from 'crypto-es'
+import data from "../../Data.ts";
 
 interface Refs {
-    inputAccountRef: React.MutableRefObject<TextField | null>
-    inputPasswordRef: React.MutableRefObject<TextField | null>
-    registerButtonRef: React.MutableRefObject<Button | null>
-    loginButtonRef: React.MutableRefObject<Button | null>
+    loginInputAccountRef: React.MutableRefObject<TextField | null>
+    loginInputPasswordRef: React.MutableRefObject<TextField | null>
     loginDialogRef: React.MutableRefObject<Dialog | null>
+    registerDialogRef: React.MutableRefObject<Dialog | null>
 }
 
 export default function LoginDialog({
-    inputAccountRef,
-    inputPasswordRef,
-    registerButtonRef,
-    loginButtonRef,
-    loginDialogRef
+    loginInputAccountRef,
+    loginInputPasswordRef,
+    loginDialogRef,
+    registerDialogRef
 }: Refs) {
-    return (
-        <mdui-dialog headline="登录" ref={loginDialogRef}>
+    const loginButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
+    const registerButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
+    useEventListener(registerButtonRef, 'click', () => registerDialogRef.current!.open = true)
+    useEventListener(loginButtonRef, 'click', async () => {
+        const account = loginInputAccountRef.current!.value
+        const password = loginInputPasswordRef.current!.value
 
-            <mdui-text-field label="账号" ref={inputAccountRef}></mdui-text-field>
+        const re = await Client.invoke("User.login", {
+            account: account,
+            password: CryptoES.SHA256(password).toString(CryptoES.Hex),
+        })
+
+        if (checkApiSuccessOrSncakbar(re, "登錄失敗")) return
+
+        data.access_token = re.data!.access_token as string
+        location.reload()
+    })
+    return (
+        <mdui-dialog headline="登錄" ref={loginDialogRef}>
+
+            <mdui-text-field label="用戶 ID / 用戶名" ref={loginInputAccountRef}></mdui-text-field>
             <div style={{
                 height: "10px",
             }}></div>
-            <mdui-text-field label="密码" ref={inputPasswordRef}></mdui-text-field>
+            <mdui-text-field label="密碼" type="password" toggle-password ref={loginInputPasswordRef}></mdui-text-field>
 
             <mdui-button slot="action" variant="text" ref={registerButtonRef}>注册</mdui-button>
             <mdui-button slot="action" variant="text" ref={loginButtonRef}>登录</mdui-button>
diff --git a/client/ui/dialog/RegisterDialog.tsx b/client/ui/dialog/RegisterDialog.tsx
new file mode 100644
index 0000000..c60f117
--- /dev/null
+++ b/client/ui/dialog/RegisterDialog.tsx
@@ -0,0 +1,67 @@
+import * as React from 'react'
+import { Button, Dialog, TextField } from "mdui";
+import useEventListener from "../useEventListener.ts";
+import Client from "../../api/Client.ts";
+import { checkApiSuccessOrSncakbar, snackbar } from "../snackbar.ts";
+
+import * as CryptoES from 'crypto-es'
+
+interface Refs {
+    loginInputAccountRef: React.MutableRefObject<TextField | null>
+    loginInputPasswordRef: React.MutableRefObject<TextField | null>
+    registerInputUserNameRef: React.MutableRefObject<TextField | null>
+    registerInputNickNameRef: React.MutableRefObject<TextField | null>
+    registerInputPasswordRef: React.MutableRefObject<TextField | null>
+    registerDialogRef: React.MutableRefObject<Dialog | null>
+}
+
+export default function RegisterDialog({
+    loginInputAccountRef,
+    loginInputPasswordRef,
+    registerInputUserNameRef,
+    registerInputNickNameRef,
+    registerInputPasswordRef,
+    registerDialogRef
+}: Refs) {
+    const registerBackButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
+    const doRegisterButtonRef: React.MutableRefObject<Button | null> = React.useRef(null)
+    useEventListener(registerBackButtonRef, 'click', () => registerDialogRef.current!.open = false)
+    useEventListener(doRegisterButtonRef, 'click', async () => {
+        const username = registerInputUserNameRef.current!.value
+        const re = await Client.invoke("User.register", {
+            username: username,
+            nickname: registerInputNickNameRef.current!.value,
+            password: CryptoES.SHA256(registerInputPasswordRef.current!.value).toString(CryptoES.Hex),
+        })
+
+        if (checkApiSuccessOrSncakbar(re, "注冊失敗")) return
+        loginInputAccountRef.current!.value = username == "" ? re.data!.userid as string : username
+        loginInputPasswordRef.current!.value = registerInputPasswordRef.current!.value
+
+        registerInputUserNameRef.current!.value = ""
+        registerInputNickNameRef.current!.value = ""
+        registerInputPasswordRef.current!.value = ""
+        registerDialogRef.current!.open = false
+        snackbar({
+            message: "注冊成功!",
+            placement: "top",
+        })
+    })
+    return (
+        <mdui-dialog headline="注冊" ref={registerDialogRef}>
+
+            <mdui-text-field label="用戶名 (可選)" ref={registerInputUserNameRef}></mdui-text-field>
+            <div style={{
+                height: "10px",
+            }}></div>
+            <mdui-text-field label="昵稱" ref={registerInputNickNameRef}></mdui-text-field>
+            <div style={{
+                height: "10px",
+            }}></div>
+            <mdui-text-field label="密码" type="password" toggle-password ref={registerInputPasswordRef}></mdui-text-field>
+
+            <mdui-button slot="action" variant="text" ref={registerBackButtonRef}>返回</mdui-button>
+            <mdui-button slot="action" variant="text" ref={doRegisterButtonRef}>注冊</mdui-button>
+        </mdui-dialog>
+    )
+}
\ No newline at end of file