package treeviewertest;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;

import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.TreeViewerColumn;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeColumn;
import org.eclipse.ui.ISharedImages;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.part.ViewPart;

public class View extends ViewPart {
  public View() {
  }

  public static final String ID = "TreeViewerTest.view";

  private TreeViewer viewer;

  class Node {
    final List<View.Node> children = new ArrayList<View.Node>();
    final int id;
    String text;

    public Node(int id, String text) {
      this.id = id;
      this.text = text;
    }

    public String getText() {
      return text;
    }

    public void setText(String text) {
      this.text = text;
    }

    public List<View.Node> getChildren() {
      return children;
    }

    public int getId() {
      return id;
    }

    @Override
    public int hashCode() {
      return id;
    }

    @Override
    public boolean equals(Object obj) {
      return id == ((Node) obj).id;
    }
  }

  class ViewContentProvider implements ITreeContentProvider {
    Node root = null;

    public void inputChanged(Viewer v, Object oldInput, Object newInput) {
      root = (Node) newInput;
    }

    public void dispose() {
    }

    public Object[] getElements(Object parent) {
      if (parent instanceof Node) {
        return ((Node) parent).getChildren().toArray();
      }
      return new Object[0];
    }

    @Override
    public Object[] getChildren(Object parentElement) {
      return ((Node) parentElement).getChildren().toArray();
    }

    @Override
    public Object getParent(Object element) {
      return null;
    }

    @Override
    public boolean hasChildren(Object element) {
      return ((Node) element).getChildren().size() > 0;
    }
  }

  class ViewLabelProvider extends LabelProvider implements ITableLabelProvider {
    public String getColumnText(Object obj, int index) {
      switch (index) {
        case 0:
          return "" + ((Node) obj).getId();
        case 1:
          return ((Node) obj).getText();
        default:
          return getText(obj);
      }
    }

    public Image getColumnImage(Object obj, int index) {
      return null;
    }

    public Image getImage(Object obj) {
      return PlatformUI.getWorkbench().getSharedImages()
          .getImage(ISharedImages.IMG_OBJ_ELEMENT);
    }
  }

  /**
   * This is a callback that will allow us to create the viewer and initialize
   * it.
   */
  public void createPartControl(Composite parent) {
    Composite composite = new Composite(parent, SWT.NONE);
    composite.setLayout(new GridLayout(1, false));

    Button btnEditElement = new Button(composite, SWT.NONE);
    btnEditElement.addSelectionListener(new SelectionAdapter() {
      @Override
      public void widgetSelected(SelectionEvent e) {
        edit();
      }
    });
    btnEditElement.setText("Edit Element 23");
    viewer = new TreeViewer(composite, SWT.MULTI | SWT.H_SCROLL | SWT.V_SCROLL
        | SWT.FULL_SELECTION);
    viewer.setUseHashlookup(true);
    Tree tree = viewer.getTree();
    tree.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1));
    tree.setHeaderVisible(true);

    TreeViewerColumn treeViewerColumn = new TreeViewerColumn(viewer, SWT.NONE);
    TreeColumn idCol = treeViewerColumn.getColumn();
    idCol.setWidth(100);
    idCol.setText("ID");

    TreeViewerColumn treeViewerColumn_1 = new TreeViewerColumn(viewer, SWT.NONE);
    TreeColumn textCol = treeViewerColumn_1.getColumn();
    textCol.setWidth(100);
    textCol.setText("Text");
    viewer.setContentProvider(new ViewContentProvider());
    viewer.setLabelProvider(new ViewLabelProvider());

    Node root = new Node(-1, "root");
    createTree(root, 1);
    viewer.setInput(root);
  }

  private int i = 0;
  Random r = new Random(1235l);
  Node el;

  private void createTree(Node parent, int depth) {
    for (int j = 0; j < 6 / depth; j++) {
      if (i > 40) {
        return;
      }
      Node child = new Node(i, "Node " + i);
      if (i == 23) {
        el = child;
      }
      i++;
      parent.getChildren().add(child);
      if (r.nextInt(11) < 6) {
        createTree(child, depth + 1);
      }
    }
  }

  public void setFocus() {
    viewer.getControl().setFocus();
  }

  private void edit() {
    if (el != null) {
      el.setText("Hallo. Wurde gerade editiert! " + (i++));
      viewer.refresh(el);
    }
  }
}