Add 'when' builder and creation of 'is' expressions

2013-04-26 13:38:57 +04:00
parent b7bc251311
commit 057528eb39
1 changed files with 109 additions and 0 deletions
@@ -275,6 +275,22 @@ public class JetPsiFactory {
        return new JetExpressionCodeFragmentImpl(project, "fragment.kt", text, context);
    }

+    @NotNull
+    public static JetIsExpression createIsExpression(Project project, @NotNull String lhs, @NotNull String rhs, boolean negated) {
+        return (JetIsExpression) createExpression(project, lhs + " " + (negated ? "!is" : "is") + " " + rhs);
+    }
+
+    @SuppressWarnings("ConstantConditions")
+    @NotNull
+    public static JetIsExpression createIsExpression(Project project, @NotNull JetExpression lhs, @NotNull JetTypeReference rhs, boolean negated) {
+        JetIsExpression isExpression = createIsExpression(project, "_", "_", negated);
+
+        isExpression = (JetIsExpression)isExpression.getLeftHandSide().replace(lhs).getParent();
+        isExpression = (JetIsExpression)isExpression.getTypeRef().replace(rhs).getParent();
+
+        return isExpression;
+    }
+
    @NotNull
    public static JetReturnExpression createReturn(Project project, @NotNull String text) {
        return (JetReturnExpression) createExpression(project, "return " + text);
@@ -312,4 +328,97 @@ public class JetPsiFactory {

        return ifExpr;
    }
+
+    public static class WhenTemplateBuilder {
+        private final StringBuilder sb = new StringBuilder("when ");
+        private boolean frozen = false;
+        private boolean inCondition = false;
+
+        public WhenTemplateBuilder(boolean addSubject) {
+            if (addSubject) {
+                sb.append("(_) ");
+            }
+            sb.append("{\n");
+        }
+
+        private void finishAndFreeze() {
+            sb.append("else -> _\n}");
+            frozen = true;
+        }
+
+        private WhenTemplateBuilder addCondition(String text) {
+            assert !frozen;
+
+            if (!inCondition) {
+                inCondition = true;
+            } else {
+                sb.append(", ");
+            }
+            sb.append(text);
+
+            return this;
+        }
+
+        public WhenTemplateBuilder addBranchWithSingleCondition() {
+            assert !frozen;
+
+            sb.append("_ -> _\n");
+
+            return this;
+        }
+
+        public WhenTemplateBuilder addBranchesWithSingleCondition(int count) {
+            for (int i = 0; i < count; i++) {
+                addBranchWithSingleCondition();
+            }
+
+            return this;
+        }
+
+        public WhenTemplateBuilder addBranchWithMultiCondition(int conditionCount) {
+            assert !frozen;
+            assert conditionCount > 0;
+
+            sb.append("_");
+            for (int i = 0; i < conditionCount - 1; i++) {
+                sb.append(", _");
+            }
+            sb.append(" -> _\n");
+
+            return this;
+        }
+
+        public WhenTemplateBuilder addExpressionCondition() {
+            return addCondition("_");
+        }
+
+        public WhenTemplateBuilder addIsCondition(boolean negated) {
+            return addCondition((negated ? "!is" : "is") + " _");
+        }
+
+        public WhenTemplateBuilder addInCondition(boolean negated) {
+            return addCondition((negated ? "!in" : "in") + " _");
+        }
+
+        public WhenTemplateBuilder finishBranch() {
+            assert !frozen;
+            assert inCondition;
+
+            inCondition = false;
+            sb.append(" -> _\n");
+
+            return this;
+        }
+
+        public JetWhenExpression toExpression(Project project) {
+            if (!frozen) {
+                finishAndFreeze();
+            }
+            return (JetWhenExpression) createExpression(project, sb.toString());
+        }
+    }
+
+    public static JetWhenExpression createWhenWithoutSubject(Project project, int positiveBranchCount) {
+        return new WhenTemplateBuilder(false).addBranchesWithSingleCondition(positiveBranchCount).toExpression(project);
+    }
 }