com.github.julior.appintrospector.ScriptCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of app-introspector Show documentation
A simple Spring plugin to diagnostic and inspect Spring beans at runtime. It provides JSON based services as well as an UI based Console to execute queries and scripts against the application.
The newest version!
package com.github.julior.appintrospector;

import org.apache.log4j.Logger;
import org.w3c.dom.Document;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.InputSource;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.xpath.XPath;
import javax.xml.xpath.XPathConstants;
import javax.xml.xpath.XPathFactory;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.Reader;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * User: rinconj
 * Date: 9/22/11 11:04 AM
 */
public class ScriptCommand{
    private final static Logger LOGGER = Logger.getLogger(ScriptCommand.class);

    private Map arguments;
    private String script;

    ScriptCommand(Map arguments, String script) {
        this.arguments = arguments;
        this.script = script;
    }

    @Override
    public String toString() {
        return "ScriptCommand{" +
                "arguments=" + arguments +
                ", script='" + script + '\'' +
                '}';
    }

    public Map getArguments() {
        return arguments;
    }

    public String getScript() {
        return script;
    }

    public static ScriptCommand fromXml(Reader reader) throws Exception{
        XPath xpath = XPathFactory.newInstance().newXPath();
        try {
            Document xml = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(new InputSource(reader));

            Map args = new HashMap();
            //parse params
            NodeList nodes = (NodeList) xpath.evaluate("/command/vars/var", xml, XPathConstants.NODESET);
            for(int i=0; i=0){
            int second = buffer.indexOf(multiSep, first + multiSep.length());
            if(second>=0){
                String replacement = "\"" + buffer.substring(first + multiSep.length(), second).replace("\r\n", "\\n").replace("\n", "\\n").replace("\"", "\\\"") + "\"";
                buffer.replace(first, second+multiSep.length(), replacement);
                first += replacement.length();
            } else break;
        }

        Map args = new HashMap();
/*
        Pattern multiLineStringsPattern = Pattern.compile("(?m)('''((.|\\r|\\n)+(.|\\r|\\n)(?=(''')))''')");
        Matcher matcher = multiLineStringsPattern.matcher(script);
        //convert multilines strings into single lines
        while (matcher.find()){
            String strConst = matcher.group(2).replace("\n", "\\\\n").replace("\"", "\\\\\"");
            matcher.appendReplacement(buffer, "\"" + strConst + "\"");
        }
        matcher.appendTail(buffer);
*/
        //LOGGER.debug("after string linearised :" + buffer.toString());

        Pattern beanRefPattern = Pattern.compile("(\"[^\"]*\")|(\\$\\{?([^\\s;}]+)\\}?)");

        //replace bean references with variables (excluding references within strings)
        Matcher matcher = beanRefPattern.matcher(buffer.toString());
        buffer = new StringBuffer();
        int i=0;
        while (matcher.find()){
            if(matcher.group(2)!=null){
                i++;
                String beanId = matcher.group(3);
                String varName = genVarName(i);
                matcher.appendReplacement(buffer, varName);
                args.put(varName, beanId);
            }
        }
        matcher.appendTail(buffer);

        return new ScriptCommand(args, buffer.toString());
    }

    public static ScriptCommand fromScript(Reader reader) throws Exception{
        //extract bean references ${beanid} from the script and replace with synthetic variables
        StringBuilder sb = null;
        try {
            sb = readContent(reader);
            return fromScript(sb.toString());
        } catch (IOException e) {
            throw new Exception("Failed reading script content", e);
        }
    }

    private static StringBuilder readContent(Reader reader) throws IOException {
        StringBuilder sb = new StringBuilder();
        BufferedReader bufferedReader = new BufferedReader(reader);
        String line = null;
        while ((line = bufferedReader.readLine())!=null){
            sb.append(line);
            sb.append("\n");
        }
        return sb;
    }

    private static String genVarName(int i){
        return "_val_" + i;
    }
}