di: shortcut to initialize constant enum fields

2012-04-07 06:32:18 +04:00
parent b596c6d81a
commit 829173d5ec
2 changed files with 6 additions and 2 deletions
@@ -132,8 +132,7 @@ public class AllInjectorsGenerator {
        generator.addPublicField(PsiClassFinderForJvm.class);
        generator.addField(false, ModuleDescriptor.class, null,
                new GivenExpression("new org.jetbrains.jet.lang.descriptors.ModuleDescriptor(\"<dummy>\")"));
-        generator.addField(false, CompilerSpecialMode.class, null,
-                new GivenExpression("CompilerSpecialMode.REGULAR"));
+        generator.addField(CompilerSpecialMode.REGULAR);

        // Parameters
        generator.addPublicParameter(Project.class);
@@ -183,6 +183,11 @@ public class DependencyInjectorGenerator {
        return addField(false, type, null, null);
    }

+    public Field addField(Enum<?> enu) {
+        Class<? extends Enum> clazz = enu.getClass();
+        return addField(false, clazz, null, new GivenExpression(clazz.getSimpleName() + "." + enu.name()));
+    }
+
    public Field addField(boolean isPublic, Class<?> type, @Nullable String name, @Nullable Expression init) {
        return addField(isPublic, new DiType(type), name, init);
    }