resolve emits types

visitor/src/lib.rs

@@ -1422,8 +1422,13 @@ where
             return;
         };
 
-        if let Some(prop_types) = self.extract_props_type(maybe_setup) {
-            inject_option(call_expr, "props", Expr::Object(prop_types));
+        let props_types = self.extract_props_type(maybe_setup);
+        let emits_types = self.extract_emits_type(maybe_setup);
+        if let Some(prop_types) = props_types {
+            inject_define_component_option(call_expr, "props", Expr::Object(prop_types));
+        }
+        if let Some(emits_type) = emits_types {
+            inject_define_component_option(call_expr, "emits", Expr::Array(emits_type));
         }
     }
 
@@ -1443,15 +1448,15 @@ where
             return;
         }
 
-        inject_option(
+        inject_define_component_option(
             call,
             "name",
             Expr::Lit(Lit::Str(quote_str!(name.sym.clone()))),
         );
     }
 }
 
-fn inject_option(call: &mut CallExpr, name: &'static str, value: Expr) {
+fn inject_define_component_option(call: &mut CallExpr, name: &'static str, value: Expr) {
     let options = call.args.get_mut(1);
     if options
         .as_ref()

visitor/tests/fixture/resolve-emits-types/disabled/input.tsx

@@ -0,0 +1,3 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+defineComponent((_, ctx: SetupContext<(e: 'foo' | 'bar') => void>) => {})

visitor/tests/fixture/resolve-emits-types/disabled/output.js

@@ -0,0 +1,2 @@
+import { defineComponent, type SetupContext } from 'vue';
+defineComponent((_, ctx: SetupContext<(e: 'foo' | 'bar') => void>)=>{});

visitor/tests/fixture/resolve-emits-types/exported-interface/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/exported-interface/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+export interface Emits { (e: 'foo' | 'bar'): void }
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/exported-interface/output.js

@@ -0,0 +1,10 @@
+import { defineComponent, type SetupContext } from 'vue';
+export interface Emits {
+    (e: 'foo' | 'bar') : void;
+}
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/exported-type-alias/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/exported-type-alias/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+export type Emits = { (e: 'foo' | 'bar'): void }
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/exported-type-alias/output.js

@@ -0,0 +1,10 @@
+import { defineComponent, type SetupContext } from 'vue';
+export type Emits = {
+    (e: 'foo' | 'bar') : void;
+};
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/fn-type/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/fn-type/input.tsx

@@ -0,0 +1,7 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+defineComponent((_, ctx: SetupContext<(e: 'foo' | 'bar') => void>) => {})
+
+defineComponent((_, ctx: SetupContext<((e: 'foo' | 'bar') => void) | ((e: 'baz', id: number) => void)>) => {})
+
+defineComponent((_, ctx: SetupContext<{(e: 'foo' | 'bar'): void; (e: 'baz', id: number): void;}>) => {})

visitor/tests/fixture/resolve-emits-types/fn-type/output.js

@@ -0,0 +1,24 @@
+import { defineComponent, type SetupContext } from 'vue';
+defineComponent((_, ctx: SetupContext<(e: 'foo' | 'bar') => void>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});
+defineComponent((_, ctx: SetupContext<((e: 'foo' | 'bar') => void) | ((e: 'baz', id: number) => void)>)=>{}, {
+    emits: [
+        "foo",
+        "bar",
+        "baz"
+    ]
+});
+defineComponent((_, ctx: SetupContext<{
+    (e: 'foo' | 'bar') : void;
+    (e: 'baz', id: number) : void;
+}>)=>{}, {
+    emits: [
+        "foo",
+        "bar",
+        "baz"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/interface-with-extends/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/interface-with-extends/input.tsx

@@ -0,0 +1,6 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+interface Base { (e: 'foo'): void }
+interface Emits extends Base { (e: 'bar'): void }
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/interface-with-extends/output.js

@@ -0,0 +1,13 @@
+import { defineComponent, type SetupContext } from 'vue';
+interface Base {
+    (e: 'foo') : void;
+}
+interface Emits extends Base {
+    (e: 'bar') : void;
+}
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "bar",
+        "foo"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/interface/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/interface/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+interface Emits { (e: 'foo' | 'bar'): void }
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/interface/output.js

@@ -0,0 +1,10 @@
+import { defineComponent, type SetupContext } from 'vue';
+interface Emits {
+    (e: 'foo' | 'bar') : void;
+}
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/property-syntax/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/property-syntax/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+defineComponent((_, ctx: SetupContext<{ foo: [], bar: [] }>) => {})
+
+defineComponent((_, ctx: SetupContext<{ 'foo:bar': [] }>) => {})

visitor/tests/fixture/resolve-emits-types/property-syntax/output.js

@@ -0,0 +1,17 @@
+import { defineComponent, type SetupContext } from 'vue';
+defineComponent((_, ctx: SetupContext<{
+    foo: [];
+    bar: [];
+}>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});
+defineComponent((_, ctx: SetupContext<{
+    'foo:bar': [];
+}>)=>{}, {
+    emits: [
+        "foo:bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/referenced-exported-fn-type/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/referenced-exported-fn-type/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+export type Emits = (e: 'foo' | 'bar') => void
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/referenced-exported-fn-type/output.js

@@ -0,0 +1,8 @@
+import { defineComponent, type SetupContext } from 'vue';
+export type Emits = (e: 'foo' | 'bar') => void;
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/referenced-fn-type/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/referenced-fn-type/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+type Emits = (e: 'foo' | 'bar') => void
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/referenced-fn-type/output.js

@@ -0,0 +1,8 @@
+import { defineComponent, type SetupContext } from 'vue';
+type Emits = (e: 'foo' | 'bar') => void;
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/type-alias/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/type-alias/input.tsx

@@ -0,0 +1,5 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+type Emits = { (e: 'foo' | 'bar'): void }
+
+defineComponent((_, ctx: SetupContext<Emits>) => {})

visitor/tests/fixture/resolve-emits-types/type-alias/output.js

@@ -0,0 +1,10 @@
+import { defineComponent, type SetupContext } from 'vue';
+type Emits = {
+    (e: 'foo' | 'bar') : void;
+};
+defineComponent((_, ctx: SetupContext<Emits>)=>{}, {
+    emits: [
+        "foo",
+        "bar"
+    ]
+});

visitor/tests/fixture/resolve-emits-types/type-references-in-union/config.json

@@ -0,0 +1,3 @@
+{
+  "resolveType": true
+}

visitor/tests/fixture/resolve-emits-types/type-references-in-union/input.tsx

@@ -0,0 +1,9 @@
+import { defineComponent, type SetupContext } from 'vue'
+
+type BaseEmit = "change"
+type Emit = "some" | "emit" | BaseEmit
+
+defineComponent((_, ctx: SetupContext<{
+  (e: Emit): void;
+  (e: "another", val: string): void;
+}>) => {})

visitor/tests/fixture/resolve-emits-types/type-references-in-union/output.js

@@ -0,0 +1,14 @@
+import { defineComponent, type SetupContext } from 'vue';
+type BaseEmit = "change";
+type Emit = "some" | "emit" | BaseEmit;
+defineComponent((_, ctx: SetupContext<{
+    (e: Emit) : void;
+    (e: "another", val: string) : void;
+}>)=>{}, {
+    emits: [
+        "some",
+        "emit",
+        "change",
+        "another"
+    ]
+});