/*
* Copyright 2000-2009 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 com.intellij.openapi.roots.ui.configuration.artifacts;
import com.intellij.ide.dnd.AdvancedDnDSource;
import com.intellij.ide.dnd.DnDAction;
import com.intellij.ide.dnd.DnDDragStartBean;
import com.intellij.ide.dnd.DnDManager;
import com.intellij.ide.dnd.aware.DnDAwareTree;
import com.intellij.ide.util.treeView.AbstractTreeBuilder;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.project.ProjectBundle;
import com.intellij.openapi.roots.ui.configuration.artifacts.nodes.PackagingElementNode;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.wm.IdeFocusManager;
import com.intellij.packaging.elements.PackagingElement;
import com.intellij.packaging.elements.RenameablePackagingElement;
import com.intellij.ui.TreeSpeedSearch;
import com.intellij.ui.TreeUIHelper;
import com.intellij.ui.treeStructure.SimpleNode;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.containers.Convertor;
import com.intellij.util.ui.tree.TreeUtil;
import org.jetbrains.annotations.Nullable;
import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreePath;
import java.awt.*;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
/**
* @author nik
*/
public class LayoutTree extends SimpleDnDAwareTree implements AdvancedDnDSource {
private static final Logger LOG = Logger.getInstance("#com.intellij.openapi.roots.ui.configuration.artifacts.LayoutTree");
private final ArtifactEditorImpl myArtifactsEditor;
public LayoutTree(ArtifactEditorImpl artifactsEditor) {
myArtifactsEditor = artifactsEditor;
setRootVisible(true);
setShowsRootHandles(false);
setCellEditor(new LayoutTreeCellEditor());
if (!ApplicationManager.getApplication().isUnitTestMode()) {
DnDManager.getInstance().registerSource(this);
}
//todo[nik,pegov] fix for tooltips in the tree. Otherwise tooltips will be ignored by DnDEnabled
setToolTipText("");
}
public void addSubtreeToUpdate(DefaultMutableTreeNode newNode) {
AbstractTreeBuilder.getBuilderFor(this).addSubtreeToUpdate(newNode);
}
@Override
protected void configureUiHelper(TreeUIHelper helper) {
final Convertor<TreePath, String> convertor = path -> {
final SimpleNode node = getNodeFor(path);
if (node instanceof PackagingElementNode) {
return ((PackagingElementNode<?>)node).getElementPresentation().getSearchName();
}
return "";
};
new TreeSpeedSearch(this, convertor, true);
}
private List<PackagingElementNode<?>> getNodesToDrag() {
return getSelection().getNodes();
}
@Override
public boolean canStartDragging(DnDAction action, Point dragOrigin) {
return !getNodesToDrag().isEmpty();
}
@Override
public DnDDragStartBean startDragging(DnDAction action, Point dragOrigin) {
return new DnDDragStartBean(new LayoutNodesDraggingObject(myArtifactsEditor, getNodesToDrag()));
}
@Override
public Pair<Image, Point> createDraggedImage(DnDAction action, Point dragOrigin) {
final List<PackagingElementNode<?>> nodes = getNodesToDrag();
if (nodes.size() == 1) {
return DnDAwareTree.getDragImage(this, getPathFor(nodes.get(0)), dragOrigin);
}
return DnDAwareTree.getDragImage(this, ProjectBundle.message("drag.n.drop.text.0.packaging.elements", nodes.size()), dragOrigin);
}
@Override
public void dragDropEnd() {
}
@Override
public void dropActionChanged(int gestureModifiers) {
}
@Override
public void dispose() {
if (!ApplicationManager.getApplication().isUnitTestMode()) {
DnDManager.getInstance().unregisterSource(this);
}
}
public LayoutTreeSelection getSelection() {
return new LayoutTreeSelection(this);
}
@Nullable
public PackagingElement<?> getElementByPath(TreePath path) {
final SimpleNode node = getNodeFor(path);
if (node instanceof PackagingElementNode) {
final List<? extends PackagingElement<?>> elements = ((PackagingElementNode<?>)node).getPackagingElements();
if (!elements.isEmpty()) {
return elements.get(0);
}
}
return null;
}
public PackagingElementNode<?> getRootPackagingNode() {
final SimpleNode node = getNodeFor(new TreePath(getRootNode()));
return node instanceof PackagingElementNode ? (PackagingElementNode<?>)node : null;
}
public DefaultMutableTreeNode getRootNode() {
return (DefaultMutableTreeNode)getModel().getRoot();
}
public List<PackagingElementNode<?>> findNodes(final Collection<? extends PackagingElement<?>> elements) {
final List<PackagingElementNode<?>> nodes = new ArrayList<>();
TreeUtil.traverseDepth(getRootNode(), node -> {
final Object userObject = ((DefaultMutableTreeNode)node).getUserObject();
if (userObject instanceof PackagingElementNode) {
final PackagingElementNode<?> packagingNode = (PackagingElementNode<?>)userObject;
final List<? extends PackagingElement<?>> nodeElements = packagingNode.getPackagingElements();
if (ContainerUtil.intersects(nodeElements, elements)) {
nodes.add(packagingNode);
}
}
return true;
});
return nodes;
}
public void addSubtreeToUpdate(final PackagingElementNode elementNode) {
final DefaultMutableTreeNode node = TreeUtil.findNodeWithObject(getRootNode(), elementNode);
if (node != null) {
addSubtreeToUpdate(node);
}
}
@Nullable
public PackagingElementNode<?> findCompositeNodeByPath(String parentPath) {
PackagingElementNode<?> node = getRootPackagingNode();
for (String name : StringUtil.split(parentPath, "/")) {
if (node == null) {
return null;
}
node = node.findCompositeChild(name);
}
return node;
}
private class LayoutTreeCellEditor extends DefaultCellEditor {
public LayoutTreeCellEditor() {
super(new JTextField());
}
@Override
public Component getTreeCellEditorComponent(JTree tree, Object value, boolean isSelected, boolean expanded, boolean leaf, int row) {
final JTextField field = (JTextField)super.getTreeCellEditorComponent(tree, value, isSelected, expanded, leaf, row);
final Object node = ((DefaultMutableTreeNode)value).getUserObject();
final PackagingElement<?> element = ((PackagingElementNode)node).getElementIfSingle();
LOG.assertTrue(element != null);
final String name = ((RenameablePackagingElement)element).getName();
field.setText(name);
int i = name.lastIndexOf('.');
field.setSelectionStart(0);
field.setSelectionEnd(i != -1 ? i : name.length());
return field;
}
@Override
public boolean stopCellEditing() {
final String newValue = ((JTextField)editorComponent).getText();
final TreePath path = getEditingPath();
final Object node = getNodeFor(path);
RenameablePackagingElement currentElement = null;
if (node instanceof PackagingElementNode) {
final PackagingElement<?> element = ((PackagingElementNode)node).getElementIfSingle();
if (element instanceof RenameablePackagingElement) {
currentElement = (RenameablePackagingElement)element;
}
}
final boolean stopped = super.stopCellEditing();
if (stopped && currentElement != null) {
final RenameablePackagingElement finalCurrentElement = currentElement;
myArtifactsEditor.getLayoutTreeComponent().editLayout(() -> finalCurrentElement.rename(newValue));
myArtifactsEditor.queueValidation();
myArtifactsEditor.getLayoutTreeComponent().updatePropertiesPanel(true);
addSubtreeToUpdate((DefaultMutableTreeNode)path.getLastPathComponent());
requestFocusToTree();
}
return stopped;
}
@Override
public void cancelCellEditing() {
super.cancelCellEditing();
requestFocusToTree();
}
private void requestFocusToTree() {
IdeFocusManager.getInstance(myArtifactsEditor.getContext().getProject()).requestFocus(LayoutTree.this, true);
}
}
}