xpath.java

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

package org.xmlsh.internal.commands;

import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.util.List;
import org.xmlsh.core.CoreException;
import org.xmlsh.core.InputPort;
import org.xmlsh.core.Namespaces;
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.IXdmItemOutputStream;
import org.xmlsh.core.io.OutputPort;
import org.xmlsh.sh.shell.SerializeOpts;
import org.xmlsh.sh.shell.Shell;
import org.xmlsh.util.Util;
import net.sf.saxon.s9api.Processor;
import net.sf.saxon.s9api.QName;
import net.sf.saxon.s9api.XPathCompiler;
import net.sf.saxon.s9api.XPathExecutable;
import net.sf.saxon.s9api.XPathSelector;
import net.sf.saxon.s9api.XdmAtomicValue;
import net.sf.saxon.s9api.XdmItem;
import net.sf.saxon.s9api.XdmNode;

public class xpath extends XCommand {

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

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

    boolean bString = opts.hasOpt("s");

    Processor processor = Shell.getProcessor();

    XPathCompiler compiler = processor.newXPathCompiler();
    XdmItem context = null;
    InputPort in = null;

    // boolean bReadStdin = false ;

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

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

    }

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

    boolean bQuiet = opts.hasOpt("e");
    boolean bBool = opts.hasOpt("b");
    if(bBool)
      bQuiet = true;

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

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

    if(opts.hasOpt("v")) {
      // Read pairs from args to set
      for(int i = 0; i < xvargs.size() / 2; i++) {
        String name = xvargs.get(i * 2).toString();

        compiler.declareVariable(new QName(name));

      }

    }

    /*
     * Add namespaces If -nons option then dont use global namespaces If -ns
     * options then add additional namespaces
     */

    Namespaces ns = null;

    if(!opts.hasOpt("nons"))
      ns = getEnv().getNamespaces();
    if(opts.hasOpt("ns")) {
      Namespaces ns2 = new Namespaces();
      if(ns != null)
        ns2.putAll(ns);

      // Add custom name spaces
      for(XValue v : opts.getOptValues("ns"))
        ns2.declare(v);

      ns = ns2;
    }

    if(ns != null) {
      for(String prefix : ns.keySet()) {
        String uri = ns.get(prefix);
        compiler.declareNamespace(prefix, uri);

      }

    }

    XPathExecutable expr = compiler.compile(xpath);

    XPathSelector eval = expr.load();
    if(context != null)
      eval.setContextItem(context);

    if(opts.hasOpt("v")) {
      // Read pairs from args to set
      for(int i = 0; i < xvargs.size() / 2; i++) {
        String name = xvargs.get(i * 2).toString();
        XValue value = xvargs.get(i * 2 + 1);

        eval.setVariable(new QName(name), value.toXdmValue());
      }
    }
    // Return the effective boolean value intead of any output
    if(bBool) {
      boolean bRet = eval.effectiveBooleanValue();
      return bRet ? 0 : 1;

    }
    else {

      OutputPort stdout = getStdout();
      IXdmItemOutputStream ser = stdout
          .asXdmItemOutputStream(getSerializeOpts());
      boolean bAnyOutput = false;
      boolean bFirst = true;

      for(XdmItem item : eval) {
        bAnyOutput = true;

        if(bQuiet)
          break;

        if(!bFirst)
          stdout.writeSequenceSeperator(getSerializeOpts()); // Thrashes
                                                             // variable
        // output !
        else {
          if(item instanceof XdmNode) {
            URI uri = ((XdmNode) item).getBaseURI();
            stdout.setSystemId(uri.toString());
          }
        }
        bFirst = false;
        if(item instanceof XdmNode) {
          XdmNode node = (XdmNode) item;
          if(bString)
            item = new XdmAtomicValue(node.getStringValue());

        }

        // Util.writeXdmValue(item, ser);
        ser.write(item);

      }
      if(!bQuiet && bAnyOutput)
        stdout.writeSequenceTerminator(getSerializeOpts());

      return bAnyOutput ? 0 : 1;
    }

  }

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