fix: correct and improve return types for single and multi select pro…

…mpts (#197)
unjs · Jun 26, 2023 · 0d03d70 · 0d03d70
1 parent dabb705
commit 0d03d70
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/prompt.ts b/src/prompt.ts
@@ -21,7 +21,7 @@ export type ConfirmOptions = {
 export type SelectOptions = {
   type: "select";
   initial?: string;
-  options: (SelectOption | string)[];
+  options: (string | SelectOption)[];
 };
 
 export type MultiSelectOptions = {
@@ -41,7 +41,11 @@ type inferPromptReturnType<T extends PromptOptions> = T extends TextOptions
   ? string
   : T extends ConfirmOptions
   ? boolean
-  : string[];
+  : T extends SelectOptions
+  ? T["options"][number]
+  : T extends MultiSelectOptions
+  ? T["options"]
+  : unknown;
 
 export async function prompt<
   _ = any,