| /* Copyright 2004 The Apache Software Foundation |
| * |
| * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.xmlbeans.impl.tool; |
| |
| import java.io.*; |
| import java.util.*; |
| import org.apache.xmlbeans.*; |
| |
| public class RunXQuery |
| { |
| public static void main ( String[] args ) throws Exception |
| { |
| CommandLine cl = |
| new CommandLine( |
| args, |
| Arrays.asList( new String[] { "q", "qf" } ) ); |
| |
| if (cl.getOpt("license") != null) |
| { |
| CommandLine.printLicense(); |
| System.exit(0); |
| return; |
| } |
| |
| args = cl.args(); |
| |
| if (args.length == 0) |
| { |
| System.out.println("Run an XQuery against an XML instance"); |
| System.out.println("Usage:"); |
| System.out.println("xquery [-verbose] [-pretty] [-q <query> | -qf query.xq] [file.xml]*"); |
| System.out.println(" -q <query> to specify a query on the command-line"); |
| System.out.println(" -qf <query> to specify a file containing a query"); |
| System.out.println(" -pretty pretty-prints the results"); |
| System.out.println(" -license prints license information"); |
| System.out.println(" the query is run on each XML file specified"); |
| System.out.println(""); |
| System.exit(0); |
| return; |
| } |
| |
| boolean verbose = cl.getOpt( "verbose" ) != null; |
| boolean pretty = cl.getOpt( "pretty" ) != null; |
| |
| // |
| // Get and compile the query |
| // |
| |
| String query = cl.getOpt( "q" ); |
| String queryfile = cl.getOpt( "qf" ); |
| |
| if (query == null && queryfile == null) |
| { |
| System.err.println( "No query specified" ); |
| System.exit(0); |
| return; |
| } |
| |
| if (query != null && queryfile != null) |
| { |
| System.err.println( "Specify -qf or -q, not both." ); |
| System.exit(0); |
| return; |
| } |
| |
| try |
| { |
| if (queryfile != null) |
| { |
| File queryFile = new File( queryfile ); |
| FileInputStream is = new FileInputStream( queryFile ); |
| InputStreamReader r = new InputStreamReader( is ); |
| |
| StringBuffer sb = new StringBuffer(); |
| |
| for ( ; ; ) |
| { |
| int ch = r.read(); |
| |
| if (ch < 0) |
| break; |
| |
| sb.append( (char) ch ); |
| } |
| |
| r.close(); |
| is.close(); |
| |
| query = sb.toString(); |
| } |
| } |
| catch ( Throwable e ) |
| { |
| System.err.println( "Cannot read query file: " + e.getMessage() ); |
| System.exit(1); |
| return; |
| } |
| |
| if (verbose) |
| { |
| System.out.println( "Compile Query:" ); |
| System.out.println( query ); |
| System.out.println(); |
| } |
| |
| try |
| { |
| query= XmlBeans.compileQuery( query ); |
| } |
| catch ( Exception e ) |
| { |
| System.err.println( "Error compiling query: " + e.getMessage() ); |
| System.exit(1); |
| return; |
| } |
| |
| // |
| // Get the instance |
| // |
| |
| File[] files = cl.getFiles(); |
| |
| for (int i = 0; i < files.length; i++) |
| { |
| XmlObject x; |
| |
| try |
| { |
| if (verbose) |
| { |
| InputStream is = new FileInputStream( files[i] ); |
| |
| for ( ; ; ) |
| { |
| int ch = is.read(); |
| |
| if (ch < 0) |
| break; |
| |
| System.out.write( ch ); |
| } |
| |
| is.close(); |
| |
| System.out.println(); |
| } |
| |
| x = XmlObject.Factory.parse( files[i] ); |
| } |
| catch ( Throwable e ) |
| { |
| System.err.println( "Error parsing instance: " + e.getMessage() ); |
| System.exit(1); |
| return; |
| } |
| |
| if (verbose) |
| { |
| System.out.println( "Executing Query..." ); |
| System.err.println(); |
| } |
| |
| XmlObject[] result = null; |
| |
| try |
| { |
| result = x.execQuery( query ); |
| } |
| catch ( Throwable e ) |
| { |
| System.err.println( "Error executing query: " + e.getMessage() ); |
| System.exit(1); |
| return; |
| } |
| |
| if (verbose) |
| { |
| System.out.println( "Query Result:" ); |
| } |
| |
| XmlOptions opts = new XmlOptions(); |
| opts.setSaveOuter(); |
| if (pretty) |
| opts.setSavePrettyPrint(); |
| |
| for (int j = 0; j < result.length; j++) |
| { |
| result[j].save( System.out, opts ); |
| System.out.println(); |
| } |
| } |
| } |
| } |