From 0e8103527dee60666c0d9bc4dfced61826b0bf94 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9E=97=E6=B0=B4=E6=BA=B6?= <shuirong1997@icloud.com>
Date: Thu, 14 Mar 2024 00:19:55 +0900
Subject: [PATCH] feat: add client ip for berry theme

---
 .../src/views/Log/component/TableHead.js      |  1 +
 web/berry/src/views/Log/component/TableRow.js |  1 +
 .../src/views/Log/component/TableToolBar.js   | 25 +++++++++++++++++++
 web/berry/src/views/Log/index.js              |  3 ++-
 web/default/src/components/LogsTable.js       |  2 +-
 5 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/web/berry/src/views/Log/component/TableHead.js b/web/berry/src/views/Log/component/TableHead.js
index 671170ce..2170d9f8 100644
--- a/web/berry/src/views/Log/component/TableHead.js
+++ b/web/berry/src/views/Log/component/TableHead.js
@@ -14,6 +14,7 @@ const LogTableHead = ({ userIsAdmin }) => {
         <TableCell>提示</TableCell>
         <TableCell>补全</TableCell>
         <TableCell>额度</TableCell>
+        <TableCell>请求端IP</TableCell>
         <TableCell>详情</TableCell>
       </TableRow>
     </TableHead>
diff --git a/web/berry/src/views/Log/component/TableRow.js b/web/berry/src/views/Log/component/TableRow.js
index 36b6b666..2fd0c7b2 100644
--- a/web/berry/src/views/Log/component/TableRow.js
+++ b/web/berry/src/views/Log/component/TableRow.js
@@ -57,6 +57,7 @@ export default function LogTableRow({ item, userIsAdmin }) {
         <TableCell>{item.prompt_tokens || ''}</TableCell>
         <TableCell>{item.completion_tokens || ''}</TableCell>
         <TableCell>{item.quota ? renderQuota(item.quota, 6) : ''}</TableCell>
+        <TableCell>{item.client_ip || ''}</TableCell>
         <TableCell>{item.content}</TableCell>
       </TableRow>
     </>
diff --git a/web/berry/src/views/Log/component/TableToolBar.js b/web/berry/src/views/Log/component/TableToolBar.js
index fec6907b..f74d24b9 100644
--- a/web/berry/src/views/Log/component/TableToolBar.js
+++ b/web/berry/src/views/Log/component/TableToolBar.js
@@ -5,6 +5,7 @@ import {
   IconKey,
   IconBrandGithubCopilot,
   IconSitemap,
+  IconMapPin
 } from "@tabler/icons-react";
 import {
   InputAdornment,
@@ -81,6 +82,30 @@ export default function TableToolBar({
           />
         </FormControl>
 
+        <FormControl>
+          <InputLabel htmlFor="channel-client_ip-label">请求端IP</InputLabel>
+          <OutlinedInput
+            id="client_ip"
+            name="client_ip"
+            sx={{
+              minWidth: "100%",
+            }}
+            label="请求端IP"
+            value={filterName.client_ip}
+            onChange={handleFilterName}
+            placeholder="请求端IP"
+            startAdornment={
+              <InputAdornment position="start">
+                <IconMapPin
+                  stroke={1.5}
+                  size="20px"
+                  color={grey500}
+                />
+              </InputAdornment>
+            }
+          />
+        </FormControl>
+
         <FormControl>
           <LocalizationProvider
             dateAdapter={AdapterDayjs}
diff --git a/web/berry/src/views/Log/index.js b/web/berry/src/views/Log/index.js
index da24b4fd..c4c00077 100644
--- a/web/berry/src/views/Log/index.js
+++ b/web/berry/src/views/Log/index.js
@@ -28,7 +28,8 @@ export default function Log() {
     start_timestamp: 0,
     end_timestamp: new Date().getTime() / 1000 + 3600,
     type: 0,
-    channel: ''
+    channel: '',
+    client_ip: ''
   };
   const [logs, setLogs] = useState([]);
   const [activePage, setActivePage] = useState(0);
diff --git a/web/default/src/components/LogsTable.js b/web/default/src/components/LogsTable.js
index 96daa8a6..c84ab170 100644
--- a/web/default/src/components/LogsTable.js
+++ b/web/default/src/components/LogsTable.js
@@ -425,7 +425,7 @@ const LogsTable = () => {
                 }}
                 width={1}
               >
-                请求IP
+                请求端IP
               </Table.HeaderCell>
               <Table.HeaderCell
                 style={{ cursor: 'pointer' }}