jsonpath.java

/**
 * $Id: $
 * $Date: $
 *
 */

package org.xmlsh.modules.json;

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintStream;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;

import org.xmlsh.core.CoreException;
import org.xmlsh.core.InputPort;
import org.xmlsh.core.Options;
import org.xmlsh.core.Options.OptionValue;
import org.xmlsh.core.XCommand;
import org.xmlsh.core.XValue;
import org.xmlsh.core.io.OutputPort;
import org.xmlsh.json.JSONUtils;
import org.xmlsh.sh.shell.SerializeOpts;
import org.xmlsh.util.Util;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.jayway.jsonpath.Configuration;
import com.jayway.jsonpath.JsonPath;
import com.jayway.jsonpath.Option;
import com.jayway.jsonpath.spi.json.JacksonJsonProvider;
import com.jayway.jsonpath.spi.json.JsonProvider;
import com.jayway.jsonpath.spi.mapper.JacksonMappingProvider;
import com.jayway.jsonpath.spi.mapper.MappingProvider;

public class jsonpath extends XCommand {

   

	@Override
	public int run(List<XValue> args) throws Exception {

		Options opts = new Options("c=context:,cf=context-file:,p=path,f=file:,i=input:,q=query:,n,e=exists,b=bool,s=string", SerializeOpts.getOptionDefs());
		opts.parse(args);


		// boolean bReadStdin = false ;

		JsonNode context = null;

		SerializeOpts serializeOpts = setSerializeOpts(opts);
		if( ! opts.hasOpt("n" ) ){ // Has XML data input
			// Order of prevelence 
			// -context
			// -context-file
			// -i

			if( opts.hasOpt("c") )
				context = opts.getOptValue("c").asJson();
			else
				if( opts.hasOpt("cf"))
					context = (getInput( XValue.newXValue(opts.getOptString("cf", "-")))).asJson(serializeOpts);
				else
					if( opts.hasOpt("i") )
						context = (getInput( opts.getOptValue("i"))).asJson(serializeOpts);
					else
						context = (getStdin()).asJson(serializeOpts);

		}

		List<XValue> xvargs = opts.getRemainingArgs();

		
		OptionValue ov = opts.getOpt("f");
		String xpath = null;
		if (ov != null)
			xpath = readString(ov.getValue(),serializeOpts);
		else {
			ov = opts.getOpt("q");
			if (ov != null)
				xpath = ov.getValue().toString();
		}

		if (xpath == null)
			xpath = xvargs.remove(0).toString();


		Configuration conf = Configuration.defaultConfiguration();
		if( opts.hasOpt("path") )
			conf = conf.addOptions( Option.AS_PATH_LIST );

		OutputPort stdout = getStdout();
		  JsonNode result = JsonPath.using(conf).parse(context).read(xpath);
		try ( JsonGenerator gen = stdout.asJsonGenerator(serializeOpts);){
			JSONUtils.writeJsonNode(gen, result);
		}

		return 0;
	}

    private String readString(XValue v, SerializeOpts opts) throws CoreException, IOException  {

		InputPort in = getInput( v );
		try (
				InputStream is = in.asInputStream(opts);
				){
			String s = Util.readString(is,opts.getInputTextEncoding());
			return s ;
		}
	}
}



/*
 * Copyright (C) 2008-2014   David A. Lee.
 * 
 * The contents of this file are subject to the "Simplified BSD License" (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.opensource.org/licenses/bsd-license.php 

 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied.
 * See the License for the specific language governing rights and limitations under the License.
 *
 * The Original Code is: all this file.
 *
 * The Initial Developer of the Original Code is David A. Lee
 *
 * Portions created by (your name) are Copyright (C) (your legal entity). All Rights Reserved.
 *
 * Contributor(s): David A. Lee
 * 
 */