From c0f70d0ff90f34702ec399f4ab3b1f010a44efb9 Mon Sep 17 00:00:00 2001
From: Maxim Shafirov <max@maxbook.local>
Date: Sun, 9 Jan 2011 15:09:55 +0300
Subject: [PATCH] Extension node getters

---
 .../org/jetbrains/jet/lang/psi/JetExtension.java    | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/idea/src/org/jetbrains/jet/lang/psi/JetExtension.java b/idea/src/org/jetbrains/jet/lang/psi/JetExtension.java
index 11cc8343aee..455353146af 100644
--- a/idea/src/org/jetbrains/jet/lang/psi/JetExtension.java
+++ b/idea/src/org/jetbrains/jet/lang/psi/JetExtension.java
@@ -2,6 +2,7 @@ package org.jetbrains.jet.lang.psi;
 
 import com.intellij.lang.ASTNode;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 import org.jetbrains.jet.JetNodeTypes;
 
 import java.util.Collections;
@@ -15,6 +16,11 @@ public class JetExtension extends JetTypeParameterListOwner {
         super(node);
     }
 
+    @Override
+    public void accept(@NotNull JetVisitor visitor) {
+        visitor.visitExtension(this);
+    }
+
     public List<JetDeclaration> getDeclarations() {
         JetClassBody body = (JetClassBody) findChildByType(JetNodeTypes.CLASS_BODY);
         if (body == null) return Collections.emptyList();
@@ -22,9 +28,8 @@ public class JetExtension extends JetTypeParameterListOwner {
         return body.getDeclarations();
     }
 
-
-    @Override
-    public void accept(@NotNull JetVisitor visitor) {
-        visitor.visitExtension(this);
+    @Nullable
+    public JetTypeReference getTargetTypeRef() {
+        return (JetTypeReference) findChildByType(JetNodeTypes.TYPE_REFERENCE);
     }
 }