@Override
public void execute(final GUI gui) {
final DialogExport dialog = new DialogExport(gui);
if (!dialog.ok()) return;
final IOFile root = new IOFile(dialog.path());
// check if existing files will be overwritten
if (root.exists()) {
IO file = null;
boolean overwrite = false;
final Data d = gui.context.data();
final IntList il = d.resources.docs();
final int is = il.size();
for (int i = 0; i < is; i++) {
file = root.merge(Token.string(d.text(il.get(i), true)));
if (file.exists()) {
if (overwrite) {
// more than one file will be overwritten; check remaining tests
file = null;
break;
}
overwrite = true;
}
}
if (overwrite) {
// show message for overwriting files or directories
final String msg = file == null ? FILES_REPLACE_X : FILE_EXISTS_X;
if (file == null) file = root;
if (!BaseXDialog.confirm(gui, Util.info(msg, file))) return;
}
}
DialogProgress.execute(gui, new Export(root.path()));
}
@Override
public final int diff(final Item it, final Collation coll, final InputInfo ii)
throws QueryException {
return it.type.isUntyped()
? coll == null ? Token.diff(string(), it.string(ii)) : coll.compare(string(), it.string(ii))
: -it.diff(this, coll, ii);
}
/**
* Constructor.
*
* @param value value
* @param type item type
* @param ii input info
* @throws QueryException query exception
*/
private Dur(final byte[] value, final Type type, final InputInfo ii) throws QueryException {
this(type);
final String val = Token.string(value).trim();
final Matcher mt = DUR.matcher(val);
if (!mt.matches() || val.endsWith("P") || val.endsWith("T")) throw dateError(value, XDURR, ii);
yearMonth(value, mt, ii);
dayTime(value, mt, 6, ii);
}
/**
* Adds the time to the specified token builder.
*
* @param tb token builder
*/
final void time(final TokenBuilder tb) {
if (sec.remainder(DAYSECONDS).signum() == 0) return;
tb.add('T');
final long h = hou();
if (h != 0) {
tb.addLong(Math.abs(h));
tb.add('H');
}
final long m = min();
if (m != 0) {
tb.addLong(Math.abs(m));
tb.add('M');
}
final BigDecimal sc = sec();
if (sc.signum() == 0) return;
tb.add(Token.chopNumber(Token.token(sc.abs().toPlainString()))).add('S');
}
@Override
public void execute(final GUI gui) {
final int pre = gui.context.marked.pres[0];
final byte[] txt = ViewData.path(gui.context.data(), pre);
// copy path to clipboard
final Clipboard clip = Toolkit.getDefaultToolkit().getSystemClipboard();
clip.setContents(new StringSelection(Token.string(txt)), null);
}
@Override
public final boolean eq(
final Item it, final Collation coll, final StaticContext sc, final InputInfo ii)
throws QueryException {
return it.type.isUntyped()
? coll == null
? Token.eq(string(), it.string(ii))
: coll.compare(string(), it.string(ii)) == 0
: it.eq(this, coll, sc, ii);
}
/**
* Initializes the date format.
*
* @param d input
* @param e example format
* @param ii input info
* @throws QueryException query exception
*/
final void date(final byte[] d, final String e, final InputInfo ii) throws QueryException {
final Matcher mt = DATE.matcher(Token.string(d).trim());
if (!mt.matches()) throw dateError(d, e, ii);
yea = toLong(mt.group(1), false, ii);
// +1 is added to BC values to simplify computations
if (yea < 0) yea++;
mon = (byte) (Strings.toInt(mt.group(3)) - 1);
day = (byte) (Strings.toInt(mt.group(4)) - 1);
if (mon < 0 || mon >= 12 || day < 0 || day >= dpm(yea, mon)) throw dateError(d, e, ii);
if (yea <= MIN_YEAR || yea > MAX_YEAR) throw DATERANGE_X_X.get(ii, type, chop(d, ii));
zone(mt, 5, d, ii);
}
@Override
public byte[] string(final InputInfo ii) {
final TokenBuilder tb = new TokenBuilder();
final boolean ymd = yea != Long.MAX_VALUE;
if (ymd) {
if (yea <= 0) tb.add('-');
prefix(tb, Math.abs(yea()), 4);
tb.add('-');
prefix(tb, mon(), 2);
tb.add('-');
prefix(tb, day(), 2);
}
if (hou >= 0) {
if (ymd) tb.add('T');
prefix(tb, hou(), 2);
tb.add(':');
prefix(tb, min(), 2);
tb.add(':');
if (sec.intValue() < 10) tb.add('0');
tb.addExt(Token.chopNumber(Token.token(sec().abs().toPlainString())));
}
zone(tb);
return tb.finish();
}
/**
* Initializes the time format.
*
* @param d input format
* @param e expected format
* @param ii input info
* @throws QueryException query exception
*/
final void time(final byte[] d, final String e, final InputInfo ii) throws QueryException {
final Matcher mt = TIME.matcher(Token.string(d).trim());
if (!mt.matches()) throw dateError(d, e, ii);
hou = (byte) Strings.toInt(mt.group(1));
min = (byte) Strings.toInt(mt.group(2));
sec = toDecimal(mt.group(3), false, ii);
if (min >= 60
|| sec.compareTo(BD60) >= 0
|| hou > 24
|| hou == 24 && (min > 0 || sec.compareTo(BigDecimal.ZERO) > 0)) throw dateError(d, e, ii);
zone(mt, 5, d, ii);
if (hou == 24) {
hou = 0;
add(DAYSECONDS);
}
}
@Override
public void execute(final GUI gui) {
final Nodes n = gui.context.marked;
final DialogInsert insert = new DialogInsert(gui);
if (!insert.ok()) return;
final StringList sl = insert.result;
final NodeType type = ANode.type(insert.kind);
String item = Token.string(type.string()) + " { " + quote(sl.get(0)) + " }";
if (type == NodeType.ATT || type == NodeType.PI) {
item += " { " + quote(sl.get(1)) + " }";
} else if (type == NodeType.ELM) {
item += " { () }";
}
gui.context.copied = null;
gui.execute(new XQuery("insert node " + item + " into " + openPre(n, 0)));
}
@Override
public final Duration toJava() {
return ADate.DF.newDuration(Token.string(string(null)));
}
@Override
public final int diff(final Item it, final Collation coll, final InputInfo ii)
throws QueryException {
return Token.diff(
binary(ii), it instanceof Bin ? ((Bin) it).binary(ii) : decode(it.string(ii), ii));
}
/**
* Prefixes the specified number of zero digits before a number.
*
* @param tb token builder
* @param number number to be printed
* @param zero maximum number of zero digits
*/
static void prefix(final TokenBuilder tb, final long number, final int zero) {
final byte[] t = Token.token(number);
for (int i = t.length; i < zero; i++) tb.add('0');
tb.add(t);
}
/**
* Functions on archives.
*
* @author BaseX Team 2005-12, BSD License
* @author Christian Gruen
*/
public class FNArchive extends StandardFunc {
/** Archive namespace. */
private static final Atts NS = new Atts(ARCHIVE, ARCHIVEURI);
/** Element: Entry. */
private static final QNm Q_ENTRY = new QNm("archive:entry", ARCHIVEURI);
/** Element: options. */
private static final QNm Q_OPTIONS = new QNm("archive:options", ARCHIVEURI);
/** Option: algorithm. */
private static final QNm Q_FORMAT = new QNm("archive:format", ARCHIVEURI);
/** Option: algorithm. */
private static final QNm Q_ALGORITHM = new QNm("archive:algorithm", ARCHIVEURI);
/** Root node test. */
private static final ExtTest TEST = new ExtTest(NodeType.ELM, Q_ENTRY);
/** Level. */
private static final QNm Q_LEVEL = new QNm("compression-level");
/** Encoding. */
private static final QNm Q_ENCODING = new QNm("encoding");
/** Last modified. */
private static final QNm Q_LAST_MOD = new QNm("last-modified");
/** Compressed size. */
private static final QNm Q_COMP_SIZE = new QNm("compressed-size");
/** Uncompressed size. */
private static final QNm Q_SIZE = new QNm("size");
/** Value. */
private static final QNm Q_VALUE = new QNm("value");
/** Option: format. */
private static final byte[] FORMAT = Token.token("format");
/** Option: algorithm. */
private static final byte[] ALGORITHM = Token.token("algorithm");
/** Option: algorithm: deflate. */
private static final byte[] DEFLATE = Token.token("deflate");
/** Option: algorithm: stored. */
private static final byte[] STORED = Token.token("stored");
/** Option: algorithm: unknown. */
private static final byte[] UNKNOWN = Token.token("unknown");
/**
* Constructor.
*
* @param ii input info
* @param f function definition
* @param e arguments
*/
public FNArchive(final InputInfo ii, final Function f, final Expr... e) {
super(ii, f, e);
}
@Override
public Iter iter(final QueryContext ctx) throws QueryException {
switch (sig) {
case _ARCHIVE_ENTRIES:
return entries(ctx);
case _ARCHIVE_EXTRACT_TEXT:
return extractText(ctx);
case _ARCHIVE_EXTRACT_BINARY:
return extractBinary(ctx);
default:
return super.iter(ctx);
}
}
@Override
public Item item(final QueryContext ctx, final InputInfo ii) throws QueryException {
checkCreate(ctx);
switch (sig) {
case _ARCHIVE_CREATE:
return create(ctx);
case _ARCHIVE_UPDATE:
return update(ctx);
case _ARCHIVE_DELETE:
return delete(ctx);
case _ARCHIVE_OPTIONS:
return options(ctx);
default:
return super.item(ctx, ii);
}
}
/**
* Creates a new archive.
*
* @param ctx query context
* @return archive
* @throws QueryException query exception
*/
private B64 create(final QueryContext ctx) throws QueryException {
final Iter entr = ctx.iter(expr[0]);
final Iter cont = ctx.iter(expr[1]);
final Item opt = expr.length > 2 ? expr[2].item(ctx, info) : null;
final TokenMap map = new FuncParams(Q_OPTIONS, info).parse(opt);
final byte[] f = map.get(FORMAT);
final String format = f != null ? string(lc(f)) : "zip";
final ArchiveOut out = ArchiveOut.get(format, info);
// check algorithm
final byte[] alg = map.get(ALGORITHM);
int level = ZipEntry.DEFLATED;
if (alg != null) {
if (format.equals("zip") && !eq(alg, STORED, DEFLATE)
|| format.equals("gzip") && !eq(alg, DEFLATE)) {
ARCH_SUPP.thrw(info, ALGORITHM, alg);
}
if (eq(alg, STORED)) level = ZipEntry.STORED;
else if (eq(alg, DEFLATE)) level = ZipEntry.DEFLATED;
}
out.level(level);
try {
int e = 0;
int c = 0;
Item en, cn;
while (true) {
en = entr.next();
cn = cont.next();
if (en == null || cn == null) break;
if (out instanceof GZIPOut && c > 0)
ARCH_ONE.thrw(info, format.toUpperCase(Locale.ENGLISH));
add(checkElmStr(en), cn, out, level);
e++;
c++;
}
// count remaining entries
if (cn != null) do c++; while (cont.next() != null);
if (en != null) do e++; while (entr.next() != null);
if (e != c) throw ARCH_DIFF.thrw(info, e, c);
} catch (final IOException ex) {
Util.debug(ex);
throw ARCH_FAIL.thrw(info, ex);
} finally {
out.close();
}
return new B64(out.toArray());
}
/**
* Returns the options of an archive.
*
* @param ctx query context
* @return entries
* @throws QueryException query exception
*/
private FElem options(final QueryContext ctx) throws QueryException {
final B64 archive = (B64) checkType(checkItem(expr[0], ctx), AtomType.B64);
String format = null;
int level = -1;
final ArchiveIn arch = ArchiveIn.get(archive.input(info), info);
try {
format = arch.format();
while (arch.more()) {
final ZipEntry ze = arch.entry();
if (ze.isDirectory()) continue;
level = ze.getMethod();
break;
}
} catch (final IOException ex) {
Util.debug(ex);
ARCH_FAIL.thrw(info, ex);
} finally {
arch.close();
}
// create result element
final FElem e = new FElem(Q_OPTIONS, NS);
if (format != null) e.add(new FElem(Q_FORMAT).add(Q_VALUE, format));
if (level >= 0) {
final byte[] lvl = level == 8 ? DEFLATE : level == 0 ? STORED : UNKNOWN;
e.add(new FElem(Q_ALGORITHM).add(Q_VALUE, lvl));
}
return e;
}
/**
* Returns the entries of an archive.
*
* @param ctx query context
* @return entries
* @throws QueryException query exception
*/
private Iter entries(final QueryContext ctx) throws QueryException {
final B64 archive = (B64) checkType(checkItem(expr[0], ctx), AtomType.B64);
final ValueBuilder vb = new ValueBuilder();
final ArchiveIn in = ArchiveIn.get(archive.input(info), info);
try {
while (in.more()) {
final ZipEntry ze = in.entry();
if (ze.isDirectory()) continue;
final FElem e = new FElem(Q_ENTRY, NS);
long s = ze.getSize();
if (s != -1) e.add(Q_SIZE, token(s));
s = ze.getTime();
if (s != -1) e.add(Q_LAST_MOD, new Dtm(s, info).string(info));
s = ze.getCompressedSize();
if (s != -1) e.add(Q_COMP_SIZE, token(s));
e.add(ze.getName());
vb.add(e);
}
return vb;
} catch (final IOException ex) {
Util.debug(ex);
throw ARCH_FAIL.thrw(info, ex);
} finally {
in.close();
}
}
/**
* Extracts text entries.
*
* @param ctx query context
* @return text entry
* @throws QueryException query exception
*/
private ValueBuilder extractText(final QueryContext ctx) throws QueryException {
final String enc = encoding(2, ARCH_ENCODING, ctx);
final ValueBuilder vb = new ValueBuilder();
for (final byte[] b : extract(ctx)) vb.add(Str.get(encode(b, enc)));
return vb;
}
/**
* Extracts binary entries.
*
* @param ctx query context
* @return binary entry
* @throws QueryException query exception
*/
private ValueBuilder extractBinary(final QueryContext ctx) throws QueryException {
final ValueBuilder vb = new ValueBuilder();
for (final byte[] b : extract(ctx)) vb.add(new B64(b));
return vb;
}
/**
* Updates an archive.
*
* @param ctx query context
* @return updated archive
* @throws QueryException query exception
*/
private B64 update(final QueryContext ctx) throws QueryException {
final B64 archive = (B64) checkType(checkItem(expr[0], ctx), AtomType.B64);
// entries to be updated
final TokenObjMap<Item[]> hm = new TokenObjMap<Item[]>();
final Iter entr = ctx.iter(expr[1]);
final Iter cont = ctx.iter(expr[2]);
int e = 0;
int c = 0;
Item en, cn;
while (true) {
en = entr.next();
cn = cont.next();
if (en == null || cn == null) break;
hm.add(checkElmStr(en).string(info), new Item[] {en, cn});
e++;
c++;
}
// count remaining entries
if (cn != null) do c++; while (cont.next() != null);
if (en != null) do e++; while (entr.next() != null);
if (e != c) ARCH_DIFF.thrw(info, e, c);
final ArchiveIn in = ArchiveIn.get(archive.input(info), info);
final ArchiveOut out = ArchiveOut.get(in.format(), info);
try {
if (in instanceof GZIPIn) ARCH_MODIFY.thrw(info, in.format().toUpperCase(Locale.ENGLISH));
// delete entries to be updated
while (in.more()) if (!hm.contains(token(in.entry().getName()))) out.write(in);
// add new and updated entries
for (final byte[] h : hm) {
if (h == null) continue;
final Item[] it = hm.get(h);
add(it[0], it[1], out, ZipEntry.DEFLATED);
}
} catch (final IOException ex) {
Util.debug(ex);
ARCH_FAIL.thrw(info, ex);
} finally {
in.close();
out.close();
}
return new B64(out.toArray());
}
/**
* Deletes files from an archive.
*
* @param ctx query context
* @return updated archive
* @throws QueryException query exception
*/
private B64 delete(final QueryContext ctx) throws QueryException {
final B64 archive = (B64) checkType(checkItem(expr[0], ctx), AtomType.B64);
// entries to be deleted
final TokenObjMap<Item[]> hm = new TokenObjMap<Item[]>();
final Iter names = ctx.iter(expr[1]);
for (Item en; (en = names.next()) != null; ) {
hm.add(checkElmStr(en).string(info), null);
}
final ArchiveIn in = ArchiveIn.get(archive.input(info), info);
final ArchiveOut out = ArchiveOut.get(in.format(), info);
try {
if (in instanceof GZIPIn) ARCH_MODIFY.thrw(info, in.format().toUpperCase(Locale.ENGLISH));
while (in.more()) if (!hm.contains(token(in.entry().getName()))) out.write(in);
} catch (final IOException ex) {
Util.debug(ex);
ARCH_FAIL.thrw(info, ex);
} finally {
in.close();
out.close();
}
return new B64(out.toArray());
}
/**
* Extracts entries from the archive.
*
* @param ctx query context
* @return text entries
* @throws QueryException query exception
*/
private TokenList extract(final QueryContext ctx) throws QueryException {
final B64 archive = (B64) checkType(checkItem(expr[0], ctx), AtomType.B64);
TokenSet hs = null;
if (expr.length > 1) {
// filter result to specified entries
hs = new TokenSet();
final Iter names = ctx.iter(expr[1]);
for (Item en; (en = names.next()) != null; ) {
hs.add(checkElmStr(en).string(info));
}
}
final TokenList tl = new TokenList();
final ArchiveIn in = ArchiveIn.get(archive.input(info), info);
try {
while (in.more()) {
final ZipEntry ze = in.entry();
if (ze.isDirectory()) continue;
if (hs == null || hs.delete(token(ze.getName())) != 0) tl.add(in.read());
}
} catch (final IOException ex) {
Util.debug(ex);
ARCH_FAIL.thrw(info, ex);
} finally {
in.close();
}
return tl;
}
/**
* Adds the specified entry to the output stream.
*
* @param entry entry descriptor
* @param con contents
* @param out output archive
* @param level default compression level
* @throws QueryException query exception
* @throws IOException I/O exception
*/
private void add(final Item entry, final Item con, final ArchiveOut out, final int level)
throws QueryException, IOException {
// create new zip entry
final String name = string(entry.string(info));
if (name.isEmpty()) ARCH_EMPTY.thrw(info);
final ZipEntry ze = new ZipEntry(name);
String en = null;
// compression level
byte[] lvl = null;
if (entry instanceof ANode) {
final ANode el = (ANode) entry;
lvl = el.attribute(Q_LEVEL);
// last modified
final byte[] mod = el.attribute(Q_LAST_MOD);
if (mod != null) {
try {
ze.setTime(new Int(new Dtm(mod, info)).itr());
} catch (final QueryException qe) {
ARCH_DATETIME.thrw(info, mod);
}
}
// encoding
final byte[] enc = el.attribute(Q_ENCODING);
if (enc != null) {
en = string(enc);
if (!Charset.isSupported(en)) ARCH_ENCODING.thrw(info, enc);
}
}
// data to be compressed
byte[] val = checkStrBin(con);
if (con instanceof AStr && en != null && en != Token.UTF8) val = encode(val, en);
try {
out.level(lvl == null ? level : toInt(lvl));
} catch (final IllegalArgumentException ex) {
ARCH_LEVEL.thrw(info, lvl);
}
out.write(ze, val);
}
/**
* Encodes the specified string to another encoding.
*
* @param val value to be encoded
* @param en encoding
* @return encoded string
* @throws QueryException query exception
*/
private byte[] encode(final byte[] val, final String en) throws QueryException {
try {
return FNConvert.toString(new ArrayInput(val), en);
} catch (final IOException ex) {
throw ARCH_ENCODE.thrw(info, ex);
}
}
/**
* Checks if the specified item is a string or element.
*
* @param it item to be checked
* @return item
* @throws QueryException query exception
*/
private Item checkElmStr(final Item it) throws QueryException {
if (it instanceof AStr || TEST.eq(it)) return it;
throw ELMSTRTYPE.thrw(info, Q_ENTRY.string(), it.type);
}
}