|
1 | 1 | import { Layers } from "constants/Layers";
|
2 |
| -import React, { ReactNode } from "react"; |
| 2 | +import React, { ReactNode, useCallback, useRef, useState, useEffect } from "react"; |
3 | 3 |
|
4 | 4 | export type CheckSelectFn = (
|
5 | 5 | item?: HTMLDivElement | null,
|
@@ -29,159 +29,154 @@ interface SectionState {
|
29 | 29 | mouseDown: boolean;
|
30 | 30 | selectionBox?: Rect;
|
31 | 31 | startPoint?: Point;
|
| 32 | + appendMode: boolean; |
32 | 33 | }
|
33 | 34 |
|
34 |
| -const InitialState = { |
| 35 | +const createInitialState = (): SectionState => ({ |
35 | 36 | mouseDown: false,
|
36 | 37 | appendMode: false,
|
37 | 38 | selectionBox: undefined,
|
38 | 39 | startPoint: undefined,
|
39 |
| -}; |
40 |
| - |
41 |
| -class DragSelectorComp extends React.Component<SectionProps, SectionState> { |
42 |
| - private readonly selectAreaRef: React.RefObject<HTMLDivElement>; |
43 |
| - |
44 |
| - constructor(props: SectionProps) { |
45 |
| - super(props); |
46 |
| - this.selectAreaRef = React.createRef<HTMLDivElement>(); |
47 |
| - this.state = InitialState; |
48 |
| - this._onMouseMove = this._onMouseMove.bind(this); |
49 |
| - this._onMouseUp = this._onMouseUp.bind(this); |
50 |
| - } |
51 |
| - |
52 |
| - componentWillUnmount(): void { |
53 |
| - window.document.removeEventListener("mousemove", this._onMouseMove); |
54 |
| - window.document.removeEventListener("mouseup", this._onMouseUp); |
55 |
| - } |
56 |
| - |
57 |
| - _onMouseDown(e: React.MouseEvent<HTMLDivElement>) { |
58 |
| - if (e.button === 2 || e.nativeEvent.which === 2) { |
59 |
| - return; |
60 |
| - } |
61 |
| - let nextState: SectionState = { mouseDown: false }; |
62 |
| - nextState.mouseDown = true; |
63 |
| - nextState.startPoint = { |
64 |
| - x: e.pageX - (this.selectAreaRef.current?.getBoundingClientRect().left ?? 0), |
65 |
| - y: e.pageY - (this.selectAreaRef.current?.getBoundingClientRect().top ?? 0), |
| 40 | +}); |
| 41 | + |
| 42 | +export const DragSelector = React.memo((props: SectionProps) => { |
| 43 | + const selectAreaRef = useRef<HTMLDivElement>(null); |
| 44 | + const [state, setState] = useState<SectionState>(createInitialState()); |
| 45 | + const mountedRef = useRef(true); |
| 46 | + |
| 47 | + // Cleanup on unmount |
| 48 | + useEffect(() => { |
| 49 | + return () => { |
| 50 | + mountedRef.current = false; |
| 51 | + // Clean up any remaining event listeners |
| 52 | + window.document.removeEventListener("mousemove", handleMouseMove); |
| 53 | + window.document.removeEventListener("mouseup", handleMouseUp); |
| 54 | + }; |
| 55 | + }, []); |
| 56 | + |
| 57 | + const handleMouseMove = useCallback((e: MouseEvent) => { |
| 58 | + if (!mountedRef.current || !state.mouseDown) return; |
| 59 | + |
| 60 | + const endPoint = { |
| 61 | + x: e.pageX - (selectAreaRef.current?.getBoundingClientRect().left ?? 0), |
| 62 | + y: e.pageY - (selectAreaRef.current?.getBoundingClientRect().top ?? 0), |
66 | 63 | };
|
67 |
| - this.setState(nextState); |
68 |
| - window.document.addEventListener("mousemove", this._onMouseMove); |
69 |
| - window.document.addEventListener("mouseup", this._onMouseUp); |
70 |
| - this.props.onMouseDown(); |
71 |
| - } |
72 |
| - |
73 |
| - _onMouseUp() { |
74 |
| - window.document.removeEventListener("mousemove", this._onMouseMove); |
75 |
| - window.document.removeEventListener("mouseup", this._onMouseUp); |
76 |
| - this.props.onMouseUp(); |
77 |
| - this.setState(InitialState); |
78 |
| - } |
79 |
| - |
80 |
| - _onMouseMove(e: MouseEvent) { |
81 |
| - if (this.state.mouseDown) { |
82 |
| - let endPoint = { |
83 |
| - x: e.pageX - (this.selectAreaRef.current?.getBoundingClientRect().left ?? 0), |
84 |
| - y: e.pageY - (this.selectAreaRef.current?.getBoundingClientRect().top ?? 0), |
85 |
| - }; |
86 |
| - this.setState({ |
87 |
| - selectionBox: this._calculateSelectionBox(this.state.startPoint, endPoint), |
88 |
| - }); |
| 64 | + |
| 65 | + setState(prevState => ({ |
| 66 | + ...prevState, |
| 67 | + selectionBox: calculateSelectionBox(prevState.startPoint, endPoint), |
| 68 | + })); |
| 69 | + |
| 70 | + // Clean up selection properly |
| 71 | + const selection = window.getSelection(); |
| 72 | + if (selection) { |
| 73 | + selection.removeAllRanges(); |
89 | 74 | }
|
90 |
| - // Disable selection of text during mouse movement |
91 |
| - var selection = window.getSelection(); |
92 |
| - selection!.removeAllRanges(); |
93 |
| - selection = null; |
94 |
| - this.props.onMouseMove(this.childrenViewCheckFunc); |
95 |
| - } |
96 |
| - |
97 |
| - rectIntersect = ( |
| 75 | + |
| 76 | + props.onMouseMove(childrenViewCheckFunc); |
| 77 | + }, [state.mouseDown, state.startPoint, props.onMouseMove]); |
| 78 | + |
| 79 | + const handleMouseUp = useCallback(() => { |
| 80 | + if (!mountedRef.current) return; |
| 81 | + |
| 82 | + window.document.removeEventListener("mousemove", handleMouseMove); |
| 83 | + window.document.removeEventListener("mouseup", handleMouseUp); |
| 84 | + props.onMouseUp(); |
| 85 | + setState(createInitialState()); |
| 86 | + }, [handleMouseMove, props.onMouseUp]); |
| 87 | + |
| 88 | + const handleMouseDown = useCallback((e: React.MouseEvent<HTMLDivElement>) => { |
| 89 | + if (!mountedRef.current || e.button === 2 || e.nativeEvent.which === 2) return; |
| 90 | + |
| 91 | + const startPoint = { |
| 92 | + x: e.pageX - (selectAreaRef.current?.getBoundingClientRect().left ?? 0), |
| 93 | + y: e.pageY - (selectAreaRef.current?.getBoundingClientRect().top ?? 0), |
| 94 | + }; |
| 95 | + |
| 96 | + setState({ |
| 97 | + mouseDown: true, |
| 98 | + startPoint, |
| 99 | + selectionBox: undefined, |
| 100 | + appendMode: false, |
| 101 | + }); |
| 102 | + |
| 103 | + window.document.addEventListener("mousemove", handleMouseMove); |
| 104 | + window.document.addEventListener("mouseup", handleMouseUp); |
| 105 | + props.onMouseDown(); |
| 106 | + }, [handleMouseMove, handleMouseUp, props.onMouseDown]); |
| 107 | + |
| 108 | + const rectIntersect = useCallback(( |
98 | 109 | selectionBox: Rect | undefined,
|
99 | 110 | item: HTMLElement | null | undefined
|
100 | 111 | ): boolean => {
|
101 |
| - if (!selectionBox || !item) { |
102 |
| - return false; |
103 |
| - } |
| 112 | + if (!selectionBox || !item || !selectAreaRef.current) return false; |
| 113 | + |
| 114 | + const containerRect = selectAreaRef.current.getBoundingClientRect(); |
104 | 115 | const itemBox = {
|
105 |
| - top: |
106 |
| - item.getBoundingClientRect().top - |
107 |
| - (this.selectAreaRef.current?.getBoundingClientRect().top ?? 0), |
108 |
| - left: |
109 |
| - item.getBoundingClientRect().left - |
110 |
| - (this.selectAreaRef.current?.getBoundingClientRect().left ?? 0), |
| 116 | + top: item.getBoundingClientRect().top - containerRect.top, |
| 117 | + left: item.getBoundingClientRect().left - containerRect.left, |
111 | 118 | width: item.getBoundingClientRect().width,
|
112 | 119 | height: item.getBoundingClientRect().height,
|
113 | 120 | };
|
| 121 | + |
114 | 122 | return (
|
115 | 123 | selectionBox.left <= itemBox.left + itemBox.width &&
|
116 | 124 | selectionBox.left + selectionBox.width >= itemBox.left &&
|
117 | 125 | selectionBox.top <= itemBox.top + itemBox.height &&
|
118 | 126 | selectionBox.top + selectionBox.height >= itemBox.top
|
119 | 127 | );
|
120 |
| - }; |
| 128 | + }, []); |
121 | 129 |
|
122 |
| - childrenViewCheckFunc = ( |
| 130 | + const childrenViewCheckFunc = useCallback(( |
123 | 131 | item?: HTMLDivElement | null,
|
124 | 132 | afterCheck?: (checkResult: boolean) => void
|
125 | 133 | ) => {
|
126 |
| - const result = this.rectIntersect(this.state.selectionBox, item); |
127 |
| - if (!!afterCheck) { |
| 134 | + const result = rectIntersect(state.selectionBox, item); |
| 135 | + if (afterCheck) { |
128 | 136 | afterCheck(result);
|
129 | 137 | }
|
130 | 138 | return result;
|
131 |
| - }; |
| 139 | + }, [state.selectionBox, rectIntersect]); |
132 | 140 |
|
133 |
| - render() { |
134 |
| - return ( |
135 |
| - <div |
136 |
| - ref={this.selectAreaRef} |
137 |
| - onMouseDown={this._onMouseDown.bind(this)} |
138 |
| - style={{ position: "relative" }} |
139 |
| - > |
140 |
| - {this.props.children} |
141 |
| - {this.renderSelectionBox()} |
142 |
| - </div> |
143 |
| - ); |
144 |
| - } |
145 |
| - |
146 |
| - renderSelectionBox() { |
147 |
| - if ( |
148 |
| - !this.state.mouseDown || |
149 |
| - !this.state.startPoint || |
150 |
| - !this.state.selectionBox || |
151 |
| - !this.selectAreaRef.current |
152 |
| - ) { |
| 141 | + const calculateSelectionBox = useCallback((startPoint: Point | undefined, endPoint: Point) => { |
| 142 | + if (!state.mouseDown || !startPoint || !endPoint) return undefined; |
| 143 | + |
| 144 | + return { |
| 145 | + left: Math.min(startPoint.x, endPoint.x), |
| 146 | + top: Math.min(startPoint.y, endPoint.y), |
| 147 | + width: Math.abs(startPoint.x - endPoint.x), |
| 148 | + height: Math.abs(startPoint.y - endPoint.y), |
| 149 | + }; |
| 150 | + }, [state.mouseDown]); |
| 151 | + |
| 152 | + const renderSelectionBox = useCallback(() => { |
| 153 | + if (!state.mouseDown || !state.startPoint || !state.selectionBox || !selectAreaRef.current) { |
153 | 154 | return null;
|
154 | 155 | }
|
| 156 | + |
155 | 157 | return (
|
156 | 158 | <div
|
157 | 159 | style={{
|
158 | 160 | background: "rgba(51, 119, 255, 0.1)",
|
159 | 161 | position: "absolute",
|
160 | 162 | zIndex: Layers.dragSelectBox,
|
161 |
| - left: this.state.selectionBox.left, |
162 |
| - top: this.state.selectionBox.top, |
163 |
| - height: this.state.selectionBox.height, |
164 |
| - width: this.state.selectionBox.width, |
| 163 | + left: state.selectionBox.left, |
| 164 | + top: state.selectionBox.top, |
| 165 | + height: state.selectionBox.height, |
| 166 | + width: state.selectionBox.width, |
165 | 167 | }}
|
166 | 168 | />
|
167 | 169 | );
|
168 |
| - } |
169 |
| - |
170 |
| - _calculateSelectionBox(startPoint: Point | undefined, endPoint: Point) { |
171 |
| - if (!this.state.mouseDown || !startPoint || !endPoint) { |
172 |
| - return undefined; |
173 |
| - } |
174 |
| - let left = Math.min(startPoint.x, endPoint.x); |
175 |
| - let top = Math.min(startPoint.y, endPoint.y); |
176 |
| - let width = Math.abs(startPoint.x - endPoint.x); |
177 |
| - let height = Math.abs(startPoint.y - endPoint.y); |
178 |
| - return { |
179 |
| - left: left, |
180 |
| - top: top, |
181 |
| - width: width, |
182 |
| - height: height, |
183 |
| - }; |
184 |
| - } |
185 |
| -} |
186 |
| - |
187 |
| -export const DragSelector = React.memo(DragSelectorComp); |
| 170 | + }, [state.mouseDown, state.startPoint, state.selectionBox]); |
| 171 | + |
| 172 | + return ( |
| 173 | + <div |
| 174 | + ref={selectAreaRef} |
| 175 | + onMouseDown={handleMouseDown} |
| 176 | + style={{ position: "relative" }} |
| 177 | + > |
| 178 | + {props.children} |
| 179 | + {renderSelectionBox()} |
| 180 | + </div> |
| 181 | + ); |
| 182 | +}); |
0 commit comments