From 799455bbb526b221c806bf8feff6b625dcf50a56 Mon Sep 17 00:00:00 2001
From: Mike Donnalley <mdonnalley@salesforce.com>
Date: Thu, 18 Aug 2022 09:01:05 -0600
Subject: [PATCH] fix: add overloads to enum flag

---
 src/flags.ts        | 10 +++++++---
 src/parser/flags.ts |  2 +-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/flags.ts b/src/flags.ts
index ccc4dcb21..339584fb0 100644
--- a/src/flags.ts
+++ b/src/flags.ts
@@ -1,4 +1,4 @@
-import {OptionFlag, Definition, BooleanFlag, EnumFlagOptions} from './interfaces'
+import {OptionFlag, Definition, BooleanFlag, EnumFlagOptions, Default} from './interfaces'
 import * as Parser from './parser'
 import Command from './command'
 
@@ -12,7 +12,11 @@ export function option<T>(options: {parse: OptionFlag<T>['parse']} & Partial<Opt
   return build<T>(options)()
 }
 
-const _enum = <T = string>(opts: EnumFlagOptions<T>): OptionFlag<T> => {
+export function _enum<T = string>(opts: EnumFlagOptions<T> & {multiple: true} & ({required: true} | { default: Default<T> })): OptionFlag<T[]>
+export function _enum<T = string>(opts: EnumFlagOptions<T> & {multiple: true}): OptionFlag<T[] | undefined>
+export function _enum<T = string>(opts: EnumFlagOptions<T> & ({required: true} | { default: Default<T> })): OptionFlag<T>
+export function _enum<T = string>(opts: EnumFlagOptions<T>): OptionFlag<T | undefined>
+export function _enum<T = string>(opts: EnumFlagOptions<T>): OptionFlag<T> | OptionFlag<T[]> | OptionFlag<T | undefined> | OptionFlag<T[] | undefined> {
   return build<T>({
     async parse(input) {
       if (!opts.options.includes(input)) throw new Error(`Expected --${this.name}=${input} to be one of: ${opts.options.join(', ')}`)
@@ -20,7 +24,7 @@ const _enum = <T = string>(opts: EnumFlagOptions<T>): OptionFlag<T> => {
     },
     helpValue: `(${opts.options.join('|')})`,
     ...opts,
-  })() as OptionFlag<T>
+  })()
 }
 
 export {_enum as enum}
diff --git a/src/parser/flags.ts b/src/parser/flags.ts
index 840a03e8e..eaa8bcd80 100644
--- a/src/parser/flags.ts
+++ b/src/parser/flags.ts
@@ -75,7 +75,7 @@ export function directory(opts: { exists?: boolean } & Partial<OptionFlag<string
 
 export function file(opts: Partial<OptionFlag<string>> & { exists?: boolean } & {multiple: true} & ({required: true} | { default: Default<string> })): OptionFlag<string[]>
 export function file(opts: Partial<OptionFlag<string>> & { exists?: boolean } & {multiple: true}): OptionFlag<string[] | undefined>
-export function file(opts: { exists?: boolean } & Partial<OptionFlag<string>> & ({required: true} | { default: string })): OptionFlag<string>
+export function file(opts: { exists?: boolean } & Partial<OptionFlag<string>> & ({required: true} | { default: Default<string> })): OptionFlag<string>
 export function file(opts?: { exists?: boolean } & Partial<OptionFlag<string>>): OptionFlag<string | undefined>
 export function file(opts: { exists?: boolean } & Partial<OptionFlag<string>> = {}): OptionFlag<string> | OptionFlag<string[]> | OptionFlag<string | undefined> | OptionFlag<string[] | undefined> {
   return build<string>({