com.subgraph.vega.impl.scanner.modules.scripting.CrawlerCallbackWrapper.java Source code

Introduction

Here is the source code for com.subgraph.vega.impl.scanner.modules.scripting.CrawlerCallbackWrapper.java

Source

/*******************************************************************************
 * Copyright (c) 2011 Subgraph.
 * 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:
 *     Subgraph - initial API and implementation
 ******************************************************************************/
package com.subgraph.vega.impl.scanner.modules.scripting;

import java.util.logging.Level;
import java.util.logging.Logger;

import org.apache.http.client.methods.HttpUriRequest;
import org.mozilla.javascript.Context;
import org.mozilla.javascript.Function;
import org.mozilla.javascript.RhinoException;
import org.mozilla.javascript.Scriptable;
import org.mozilla.javascript.WrappedException;

import com.subgraph.vega.api.crawler.ICrawlerResponseProcessor;
import com.subgraph.vega.api.crawler.IWebCrawler;
import com.subgraph.vega.api.http.requests.IHttpResponse;
import com.subgraph.vega.api.scanner.IInjectionModuleContext;

public class CrawlerCallbackWrapper implements ICrawlerResponseProcessor {
    private static final Logger logger = Logger.getLogger("scanner");
    private static final long serialVersionUID = 1L;

    private Function callbackFunction;

    public CrawlerCallbackWrapper(Function callback) {
        callbackFunction = callback;
    }

    @Override
    public void processResponse(IWebCrawler crawler, HttpUriRequest request, IHttpResponse response,
            Object argument) {
        if (!(argument instanceof IInjectionModuleContext)) {

        }
        final IInjectionModuleContext ctx = (IInjectionModuleContext) argument;
        final Scriptable scope = callbackFunction.getParentScope();
        try {
            final Context cx = Context.enter();
            final Object[] arguments = { request, createResponse(response, cx, scope), new ModuleContextJS(ctx) };
            callbackFunction.call(cx, scope, scope, arguments);
        } catch (WrappedException e) {
            logger.log(Level.WARNING,
                    new RhinoExceptionFormatter("Wrapped exception running module script", e).toString());
            e.printStackTrace();
        } catch (RhinoException e) {
            logger.warning(new RhinoExceptionFormatter("Exception running module script.", e).toString());
        } finally {
            Context.exit();
        }
    }

    private Scriptable createResponse(IHttpResponse response, Context cx, Scriptable scope) {
        Object responseOb = Context.javaToJS(response, scope);
        Object[] args = { responseOb };
        return cx.newObject(scope, "Response", args);
    }
}