feat: add component inspector to support component tree navigable (#200)

alexzhang1030 · webfansplz · web-flow · commit 3e730aec6354 · 2023-08-13T17:30:40.000+08:00
---------

Co-authored-by: webfansplz &lt;308241863@qq.com&gt;
diff --git a/packages/client/components/ComponentTreeNode.vue b/packages/client/components/ComponentTreeNode.vue
@@ -1,4 +1,6 @@
 <script setup lang="ts">
+import scrollIntoView from 'scroll-into-view-if-needed'
+
 /* eslint-disable @typescript-eslint/consistent-type-imports */
 import type { ComponentTreeNode } from '~/types'
 
@@ -11,6 +13,23 @@ const props = withDefaults(defineProps<{
 
 const { isSelected, select, isExpanded, toggleExpand } = useComponent(props.data)
 const { highlight, unhighlight } = useHighlightComponent(props.data)
+
+const toggleEl = ref<HTMLElement>()
+
+function autoScroll() {
+  if (isSelected.value && toggleEl.value) {
+    const el = toggleEl.value
+    scrollIntoView(el, {
+      scrollMode: 'if-needed',
+      block: 'center',
+      behavior: 'smooth',
+      inline: 'nearest',
+    })
+  }
+}
+
+watch(isSelected, () => autoScroll())
+watch(toggleEl, () => autoScroll())
 </script>
 
 <template>
@@ -24,7 +43,7 @@ const { highlight, unhighlight } = useHighlightComponent(props.data)
     @mouseover="highlight"
     @mouseleave="unhighlight"
   >
-    <h3 vue-block-title @click="data.hasChildren ? toggleExpand(data.id) : () => {}">
+    <h3 ref="toggleEl" vue-block-title @click="data.hasChildren ? toggleExpand(data.id) : () => {}">
       <VDExpandIcon v-if="data.hasChildren" :value="isExpanded" />
       <i v-else inline-block h-6 w-6 />
       <span
diff --git a/packages/client/composables/component.ts b/packages/client/composables/component.ts
@@ -1,5 +1,6 @@
 import type { ComponentInternalInstance } from 'vue'
-import { InstanceMap, getInstanceDetails, getInstanceName, getInstanceOrVnodeRect, getRootElementsFromComponentInstance } from '~/logic/components'
+import { getInstanceName } from '@vite-plugin-vue-devtools/core'
+import { InstanceMap, getInstanceDetails, getInstanceOrVnodeRect, getRootElementsFromComponentInstance } from '~/logic/components'
 import { useDevToolsClient } from '~/logic/client'
 import type { ComponentTreeNode } from '~/types'
 
@@ -13,23 +14,29 @@ const expandedMap = ref<Record<ComponentTreeNode['id'], boolean>>({
 
 export const selectedComponent = ref<ComponentInternalInstance>()
 export const selectedComponentState = shallowRef<Record<string, any>[]>([])
+
+export function selectComponentTreeNode(data: ComponentTreeNode) {
+  selected.value = data.id
+  selectedComponentName.value = data.name
+  // TODO (Refactor): get instance state way
+  selectedComponentState.value = InstanceMap.get(data.id)
+  selectedComponentNode.value = data
+  // selectedComponent.value = instance.instance
+  // selectedComponentState.value = getInstanceState(instance.instance!)
+}
+
+export function setExpanded(id: string, expanded: boolean) {
+  expandedMap.value[id] = expanded
+}
+
 export function useComponent(instance: ComponentTreeNode & { instance?: ComponentInternalInstance }) {
-  function select(data: ComponentTreeNode) {
-    selected.value = data.id
-    selectedComponentName.value = data.name
-    // TODO (Refactor): get instance state way
-    selectedComponentState.value = InstanceMap.get(data.id)
-    selectedComponentNode.value = data
-    // selectedComponent.value = instance.instance
-    // selectedComponentState.value = getInstanceState(instance.instance!)
-  }
   function toggleExpand(id: string) {
     expandedMap.value[id] = !expandedMap.value[id]
   }
   const isSelected = computed(() => selected.value === instance.id)
   const isExpanded = computed(() => expandedMap.value[instance.id])
 
-  return { isSelected, select, isExpanded, toggleExpand }
+  return { isSelected, select: selectComponentTreeNode, isExpanded, toggleExpand }
 }
 
 export function useHighlightComponent(node: ComponentTreeNode): {
diff --git a/packages/client/logic/client.ts b/packages/client/logic/client.ts
@@ -9,7 +9,9 @@ const client = ref<VueDevtoolsHostClient>({
   componentInspector: {
     highlight: () => {},
     unHighlight: () => {},
-    scrollToComponent: () => {},
+    scrollToComponent: () => { },
+    startInspect: () => { },
+    stopInspect: () => { },
   },
   rerenderHighlight: {
     updateInfo: () => {},
diff --git a/packages/client/logic/components/data.ts b/packages/client/logic/components/data.ts
@@ -1,6 +1,7 @@
 /* eslint-disable @typescript-eslint/indent */
 import type { ComponentInternalInstance } from 'vue'
-import { camelize, getInstanceName, getUniqueComponentId, returnError } from './util'
+import { getInstanceName } from '@vite-plugin-vue-devtools/core'
+import { camelize, getUniqueComponentId, returnError } from './util'
 
 const vueBuiltins = [
   'nextTick',
diff --git a/packages/client/logic/components/filter.ts b/packages/client/logic/components/filter.ts
@@ -1,5 +1,6 @@
 import type { ComponentInternalInstance } from 'vue'
-import { classify, getInstanceName, kebabize } from './util'
+import { getInstanceName } from '@vite-plugin-vue-devtools/core'
+import { classify, kebabize } from './util'
 
 export class ComponentFilter {
   private filter: string
diff --git a/packages/client/logic/components/index.ts b/packages/client/logic/components/index.ts
@@ -1,4 +1,3 @@
 export { ComponentWalker, InstanceMap } from './tree'
 export { getInstanceState, processSetupState, getInstanceDetails, getSetupStateInfo } from './data'
 export { getInstanceOrVnodeRect, getRootElementsFromComponentInstance } from './el'
-export { getInstanceName } from './util'
diff --git a/packages/client/logic/components/tree.ts b/packages/client/logic/components/tree.ts
@@ -1,10 +1,14 @@
 import type { ComponentInternalInstance, SuspenseBoundary } from 'vue'
-import { getInstanceName, getRenderKey, getUniqueComponentId, isBeingDestroyed, isFragment } from './util'
+import { getInstanceName } from '@vite-plugin-vue-devtools/core'
+import { getRenderKey, getUniqueComponentId, isBeingDestroyed, isFragment } from './util'
 import { ComponentFilter } from './filter'
 import { getRootElementsFromComponentInstance } from './el'
 import { getInstanceState } from './data'
+import type { ComponentTreeNode } from '~/types'
 
 export const InstanceMap = new Map()
+export const UidToTreeNodeMap = new Map<number, ComponentTreeNode>()
+
 export class ComponentWalker {
   maxDepth: number
   recursively: boolean
@@ -27,7 +31,7 @@ export class ComponentWalker {
 
   getComponentParents(instance: ComponentInternalInstance) {
     this.captureIds = new Map()
-    const parents = []
+    const parents: ComponentInternalInstance[] = []
     this.captureId(instance)
     let parent = instance
     // eslint-disable-next-line no-cond-assign
@@ -205,6 +209,7 @@ export class ComponentWalker {
     // }
 
     InstanceMap.set(treeNode.id, getInstanceState(instance))
+    UidToTreeNodeMap.set(treeNode.uid, treeNode)
     treeNode.instance = instance
 
     return treeNode
diff --git a/packages/client/logic/components/util.ts b/packages/client/logic/components/util.ts
@@ -9,53 +9,6 @@ export function isFragment(instance: ComponentInternalInstance) {
   return Fragment === instance.subTree?.type
 }
 
-/**
- * Get the appropriate display name for an instance.
- *
- * @param {Vue} instance
- * @return {String}
- */
-export function getInstanceName(instance: any) {
-  const name = getComponentTypeName(instance.type || {})
-  if (name)
-    return name
-  if (instance.root === instance)
-    return 'Root'
-  for (const key in instance.parent?.type?.components) {
-    if (instance.parent.type.components[key] === instance.type)
-      return saveComponentName(instance, key)
-  }
-
-  for (const key in instance.appContext?.components) {
-    if (instance.appContext.components[key] === instance.type)
-      return saveComponentName(instance, key)
-  }
-
-  const fileName = getComponentFileName(instance.type || {})
-  if (fileName)
-    return fileName
-
-  return 'Anonymous Component'
-}
-
-function saveComponentName(instance: ComponentInternalInstance, key: string) {
-  return key
-}
-
-function getComponentTypeName(options: any) {
-  return options.name || options._componentTag || options.__vdevtools_guessedName || options.__name
-}
-
-export function getComponentFileName(options: any) {
-  const file = options.__file // injected by vite
-  // TODO: classify
-  if (file) {
-    const filename = options.__file?.match(/\/?([^/]+?)(\.[^/.]+)?$/)?.[1]
-    return filename ?? file
-  }
-  // return classify(basename(file, '.vue'))
-}
-
 /**
  * Returns a devtools unique id for instance.
  * @param {Vue} instance
diff --git a/packages/client/logic/timeline.ts b/packages/client/logic/timeline.ts
@@ -1,5 +1,5 @@
 import { nanoid } from 'nanoid'
-import { getComponentFileName } from './components/util'
+import { getComponentFileName } from '@vite-plugin-vue-devtools/core'
 import { useDevToolsClient } from './client'
 
 interface TimelineLayer {
diff --git a/packages/client/package.json b/packages/client/package.json
@@ -37,6 +37,7 @@
     "json-editor-vue": "^0.10.6",
     "minimatch": "^9.0.3",
     "nanoid": "^4.0.2",
+    "scroll-into-view-if-needed": "^3.0.10",
     "splitpanes": "^3.1.5",
     "vanilla-jsoneditor": "^0.17.8",
     "vite-hot-client": "^0.2.1",
diff --git a/packages/client/pages/components.vue b/packages/client/pages/components.vue
@@ -1,13 +1,16 @@
 <script setup lang="ts">
 import { Pane, Splitpanes } from 'splitpanes'
-import { scrollToComponent, selected, selectedComponentName, selectedComponentNode, selectedComponentNodeFilePath } from '../composables/component'
+import type { ComponentInternalInstance } from 'vue'
+import { scrollToComponent, selectComponentTreeNode, selected, selectedComponentName, selectedComponentNode, selectedComponentNodeFilePath } from '../composables/component'
 
 /* eslint-disable @typescript-eslint/consistent-type-imports */
+import { UidToTreeNodeMap } from '../logic/components/tree'
 import type { ComponentTreeNode } from '~/types'
 import { ComponentWalker, getInstanceState } from '~/logic/components'
 import { useDevToolsClient } from '~/logic/client'
 import { instance, onVueInstanceUpdate } from '~/logic/app'
 import { rootPath } from '~/logic/global'
+import { getUniqueComponentId } from '~/logic/components/util'
 
 const componentTree = ref<ComponentTreeNode[]>([])
 const filterName = ref('')
@@ -82,14 +85,60 @@ function openInEditor() {
   const client = useDevToolsClient()
   client.value.openInEditor(selectedComponentNodeFilePath.value)
 }
+
+const client = useDevToolsClient()
+
+const inspectorEnabled = ref(false)
+
+function inspectComponentClick(instance: ComponentInternalInstance) {
+  inspectorEnabled.value = false
+  const treeNode = UidToTreeNodeMap.get(instance.uid)
+  if (treeNode) {
+    selectComponentTreeNode(treeNode)
+    const walker = new ComponentWalker(0, null, false)
+    const parents = walker.getComponentParents(instance)
+    parents.reverse().forEach((instance) => {
+      const id = getUniqueComponentId(instance)
+      // Ignore root
+      if (id.endsWith('root'))
+        return
+      setExpanded(id, true)
+    })
+  }
+}
+
+function toggleInspector(target?: boolean) {
+  inspectorEnabled.value = target ?? !inspectorEnabled.value
+  if (inspectorEnabled.value)
+    client.value.componentInspector.startInspect(inspectComponentClick)
+
+  else client.value.componentInspector.stopInspect()
+}
+
+const { control, c, escape } = useMagicKeys()
+
+watchEffect(() => {
+  if ((control.value && c.value) || (escape.value))
+    toggleInspector(false)
+})
 </script>
 
 <template>
   <div h-screen n-panel-grids>
     <Splitpanes>
       <Pane border="r base">
-        <div v-if="componentWalker" w-full px10px py12px>
-          <VDTextInput v-model="filterName" placeholder="Find components..." />
+        <div v-if="componentWalker" sticky left-0 top-0 z-300 w-full flex gap2 px10px py12px bg-base>
+          <VDTextInput v-model="filterName" placeholder="Find components..." flex-1 />
+          <button p2 @click="() => toggleInspector()">
+            <svg
+              xmlns="http://www.w3.org/2000/svg"
+              style="height: 1.1em; width: 1.1em; opacity:0.5;"
+              :style="inspectorEnabled ? 'opacity:1;color:#00dc82' : ''"
+              viewBox="0 0 24 24"
+            >
+              <g fill="none" stroke="currentColor" stroke-linecap="round" stroke-linejoin="round" stroke-width="2"><circle cx="12" cy="12" r=".5" fill="currentColor" /><path d="M5 12a7 7 0 1 0 14 0a7 7 0 1 0-14 0m7-9v2m-9 7h2m7 7v2m7-9h2" /></g>
+            </svg>
+          </button>
         </div>
         <div h-screen select-none overflow-scroll p-2 class="no-scrollbar">
           <ComponentTreeNode v-for="(item) in componentTree" :key="item.id" :data="item" />
diff --git a/packages/client/pages/rerender-trace.vue b/packages/client/pages/rerender-trace.vue
@@ -1,11 +1,10 @@
 <script setup lang="ts">
 import dayjs from 'dayjs'
-import { DevToolsHooks } from '@vite-plugin-vue-devtools/core'
+import { DevToolsHooks, getInstanceName } from '@vite-plugin-vue-devtools/core'
 import type { ComponentInternalInstance, DebuggerEvent, Ref } from 'vue'
 import { useDevToolsClient } from '~/logic/client'
 import { rootPath } from '~/logic/global'
 import { getSetupStateInfo, toRaw } from '~/logic/components/data'
-import { getInstanceName } from '~/logic/components'
 
 type ComponentInstance = ComponentInternalInstance & {
   devtoolsRawSetupState: Record<string, unknown>
diff --git a/packages/client/types.ts b/packages/client/types.ts
@@ -1,3 +1,4 @@
+import type { ComponentInternalInstance } from 'vue'
 import type { Router } from 'vue-router'
 
 type WithOptional<T, K extends keyof T> = Omit<T, K> & Partial<Pick<T, K>>
@@ -27,6 +28,8 @@ export interface VueDevtoolsHostClient {
   componentInspector: {
     highlight: (_name: string, _bounds: ComponentInspectorBounds) => void
     unHighlight: () => void
+    startInspect(cb?: (instance: ComponentInternalInstance) => void): void
+    stopInspect(): void
     scrollToComponent: (_bounds: ComponentInspectorBounds) => void
   }
   rerenderHighlight: {
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -44,6 +44,7 @@
   },
   "devDependencies": {
     "@babel/types": "^7.22.5",
-    "@vue/compiler-sfc": "^3.3.4"
+    "@vue/compiler-sfc": "^3.3.4",
+    "vue": "^3.3.4"
   }
 }
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
@@ -2,3 +2,4 @@ export * from './hook'
 export * from './host'
 export * from './rpc'
 export * from './constant'
+export * from './shared'
diff --git a/packages/core/src/shared.ts b/packages/core/src/shared.ts
@@ -0,0 +1,48 @@
+import type { ComponentInternalInstance } from 'vue'
+
+export function getComponentTypeName(options: any) {
+  return options.name || options._componentTag || options.__vdevtools_guessedName || options.__name
+}
+
+function saveComponentName(instance: ComponentInternalInstance, key: string) {
+  return key
+}
+
+export function getComponentFileName(options: any) {
+  const file = options.__file // injected by vite
+  // TODO: classify
+  if (file) {
+    const filename = options.__file?.match(/\/?([^/]+?)(\.[^/.]+)?$/)?.[1]
+    return filename ?? file
+  }
+  // return classify(basename(file, '.vue'))
+}
+
+/**
+ * Get the appropriate display name for an instance.
+ *
+ * @param {Vue} instance
+ * @return {String}
+ */
+export function getInstanceName(instance: any) {
+  const name = getComponentTypeName(instance.type || {})
+  if (name)
+    return name
+  if (instance.root === instance)
+    return 'Root'
+  for (const key in instance.parent?.type?.components) {
+    if (instance.parent.type.components[key] === instance.type)
+      return saveComponentName(instance, key)
+  }
+
+  for (const key in instance.appContext?.components) {
+    if (instance.appContext.components[key] === instance.type)
+      return saveComponentName(instance, key)
+  }
+
+  const fileName = getComponentFileName(instance.type || {})
+  if (fileName)
+    return fileName
+
+  return 'Anonymous Component'
+}
diff --git a/packages/node/src/views/Main.vue b/packages/node/src/views/Main.vue
diff --git a/packages/node/src/views/composables.ts b/packages/node/src/views/composables.ts
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
