/**
    * Copyright (C) 2008 Atlassian
    * 
    * 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.atlassian.theplugin.idea.crucible.tree;
  
  import com.atlassian.theplugin.idea.ui.tree.AtlassianTree;
  import com.atlassian.theplugin.idea.ui.tree.AtlassianTreeModel;
  import com.atlassian.theplugin.idea.ui.tree.AtlassianTreeNode;
  import com.intellij.openapi.util.IconLoader;
  import com.intellij.util.Icons;
  import org.jetbrains.annotations.NotNull;
  
  import javax.swing.*;
  import javax.swing.tree.TreePath;
  import java.awt.*;
  
  /**
   * @author Lukasz Guminski
   */
  public class AtlassianTreeWithToolbar extends ComponentWithToolbar {
  	private AtlassianTree tree;
  	private ModelProvider modelProvider = ModelProvider.EMPTY_MODEL_PROVIDER;
  	private STATE state = STATE.DIRED;
  
  
  	private VIEW_STATE viewState = VIEW_STATE.EXPANDED;
  
  	public AtlassianTreeWithToolbar(String toolbar, final ModelProvider modelProvider) {
  		this(toolbar);
  		setModelProvider(modelProvider);
  	}
  
  	public AtlassianTreeWithToolbar(final String toolbarName) {
  		super(toolbarName);
  	}
  
  	@Override
  	public Component getTreeComponent() {
  		if (tree == null) {
  			tree = new AtlassianTree();
  		}
  		return tree;
  	}
  
  
  	public void setRootVisible(final boolean isVisible) {
  		tree.setRootVisible(isVisible);
  	}
  
  	public void setModel(final AtlassianTreeModel model) {
  		tree.setModel(model);
  		setViewState(getViewState());
  	}
  
  	public void expandAll() {
  		tree.expandAll();
  	}
  
  	public void setModelProvider(final ModelProvider modelProvider) {
  		this.modelProvider = modelProvider;
  		triggerModelUpdated();
  	}
  
  	public ModelProvider getModelProvider() {
  		return modelProvider;
  	}
  
  	public STATE getState() {
  		return state;
  	}
  
  	public void setState(final STATE state) {
  		this.state = state;
  		triggerModelUpdated();
  	}
  
  	public void triggerModelUpdated() {
  		setModel(modelProvider.getModel(state));
  	}
  
  	public void changeState() {
  		setState(getState().getNextState());
  	}
  
  	public VIEW_STATE getViewState() {
  		return tree.isExpanded(0) ? VIEW_STATE.EXPANDED : VIEW_STATE.COLLAPSED;
 	}
 
 	public void setViewState(final VIEW_STATE viewState) {
 		this.viewState = viewState;
 		switch (viewState) {
 			case COLLAPSED:
 				collapseAll();
 				break;
 			case EXPANDED:
 				expandAll();
 				break;
 			default:
 				throw new IllegalStateException("Unknown state of tree: " + viewState.toString());
 		}
 	}
 
 	public void collapseAll() {
 		tree.collapseAll();
 	}
 
 	public AtlassianTreeNode getSelectedTreeNode() {
 		if (tree != null) {
 			TreePath path = tree.getSelectionPath();
 			if (path != null) {
 				return (AtlassianTreeNode) path.getLastPathComponent();
 			}
 		}
 		return null;
 	}
 
 	@NotNull
 	public AtlassianTreeModel getModel() {
 		return (AtlassianTreeModel) tree.getModel();
 	}
 
 	public void focusOnNode(final AtlassianTreeNode node) {
 		tree.focusOnNode(node);
 	}
 
 	public boolean isEmpty() {
 		return !tree.isRootVisible() && (tree.getModel().getChildCount(tree.getModel().getRoot()) == 0);
 	}
 
 	public enum STATE {
 		FLAT(Icons.DIRECTORY_CLOSED_ICON, "Show flat") {
 			@Override
 			public STATE getNextState() {
 				return DIRED;
 			}},
 		DIRED(Icons.DIRECTORY_OPEN_ICON, "Show dired") {
 			public STATE getNextState() {
 				return FLAT;
 			}};
 
 		private Icon icon;
 		private String string;
 
 		STATE(final Icon icon, final String string) {
 			this.icon = icon;
 			this.string = string;
 		}
 
 		public abstract STATE getNextState();
 
 		public Icon getIcon() {
 			return icon;
 		}
 
 		@Override
 		public String toString() {
 			return string;
 		}
 	}
 
 	public enum VIEW_STATE {
 		COLLAPSED(IconLoader.getIcon("/actions/collapseall.png"), "Collapse all") {
 
 			public VIEW_STATE getNextState() {
 				return EXPANDED;
 			}
 		},
 		EXPANDED(IconLoader.getIcon("/actions/expandall.png"), "Expand all") {
 
 			public VIEW_STATE getNextState() {
 				return COLLAPSED;
 			}
 		};
 
 		private Icon icon;
 		private String string;
 
 		VIEW_STATE(final Icon icon, final String string) {
 			this.icon = icon;
 			this.string = string;
 		}
 
 		public abstract VIEW_STATE getNextState();
 
 		public Icon getIcon() {
 			return icon;
 		}
 
 		@Override
 		public String toString() {
 			return string;
 		}
 	}
 }