From 221dabc823c84a16fa2b90ac04cdbdd93686ed5a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90=20Kevin=20Deng?=
 <sxzz@sxzz.moe>
Date: Fri, 31 Mar 2023 22:21:20 +0800
Subject: [PATCH] wip: defineModel

---
 packages/compiler-sfc/src/compileScript.ts    |  6 ++++
 packages/dts-test/setupHelpers.test-d.ts      | 25 +++++++++++++-
 packages/runtime-core/src/apiSetupHelpers.ts  | 33 +++++++++++++++++--
 packages/runtime-core/src/index.ts            |  1 +
 .../types/scriptSetupHelpers.d.ts             |  2 ++
 5 files changed, 64 insertions(+), 3 deletions(-)

diff --git a/packages/compiler-sfc/src/compileScript.ts b/packages/compiler-sfc/src/compileScript.ts
index 8d22d7e1348..d980835044f 100644
--- a/packages/compiler-sfc/src/compileScript.ts
+++ b/packages/compiler-sfc/src/compileScript.ts
@@ -151,6 +151,11 @@ type EmitsDeclType = FromNormalScript<
   TSFunctionType | TSTypeLiteral | TSInterfaceBody
 >
 
+export interface ModelDecl {
+  type: TSType
+  option: Node
+}
+
 /**
  * Compile `<script setup>`
  * It requires the whole SFC descriptor because we need to handle and merge
@@ -325,6 +330,7 @@ export function compileScript(
   let emitsTypeDecl: EmitsDeclType | undefined
   let emitIdentifier: string | undefined
   let optionsRuntimeDecl: Node | undefined
+  // let modelDeclMap: Record<string, ModelDecl> = {}
   let hasAwait = false
   let hasInlinedSsrRenderFn = false
   // props/emits declared via types
diff --git a/packages/dts-test/setupHelpers.test-d.ts b/packages/dts-test/setupHelpers.test-d.ts
index 13d80947c83..14229d168c3 100644
--- a/packages/dts-test/setupHelpers.test-d.ts
+++ b/packages/dts-test/setupHelpers.test-d.ts
@@ -6,7 +6,9 @@ import {
   withDefaults,
   Slots,
   defineSlots,
-  VNode
+  VNode,
+  WritableComputedRef,
+  defineModel
 } from 'vue'
 import { describe, expectType } from './utils'
 
@@ -202,6 +204,27 @@ describe('defineSlots', () => {
   expectType<Slots>(slotsUntype)
 })
 
+describe('defineModel', () => {
+  const modelValue = defineModel<string>()
+  expectType<WritableComputedRef<string | undefined>>(modelValue)
+  modelValue.value = 'new value'
+
+  const modelValueRequired = defineModel<boolean>({ required: true })
+  expectType<WritableComputedRef<boolean>>(modelValueRequired)
+
+  const modelValueDefault = defineModel<boolean>({ default: true })
+  expectType<WritableComputedRef<boolean>>(modelValueDefault)
+
+  const count = defineModel<number>('count')
+  expectType<WritableComputedRef<number | undefined>>(count)
+
+  const countRequired = defineModel<number>('count', { required: true })
+  expectType<WritableComputedRef<number>>(countRequired)
+
+  const countDefault = defineModel<number>('count', { default: 1 })
+  expectType<WritableComputedRef<number>>(countDefault)
+})
+
 describe('useAttrs', () => {
   const attrs = useAttrs()
   expectType<Record<string, unknown>>(attrs)
diff --git a/packages/runtime-core/src/apiSetupHelpers.ts b/packages/runtime-core/src/apiSetupHelpers.ts
index 1c60416c6ea..e5030c043d6 100644
--- a/packages/runtime-core/src/apiSetupHelpers.ts
+++ b/packages/runtime-core/src/apiSetupHelpers.ts
@@ -26,6 +26,7 @@ import {
 } from './componentProps'
 import { warn } from './warning'
 import { SlotsType, TypedSlots } from './componentSlots'
+import { WritableComputedRef } from '@vue/reactivity'
 
 // dev only
 const warnRuntimeUsage = (method: string) =>
@@ -200,11 +201,39 @@ export function defineOptions<
 
 export function defineSlots<
   S extends Record<string, any> = Record<string, any>
->(): // @ts-expect-error
-TypedSlots<SlotsType<S>> {
+>(): TypedSlots<SlotsType<S>> {
   if (__DEV__) {
     warnRuntimeUsage(`defineSlots`)
   }
+  return null as any
+}
+
+export function defineModel<T>(
+  options: { required: true } & Record<string, unknown>
+): WritableComputedRef<T>
+export function defineModel<T>(
+  options: { default: any } & Record<string, unknown>
+): WritableComputedRef<T>
+export function defineModel<T>(
+  options?: Record<string, unknown>
+): WritableComputedRef<T | undefined>
+export function defineModel<T>(
+  name: string,
+  options: { required: true } & Record<string, unknown>
+): WritableComputedRef<T>
+export function defineModel<T>(
+  name: string,
+  options: { default: any } & Record<string, unknown>
+): WritableComputedRef<T>
+export function defineModel<T>(
+  name: string,
+  options?: Record<string, unknown>
+): WritableComputedRef<T | undefined>
+export function defineModel() {
+  if (__DEV__) {
+    warnRuntimeUsage('defineModel')
+  }
+  return null as any
 }
 
 type NotUndefined<T> = T extends undefined ? never : T
diff --git a/packages/runtime-core/src/index.ts b/packages/runtime-core/src/index.ts
index 428f3683d5a..34f2eb910e5 100644
--- a/packages/runtime-core/src/index.ts
+++ b/packages/runtime-core/src/index.ts
@@ -71,6 +71,7 @@ export {
   defineExpose,
   defineOptions,
   defineSlots,
+  defineModel,
   withDefaults,
   // internal
   mergeDefaults,
diff --git a/packages/runtime-core/types/scriptSetupHelpers.d.ts b/packages/runtime-core/types/scriptSetupHelpers.d.ts
index 4ffc2f725ae..a9ba734a53d 100644
--- a/packages/runtime-core/types/scriptSetupHelpers.d.ts
+++ b/packages/runtime-core/types/scriptSetupHelpers.d.ts
@@ -5,6 +5,7 @@ type _defineEmits = typeof defineEmits
 type _defineExpose = typeof defineExpose
 type _defineOptions = typeof defineOptions
 type _defineSlots = typeof defineSlots
+type _defineModel = typeof defineModel
 type _withDefaults = typeof withDefaults
 
 declare global {
@@ -13,5 +14,6 @@ declare global {
   const defineExpose: _defineExpose
   const defineOptions: _defineOptions
   const defineSlots: _defineSlots
+  const defineModel: _defineModel
   const withDefaults: _withDefaults
 }