From 9c63bb52e9a8a15574dd16f2856174d197ac7713 Mon Sep 17 00:00:00 2001
From: Peter Hale <peter.hale@salesforce.com>
Date: Tue, 6 Dec 2022 13:01:58 -0700
Subject: [PATCH] feat: handle custom parser nested array for multiple flag

@W-12165862@
---
 src/interfaces/parser.ts  |  2 +-
 src/parser/parse.ts       |  2 +-
 test/parser/parse.test.ts | 16 ++++++++++++++++
 3 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/interfaces/parser.ts b/src/interfaces/parser.ts
index 6b479627..797de0f9 100644
--- a/src/interfaces/parser.ts
+++ b/src/interfaces/parser.ts
@@ -175,7 +175,7 @@ export type OptionFlagProps = FlagProps & {
   multiple?: boolean;
 }
 
-export type FlagParser<T, I, P = any> = (input: I, context: any, opts: P & OptionFlag<T>) => Promise<T>
+export type FlagParser<T, I, P = any> = (input: I, context: any, opts: P & OptionFlag<T>) => Promise<T | T[]>
 
 export type FlagBase<T, I, P = any> = FlagProps & {
   parse: FlagParser<T, I, P>;
diff --git a/src/parser/parse.ts b/src/parser/parse.ts
index 58557880..0400d649 100644
--- a/src/parser/parse.ts
+++ b/src/parser/parse.ts
@@ -201,7 +201,7 @@ export class Parser<T extends ParserInput, TFlags extends OutputFlags<T['flags']
         const value = flag.parse ? await flag.parse(input, this.context, flag) : input
         if (flag.multiple) {
           flags[token.flag] = flags[token.flag] || []
-          flags[token.flag].push(value)
+          flags[token.flag].push(...(Array.isArray(value) ? value : [value]))
         } else {
           flags[token.flag] = value
         }
diff --git a/test/parser/parse.test.ts b/test/parser/parse.test.ts
index 54599390..9179c6db 100644
--- a/test/parser/parse.test.ts
+++ b/test/parser/parse.test.ts
@@ -634,6 +634,22 @@ See more help with --help`)
       })
       expect(out.args).to.deep.include({num: '15'})
     })
+    it('flag multiple with arguments, custom parser', async () => {
+      const out = await parse(
+        ['--foo', './a.txt,./b.txt', '--foo', './c.txt', '--', '15'],
+        {
+          args: [{name: 'num'}],
+          flags: {foo: flags.string({
+            multiple: true,
+            parse: async input => input.split(',').map(i => i.trim()),
+          })},
+        },
+      )
+      expect(out.flags).to.deep.include({
+        foo: ['./a.txt', './b.txt', './c.txt'],
+      })
+      expect(out.args).to.deep.include({num: '15'})
+    })
   })
 
   describe('defaults', () => {