element-plus/packages/hooks/use-attrs/index.ts
三咲智子 6503e55277
refactor(utils): migrate utils (#5949)
* refactor(utils-v2): migrate utils

* refactor(utils-v2): migrate utils

* refactor(utils-v2): migrate utils

* refactor(utils): remove

* refactor(utils): rename

* refactor(utils): move EVENT_CODE to constants

* refactor: remove generic
2022-02-11 11:03:15 +08:00

40 lines
1.0 KiB
TypeScript

import { getCurrentInstance, computed } 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))
)
)
)
}