element-plus/packages/hooks/use-attrs/index.ts
2022-03-25 15:35:56 +08:00

40 lines
1.0 KiB
TypeScript

import { computed, getCurrentInstance } from 'vue'
import { fromPairs } from 'lodash-unified'
import { debugWarn } from '@element-plus/utils'
import type { ComputedRef } from 'vue'
interface Params {
excludeListeners?: boolean
excludeKeys?: string[]
}
const DEFAULT_EXCLUDE_KEYS = ['class', 'style']
const LISTENER_PREFIX = /^on[A-Z]/
export const useAttrs = (
params: Params = {}
): ComputedRef<Record<string, unknown>> => {
const { excludeListeners = false, excludeKeys = [] } = params
const allExcludeKeys = excludeKeys.concat(DEFAULT_EXCLUDE_KEYS)
const instance = getCurrentInstance()
if (!instance) {
debugWarn(
'use-attrs',
'getCurrentInstance() returned null. useAttrs() must be called at the top of a setup function'
)
return computed(() => ({}))
}
return computed(() =>
fromPairs(
Object.entries(instance.proxy?.$attrs!).filter(
([key]) =>
!allExcludeKeys.includes(key) &&
!(excludeListeners && LISTENER_PREFIX.test(key))
)
)
)
}