/*
* ModeShape (http://www.modeshape.org)
*
* 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 org.modeshape.sequencer.ddl.node;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.modeshape.common.annotation.NotThreadSafe;
import org.modeshape.common.util.CheckArg;
import org.modeshape.jcr.api.JcrConstants;
/**
* Utility object class designed to facilitate constructing an AST or Abstract Syntax Tree representing nodes and properties that
* are compatible with ModeShape graph component structure.
*/
@NotThreadSafe
public final class AstNode implements Iterable<AstNode>, Cloneable {
private AstNode parent;
private final String name;
private final Map<String, Object> properties = new HashMap<String, Object>();
private final LinkedList<AstNode> children = new LinkedList<AstNode>();
private final List<AstNode> childrenView = Collections.unmodifiableList(children);
/**
* Construct a node with the supplied name but without a parent.
*
* @param name the name of the node; may not be null
*/
AstNode( String name ) {
this(null, name);
}
/**
* Construct a node with the supplied name and parent.
*
* @param parent the parent node; may be null if the new node is to be a root
* @param name the name of the node; may not be null
*/
public AstNode( AstNode parent,
String name ) {
CheckArg.isNotNull(name, "name");
this.name = name;
if (parent != null) {
this.parent = parent;
this.parent.children.add(this);
}
}
/**
* @param mixin the mixin being added (cannot be <code>null</code> or empty)
* @return <code>true</code> if mixin was added
*/
public boolean addMixin( final String mixin ) {
CheckArg.isNotEmpty(mixin, "mixin");
final List<String> mixins = getMixins();
if (!mixins.contains(mixin)) {
if (mixins.add(mixin)) {
setProperty(JcrConstants.JCR_MIXIN_TYPES, mixins);
}
}
return false;
}
/**
* @param mixin the mixin to look for (cannot be <code>null</code> or empty)
* @return <code>true</code> if the node has the specified mixin
*/
public boolean hasMixin( final String mixin ) {
CheckArg.isNotEmpty(mixin, "mixin");
return getMixins().contains(mixin);
}
/**
* Get the name of the node.
*
* @return the node's name; never null
*/
public String getName() {
return name;
}
public String getPrimaryType() {
return (String)properties.get(JcrConstants.JCR_PRIMARY_TYPE);
}
/**
* Get the current same-name-sibling index.
*
* @return the SNS index, or 1 if this is the first sibling with the same name
*/
public int getSameNameSiblingIndex() {
int snsIndex = 1;
if (this.parent == null) {
return snsIndex;
}
// Go through all the children ...
for (AstNode sibling : this.parent.getChildren()) {
if (sibling == this) {
break;
}
if (sibling.getName().equals(this.name)) {
++snsIndex;
}
}
return snsIndex;
}
/**
* Get the current path of this node
*
* @return the path of this node; never null
*/
public String getAbsolutePath() {
StringBuilder pathBuilder = new StringBuilder("/").append(this.getName());
AstNode parent = this.getParent();
while (parent != null) {
pathBuilder.insert(0, "/" + parent.getName());
parent = parent.getParent();
}
return pathBuilder.toString();
}
/**
* Get the property with the supplied name.
*
* @param name the property name; never null
* @return the property, or null if no such property exists on the node
*/
public Object getProperty( String name ) {
return properties.get(name);
}
/**
* Set the property with the given name to the supplied value. Any existing property with the same name will be replaced.
*
* @param name the name of the property; may not be null
* @param value the value of the property; may not be null
* @return this node, for method chaining purposes
*/
public AstNode setProperty( String name,
Object value ) {
CheckArg.isNotNull(name, "name");
CheckArg.isNotNull(value, "value");
properties.put(name, value);
return this;
}
/**
* Set the property with the given name to the supplied values. If there is at least one value, the new property will replace
* any existing property with the same name. This method does nothing if zero values are supplied.
*
* @param name the name of the property; may not be null
* @param values the values of the property
* @return this node, for method chaining purposes
*/
public AstNode setProperty( String name,
Object... values ) {
CheckArg.isNotNull(name, "name");
CheckArg.isNotNull(values, "value");
if (values.length != 0) {
properties.put(name, Arrays.asList(values));
}
return this;
}
/**
* Remove and return the property with the supplied name.
*
* @param name the property name; may not be null
* @return the list of values of the property that was removed, or null if there was no such property
*/
public Object removeProperty( String name ) {
return properties.remove(name);
}
/**
* Return the list of property names for this node.
*
* @return the list of strings.
*/
public List<String> getPropertyNames() {
return new ArrayList<String>(properties.keySet());
}
@SuppressWarnings( "unchecked" )
public List<String> getMixins() {
Object mixinValues = getProperty(JcrConstants.JCR_MIXIN_TYPES);
List<String> result = new ArrayList<String>();
if (mixinValues instanceof Collection) {
result.addAll((Collection<? extends String>)mixinValues);
} else if (mixinValues != null) {
result.add(mixinValues.toString());
}
return result;
}
/**
* Get the parent of this node.
*
* @return the parent node, or null if this node has no parent
*/
public AstNode getParent() {
return parent;
}
/**
* Set the parent for this node. If this node already has a parent, this method will remove this node from the current parent.
* If the supplied parent is not null, then this node will be added to the supplied parent's children.
*
* @param parent the new parent, or null if this node is to have no parent
*/
public void setParent( AstNode parent ) {
removeFromParent();
if (parent != null) {
this.parent = parent;
this.parent.children.add(this);
}
}
/**
* Insert the supplied node into the plan node tree immediately above this node. If this node has a parent when this method is
* called, the new parent essentially takes the place of this node within the list of children of the old parent. This method
* does nothing if the supplied new parent is null.
* <p>
* For example, consider a plan node tree before this method is called:
*
* <pre>
* A
* / | \
* / | \
* B C D
* </pre>
*
* Then after this method is called with <code>c.insertAsParent(e)</code>, the resulting plan node tree will be:
*
* <pre>
* A
* / | \
* / | \
* B E D
* |
* |
* C
* </pre>
*
* </p>
* <p>
* Also note that the node on which this method is called ('C' in the example above) will always be added as the
* {@link #addLastChild(AstNode) last child} to the new parent. This allows the new parent to already have children before
* this method is called.
* </p>
*
* @param newParent the new parent; method does nothing if this is null
*/
public void insertAsParent( AstNode newParent ) {
if (newParent == null) {
return;
}
newParent.removeFromParent();
if (this.parent != null) {
this.parent.replaceChild(this, newParent);
}
newParent.addLastChild(this);
}
/**
* Remove this node from its parent, and return the node that used to be the parent of this node. Note that this method
* removes the entire subgraph under this node.
*
* @return the node that was the parent of this node, or null if this node had no parent
* @see #extractChild(AstNode)
* @see #extractFromParent()
*/
public AstNode removeFromParent() {
AstNode result = this.parent;
if (this.parent != null) {
// Remove this node from its current parent ...
this.parent.children.remove(this);
this.parent = null;
}
return result;
}
/**
* Replace the supplied child with another node. If the replacement is already a child of this node, this method effectively
* swaps the position of the child and replacement nodes.
*
* @param child the node that is already a child and that is to be replaced; may not be null and must be a child
* @param replacement the node that is to replace the 'child' node; may not be null
* @return true if the child was successfully replaced
*/
public boolean replaceChild( AstNode child,
AstNode replacement ) {
assert child != null;
assert replacement != null;
if (child.parent == this) {
int i = this.children.indexOf(child);
if (replacement.parent == this) {
// Swapping the positions ...
int j = this.children.indexOf(replacement);
this.children.set(i, replacement);
this.children.set(j, child);
return true;
}
// The replacement is not yet a child ...
this.children.set(i, replacement);
replacement.removeFromParent();
replacement.parent = this;
child.parent = null;
return true;
}
return false;
}
/**
* Get the number of child nodes.
*
* @return the number of children; never negative
*/
public int getChildCount() {
return this.children.size();
}
/**
* Get the first child.
*
* @return the first child, or null if there are no children
*/
public AstNode getFirstChild() {
return this.children.isEmpty() ? null : this.children.getFirst();
}
/**
* Get the last child.
*
* @return the last child, or null if there are no children
*/
public AstNode getLastChild() {
return this.children.isEmpty() ? null : this.children.getLast();
}
/**
* @param name the name of the child being requested (cannot be <code>null</code> or empty)
* @return a collection of children with the specified name (never <code>null</code> but can be empty)
*/
public List<AstNode> childrenWithName( final String name ) {
CheckArg.isNotEmpty(name, "name");
if (this.children.isEmpty()) {
return Collections.emptyList();
}
final List<AstNode> matches = new ArrayList<AstNode>();
for (final AstNode kid : this.children) {
if (name.equals(kid.getName())) {
matches.add(kid);
}
}
return matches;
}
/**
* Get the child at the supplied index.
*
* @param index the index
* @return the child, or null if there are no children
* @throws IndexOutOfBoundsException if the index is not valid given the number of children
*/
public AstNode getChild( int index ) {
return this.children.isEmpty() ? null : this.children.get(index);
}
/**
* Add the supplied node to the front of the list of children.
*
* @param child the node that should be added as the first child; may not be null
*/
public void addFirstChild( AstNode child ) {
assert child != null;
this.children.addFirst(child);
child.removeFromParent();
child.parent = this;
}
/**
* Add the supplied node to the end of the list of children.
*
* @param child the node that should be added as the last child; may not be null
*/
public void addLastChild( AstNode child ) {
assert child != null;
this.children.addLast(child);
child.removeFromParent();
child.parent = this;
}
/**
* Add the supplied nodes at the end of the list of children.
*
* @param otherChildren the children to add; may not be null
*/
public void addChildren( Iterable<AstNode> otherChildren ) {
assert otherChildren != null;
for (AstNode planNode : otherChildren) {
this.addLastChild(planNode);
}
}
/**
* Add the supplied nodes at the end of the list of children.
*
* @param first the first child to add
* @param second the second child to add
*/
public void addChildren( AstNode first,
AstNode second ) {
if (first != null) {
this.addLastChild(first);
}
if (second != null) {
this.addLastChild(second);
}
}
/**
* Add the supplied nodes at the end of the list of children.
*
* @param first the first child to add
* @param second the second child to add
* @param third the third child to add
*/
public void addChildren( AstNode first,
AstNode second,
AstNode third ) {
if (first != null) {
this.addLastChild(first);
}
if (second != null) {
this.addLastChild(second);
}
if (third != null) {
this.addLastChild(third);
}
}
/**
* Remove the node from this node.
*
* @param child the child node; may not be null
* @return true if the child was removed from this node, or false if the supplied node was not a child of this node
*/
public boolean removeChild( AstNode child ) {
boolean result = this.children.remove(child);
if (result) {
child.parent = null;
}
return result;
}
/**
* Remove the child node from this node, and replace that child with its first child (if there is one).
*
* @param child the child to be extracted; may not be null and must have at most 1 child
* @see #extractFromParent()
*/
public void extractChild( AstNode child ) {
if (child.getChildCount() == 0) {
removeChild(child);
} else {
AstNode grandChild = child.getFirstChild();
replaceChild(child, grandChild);
}
}
/**
* Extract this node from its parent, but replace this node with its child (if there is one).
*
* @see #extractChild(AstNode)
*/
public void extractFromParent() {
this.parent.extractChild(this);
}
/**
* Get the unmodifiable list of child nodes. This list will immediately reflect any changes made to the children (via other
* methods), but this list cannot be used to add or remove children.
*
* @return the list of children, which immediately reflects changes but which cannot be modified directly; never null
*/
public List<AstNode> getChildren() {
return childrenView;
}
/**
* @param mixin the mixin to match children with (cannot be <code>null</code> or empty)
* @return the children having the specified mixin (never <code>null</code>)
*/
public List<AstNode> getChildren( final String mixin ) {
final List<AstNode> result = new ArrayList<AstNode>();
for (final AstNode kid : getChildren()) {
if (kid.getMixins().contains(mixin)) {
result.add(kid);
}
}
return result;
}
/**
* {@inheritDoc}
* <p>
* This iterator is immutable.
* </p>
*
* @see java.lang.Iterable#iterator()
*/
@Override
public Iterator<AstNode> iterator() {
return childrenView.iterator();
}
/**
* Remove all children from this node. All nodes immediately become orphaned. The resulting list will be mutable.
*
* @return a copy of all the of the children that were removed (and which have no parent); never null
*/
public List<AstNode> removeAllChildren() {
if (this.children.isEmpty()) {
return new ArrayList<AstNode>(0);
}
List<AstNode> copyOfChildren = new ArrayList<AstNode>(this.children);
for (Iterator<AstNode> childIter = this.children.iterator(); childIter.hasNext();) {
AstNode child = childIter.next();
childIter.remove();
child.parent = null;
}
return copyOfChildren;
}
/**
* Determine whether the supplied plan is equivalent to this plan.
*
* @param other the other plan to compare with this instance
* @return true if the two plans are equivalent, or false otherwise
*/
public boolean isSameAs( AstNode other ) {
if (other == null) {
return false;
}
if (!this.name.equals(other.name)) {
return false;
}
if (!this.properties.equals(other.properties)) {
return false;
}
if (this.getChildCount() != other.getChildCount()) {
return false;
}
Iterator<AstNode> thisChildren = this.getChildren().iterator();
Iterator<AstNode> thatChildren = other.getChildren().iterator();
while (thisChildren.hasNext() && thatChildren.hasNext()) {
if (!thisChildren.next().isSameAs(thatChildren.next())) {
return false;
}
}
return true;
}
/**
* {@inheritDoc}
* <p>
* This class returns a new clone of the plan tree rooted at this node. However, the top node of the resulting plan tree (that
* is, the node returned from this method) has no parent.
* </p>
*
* @see java.lang.Object#clone()
*/
@Override
public AstNode clone() {
return cloneWithoutNewParent();
}
protected AstNode cloneWithoutNewParent() {
AstNode result = new AstNode(this.name);
result.properties.putAll(this.properties);
// Clone the children ...
for (AstNode child : children) {
AstNode childClone = child.cloneWithoutNewParent();
// The child has no parent, so add the child to the new result ...
result.addLastChild(childClone);
}
return result;
}
@Override
public String toString() {
StringBuilder stringBuilder = new StringBuilder();
stringBuilder.append(getAbsolutePath());
stringBuilder.append("[");
for (Iterator<String> propertyIt = properties.keySet().iterator(); propertyIt.hasNext();) {
String propertyName = propertyIt.next();
stringBuilder.append(propertyName).append(":").append(properties.get(propertyName));
if (propertyIt.hasNext()) {
{
stringBuilder.append(", ");
}
}
}
stringBuilder.append("]");
return stringBuilder.toString();
}
}