71 lines
1.8 KiB
TypeScript
71 lines
1.8 KiB
TypeScript
import { EMPTY_OBJ, NOOP } from './utils'
|
|
import { computed, stop, ComputedGetter } from '@vue/observer'
|
|
import { ComponentClass, ComponentInstance } from './component'
|
|
import { ComponentComputedOptions } from './componentOptions'
|
|
|
|
const extractionCache: WeakMap<
|
|
ComponentClass,
|
|
ComponentComputedOptions
|
|
> = new WeakMap()
|
|
|
|
export function getComputedOptions(
|
|
comp: ComponentClass
|
|
): ComponentComputedOptions {
|
|
let computedOptions = extractionCache.get(comp)
|
|
if (computedOptions) {
|
|
return computedOptions
|
|
}
|
|
computedOptions = {}
|
|
const descriptors = Object.getOwnPropertyDescriptors(comp.prototype as any)
|
|
for (const key in descriptors) {
|
|
const d = descriptors[key]
|
|
if (d.get) {
|
|
computedOptions[key] = d.get
|
|
// there's no need to do anything for the setter
|
|
// as it's already defined on the prototype
|
|
}
|
|
}
|
|
extractionCache.set(comp, computedOptions)
|
|
return computedOptions
|
|
}
|
|
|
|
export function initializeComputed(
|
|
instance: ComponentInstance,
|
|
computedOptions: ComponentComputedOptions | undefined
|
|
) {
|
|
if (!computedOptions) {
|
|
instance.$computed = EMPTY_OBJ
|
|
return
|
|
}
|
|
const handles: Record<
|
|
string,
|
|
ComputedGetter
|
|
> = (instance._computedGetters = {})
|
|
const proxy = instance.$proxy
|
|
for (const key in computedOptions) {
|
|
const option = computedOptions[key]
|
|
const getter = typeof option === 'function' ? option : option.get || NOOP
|
|
handles[key] = computed(getter, proxy)
|
|
}
|
|
instance.$computed = new Proxy(
|
|
{},
|
|
{
|
|
get(_, key: any) {
|
|
if (handles.hasOwnProperty(key)) {
|
|
return handles[key]()
|
|
}
|
|
}
|
|
// TODO should be readonly
|
|
}
|
|
)
|
|
}
|
|
|
|
export function teardownComputed(instance: ComponentInstance) {
|
|
const handles = instance._computedGetters
|
|
if (handles !== null) {
|
|
for (const key in handles) {
|
|
stop(handles[key].runner)
|
|
}
|
|
}
|
|
}
|