aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/firrtl/options/OptionParser.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/firrtl/options/OptionParser.scala')
-rw-r--r--src/main/scala/firrtl/options/OptionParser.scala39
1 files changed, 39 insertions, 0 deletions
diff --git a/src/main/scala/firrtl/options/OptionParser.scala b/src/main/scala/firrtl/options/OptionParser.scala
new file mode 100644
index 00000000..6d8095c0
--- /dev/null
+++ b/src/main/scala/firrtl/options/OptionParser.scala
@@ -0,0 +1,39 @@
+// See LICENSE for license details.
+
+package firrtl.options
+
+import firrtl.{FIRRTLException, AnnotationSeq}
+
+import scopt.OptionParser
+
+/** Causes an OptionParser to not call exit (call `sys.exit`) if the `--help` option is passed
+ */
+trait DoNotTerminateOnExit { this: OptionParser[_] =>
+ override def terminate(exitState: Either[String, Unit]): Unit = Unit
+}
+
+/** A modified OptionParser with mutable termination and additional checks
+ */
+trait DuplicateHandling extends OptionParser[AnnotationSeq] {
+
+ override def parse(args: Seq[String], init: AnnotationSeq): Option[AnnotationSeq] = {
+
+ /** Message for found duplicate options */
+ def msg(x: String, y: String) = s"""Duplicate $x "$y" (did your custom Transform or OptionsManager add this?)"""
+
+ val longDups = options.map(_.name).groupBy(identity).collect{ case (k, v) if v.size > 1 && k != "" => k }
+ val shortDups = options.map(_.shortOpt).flatten.groupBy(identity).collect{ case (k, v) if v.size > 1 => k }
+
+
+ if (longDups.nonEmpty) {
+ throw new OptionsException(msg("long option", longDups.map("--" + _).mkString(",")), new IllegalArgumentException)
+ }
+
+ if (shortDups.nonEmpty) {
+ throw new OptionsException(msg("short option", shortDups.map("-" + _).mkString(",")), new IllegalArgumentException)
+ }
+
+ super.parse(args, init)
+ }
+
+}