query.java

package org.xmlsh.marklogic;


import java.io.InputStream;
import java.util.List;

import org.xmlsh.core.InputPort;
import org.xmlsh.core.Options;
import org.xmlsh.core.Options.OptionValue;
import org.xmlsh.core.io.OutputPort;
import org.xmlsh.core.XValue;
import org.xmlsh.marklogic.util.MLCommand;
import org.xmlsh.marklogic.util.MLUtil;
import org.xmlsh.sh.shell.SerializeOpts;
import org.xmlsh.util.Util;

import com.marklogic.xcc.AdhocQuery;
import com.marklogic.xcc.RequestOptions;
import com.marklogic.xcc.ResultSequence;
import com.marklogic.xcc.types.XdmVariable;

public class query extends MLCommand {

	private AdhocQuery request;
	private RequestOptions options;

	@Override
	public int run(List<XValue> args) throws Exception {
		
		Options opts = new Options("c=connect:,q:,v,f:,t=text,b=bool,binary",SerializeOpts.getOptionDefs());
		opts.parse(args);
		args = opts.getRemainingArgs();
		
		mContentSource = getConnection(opts);
	
	    OutputPort out = getStdout();
	    String query = null ;
		if( opts.hasOpt("q"))
			query = opts.getOpt("q").getValue().toString();
		boolean asText = opts.hasOpt("t");
		boolean bBinary = opts.hasOpt("binary");
		boolean bBool = opts.hasOpt("b");
	
		mSession = mContentSource.newSession();
		request = mSession.newAdhocQuery (null);

		OptionValue ov = opts.getOpt("f");
		SerializeOpts serializeOpts = getSerializeOpts(opts);
		if(ov != null) {
			if(query != null)
				throwInvalidArg("Cannot specifify both -q and -f");

			InputPort qin = getInput(ov.getValue());
			try (InputStream is = qin.asInputStream(serializeOpts)) {
				query = Util.readString(is, serializeOpts.getInputTextEncoding());
			}

		}
		if( query == null && args.size() < 1 )
			throwInvalidArg("No query specified");
		else
		if( query == null )
			query = args.remove(0).toString();
			
			
		

      /*
       *  Add Variables - for now only handle string variables
       */

		if( opts.hasOpt("v")){
			// Read pairs from args to set
			for( int i = 0 ; i < args.size()/2 ; i++ ){
				
				
				String name = args.get(i*2).toString();
				XValue value = args.get(i*2+1);
				XdmVariable var = MLUtil.newVariable(name, value, serializeOpts);
				request.setVariable(var);
					
				
			}
				
			
		}
		
		request.setQuery (query);
		request.setOptions (options);
		
		
	    ResultSequence rs = mSession.submitRequest (request);
	    
	    int ret = 0;
	    boolean bOutput = false ;
	    if( bBool ){
	    	// Effective boolean value : 
	    	ret = MLUtil.effectiveBoolean(rs) ? 0 : 1;
	    	
	    	
	    }
	    else
	    	bOutput = MLUtil.writeResult(rs, out, serializeOpts,asText, bBinary );
        rs.close();
		
        mSession.close();

        if( !bBool  && ! asText && bOutput  )
        	out.writeSequenceTerminator(serializeOpts);
	
		
		return ret;
	}

	
	

	
	

}



//
//
//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): none.
//