ar.com.zauber.leviathan.scrapper.transformation.XSLTTransformer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of leviathan-scrapper Show documentation
The newest version!
/**
 * Copyright (c) 2009-2015 Zauber S.A. 
 *
 * 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 ar.com.zauber.leviathan.scrapper.transformation;

import java.io.Reader;
import java.io.StringReader;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang.Validate;

import ar.com.zauber.commons.dao.Transformer;
import ar.com.zauber.commons.web.transformation.processors.impl.TidyScrapper;
import ar.com.zauber.leviathan.api.URIFetcherResponse;
import ar.com.zauber.leviathan.api.URIFetcherResponse.URIAndCtx;
import ar.com.zauber.leviathan.scrapper.utils.ContextualResponse;

/**
 * Transformaci?n que aplica a un {@link URIFetcherResponse} una transformaci?n
 * XSL dada y devuelve una {@link ContextualResponse} con la respuesta y el
 * {@link URIAndCtx} de contexto
 * 
 * @author Marcelo Turrin
 * @since Sep 9, 2010
 */
public class XSLTTransformer 
    implements Transformer, 
                            ContextualResponse> {

    private final TidyScrapper tidyScrapper;
    private final Map extraModel = new HashMap();
    

    /** Creates the XSLTTransformer. */
    public XSLTTransformer(final TidyScrapper tidyScrapper) {
        this(tidyScrapper, null);
    }
    /** Creates the XSLTTransformer. */
    public XSLTTransformer(final TidyScrapper tidyScrapper, 
            final Map extraModel) {
        super();
        Validate.notNull(tidyScrapper);
        
        this.tidyScrapper = tidyScrapper;
        if(extraModel != null) {
            this.extraModel.putAll(extraModel);
        }
    }

    /** @see Transformer#transform(java.lang.Object) */
    public final ContextualResponse transform(
            final ContextualResponse input) {
        Validate.notNull(input);
        final StringWriter sw = new StringWriter();
        final Map model = new HashMap();
        model.putAll(extraModel);
        model.putAll(input.getContext().getCtx());
        tidyScrapper.scrap(input.getResponse(), sw, model);
        
        return new ContextualResponse(input.getContext(),
                                               new  StringReader(sw.toString()));
    }
}