/*****************************************************************
 *   Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you 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 org.apache.cayenne.modeler.dialog.datadomain;

import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

import javax.swing.JTree;
import javax.swing.tree.TreeNode;
import javax.swing.tree.TreePath;

import org.apache.cayenne.modeler.ProjectController;
import org.apache.cayenne.modeler.ProjectTreeModel;
import org.apache.cayenne.modeler.ProjectTreeView;

public class FilterController {
	
	private Map<String,Boolean> filterMap = new HashMap<>();
	private ProjectTreeView tree;
	private ProjectController eventController;
	private ProjectTreeModel treeModel;
	
	public ProjectTreeView getTree() {
		return tree;
	}

	public ProjectTreeModel getTreeModel() {
		return treeModel;
	}

	public ProjectController getEventController() {
		return eventController;
	}

	public Map<String, Boolean> getFilterMap() {
		return filterMap;
	}

	public FilterController(ProjectController eventController, ProjectTreeView treePanel) {
	
		this.eventController = eventController;
		this.tree = treePanel;
		this.treeModel = tree.getProjectModel();
		
		filterMap.put("dbEntity",	true);
		filterMap.put("objEntity",	true);
		filterMap.put("embeddable",	true);
		filterMap.put("procedure",	true);
		filterMap.put("query",		true);
	}
	
	
	public void treeExpOrCollPath(String action) {
		TreeNode root = (TreeNode) treeModel.getRoot();
		expandAll(tree, new TreePath(root),action);
	}

	private void expandAll(JTree tree, TreePath parent, String action) {
		TreeNode node = (TreeNode) parent.getLastPathComponent();
		
		if (node.getChildCount() >= 0) {
			for (Enumeration e = node.children(); e.hasMoreElements();) {
				TreeNode n = (TreeNode) e.nextElement();
				TreePath path = parent.pathByAddingChild(n);
				expandAll(tree, path, action);
			}
		}

		if("expand".equals(action)) {
			tree.expandPath(parent);
		} else if("collapse".equals(action)) {
			treeModel.reload(treeModel.getRootNode());
		}
	}
	
}