JetMultiDeclarationEntry implements JetVariableDeclaration

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetMultiDeclaration.java

@@ -30,30 +30,11 @@ import static org.jetbrains.jet.lexer.JetTokens.VAR_KEYWORD;
 /**
  * @author abreslav
  */
-public class JetMultiDeclaration extends JetDeclarationImpl implements JetVariableDeclaration  {
+public class JetMultiDeclaration extends JetDeclarationImpl {
     public JetMultiDeclaration(@NotNull ASTNode node) {
         super(node);
     }
 
-    @Override
-    public boolean isVar() {
-        return getNode().findChildByType(JetTokens.VAR_KEYWORD) != null;
-    }
-
-    @Nullable
-    @Override
-    public JetExpression getInitializer() {
-        return PsiTreeUtil.getNextSiblingOfType(findChildByType(EQ), JetExpression.class);
-    }
-
-    @NotNull
-    @Override
-    public ASTNode getValOrVarNode() {
-        ASTNode node = getNode().findChildByType(TokenSet.create(VAL_KEYWORD, VAR_KEYWORD));
-        assert node != null : "Val or var should always exist for property";
-        return node;
-    }
-
     @Override
     public void accept(@NotNull JetVisitorVoid visitor) {
         visitor.visitMultiDeclaration(this);

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetMultiDeclarationEntry.java

@@ -17,20 +17,27 @@
 package org.jetbrains.jet.lang.psi;
 
 import com.intellij.lang.ASTNode;
+import com.intellij.psi.tree.TokenSet;
+import com.intellij.psi.util.PsiTreeUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.jetbrains.jet.JetNodeTypes;
+import org.jetbrains.jet.lexer.JetTokens;
+
+import static org.jetbrains.jet.lexer.JetTokens.EQ;
+import static org.jetbrains.jet.lexer.JetTokens.VAL_KEYWORD;
+import static org.jetbrains.jet.lexer.JetTokens.VAR_KEYWORD;
 
 /**
  * @author abreslav
  */
-public class JetMultiDeclarationEntry extends JetNamedDeclarationNotStubbed {
+public class JetMultiDeclarationEntry extends JetNamedDeclarationNotStubbed implements JetVariableDeclaration {
     public JetMultiDeclarationEntry(@NotNull ASTNode node) {
         super(node);
     }
 
-    @Nullable
-    public JetTypeReference getTypeRef() {
+    @Override
+    public JetTypeReference getPropertyTypeRef() {
         return (JetTypeReference) findChildByType(JetNodeTypes.TYPE_REFERENCE);
     }
 
@@ -44,5 +51,33 @@ public class JetMultiDeclarationEntry extends JetNamedDeclarationNotStubbed {
         return visitor.visitMultiDeclarationEntry(this, data);
     }
 
+    @Override
+    public boolean isVar() {
+        return getParentNode().findChildByType(JetTokens.VAR_KEYWORD) != null;
+    }
 
+    @Nullable
+    @Override
+    public JetExpression getInitializer() {
+        ASTNode eqNode = getParentNode().findChildByType(EQ);
+        if (eqNode == null) {
+            return null;
+        }
+        return PsiTreeUtil.getNextSiblingOfType(eqNode.getPsi(), JetExpression.class);
+    }
+
+    @NotNull
+    private ASTNode getParentNode() {
+        ASTNode parent = getNode().getTreeParent();
+        assert parent.getElementType() == JetNodeTypes.MULTI_VARIABLE_DECLARATION;
+        return parent;
+    }
+
+    @NotNull
+    @Override
+    public ASTNode getValOrVarNode() {
+        ASTNode node = getParentNode().findChildByType(TokenSet.create(VAL_KEYWORD, VAR_KEYWORD));
+        assert node != null : "Val or var should always exist for property";
+        return node;
+    }
 }

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetProperty.java

@@ -114,6 +114,7 @@ public class JetProperty extends JetTypeParameterListOwnerStub<PsiJetPropertyStu
         return null;
     }
 
+    @Override
     @Nullable
     public JetTypeReference getPropertyTypeRef() {
         ASTNode node = getNode().getFirstChildNode();

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetVariableDeclaration.java

@@ -1,7 +1,24 @@
+/*
+ * Copyright 2010-2012 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package org.jetbrains.jet.lang.psi;
 
 import com.intellij.lang.ASTNode;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 /**
  * @author abreslav
@@ -11,4 +28,7 @@ public interface JetVariableDeclaration extends JetDeclaration, JetWithExpressio
 
     @NotNull
     ASTNode getValOrVarNode();
+
+    @Nullable
+    JetTypeReference getPropertyTypeRef();
 }