From 0617722410f21916ae96d40d30e0778e2239217d Mon Sep 17 00:00:00 2001
From: Junjie <xjj@123>
Date: 星期五, 26 四月 2024 12:55:04 +0800
Subject: [PATCH] #

---
 zy-asrs-flow/src/pages/map/index.jsx |   93 +++++++++++++++++++++++++++++++++++-----------
 1 files changed, 70 insertions(+), 23 deletions(-)

diff --git a/zy-asrs-flow/src/pages/map/index.jsx b/zy-asrs-flow/src/pages/map/index.jsx
index b8515e5..7c039da 100644
--- a/zy-asrs-flow/src/pages/map/index.jsx
+++ b/zy-asrs-flow/src/pages/map/index.jsx
@@ -1,5 +1,5 @@
 import * as React from 'react'
-import * as PIXI from 'pixi.js';
+import * as TWEEDLE from 'tweedle.js';
 import { FormattedMessage, useIntl, useModel } from '@umijs/max';
 import { Layout, Button, Flex, Row, Col, FloatButton, Select, notification, Segmented } from 'antd';
 const { Header, Content } = Layout;
@@ -8,17 +8,18 @@
     FileAddOutlined,
     CompressOutlined,
     SettingOutlined,
-    CloseOutlined
 } from '@ant-design/icons';
 import './index.css'
 import { createStyles } from 'antd-style';
 import Edit from './components/device';
 import Settings from './components/settings'
 import * as Utils from './utils'
+import WebSocketClient from './websocket'
 import Player from './player';
 import MapSearch from './header/search';
 import MapFloor from './header/floor';
 import MapDrawer from './drawer';
+import BatchDrawer from './batch';
 
 const useStyles = createStyles(({ token }) => {
     let dark = token.colorBgBase === '#000';
@@ -59,7 +60,7 @@
     SETTINGS_MODEL: "3",
 })
 
-let player;
+let player, websocket;
 
 const Map = () => {
     const intl = useIntl();
@@ -78,26 +79,36 @@
     });
     const [app, setApp] = React.useState(null);
     const [mapContainer, setMapContainer] = React.useState(null);
-    const [didClickSprite, setDidClickSprite] = React.useState(false);
     const [spriteBySettings, setSpriteBySettings] = React.useState(null);
     const prevSpriteBySettingsRef = React.useRef();
     const [drawerVisible, setDrawerVisible] = React.useState(false);
     const [dataFetched, setDataFetched] = React.useState(false);
     const [curSprite, setCurSPrite] = React.useState(null);
     const prevCurSpriteRef = React.useRef();
-    const hasFloor = true;
-    // const [hasFloor, setHasFloor] = React.useState(true);
-    const [curFloor, setCurFloor] = React.useState(1);
+    const [floorList, setFloorList] = React.useState([]);
+    const [curFloor, setCurFloor] = React.useState(() => {
+        const storedValue = localStorage.getItem('curFloor');
+        return storedValue !== null ? JSON.parse(storedValue) : null;
+    });
+    const curFloorRef = React.useRef(curFloor);
+    const [batchSprites, setBatchSprites] = React.useState([]);
+    const [batchDrawerVisible, setBatchDrawerVisible] = React.useState(false);
 
     // init func
     React.useEffect(() => {
         const initialize = async () => {
-            player = new Player(mapRef.current, styles.dark, didClickSprite);
+            player = new Player(mapRef.current, styles.dark);
             setApp(player.app);
             setMapContainer(player.mapContainer);
             Utils.syncApp(player.app);
             Utils.syncMapContainer(player.mapContainer);
             Utils.syncNotify(notify);
+
+            websocket = new WebSocketClient('/ws/map/websocket');
+            websocket.connect();
+            websocket.onMessage = (data) => {
+                Utils.updateMapStatusInRealTime(data, () => curFloorRef.current, setCurSPrite);
+            }
 
             const handleResize = () => {
                 setWindowSize({
@@ -107,7 +118,11 @@
             };
             window.addEventListener('resize', handleResize);
 
-            await Utils.fetchMapData(curFloor);
+            const mapFloorData = await Utils.fetchMapFloor();
+            setFloorList(mapFloorData);
+            let defaultFloor = curFloor || mapFloorData?.[0]?.value;
+            setCurFloor(defaultFloor);
+            await Utils.fetchMapData(defaultFloor);
             setDataFetched(true);
             setModel(MapModel.OBSERVER_MODEL)
             setTimeout(() => {
@@ -116,6 +131,13 @@
             }, 200)
         }
         initialize();
+
+        return () => {
+            websocket.onMessage = (data) => {}
+            if (websocket) {
+                websocket.close();
+            }
+        }
     }, []);
 
     // resize
@@ -139,6 +161,8 @@
         setDrawerVisible(false);
         setSpriteBySettings(null);
         setSettingsVisible(false);
+        setBatchSprites([]);
+        setBatchDrawerVisible(false);
 
         switch (model) {
             case MapModel.OBSERVER_MODEL:
@@ -146,7 +170,9 @@
                 player.hideStarryBackground();
 
                 setDeviceVisible(false);
-                player.activateMapEvent(null);
+                player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, (selectedSprites, resetFn) => {
+                    setBatchSprites(selectedSprites);
+                });
 
                 mapContainer.children.forEach(child => {
                     Utils.viewFeature(child, setCurSPrite);
@@ -163,7 +189,7 @@
                 });
 
                 mapContainer.children.forEach(child => {
-                    Utils.beMovable(child, setDidClickSprite);
+                    Utils.beMovable(child);
                 })
                 break
             case MapModel.SETTINGS_MODEL:
@@ -172,10 +198,7 @@
 
                 setDeviceVisible(false);
                 player.activateMapEvent(Utils.MapEvent.SELECTION_BOX, (selectedSprites, resetFn) => {
-                    console.log(selectedSprites);
-                    if (selectedSprites.length > 0) {
-                        setDrawerVisible(true);
-                    }
+                    setBatchSprites(selectedSprites);
                 });
 
                 mapContainer.children.forEach(child => {
@@ -203,7 +226,7 @@
 
         Utils.initSprite(sprite, type);
         mapContainer.addChild(sprite);
-        Utils.beMovable(sprite, setDidClickSprite);
+        Utils.beMovable(sprite);
     };
 
     // watch curSprite
@@ -239,16 +262,13 @@
 
     // watch curFloor
     React.useEffect(() => {
+        curFloorRef.current = curFloor;
         if (!mapContainer && !dataFetched) {
             return;
         }
         switchFloor(curFloor);
+        localStorage.setItem('curFloor', JSON.stringify(curFloor));
     }, [curFloor]);
-
-    // didClickSprite, stop triggers both sprite click and play's selection boxs
-    React.useEffect(() => {
-        player.updateDidClickSprite(didClickSprite);
-    }, [didClickSprite])
 
     // watch spriteBySettings
     React.useEffect(() => {
@@ -268,6 +288,19 @@
     }, [spriteBySettings])
     const prevSpriteBySettings = prevSpriteBySettingsRef.current;
 
+    // watch batchSprites
+    React.useEffect(() => {
+        if (!mapContainer) {
+            return;
+        }
+        if (batchSprites?.length > 0) {
+            setBatchDrawerVisible(true)
+        } else {
+            player.clearSelectedSprites();
+            setBatchDrawerVisible(false)
+        }
+    }, [batchSprites])
+
     return (
         <>
             {contextHolder}
@@ -280,6 +313,7 @@
                                     model={model}
                                     setModel={setModel}
                                     ModelEnum={MapModel}
+                                    curFloor={curFloor}
                                     curSprite={curSprite}
                                     setCurSPrite={setCurSPrite}
                                     setSpriteBySettings={setSpriteBySettings}
@@ -352,8 +386,9 @@
                 <Content ref={contentRef} className={styles.content}>
                     <div ref={mapRef} />
 
-                    {hasFloor && (
+                    {floorList.length > 0 && (
                         <MapFloor
+                            floorList={floorList}
                             curFloor={curFloor}
                             setCurFloor={setCurFloor}
                         />
@@ -412,10 +447,23 @@
             <MapDrawer
                 open={drawerVisible}
                 curSprite={curSprite}
+                curFloor={curFloor}
                 refCurr={mapRef.current}
                 onCancel={() => {
                     setCurSPrite(null);
                     setDrawerVisible(false);
+                }}
+            />
+
+            <BatchDrawer
+                open={batchDrawerVisible}
+                batchSprites={batchSprites}
+                refCurr={mapRef.current}
+                model={model}
+                ModelEnum={MapModel}
+                onCancel={() => {
+                    setBatchSprites([]);
+                    setBatchDrawerVisible(false);
                 }}
             />
 
@@ -436,7 +484,6 @@
                     setSpriteBySettings(null);
                 }}
                 setSpriteBySettings={setSpriteBySettings}
-                setDidClickSprite={setDidClickSprite}
                 refCurr={mapRef.current}
             />
         </>

--
Gitblit v1.9.1