001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.lang.psi;
018
019 import com.intellij.extapi.psi.PsiFileBase;
020 import com.intellij.lang.ASTNode;
021 import com.intellij.lang.FileASTNode;
022 import com.intellij.openapi.application.ApplicationManager;
023 import com.intellij.openapi.fileTypes.FileType;
024 import com.intellij.openapi.util.text.StringUtil;
025 import com.intellij.openapi.vfs.VirtualFile;
026 import com.intellij.openapi.vfs.VirtualFileWithId;
027 import com.intellij.psi.FileViewProvider;
028 import com.intellij.psi.PsiElementVisitor;
029 import com.intellij.psi.PsiFile;
030 import com.intellij.psi.PsiLock;
031 import com.intellij.psi.stubs.ObjectStubTree;
032 import com.intellij.psi.stubs.StubBase;
033 import com.intellij.psi.stubs.StubTree;
034 import com.intellij.psi.stubs.StubTreeLoader;
035 import com.intellij.psi.tree.IElementType;
036 import com.intellij.psi.tree.IStubFileElementType;
037 import com.intellij.psi.util.PsiTreeUtil;
038 import org.jetbrains.annotations.NotNull;
039 import org.jetbrains.annotations.Nullable;
040 import org.jetbrains.jet.JetNodeTypes;
041 import org.jetbrains.jet.lang.psi.stubs.PsiJetFileStub;
042 import org.jetbrains.jet.plugin.JetFileType;
043 import org.jetbrains.jet.plugin.JetLanguage;
044
045 import java.util.List;
046
047 public class JetFile extends PsiFileBase implements JetDeclarationContainer, JetElement {
048
049 private final boolean isCompiled;
050
051 public JetFile(FileViewProvider viewProvider, boolean compiled) {
052 super(viewProvider, JetLanguage.INSTANCE);
053 this.isCompiled = compiled;
054 }
055
056 @Override
057 public FileASTNode getNode() {
058 return super.getNode();
059 }
060
061 public boolean isCompiled() {
062 return isCompiled;
063 }
064
065 @Override
066 @NotNull
067 public FileType getFileType() {
068 return JetFileType.INSTANCE;
069 }
070
071 @Override
072 public String toString() {
073 return "JetFile: " + getName();
074 }
075
076 @NotNull
077 @Override
078 public List<JetDeclaration> getDeclarations() {
079 return PsiTreeUtil.getChildrenOfTypeAsList(this, JetDeclaration.class);
080 }
081
082 public List<JetImportDirective> getImportDirectives() {
083 return PsiTreeUtil.getChildrenOfTypeAsList(this, JetImportDirective.class);
084 }
085
086 @Nullable
087 public JetImportDirective findImportByAlias(@NotNull String name) {
088 for (JetImportDirective directive : getImportDirectives()) {
089 if (name.equals(directive.getAliasName())) {
090 return directive;
091 }
092 }
093 return null;
094 }
095
096 // scripts has no namespace header
097 @Nullable
098 public JetNamespaceHeader getNamespaceHeader() {
099 ASTNode ast = getNode().findChildByType(JetNodeTypes.NAMESPACE_HEADER);
100 return ast != null ? (JetNamespaceHeader) ast.getPsi() : null;
101 }
102
103 @Nullable
104 public String getPackageName() {
105 PsiJetFileStub stub = (PsiJetFileStub) getStub();
106 if (stub != null) {
107 return stub.getPackageName();
108 }
109
110 JetNamespaceHeader statement = getNamespaceHeader();
111 return statement != null ? statement.getQualifiedName() : null;
112 }
113
114 @Nullable
115 public JetScript getScript() {
116 return PsiTreeUtil.getChildOfType(this, JetScript.class);
117 }
118
119 public boolean isScript() {
120 PsiJetFileStub stub = (PsiJetFileStub) getStub();
121 if (stub != null) {
122 return stub.isScript();
123 }
124
125 return getScript() != null;
126 }
127
128 @NotNull
129 @Override
130 public String getName() {
131 return super.getName(); // TODO
132 }
133
134 @Override
135 public void accept(@NotNull PsiElementVisitor visitor) {
136 if (visitor instanceof JetVisitorVoid) {
137 accept((JetVisitorVoid) visitor);
138 }
139 else {
140 visitor.visitFile(this);
141 }
142 }
143
144 @Override
145 public <D> void acceptChildren(@NotNull JetTreeVisitor<D> visitor, D data) {
146 JetPsiUtil.visitChildren(this, visitor, data);
147 }
148
149 @Override
150 public void accept(@NotNull JetVisitorVoid visitor) {
151 visitor.visitJetFile(this);
152 }
153
154 @Override
155 public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
156 return visitor.visitJetFile(this, data);
157 }
158 }