/** Initializes the output. Sets the expected encoding and content type. */
public void initResponse() {
// set content type and encoding
final SerializerOptions opts = sopts();
final String enc = opts.get(SerializerOptions.ENCODING);
res.setCharacterEncoding(enc);
final String ct = mediaType(opts);
res.setContentType(new TokenBuilder(ct).add(CHARSET).add(enc).toString());
}
/**
* Runs a query and checks the serialization parameters.
*
* @throws IOException I/O exception
*/
@Test
public void queryOptions() throws IOException {
final Query query = session.query("declare option output:encoding 'US-ASCII';()");
query.execute();
final SerializerOptions sp = new SerializerOptions();
sp.parse(query.options());
assertEquals("US-ASCII", sp.get(SerializerOptions.ENCODING));
query.close();
}
/**
* Returns the media type defined in the specified serialization parameters.
*
* @param sopts serialization parameters
* @return media type
*/
public static String mediaType(final SerializerOptions sopts) {
// set content type
final String type = sopts.get(SerializerOptions.MEDIA_TYPE);
if (!type.isEmpty()) return type;
// determine content type dependent on output method
final SerialMethod sm = sopts.get(SerializerOptions.METHOD);
if (sm == SerialMethod.RAW) return APP_OCTET;
if (sm == SerialMethod.XML) return APP_XML;
if (sm == SerialMethod.XHTML || sm == SerialMethod.HTML) return TEXT_HTML;
if (sm == SerialMethod.JSON) {
final JsonSerialOptions jprop = sopts.get(SerializerOptions.JSON);
return jprop.get(JsonOptions.FORMAT) == JsonFormat.JSONML ? APP_JSONML : APP_JSON;
}
return TEXT_PLAIN;
}
/**
* Returns the serialization parameters.
*
* @param ctx context
* @return serialization parameters
*/
public String parameters(final Context ctx) {
try {
qp(args[0], ctx);
parse(null);
return qp.qc.serParams().toString();
} catch (final QueryException ex) {
error(Util.message(ex));
} finally {
qp = null;
}
return SerializerOptions.get(true).toString();
}
/**
* Constructor.
*
* @param os output stream
* @param sopts serializer options
* @throws QueryIOException query I/O exception
*/
protected OutputSerializer(final OutputStream os, final SerializerOptions sopts)
throws QueryIOException {
this.sopts = sopts;
indent = sopts.yes(INDENT);
// project-specific options
indents = sopts.get(INDENTS);
tab = sopts.yes(TABULATOR) ? '\t' : ' ';
encoding = Strings.normEncoding(sopts.get(ENCODING), true);
PrintOutput po;
if (encoding == Strings.UTF8) {
po = PrintOutput.get(os);
} else {
try {
po = new EncoderOutput(os, Charset.forName(encoding));
} catch (final Exception ex) {
throw SERENCODING_X.getIO(encoding);
}
}
final int limit = sopts.get(LIMIT);
if (limit != -1) po.setLimit(limit);
final byte[] nl = token(sopts.get(NEWLINE).newline());
if (nl.length != 1 || nl[0] != '\n') po = new NewlineOutput(po, nl);
out = po;
}
/** Tests the namespace stripping option (Option {@link MainOptions#STRIPNS}). */
@Test
public void parse() {
set(MainOptions.STRIPNS, true);
set(MainOptions.SERIALIZER, SerializerOptions.get(false));
final String doc = "<e xmlns='A'><b:f xmlns:b='B'/></e>";
for (final boolean b : new boolean[] {false, true}) {
set(MainOptions.INTPARSE, b);
execute(new CreateDB(NAME, doc));
assertEquals("<e><f/></e>", query("."));
assertEquals("<f/>", query("e/f"));
}
}
/** Tests the xml:space attribute. */
@Test
public void xmlSpace() {
set(MainOptions.SERIALIZER, SerializerOptions.get(false));
final String in =
"<x><a xml:space='default'> </a><a> </a>" + "<a xml:space='preserve'> </a></x>";
final String out = "<x><a xml:space=\"default\"/><a/>" + "<a xml:space=\"preserve\"> </a></x>";
for (final boolean b : new boolean[] {true, false}) {
set(MainOptions.INTPARSE, b);
execute(new CreateDB(NAME, in));
assertEquals("Internal parser: " + b, out, query("."));
}
}
/**
* Returns the current serialization options.
*
* @param mth consider specified serialization method (may be {@code null})
* @return options
* @throws BaseXException database exception
*/
private SerializerOptions options(final SerialMethod mth) throws BaseXException {
final SerializerOptions sopts = new SerializerOptions();
sopts.assign(params.getText());
sopts.set(SerializerOptions.METHOD, SerialMethod.valueOf(method.getSelectedItem()));
sopts.set(SerializerOptions.ENCODING, encoding.getSelectedItem());
if (mth == SerialMethod.JSON) {
final JsonSerialOptions jopts = new JsonSerialOptions();
jopts.assign(mparams.getText());
sopts.set(SerializerOptions.JSON, jopts);
} else if (mth == SerialMethod.CSV) {
final CsvOptions copts = new CsvOptions();
copts.assign(mparams.getText());
sopts.set(SerializerOptions.CSV, copts);
}
return sopts;
}
/**
* Default constructor.
*
* @param main reference to the main window
*/
public DialogExport(final GUI main) {
super(main, EXPORT);
// create checkboxes
final BaseXBack p = new BaseXBack(new TableLayout(4, 1, 0, 0));
p.add(new BaseXLabel(OUTPUT_DIR + COL, true, true).border(0, 0, 6, 0));
// output label
BaseXBack pp = new BaseXBack(new TableLayout(1, 2, 8, 0));
path = new BaseXTextField(main.gopts.get(GUIOptions.INPUTPATH), this);
pp.add(path.history(GUIOptions.INPUTS, this));
final BaseXButton browse = new BaseXButton(BROWSE_D, this);
browse.addActionListener(
new ActionListener() {
@Override
public void actionPerformed(final ActionEvent e) {
choose();
}
});
pp.add(browse);
p.add(pp);
// provide components for method and encoding
final MainOptions opts = gui.context.options;
final SerializerOptions sopts = opts.get(MainOptions.EXPORTER);
// method (ignore last entry)
final StringList sl = new StringList();
for (final SerialMethod sm : SerialMethod.values()) sl.add(sm.name());
sl.remove(sl.size() - 1);
method = new BaseXCombo(this, sl.finish());
final SerialMethod sm = sopts.get(SerializerOptions.METHOD);
method.setSelectedItem((sm == null ? SerialMethod.BASEX : sm).name());
mparams = new BaseXTextField(this);
mparams.setColumns(24);
final BaseXBack mth = new BaseXBack(new TableLayout(1, 2, 8, 0));
mth.add(method);
mth.add(mparams);
encoding = new BaseXCombo(this, ENCODINGS);
String enc = sopts.get(SerializerOptions.ENCODING);
boolean f = false;
for (final String s : ENCODINGS) f |= s.equals(enc);
if (!f) {
enc = enc.toUpperCase(Locale.ENGLISH);
for (final String s : ENCODINGS) f |= s.equals(enc);
}
encoding.setSelectedItem(f ? enc : sopts.get(SerializerOptions.ENCODING));
params = new BaseXTextField(sopts.toString(), this);
params.setToolTipText(tooltip(SerializerMode.DEFAULT.get()));
pp = new BaseXBack(new TableLayout(3, 2, 16, 6)).border(8, 0, 8, 0);
pp.add(new BaseXLabel(METHOD + COL, true, true));
pp.add(mth);
pp.add(new BaseXLabel(ENCODING + COL, true, true));
pp.add(encoding);
pp.add(new BaseXLabel(PARAMETERS + COL, true, true));
pp.add(params);
p.add(pp);
info = new BaseXLabel(" ").border(8, 0, 0, 0);
p.add(info);
// indentation
set(p, BorderLayout.CENTER);
// buttons
pp = new BaseXBack(new BorderLayout());
buttons = okCancel();
pp.add(buttons, BorderLayout.EAST);
set(pp, BorderLayout.SOUTH);
action(method);
finish(null);
}
/** Creates the database context. */
@BeforeClass
public static void start() {
// turn off pretty printing
set(MainOptions.SERIALIZER, SerializerOptions.get(false));
}
/**
* Sets the item separator.
*
* @param def default separator
*/
protected final void itemsep(final String def) {
final String is = sopts.get(ITEM_SEPARATOR);
itemsep = sopts.contains(ITEM_SEPARATOR) ? token(is) : def == null ? null : token(def);
}
/**
* 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();
}
}
