fix: restore to last cursor not default

antvis · Sep 15, 2023 · 91326a3 · 91326a3
1 parent 20d663c
commit 91326a3
Showing 1 changed file with 21 additions and 9 deletions.
diff --git a/src/interaction/utils.ts b/src/interaction/utils.ts
@@ -447,16 +447,28 @@ export function renderBackground({
   return [append, remove, is] as const;
 }
 
-export function setCursor(root, cursor) {
-  // @ts-ignore
-  const canvas = root.getRootNode().defaultView;
-  const dom = canvas.getContextService().getDomElement();
-  if (dom?.style) dom.style.cursor = cursor;
-}
+export const { setCursor, restoreCursor } = (function () {
+  let originCursor;
+
+  function setCursor(root, cursor) {
+    // @ts-ignore
+    const canvas = root.getRootNode().defaultView;
+    const dom = canvas.getContextService().getDomElement();
+    if (dom?.style) {
+      originCursor = dom.style.cursor;
+      dom.style.cursor = cursor;
+    }
+  }
 
-export function restoreCursor(root) {
-  setCursor(root, 'default');
-}
+  function restoreCursor(root) {
+    setCursor(root, originCursor);
+  }
+
+  return {
+    setCursor,
+    restoreCursor,
+  };
+})();
 
 export function selectElementByData(elements, data, datum) {
   return elements.find((d) =>