87 lines
2.0 KiB
TypeScript
87 lines
2.0 KiB
TypeScript
import { track, trigger } from './effect'
|
|
import { OperationTypes } from './operations'
|
|
import { isObject } from '@vue/shared'
|
|
import { reactive } from './reactive'
|
|
import { ComputedRef } from './computed'
|
|
import { CollectionTypes } from './collectionHandlers'
|
|
|
|
export interface Ref<T = any> {
|
|
_isRef: true
|
|
value: UnwrapRef<T>
|
|
}
|
|
|
|
const convert = <T extends unknown>(val: T): T =>
|
|
isObject(val) ? reactive(val) : val
|
|
|
|
export function ref<T extends Ref>(raw: T): T
|
|
export function ref<T>(raw: T): Ref<T>
|
|
export function ref<T = any>(): Ref<T>
|
|
export function ref(raw?: unknown) {
|
|
if (isRef(raw)) {
|
|
return raw
|
|
}
|
|
raw = convert(raw)
|
|
const r = {
|
|
_isRef: true,
|
|
get value() {
|
|
track(r, OperationTypes.GET, 'value')
|
|
return raw
|
|
},
|
|
set value(newVal) {
|
|
raw = convert(newVal)
|
|
trigger(
|
|
r,
|
|
OperationTypes.SET,
|
|
'value',
|
|
__DEV__ ? { newValue: newVal } : void 0
|
|
)
|
|
}
|
|
}
|
|
return r as Ref
|
|
}
|
|
|
|
export function isRef(r: any): r is Ref {
|
|
return r ? r._isRef === true : false
|
|
}
|
|
|
|
export function toRefs<T extends object>(
|
|
object: T
|
|
): { [K in keyof T]: Ref<T[K]> } {
|
|
const ret: any = {}
|
|
for (const key in object) {
|
|
ret[key] = toProxyRef(object, key)
|
|
}
|
|
return ret
|
|
}
|
|
|
|
function toProxyRef<T extends object, K extends keyof T>(
|
|
object: T,
|
|
key: K
|
|
): Ref<T[K]> {
|
|
return {
|
|
_isRef: true,
|
|
get value(): any {
|
|
return object[key]
|
|
},
|
|
set value(newVal) {
|
|
object[key] = newVal
|
|
}
|
|
}
|
|
}
|
|
|
|
// Recursively unwraps nested value bindings.
|
|
export type UnwrapRef<T> = {
|
|
cRef: T extends ComputedRef<infer V> ? UnwrapRef<V> : T
|
|
ref: T extends Ref<infer V> ? UnwrapRef<V> : T
|
|
array: T extends Array<infer V> ? Array<UnwrapRef<V>> : T
|
|
object: { [K in keyof T]: UnwrapRef<T[K]> }
|
|
}[T extends ComputedRef<any>
|
|
? 'cRef'
|
|
: T extends Ref
|
|
? 'ref'
|
|
: T extends Array<any>
|
|
? 'array'
|
|
: T extends Function | CollectionTypes
|
|
? 'ref' // bail out on types that shouldn't be unwrapped
|
|
: T extends object ? 'object' : 'ref']
|