compiler argument parsing cleanup

2012-04-08 02:08:26 +04:00
parent a43c0c19a8
commit 0a610d81ad
1 changed files with 13 additions and 6 deletions
@@ -106,12 +106,19 @@ public class KotlinCompiler {
        CompilerSpecialMode mode;
        if (arguments.mode == null) {
            mode = CompilerSpecialMode.REGULAR;
-        } else if (arguments.mode.equals("jdkHeaders")) {
-            mode = CompilerSpecialMode.JDK_HEADERS;
-        } else if (arguments.mode.equals("builtins")) {
-            mode = CompilerSpecialMode.BUILTINS;
-        } else {
-            throw new IllegalArgumentException("unknown compiler mode: " + arguments.mode);
+        }
+        else {
+            computeMode:
+            {
+                for (CompilerSpecialMode variant : CompilerSpecialMode.values()) {
+                    if (arguments.mode.equalsIgnoreCase(variant.name().replaceAll("_", ""))) {
+                        mode = variant;
+                        break computeMode;
+                    }
+                }
+                // TODO: report properly
+                throw new IllegalArgumentException("unknown compiler mode: " + arguments.mode);
+            }
        }

        CompileEnvironment environment = new CompileEnvironment(messageRenderer, arguments.verbose, mode);