/**
* Removes comments from the specified string and returns the first characters of a query.
*
* @param qu query string
* @param max maximum length of string to return
* @return result
*/
public static String removeComments(final String qu, final int max) {
final StringBuilder sb = new StringBuilder();
int m = 0;
boolean s = false;
final int cl = qu.length();
for (int c = 0; c < cl && sb.length() < max; ++c) {
final char ch = qu.charAt(c);
if (ch == 0x0d) continue;
if (ch == '(' && c + 1 < cl && qu.charAt(c + 1) == ':') {
if (m == 0 && !s) {
sb.append(' ');
s = true;
}
++m;
++c;
} else if (m != 0 && ch == ':' && c + 1 < cl && qu.charAt(c + 1) == ')') {
--m;
++c;
} else if (m == 0) {
if (ch > ' ') sb.append(ch);
else if (!s) sb.append(' ');
s = ch <= ' ';
}
}
if (sb.length() >= max) sb.append("...");
return sb.toString().trim();
}
@Override
public String getHeaderField(final String field) {
final List<String> values = headers.get(field);
final StringBuilder sb = new StringBuilder();
for (final String v : values) sb.append(v).append(';');
return sb.substring(0, sb.length() - 1);
}
/**
* Returns a map with variable bindings.
*
* @param opts main options
* @return bindings
*/
public static HashMap<String, String> bindings(final MainOptions opts) {
final HashMap<String, String> bindings = new HashMap<>();
final String bind = opts.get(MainOptions.BINDINGS).trim();
final StringBuilder key = new StringBuilder();
final StringBuilder val = new StringBuilder();
boolean first = true;
final int sl = bind.length();
for (int s = 0; s < sl; s++) {
final char ch = bind.charAt(s);
if (first) {
if (ch == '=') {
first = false;
} else {
key.append(ch);
}
} else {
if (ch == ',') {
if (s + 1 == sl || bind.charAt(s + 1) != ',') {
bindings.put(key.toString().trim(), val.toString());
key.setLength(0);
val.setLength(0);
first = true;
continue;
}
// literal commas are escaped by a second comma
s++;
}
val.append(ch);
}
}
if (key.length() != 0) bindings.put(key.toString().trim(), val.toString());
return bindings;
}
/**
* Returns a string representation of all found arguments.
*
* @param args array with arguments
* @return string representation
*/
static String foundArgs(final Value[] args) {
// compose found arguments
final StringBuilder sb = new StringBuilder();
for (final Value v : args) {
if (sb.length() != 0) sb.append(", ");
sb.append(v instanceof Jav ? Util.className(((Jav) v).toJava()) : v.seqType());
}
return sb.toString();
}
@Override
public final String toString() {
final StringBuilder sb = new StringBuilder();
if (root != null) sb.append(root);
for (final Expr s : steps) {
if (sb.length() != 0) sb.append(s instanceof Bang ? " ! " : "/");
sb.append(s);
}
return sb.toString();
}
/**
* Splits the string and returns an array.
*
* @param str string to be split
* @return array
*/
private static String[] split(final String str) {
final int l = str.length();
final String[] split = new String[l];
int s = 0;
char delim = 0;
final StringBuilder sb = new StringBuilder();
for (int i = 0; i < l; ++i) {
final char c = str.charAt(i);
if (delim == 0) {
if (c == '\'' || c == '"') {
delim = c;
} else if (!XMLToken.isChar(c)
&& c != '@'
&& c != '='
&& c != '<'
&& c != '>'
&& c != '~') {
if (sb.length() != 0) {
split[s++] = sb.toString();
sb.setLength(0);
}
} else {
sb.append(c);
}
} else {
if (c == delim) {
delim = 0;
if (sb.length() != 0) {
split[s++] = sb.toString();
sb.setLength(0);
}
} else {
if (c != '\'' && c != '"') sb.append(c);
}
}
}
if (sb.length() != 0) split[s++] = sb.toString();
return Array.copyOf(split, s);
}
@Override
public void action(final Object cmp) {
final StringBuilder bind = new StringBuilder();
final int cl = context.length;
for (int c = 0; c < cl; c += 2) {
final String key = context[c].getText().replaceAll("^\\$", "");
if (key.isEmpty()) continue;
if (bind.length() != 0) bind.append(',');
bind.append(key.replaceAll(",", ",,")).append('=');
bind.append(context[c + 1].getText().replaceAll(",", ",,"));
}
gui.context.options.set(MainOptions.BINDINGS, bind.toString());
}
/** Tests the internal parser (Option {@link MainOptions#INTPARSE}). */
@Test
public void intParse() {
set(MainOptions.CHOP, false);
final StringBuilder sb = new StringBuilder();
final String[] docs = {
"<x/>",
" <x/> ",
"<x></x>",
"<x>A</x>",
"<x><x>",
"<x/><x/>",
"<x></x><x/>",
"<x>",
"</x>",
"<x></x></x>",
"x<x>",
"<x>x",
"<x><![CDATA[ ]]></x>",
};
for (final String doc : docs) {
// parse document with default parser (expected to yield correct result)
set(MainOptions.INTPARSE, false);
boolean def = true;
try {
new CreateDB(NAME, doc).execute(context);
} catch (final BaseXException ex) {
def = false;
}
// parse document with internal parser
set(MainOptions.INTPARSE, true);
boolean cust = true;
try {
new CreateDB(NAME, doc).execute(context);
} catch (final BaseXException ex) {
cust = false;
}
// compare results
if (def != cust) {
sb.append('\n').append(def ? "- not accepted: " : "- not rejected: ").append(doc);
}
}
// list all errors
if (sb.length() != 0) fail(sb.toString());
set(MainOptions.MAINMEM, false);
}
/**
* Parses and returns a string result.
*
* @param input input string or {@code null} if invalid
* @param cmd referring command; if specified, the result must not be empty
* @return string result or {@code null}
* @throws QueryException query exception
*/
private String finish(final StringBuilder input, final Cmd cmd) throws QueryException {
if (input != null && input.length() != 0) return input.toString();
if (cmd != null) throw help(null, cmd);
return null;
}
/**
* Constructor.
*
* @param args command-line arguments
* @throws IOException I/O exception
*/
public BaseX(final String... args) throws IOException {
super(args);
// create session to show optional login request
session();
console = true;
try {
// loop through all commands
final StringBuilder bind = new StringBuilder();
SerializerOptions sopts = null;
boolean v = false, qi = false, qp = false;
final int os = ops.size();
for (int o = 0; o < os; o++) {
final int c = ops.get(o);
String val = vals.get(o);
if (c == 'b') {
// set/add variable binding
if (bind.length() != 0) bind.append(',');
// commas are escaped by a second comma
val = bind.append(val.replaceAll(",", ",,")).toString();
execute(new Set(MainOptions.BINDINGS, val), false);
} else if (c == 'c') {
// evaluate commands
final IO io = IO.get(val);
String base = ".";
if (io.exists() && !io.isDir()) {
val = io.string();
base = io.path();
}
execute(new Set(MainOptions.QUERYPATH, base), false);
execute(val);
execute(new Set(MainOptions.QUERYPATH, ""), false);
console = false;
} else if (c == 'D') {
// hidden option: show/hide dot query graph
execute(new Set(MainOptions.DOTPLAN, null), false);
} else if (c == 'i') {
// open database or create main memory representation
execute(new Set(MainOptions.MAINMEM, true), false);
execute(new Check(val), verbose);
execute(new Set(MainOptions.MAINMEM, false), false);
} else if (c == 'I') {
// set/add variable binding
if (bind.length() != 0) bind.append(',');
// commas are escaped by a second comma
val = bind.append("=").append(val.replaceAll(",", ",,")).toString();
execute(new Set(MainOptions.BINDINGS, val), false);
} else if (c == 'o') {
// change output stream
if (out != System.out) out.close();
out = new PrintOutput(val);
session().setOutputStream(out);
} else if (c == 'q') {
// evaluate query
execute(new XQuery(val), verbose);
console = false;
} else if (c == 'Q') {
// evaluate file contents or string as query
final IO io = IO.get(val);
String base = ".";
if (io.exists() && !io.isDir()) {
val = io.string();
base = io.path();
}
execute(new Set(MainOptions.QUERYPATH, base), false);
execute(new XQuery(val), verbose);
execute(new Set(MainOptions.QUERYPATH, ""), false);
console = false;
} else if (c == 'r') {
// parse number of runs
execute(new Set(MainOptions.RUNS, Strings.toInt(val)), false);
} else if (c == 'R') {
// toggle query evaluation
execute(new Set(MainOptions.RUNQUERY, null), false);
} else if (c == 's') {
// set/add serialization parameter
if (sopts == null) sopts = new SerializerOptions();
final String[] kv = val.split("=", 2);
sopts.assign(kv[0], kv.length > 1 ? kv[1] : "");
execute(new Set(MainOptions.SERIALIZER, sopts), false);
} else if (c == 't') {
// evaluate query
execute(new Test(val), verbose);
console = false;
} else if (c == 'u') {
// (de)activate write-back for updates
execute(new Set(MainOptions.WRITEBACK, null), false);
} else if (c == 'v') {
// show/hide verbose mode
v ^= true;
} else if (c == 'V') {
// show/hide query info
qi ^= true;
execute(new Set(MainOptions.QUERYINFO, null), false);
} else if (c == 'w') {
// toggle chopping of whitespaces
execute(new Set(MainOptions.CHOP, null), false);
} else if (c == 'x') {
// show/hide xml query plan
execute(new Set(MainOptions.XMLPLAN, null), false);
qp ^= true;
} else if (c == 'X') {
// show query plan before/after query compilation
execute(new Set(MainOptions.COMPPLAN, null), false);
} else if (c == 'z') {
// toggle result serialization
execute(new Set(MainOptions.SERIALIZE, null), false);
}
verbose = qi || qp || v;
}
if (console) console();
} finally {
quit();
}
}