fix: handle more types, close #5

so1ve · Aug 27, 2023 · c3fe6b9 · c3fe6b9
1 parent 705fcc5
commit c3fe6b9
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/src/core/utils.ts b/src/core/utils.ts
@@ -50,8 +50,18 @@ export function printVueAcceptableTypeLiteralFromSymbols(
 		for (const decl of symbol.getDeclarations() ?? []) {
 			if (ts.isPropertySignature(decl)) {
 				const type = typeChecker.getTypeAtLocation(decl);
-				const typeString = typeChecker.typeToString(type);
-				push(`${decl.name.getText()}: ${toVueAcceptableType(typeString)};`);
+				const typeString = typeChecker.typeToString(
+					typeChecker.getBaseTypeOfLiteralType(type),
+				);
+				let questionMark = "";
+				if (decl.questionToken) {
+					questionMark = "?";
+				}
+				push(
+					`${decl.name.getText()}${questionMark}: ${toVueAcceptableType(
+						typeString,
+					)};`,
+				);
 			}
 		}
 	}

diff --git a/test/__fixtures__/a.vue b/test/__fixtures__/a.vue
@@ -4,6 +4,8 @@ import type { A } from "./foo";
 defineProps<
 	{
 		msg: string;
+		aaa?: string;
+		bbb: "a" | "b";
 	} & A
 >();
 </script>
diff --git a/test/__snapshots__/index.test.ts.snap b/test/__snapshots__/index.test.ts.snap
@@ -8,6 +8,8 @@ import type { A } from \\"./foo\\";
 defineProps<
 	{
   msg: string;
+  aaa?: string;
+  bbb: string;
   foo: object;
   bar: number;
   baz: object;
@@ -17,7 +19,7 @@ defineProps<
 ",
   "map": SourceMap {
     "file": undefined,
-    "mappings": "AAAA,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACxB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/B;AACA,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACZ,CAAC;;;;;CAEK;AACN,CAAC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;",
+    "mappings": "AAAA,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACxB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/B;AACA,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACZ,CAAC;;;;;;;CAIK;AACN,CAAC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;",
     "names": [],
     "sources": [
       "",