init

uni_modules/nutui-uni/components/_hooks/useProvide.ts

@@ -0,0 +1,93 @@
+import { getCurrentInstance, markRaw, provide, shallowReactive } from 'vue'
+import type {
+  ComponentInternalInstance,
+  ConcreteComponent,
+  InjectionKey,
+  VNode,
+  VNodeNormalizedChildren,
+} from 'vue'
+
+// TODO: uniapp 不支持 vue 直接导出的 isVNode
+export function isVNode(value: any): value is VNode {
+  return value ? value.__v_isVNode === true : false
+}
+
+export function flattenVNodes(shouldTraverseChildren: VNodeNormalizedChildren, childName?: string) {
+  const result: VNode[] = []
+
+  const traverse = (children: VNodeNormalizedChildren) => {
+    if (!Array.isArray(children))
+      return
+    children.forEach((child) => {
+      if (!isVNode(child))
+        return
+
+      if (childName) {
+        if (child.type && (child.type as ConcreteComponent).name === childName) {
+          result.push(child)
+          return
+        }
+      }
+      else {
+        result.push(child)
+      }
+
+      if (child.component?.subTree)
+        traverse(child.component.subTree.children)
+
+      if (child.children)
+        traverse(child.children)
+    })
+  }
+
+  traverse(shouldTraverseChildren)
+
+  return result
+}
+
+export function sortChildren(
+  parent: ComponentInternalInstance,
+  internalChildren: ComponentInternalInstance[],
+  childName?: string,
+) {
+  const vnodes = flattenVNodes(parent && parent.subTree && parent.subTree.children, childName)
+  internalChildren.sort((a, b) => {
+    return vnodes.indexOf(a.vnode) - vnodes.indexOf(b.vnode)
+  })
+}
+
+// 如果指定组件名称，则只查找此组件并且查到后结束。也就是不关心此组件下的内容，在大部分场景下节省查找消耗。
+export function useProvide<ProvideValue>(key: InjectionKey<ProvideValue>, childName?: string) {
+  const internalChildren: ComponentInternalInstance[] = shallowReactive([])
+  const publicChildren = shallowReactive<any[]>([])
+  const parent = getCurrentInstance()!
+
+  const add = (child: ComponentInternalInstance) => {
+    if (!child.proxy)
+      return
+    internalChildren.push(markRaw(child))
+    publicChildren.push(markRaw(child.proxy))
+    sortChildren(parent, internalChildren, childName)
+  }
+
+  const remove = (child: ComponentInternalInstance) => {
+    if (child.proxy) {
+      internalChildren.splice(internalChildren.indexOf(markRaw(child)), 1)
+      publicChildren.splice(publicChildren.indexOf(markRaw(child.proxy)), 1)
+    }
+  }
+
+  return (value?: ProvideValue) => {
+    provide(key, {
+      add,
+      remove,
+      internalChildren,
+      ...value,
+    } as any)
+
+    return {
+      internalChildren,
+      children: publicChildren,
+    }
+  }
+}