fix(types): infer setup attrs

Author: rudyxu1102

packages/runtime-core/src/componentOptions.ts

@@ -120,7 +120,8 @@ export interface ComponentOptionsBase<
   EE extends string = string,
   Defaults = {},
   I extends ComponentInjectOptions = {},
-  II extends string = string
+  II extends string = string,
+  Attrs = {}
 > extends LegacyOptions<Props, D, C, M, Mixin, Extends, I, II>,
     ComponentInternalOptions,
     ComponentCustomOptions {
@@ -133,7 +134,7 @@ export interface ComponentOptionsBase<
           UnionToIntersection<ExtractOptionProp<Extends>>
       >
     >,
-    ctx: SetupContext<E>
+    ctx: SetupContext<E, Attrs>
   ) => Promise<RawBindings> | RawBindings | RenderFunction | void
   name?: string
   template?: string | object // can be a direct DOM node
@@ -243,7 +244,8 @@ export type ComponentOptionsWithoutProps<
   EE,
   {},
   I,
-  II
+  II,
+  Attrs
 > & {
   props?: undefined
 } & ThisType<
@@ -290,7 +292,8 @@ export type ComponentOptionsWithArrayProps<
   EE,
   {},
   I,
-  II
+  II,
+  Attrs
 > & {
   props: PropNames[]
 } & ThisType<
@@ -338,7 +341,8 @@ export type ComponentOptionsWithObjectProps<
   EE,
   Defaults,
   I,
-  II
+  II,
+  Attrs
 > & {
   props: PropsOptions & ThisType<void>
 } & ThisType<

test-dts/defineComponent.test-d.tsx

@@ -1254,7 +1254,33 @@ describe('define attrs', () => {
     expectType<JSX.Element>(<MyComp bar={1} />)
   })
 
-  test('define attrs w/ function component', () => {
+  test('define attrs w/ composition api', () => {
+    type CompAttrs = {
+      bar: number
+      baz?: string
+    }
+    const MyComp = defineComponent(
+      {
+        props: {
+          foo: {
+            type: String,
+            required: true
+          }
+        },
+        setup(props, { attrs }) {
+          expectType<string>(props.foo)
+          expectType<number>(attrs.bar)
+          expectType<string | undefined>(attrs.baz)
+        }
+      },
+      {
+        attrs: {} as CompAttrs
+      }
+    )
+    expectType<JSX.Element>(<MyComp foo="1" bar={1} />)
+  })
+
+  test('define attrs w/ functional component', () => {
     type CompAttrs = {
       bar: number
       baz?: string