/*
 * Copyright 2000-2006 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.communicator.util;

import com.intellij.util.ui.Tree;
import org.apache.log4j.Logger;

import javax.swing.*;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreePath;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.KeyEvent;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;

/**
 * @author Kir
 */
public class KirTree extends Tree {
  private static final Logger LOG = Logger.getLogger(KirTree.class);

  public KirTree() {
    setOpaque(true);
    setBackground(Color.white);

    setRootVisible(false);
    setShowsRootHandles(true);

    ToolTipManager.sharedInstance().registerComponent(this);

    setCellRenderer(new DefaultTreeCellRenderer(){
      public Component getTreeCellRendererComponent(JTree tree1, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus1) {
        Component renderer = super.getTreeCellRendererComponent(tree1, value, sel, expanded, leaf, row, hasFocus1);
        if (renderer instanceof JLabel && value instanceof KirTreeNode) {
          JLabel label = (JLabel) renderer;
          renderer = ((KirTreeNode) value).renderIn(label, sel, hasFocus);
        }
        return renderer;
      }
    });

    addMouseListener(new MouseAdapter() {
      public void mouseReleased(MouseEvent e) {
        // Disable selection on mouse release
        //e.consume();
      }

      public void mousePressed(final MouseEvent e) {
        final TreePath closestPath = getPathForLocation(e.getX(), e.getY());
        if (e.getClickCount() == 2) {
          TreePath path = getSelectionPath();
          if (path != null && path.equals(closestPath)) {
            onDblClick(path, path.getLastPathComponent(), e);
          }
        }
        else if (e.getClickCount() == 1) {

          if (e.getButton() == MouseEvent.BUTTON1) {
            UIUtil.invokeLater(new Runnable() {
              public void run() {
                TreePath path = getSelectionPath();
                if (path != null && path.equals(closestPath)) {
                  onClick(path, path.getLastPathComponent(), e);
                }
              }
            });
          }
          else {
            if (!isPathSelected(closestPath)) {
              setSelectionPath(closestPath);
            }
          }
        }
      }
    });

    getInputMap().put(KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0), "Enter");
    getActionMap().put("Enter", new AbstractAction() {
      public void actionPerformed(ActionEvent e) {
        onEnter();
      }
    });
  }

  protected void onEnter() {
  }

  protected void onDblClick(TreePath path, Object pathComponent, MouseEvent e) {
    onEnter();
  }

  protected void onClick(TreePath path, Object pathComponent, MouseEvent e) {
    LOG.debug("Click on " + pathComponent);
  }
}