make FlagExclusivity.chooseFirst work (#67)

Author: glessard

Sources/ArgumentParser/Parsing/ArgumentSet.swift

@@ -176,12 +176,14 @@ extension ArgumentSet {
     case(true, let previous?, .exclusive):
       // This value has already been set.
       throw ParserError.duplicateExclusiveValues(previous: previous.inputOrigin, duplicate: origin, originalInput: values.originalInput)
+    case (true, _, .chooseFirst):
+      values.update(forKey: key, inputOrigin: origin, initial: value, closure: { _ in })
     case (false, _, _), (_, _, .chooseLast):
       values.set(value, forKey: key, inputOrigin: origin)
-      return true
     default:
-      return hasUpdated
+      break
     }
+    return true
   }
 
   /// Creates an argument set for a pair of inverted Boolean flags.

Tests/EndToEndTests/FlagsEndToEndTests.swift

@@ -80,10 +80,9 @@ extension FlagsEndToEndTests {
     AssertParse(Bar.self, ["--enable-logging"]) { options in
       XCTAssertEqual(options.logging, true)
     }
-// Can't test this yet, because .chooseFirst flags don't work
-//    AssertParse(Bar.self, ["--enable-logging", "--disable-logging"]) { options in
-//      XCTAssertEqual(options.logging, false)
-//    }
+    AssertParse(Bar.self, ["--disable-logging", "--enable-logging"]) { options in
+      XCTAssertEqual(options.logging, false)
+    }
   }
 }
 
@@ -263,3 +262,25 @@ extension FlagsEndToEndTests {
     XCTAssertThrowsError(try Qux.parse(["--pink", "--small", "--bloop"]))
   }
 }
+
+fileprivate struct RepeatOK: ParsableArguments {
+  @Flag(exclusivity: .chooseFirst)
+  var color: Color
+
+  @Flag(exclusivity: .chooseLast)
+  var shape: Shape
+}
+
+extension FlagsEndToEndTests {
+  func testParsingCaseIterable_RepeatableFlags() throws {
+    AssertParse(RepeatOK.self, ["--pink", "--purple", "--square"]) { options in
+      XCTAssertEqual(options.color, .pink)
+      XCTAssertEqual(options.shape, .square)
+    }
+
+    AssertParse(RepeatOK.self, ["--round", "--oblong", "--silver"]) { options in
+      XCTAssertEqual(options.color, .silver)
+      XCTAssertEqual(options.shape, .oblong)
+    }
+  }
+}