From 6497c2013f977d607f87acd09e5e4009edaf42c4 Mon Sep 17 00:00:00 2001
From: redhoodsu <redhoodsu@tencent.com>
Date: Mon, 25 Dec 2023 19:01:12 +0800
Subject: [PATCH] release(dom-viewer): v1.3.0

---
 src/dom-viewer/CHANGELOG.md |   4 ++
 src/dom-viewer/README.md    |   1 +
 src/dom-viewer/index.ts     | 103 +++++++++++++++++++++++++++++++++---
 src/dom-viewer/package.json |   2 +-
 4 files changed, 103 insertions(+), 7 deletions(-)

diff --git a/src/dom-viewer/CHANGELOG.md b/src/dom-viewer/CHANGELOG.md
index e365e41..cfcb6f4 100644
--- a/src/dom-viewer/CHANGELOG.md
+++ b/src/dom-viewer/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 1.3.0 (25 Dec 2023)
+
+* feat: hotkey for navigation
+
 ## 1.2.4 (19 Apr 2023)
 
 * fix: module is not defined [#11](https://github.com/liriliri/luna/issues/11)
diff --git a/src/dom-viewer/README.md b/src/dom-viewer/README.md
index f24ab62..de06555 100644
--- a/src/dom-viewer/README.md
+++ b/src/dom-viewer/README.md
@@ -36,6 +36,7 @@ domViewer.expand()
 
 ## Configuration
 
+* hotkey(boolean): Enable hotkey.
 * ignore(AnyFn): Predicate function which removes the matching child nodes.
 * node(ChildNode): Html element to navigate.
 
diff --git a/src/dom-viewer/index.ts b/src/dom-viewer/index.ts
index 02d0ab7..4eba6cf 100644
--- a/src/dom-viewer/index.ts
+++ b/src/dom-viewer/index.ts
@@ -10,10 +10,12 @@ import MutationObserver from 'licia/MutationObserver'
 import contain from 'licia/contain'
 import highlight from 'licia/highlight'
 import truncate from 'licia/truncate'
+import last from 'licia/last'
 import types from 'licia/types'
 import escape from 'licia/escape'
 import trim from 'licia/trim'
 import every from 'licia/every'
+import hotkey from 'licia/hotkey'
 import { exportCjs, getPlatform, hasTouchSupport } from '../share/util'
 
 const emptyHighlightStyle = {
@@ -30,6 +32,8 @@ export interface IOptions extends IComponentOptions {
   node?: ChildNode
   /** Predicate function which removes the matching child nodes. */
   ignore?: types.AnyFn
+  /** Enable hotkey. */
+  hotkey?: boolean
   parent?: DomViewer | null
   isEndTag?: boolean
   rootContainer?: HTMLElement
@@ -45,13 +49,13 @@ export interface IOptions extends IComponentOptions {
  * domViewer.expand()
  */
 export default class DomViewer extends Component<IOptions> {
+  isExpanded = false
+  childNodes: ChildNode[] = []
+  childNodeDomViewers: DomViewer[] = []
+  endTagDomViewer?: DomViewer
   private $tag: $.$
   private $children: $.$
-  private isExpanded = false
   private observer: MutationObserver
-  private childNodes: ChildNode[] = []
-  private childNodeDomViewers: DomViewer[] = []
-  private endTagDomViewer?: DomViewer
   constructor(container: HTMLElement, options: IOptions = {}) {
     super(container, { compName: 'dom-viewer' }, options)
 
@@ -62,6 +66,7 @@ export default class DomViewer extends Component<IOptions> {
       rootContainer: container,
       rootDomViewer: this,
       ignore: () => false,
+      hotkey: true,
     })
 
     this.initTpl()
@@ -121,7 +126,7 @@ export default class DomViewer extends Component<IOptions> {
       this.$children.remove()
     }
   }
-  expand() {
+  expand = () => {
     if (!this.isExpandable() || this.isExpanded) {
       return
     }
@@ -131,7 +136,7 @@ export default class DomViewer extends Component<IOptions> {
 
     this.renderChildNodes()
   }
-  collapse() {
+  collapse = () => {
     if (!this.isExpandable() || !this.isExpanded) {
       return
     }
@@ -258,6 +263,92 @@ export default class DomViewer extends Component<IOptions> {
     } else {
       $tag.on('mousedown', () => this.select())
     }
+
+    if (this.options.hotkey) {
+      const options = { element: $tag.get(0) as HTMLElement }
+      hotkey.on('right', options, this.onKeyRight)
+      hotkey.on('left', options, this.onKeyLeft)
+      hotkey.on('down', options, this.onKeyDown)
+      hotkey.on('up', options, this.onKeyUp)
+    }
+  }
+  private onKeyRight = () => {
+    if (this.isExpanded) {
+      this.childNodeDomViewers[0].select()
+    } else {
+      this.expand()
+    }
+  }
+  private onKeyLeft = () => {
+    if (this.isExpanded) {
+      this.collapse()
+    } else {
+      this.options.parent?.select()
+    }
+  }
+  private onKeyDown = () => {
+    const { options } = this
+
+    if (this.isExpanded) {
+      this.childNodeDomViewers[0].select()
+      return
+    }
+
+    let { parent } = options
+    if (!parent) {
+      return
+    }
+
+    if (options.isEndTag) {
+      parent = parent.getOption('parent')
+      if (!parent) {
+        return
+      }
+
+      const { childNodes, childNodeDomViewers, endTagDomViewer } = parent
+      const idx = childNodes.indexOf(options.node)
+      if (childNodes[idx + 1]) {
+        childNodeDomViewers[idx + 1].select()
+      } else if (endTagDomViewer) {
+        endTagDomViewer.select()
+      }
+    } else {
+      const { childNodeDomViewers, endTagDomViewer } = parent
+      const idx = childNodeDomViewers.indexOf(this)
+      if (childNodeDomViewers[idx + 1]) {
+        childNodeDomViewers[idx + 1].select()
+      } else if (endTagDomViewer) {
+        endTagDomViewer.select()
+      }
+    }
+  }
+  private onKeyUp = () => {
+    const { options } = this
+
+    const parent = options.parent
+    if (!parent) {
+      return
+    }
+
+    let domViewer
+    if (options.isEndTag) {
+      domViewer = last(parent.childNodeDomViewers)
+    } else {
+      const idx = parent.childNodeDomViewers.indexOf(this)
+      if (idx < 1) {
+        parent.select()
+      } else {
+        domViewer = parent.childNodeDomViewers[idx - 1]
+      }
+    }
+
+    if (domViewer) {
+      if (domViewer.isExpanded) {
+        domViewer.endTagDomViewer?.select()
+      } else {
+        domViewer.select()
+      }
+    }
   }
   private isExpandable() {
     const { node } = this.options
diff --git a/src/dom-viewer/package.json b/src/dom-viewer/package.json
index 75b39f5..1f89375 100644
--- a/src/dom-viewer/package.json
+++ b/src/dom-viewer/package.json
@@ -1,6 +1,6 @@
 {
   "name": "dom-viewer",
-  "version": "1.2.4",
+  "version": "1.3.0",
   "description": "Dom tree navigator",
   "luna": {
     "icon": true