BatchReferenceableElementsLocator.java

/*******************************************************************************
 * Copyright (c) 2009, 2011 Spring IDE Developers
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Spring IDE Developers - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.batch.ui.namespaces;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.eclipse.core.resources.IFile;
import org.springframework.ide.eclipse.beans.ui.editor.namespaces.IReferenceableElementsLocator;
import org.springframework.ide.eclipse.beans.ui.editor.util.BeansEditorUtils;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

/**
 * {@link IReferenceableElementsLocator} for steps and other Spring Batch related artifacts.
 * @author Christian Dupuis
 * @author Leo Dos Santos
 * @author Terry Denney
 * @since 2.2.3
 */
public class BatchReferenceableElementsLocator implements IReferenceableElementsLocator {

	public Map<String, Set<Node>> getReferenceableElements(Document document,
			IFile file) {
		return getReferenceableElementsHelper(document.getDocumentElement());
	}
	
	private Map<String, Set<Node>> getReferenceableElementsHelper(Node parent) {
		Map<String, Set<Node>> nodes = new HashMap<String, Set<Node>>();
		NodeList childNodes = parent.getChildNodes();
		for (int i = 0; i < childNodes.getLength(); i++) {
			Node node = childNodes.item(i);
			if (BeansEditorUtils.hasAttribute(node, "id")) {
				String id = BeansEditorUtils.getAttribute(node, "id");
				Set<Node> matchedNodes = nodes.get(id);
				if (matchedNodes == null) {
					matchedNodes = new HashSet<Node>();
					nodes.put(id, matchedNodes);
				}
				matchedNodes.add(node);
			}
			if (node.hasChildNodes()) {
				Map<String, Set<Node>> tempNodes = getReferenceableElementsHelper(node);
				for(String id: tempNodes.keySet()) {
					Set<Node> matchedNodes = nodes.get(id);
					if (matchedNodes == null) {
						nodes.put(id, tempNodes.get(id));
					} else {
						matchedNodes.addAll(tempNodes.get(id));
					}
				}
			}
		}
		return nodes;
	}

}