org.attoparser.select.AttributeSelectionMarkingMarkupHandler Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of attoparser Show documentation
Show all versions of attoparser Show documentation
Powerful, fast and easy to use HTML and XML parser for Java
/*
* =============================================================================
*
* Copyright (c) 2012-2014, The ATTOPARSER team (http://www.attoparser.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.attoparser.select;
import org.attoparser.AbstractChainedMarkupHandler;
import org.attoparser.IMarkupHandler;
import org.attoparser.ParseException;
/**
*
* Implementation of the {@link org.attoparser.IMarkupHandler} that
* adds an attribute (with a user-specified name) to all elements that match one or more selectors,
* as determined by a {@link org.attoparser.select.BlockSelectorMarkupHandler} or
* {@link org.attoparser.select.NodeSelectorMarkupHandler} handler.
*
*
* So for example, given an instance of this handler configured to use "selectors" as its attribute name
* and an <img href="logo.png"> tag that matches both selectors "//img" and
* "div/img", this handler would transform such tag in:
* <img href="logo.png" selectors="[//img,div/img]">
*
*
* @author Daniel Fernández
*
* @since 2.0.0
*
*/
public final class AttributeSelectionMarkingMarkupHandler
extends AbstractChainedMarkupHandler {
private static final char[] INNER_WHITESPACE_BUFFER = " ".toCharArray();
private final char[] selectorAttributeName;
private final int selectorAttributeNameLen;
private ParseSelection selection;
private boolean lastWasInnerWhiteSpace = false;
private char[] selectorAttributeBuffer;
/**
*
* Build a new instance of this class, specifying the name of the attribute to be added to the matching elements
* and also the handler all events should be delegated to.
*
*
* @param selectorAttributeName the name of the marking attribute.
* @param handler the handler to delegate events to.
*/
public AttributeSelectionMarkingMarkupHandler(final String selectorAttributeName, final IMarkupHandler handler) {
super(handler);
if (selectorAttributeName == null || selectorAttributeName.trim().length() == 0) {
throw new IllegalArgumentException("Selector attribute name cannot be null or empty");
}
this.selectorAttributeName = selectorAttributeName.toCharArray();
this.selectorAttributeNameLen = this.selectorAttributeName.length;
this.selectorAttributeBuffer = new char[this.selectorAttributeNameLen + 20];
System.arraycopy(this.selectorAttributeName, 0, this.selectorAttributeBuffer, 0, this.selectorAttributeNameLen);
this.selectorAttributeBuffer[this.selectorAttributeNameLen] = '=';
this.selectorAttributeBuffer[this.selectorAttributeNameLen + 1] = '\"';
}
@Override
public void setParseSelection(final ParseSelection selection) {
this.selection = selection;
super.setParseSelection(selection);
}
@Override
public void handleAttribute(
final char[] buffer,
final int nameOffset, final int nameLen,
final int nameLine, final int nameCol,
final int operatorOffset, final int operatorLen,
final int operatorLine, final int operatorCol,
final int valueContentOffset, final int valueContentLen,
final int valueOuterOffset, final int valueOuterLen,
final int valueLine, final int valueCol)
throws ParseException {
this.lastWasInnerWhiteSpace = false;
getNext().handleAttribute(
buffer,
nameOffset, nameLen, nameLine, nameCol,
operatorOffset, operatorLen, operatorLine, operatorCol,
valueContentOffset, valueContentLen, valueOuterOffset, valueOuterLen, valueLine, valueCol);
}
@Override
public void handleStandaloneElementEnd(
final char[] buffer,
final int nameOffset, final int nameLen,
final boolean minimized,
final int line, final int col)
throws ParseException {
if (this.selection.isMatchingAny()) {
if (!this.lastWasInnerWhiteSpace) {
getNext().handleInnerWhiteSpace(INNER_WHITESPACE_BUFFER, 0, INNER_WHITESPACE_BUFFER.length, line, col);
this.lastWasInnerWhiteSpace = true;
}
final String selectorValues = this.selection.toString();
final int selectorValuesLen = selectorValues.length();
checkSelectorAttributeLen(selectorValuesLen);
selectorValues.getChars(0, selectorValuesLen, this.selectorAttributeBuffer, this.selectorAttributeNameLen + 2);
this.selectorAttributeBuffer[this.selectorAttributeNameLen + 2 + selectorValuesLen] = '\"';
getNext().handleAttribute(
this.selectorAttributeBuffer,
0, this.selectorAttributeNameLen, line, col,
this.selectorAttributeNameLen, 1, line, col,
this.selectorAttributeNameLen + 2, selectorValuesLen,
this.selectorAttributeNameLen + 1, selectorValuesLen + 2, line, col);
}
this.lastWasInnerWhiteSpace = false;
getNext().handleStandaloneElementEnd(buffer, nameOffset, nameLen, minimized, line, col);
}
@Override
public void handleOpenElementEnd(
final char[] buffer,
final int nameOffset, final int nameLen,
final int line, final int col)
throws ParseException {
if (this.selection.isMatchingAny()) {
if (!this.lastWasInnerWhiteSpace) {
getNext().handleInnerWhiteSpace(INNER_WHITESPACE_BUFFER, 0, INNER_WHITESPACE_BUFFER.length, line, col);
this.lastWasInnerWhiteSpace = true;
}
final String selectorValues = this.selection.toString();
final int selectorValuesLen = selectorValues.length();
checkSelectorAttributeLen(selectorValuesLen);
selectorValues.getChars(0, selectorValuesLen, this.selectorAttributeBuffer, this.selectorAttributeNameLen + 2);
this.selectorAttributeBuffer[this.selectorAttributeNameLen + 2 + selectorValuesLen] = '\"';
getNext().handleAttribute(
this.selectorAttributeBuffer,
0, this.selectorAttributeNameLen, line, col,
this.selectorAttributeNameLen, 1, line, col,
this.selectorAttributeNameLen + 2, selectorValuesLen,
this.selectorAttributeNameLen + 1, selectorValuesLen + 2, line, col);
}
this.lastWasInnerWhiteSpace = false;
getNext().handleOpenElementEnd(buffer, nameOffset, nameLen, line, col);
}
@Override
public void handleInnerWhiteSpace(
final char[] buffer,
final int offset, final int len,
final int line, final int col)
throws ParseException {
this.lastWasInnerWhiteSpace = true;
getNext().handleInnerWhiteSpace(buffer, offset, len, line, col);
}
private void checkSelectorAttributeLen(final int valueLen) {
final int totalLenRequired = this.selectorAttributeNameLen + 3 + valueLen;
if (this.selectorAttributeBuffer.length < totalLenRequired) {
final char[] newSelectorAttributeBuffer = new char[totalLenRequired];
System.arraycopy(this.selectorAttributeBuffer, 0, newSelectorAttributeBuffer, 0, this.selectorAttributeBuffer.length);
this.selectorAttributeBuffer = newSelectorAttributeBuffer;
}
}
}