wip: make computed implementation consistent

packages/reactivity/src/computed.ts

@@ -1,17 +1,29 @@
 import { effect, ReactiveEffect, activeReactiveEffectStack } from './effect'
 import { UnwrapNestedRefs, knownRefs } from './ref'
+import { isFunction } from '@vue/shared'
 
 export interface ComputedRef<T> {
   readonly value: UnwrapNestedRefs<T>
   readonly effect: ReactiveEffect
 }
 
+export interface ComputedOptions<T> {
+  get: () => T
+  set: (v: T) => void
+}
+
 export function computed<T>(
-  getter: () => T,
-  setter?: (v: T) => void
+  getterOrOptions: (() => T) | ComputedOptions<T>
 ): ComputedRef<T> {
+  const isReadonly = isFunction(getterOrOptions)
+  const getter = isReadonly
+    ? (getterOrOptions as (() => T))
+    : (getterOrOptions as ComputedOptions<T>).get
+  const setter = isReadonly ? null : (getterOrOptions as ComputedOptions<T>).set
+
   let dirty: boolean = true
   let value: any = undefined
+
   const runner = effect(getter, {
     lazy: true,
     // mark effect as computed so that it gets priority during trigger

packages/reactivity/src/index.ts

@@ -8,7 +8,7 @@ export {
   markImmutable,
   markNonReactive
 } from './reactive'
-export { computed, ComputedRef } from './computed'
+export { computed, ComputedRef, ComputedOptions } from './computed'
 export {
   effect,
   stop,

packages/runtime-core/src/apiReactivity.ts

@@ -23,11 +23,11 @@ export {
 import {
   computed as _computed,
   ComputedRef,
+  ComputedOptions,
   ReactiveEffect
 } from '@vue/reactivity'
 
 import { currentInstance } from './component'
-import { isFunction } from '@vue/shared'
 
 // record effects created during a component's setup() so that they can be
 // stopped when the component unmounts
@@ -37,20 +37,10 @@ export function recordEffect(effect: ReactiveEffect) {
   }
 }
 
-interface ComputedOptions<T> {
-  get: () => T
-  set: (v: T) => void
-}
-
 export function computed<T>(
   getterOrOptions: (() => T) | ComputedOptions<T>
 ): ComputedRef<T> {
-  let c: ComputedRef<T>
-  if (isFunction(getterOrOptions)) {
-    c = _computed(getterOrOptions)
-  } else {
-    c = _computed(getterOrOptions.get, getterOrOptions.set)
-  }
+  const c = _computed(getterOrOptions)
   recordEffect(c.effect)
   return c
 }