/*
* Copyright 2003-2016 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 jetbrains.mps.ide.projectPane;
import com.intellij.ide.SelectInContext;
import com.intellij.openapi.fileEditor.FileEditor;
import com.intellij.openapi.fileEditor.FileEditorManager;
import com.intellij.openapi.vfs.VirtualFile;
import jetbrains.mps.fileTypes.MPSFileTypesManager;
import jetbrains.mps.ide.editor.MPSFileNodeEditor;
import jetbrains.mps.ide.vfs.IdeaFile;
import jetbrains.mps.ide.vfs.IdeaFileSystem;
import jetbrains.mps.nodefs.MPSNodeVirtualFile;
import jetbrains.mps.project.MPSProject;
import jetbrains.mps.smodel.ModuleFileTracker;
import jetbrains.mps.smodel.SModelFileTracker;
import jetbrains.mps.vfs.FileSystem;
import jetbrains.mps.vfs.IFile;
import org.jetbrains.mps.openapi.model.SModel;
import org.jetbrains.mps.openapi.model.SNode;
import org.jetbrains.mps.openapi.module.SModule;
public class ProjectPaneSelectInTarget extends AbstractProjectViewSelectInTarget {
private static final IdeaFileSystem ourFileSystem = (IdeaFileSystem) FileSystem.getInstance();
private MPSProject myProject;
private boolean mySelectRoot;
public ProjectPaneSelectInTarget(MPSProject p, boolean selectRoot) {
super(p.getProject(), ProjectPane.ID, 0, selectRoot ? "Logical View" : "Project Tree");
myProject = p;
mySelectRoot = selectRoot;
}
@Override
public boolean canSelect(SelectInContext context) {
return getNode(context) != null || getModel(context)!=null || getModule(context)!=null;
}
@Override
protected void doSelectIn(SelectInContext context, boolean requestFocus) {
SNode node = getNode(context);
if (node != null) {
ProjectPane.getInstance(myProject).selectNode(node, requestFocus);
return;
}
SModel model = getModel(context);
if (model != null) {
ProjectPane.getInstance(myProject).selectModel(model, requestFocus);
return;
}
SModule module = getModule(context);
if (module != null) {
ProjectPane.getInstance(myProject).selectModule(module, requestFocus);
}
}
private SModel getModel(SelectInContext context) {
VirtualFile virtualFile = context.getVirtualFile();
if (!MPSFileTypesManager.isModelFile(virtualFile)) {
return null;
}
IFile modelFile = new IdeaFile(ourFileSystem, virtualFile);
return SModelFileTracker.getInstance(myProject.getRepository()).findModel(modelFile);
}
private SModule getModule(SelectInContext context) {
VirtualFile virtualFile = context.getVirtualFile();
if (!MPSFileTypesManager.isModuleFile(virtualFile)) {
return null;
}
IFile moduleFile = new IdeaFile(ourFileSystem, virtualFile);
// XXX why don't we obtain model read here? ModuleFileTracker might need to walk repository to find out actual modules and their files
return ModuleFileTracker.getInstance(myProject.getRepository()).getModuleByFile(moduleFile);
}
private SNode getNode(SelectInContext context) {
VirtualFile virtualFile = context.getVirtualFile();
if (!(virtualFile instanceof MPSNodeVirtualFile)) {
return null;
}
MPSNodeVirtualFile file = (MPSNodeVirtualFile) virtualFile;
FileEditor[] editors = FileEditorManager.getInstance(myProject.getProject()).getEditors(file);
if (editors.length != 0) {
FileEditor editor = editors[0];
if (!(editor instanceof MPSFileNodeEditor)) return null;
jetbrains.mps.openapi.editor.EditorComponent editorComponent = ((MPSFileNodeEditor) editor).getNodeEditor().getCurrentEditorComponent();
if (editorComponent == null) return null;
return mySelectRoot ? editorComponent.getEditedNode() : editorComponent.getSelectedNode();
} else {
return file.getNode();
}
}
}