Mercurial > hg4j
view src/org/tmatesoft/hg/repo/HgRemoteRepository.java @ 655:bcbcc318f250
Performance: reuse unzip output buffer
author | Artem Tikhomirov <tikhomirov.artem@gmail.com> |
---|---|
date | Thu, 04 Jul 2013 18:36:38 +0200 |
parents | cd77bf51b562 |
children | 545b1d4cc11d |
line wrap: on
line source
/* * Copyright (c) 2011-2013 TMate Software Ltd * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; version 2 of the License. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * For information on how to redistribute this software under * the terms of a license other than GNU General Public License * contact TMate Software at support@hg4j.com */ package org.tmatesoft.hg.repo; import static org.tmatesoft.hg.util.LogFacility.Severity.Info; import static org.tmatesoft.hg.util.Outcome.Kind.Failure; import static org.tmatesoft.hg.util.Outcome.Kind.Success; import java.io.BufferedReader; import java.io.ByteArrayOutputStream; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.OutputStream; import java.io.StreamTokenizer; import java.net.ContentHandler; import java.net.ContentHandlerFactory; import java.net.HttpURLConnection; import java.net.MalformedURLException; import java.net.URL; import java.net.URLConnection; import java.security.cert.CertificateException; import java.security.cert.X509Certificate; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.HashSet; import java.util.Iterator; import java.util.LinkedHashMap; import java.util.LinkedList; import java.util.List; import java.util.Map; import java.util.Set; import java.util.prefs.BackingStoreException; import java.util.prefs.Preferences; import java.util.zip.InflaterInputStream; import javax.net.ssl.HttpsURLConnection; import javax.net.ssl.SSLContext; import javax.net.ssl.TrustManager; import javax.net.ssl.X509TrustManager; import org.tmatesoft.hg.core.HgBadArgumentException; import org.tmatesoft.hg.core.HgIOException; import org.tmatesoft.hg.core.HgRemoteConnectionException; import org.tmatesoft.hg.core.HgRepositoryNotFoundException; import org.tmatesoft.hg.core.Nodeid; import org.tmatesoft.hg.core.SessionContext; import org.tmatesoft.hg.internal.DataSerializer; import org.tmatesoft.hg.internal.DataSerializer.OutputStreamSerializer; import org.tmatesoft.hg.internal.EncodingHelper; import org.tmatesoft.hg.internal.Internals; import org.tmatesoft.hg.internal.PropertyMarshal; import org.tmatesoft.hg.util.LogFacility.Severity; import org.tmatesoft.hg.util.Outcome; import org.tmatesoft.hg.util.Pair; /** * WORK IN PROGRESS, DO NOT USE * * @see http://mercurial.selenic.com/wiki/WireProtocol * @see http://mercurial.selenic.com/wiki/HttpCommandProtocol * * @author Artem Tikhomirov * @author TMate Software Ltd. */ public class HgRemoteRepository implements SessionContext.Source { private final URL url; private final SSLContext sslContext; private final String authInfo; private final boolean debug; private HgLookup lookupHelper; private final SessionContext sessionContext; private Set<String> remoteCapabilities; static { URLConnection.setContentHandlerFactory(new ContentHandlerFactory() { public ContentHandler createContentHandler(String mimetype) { if ("application/mercurial-0.1".equals(mimetype)) { return new ContentHandler() { @Override public Object getContent(URLConnection urlc) throws IOException { if (urlc.getContentLength() > 0) { ByteArrayOutputStream bos = new ByteArrayOutputStream(); InputStream is = urlc.getInputStream(); int r; while ((r = is.read()) != -1) { bos.write(r); } return new String(bos.toByteArray()); } return "<empty>"; } }; } return null; } }); } HgRemoteRepository(SessionContext ctx, URL url) throws HgBadArgumentException { if (url == null || ctx == null) { throw new IllegalArgumentException(); } this.url = url; sessionContext = ctx; debug = new PropertyMarshal(ctx).getBoolean("hg4j.remote.debug", false); if ("https".equals(url.getProtocol())) { try { sslContext = SSLContext.getInstance("SSL"); class TrustEveryone implements X509TrustManager { public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException { if (debug) { System.out.println("checkClientTrusted:" + authType); } } public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException { if (debug) { System.out.println("checkServerTrusted:" + authType); } } public X509Certificate[] getAcceptedIssuers() { return new X509Certificate[0]; } }; sslContext.init(null, new TrustManager[] { new TrustEveryone() }, null); } catch (Exception ex) { throw new HgBadArgumentException("Can't initialize secure connection", ex); } } else { sslContext = null; } if (url.getUserInfo() != null) { String ai = null; try { // Hack to get Base64-encoded credentials Preferences tempNode = Preferences.userRoot().node("xxx"); tempNode.putByteArray("xxx", url.getUserInfo().getBytes()); ai = tempNode.get("xxx", null); tempNode.removeNode(); } catch (BackingStoreException ex) { sessionContext.getLog().dump(getClass(), Info, ex, null); // IGNORE } authInfo = ai; } else { authInfo = null; } } public boolean isInvalid() throws HgRemoteConnectionException { initCapabilities(); return remoteCapabilities.isEmpty(); } /** * @return human-readable address of the server, without user credentials or any other security information */ public String getLocation() { if (url.getUserInfo() == null) { return url.toExternalForm(); } if (url.getPort() != -1) { return String.format("%s://%s:%d%s", url.getProtocol(), url.getHost(), url.getPort(), url.getPath()); } else { return String.format("%s://%s%s", url.getProtocol(), url.getHost(), url.getPath()); } } public SessionContext getSessionContext() { return sessionContext; } public List<Nodeid> heads() throws HgRemoteConnectionException { HttpURLConnection c = null; try { URL u = new URL(url, url.getPath() + "?cmd=heads"); c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } InputStreamReader is = new InputStreamReader(c.getInputStream(), "US-ASCII"); StreamTokenizer st = new StreamTokenizer(is); st.ordinaryChars('0', '9'); // wordChars performs |, hence need to 0 first st.wordChars('0', '9'); st.eolIsSignificant(false); LinkedList<Nodeid> parseResult = new LinkedList<Nodeid>(); while (st.nextToken() != StreamTokenizer.TT_EOF) { parseResult.add(Nodeid.fromAscii(st.sval)); } return parseResult; } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("heads").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("heads").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } public List<Nodeid> between(Nodeid tip, Nodeid base) throws HgRemoteConnectionException { Range r = new Range(base, tip); // XXX shall handle errors like no range key in the returned map, not sure how. return between(Collections.singletonList(r)).get(r); } /** * @param ranges * @return map, where keys are input instances, values are corresponding server reply * @throws HgRemoteConnectionException */ public Map<Range, List<Nodeid>> between(Collection<Range> ranges) throws HgRemoteConnectionException { if (ranges.isEmpty()) { return Collections.emptyMap(); } // if fact, shall do other way round, this method shall send LinkedHashMap<Range, List<Nodeid>> rv = new LinkedHashMap<HgRemoteRepository.Range, List<Nodeid>>(ranges.size() * 4 / 3); StringBuilder sb = new StringBuilder(20 + ranges.size() * 82); sb.append("pairs="); for (Range r : ranges) { sb.append(r.end.toString()); sb.append('-'); sb.append(r.start.toString()); sb.append('+'); } if (sb.charAt(sb.length() - 1) == '+') { // strip last space sb.setLength(sb.length() - 1); } HttpURLConnection c = null; try { boolean usePOST = ranges.size() > 3; URL u = new URL(url, url.getPath() + "?cmd=between" + (usePOST ? "" : '&' + sb.toString())); c = setupConnection(u.openConnection()); if (usePOST) { c.setRequestMethod("POST"); c.setRequestProperty("Content-Length", String.valueOf(sb.length()/*nodeids are ASCII, bytes == characters */)); c.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); c.setDoOutput(true); c.connect(); OutputStream os = c.getOutputStream(); os.write(sb.toString().getBytes()); os.flush(); os.close(); } else { c.connect(); } if (debug) { System.out.printf("%d ranges, method:%s \n", ranges.size(), c.getRequestMethod()); dumpResponseHeader(u, c); } InputStreamReader is = new InputStreamReader(c.getInputStream(), "US-ASCII"); StreamTokenizer st = new StreamTokenizer(is); st.ordinaryChars('0', '9'); st.wordChars('0', '9'); st.eolIsSignificant(true); Iterator<Range> rangeItr = ranges.iterator(); LinkedList<Nodeid> currRangeList = null; Range currRange = null; boolean possiblyEmptyNextLine = true; while (st.nextToken() != StreamTokenizer.TT_EOF) { if (st.ttype == StreamTokenizer.TT_EOL) { if (possiblyEmptyNextLine) { // newline follows newline; assert currRange == null; assert currRangeList == null; if (!rangeItr.hasNext()) { throw new HgInvalidStateException("Internal error"); // TODO revisit-1.1 } rv.put(rangeItr.next(), Collections.<Nodeid>emptyList()); } else { if (currRange == null || currRangeList == null) { throw new HgInvalidStateException("Internal error"); // TODO revisit-1.1 } // indicate next range value is needed currRange = null; currRangeList = null; possiblyEmptyNextLine = true; } } else { possiblyEmptyNextLine = false; if (currRange == null) { if (!rangeItr.hasNext()) { throw new HgInvalidStateException("Internal error"); // TODO revisit-1.1 } currRange = rangeItr.next(); currRangeList = new LinkedList<Nodeid>(); rv.put(currRange, currRangeList); } Nodeid nid = Nodeid.fromAscii(st.sval); currRangeList.addLast(nid); } } is.close(); return rv; } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("between").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("between").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } public List<RemoteBranch> branches(List<Nodeid> nodes) throws HgRemoteConnectionException { StringBuilder sb = appendNodeidListArgument("nodes", nodes, null); HttpURLConnection c = null; try { URL u = new URL(url, url.getPath() + "?cmd=branches&" + sb.toString()); c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } InputStreamReader is = new InputStreamReader(c.getInputStream(), "US-ASCII"); StreamTokenizer st = new StreamTokenizer(is); st.ordinaryChars('0', '9'); st.wordChars('0', '9'); st.eolIsSignificant(false); ArrayList<Nodeid> parseResult = new ArrayList<Nodeid>(nodes.size() * 4); while (st.nextToken() != StreamTokenizer.TT_EOF) { parseResult.add(Nodeid.fromAscii(st.sval)); } if (parseResult.size() != nodes.size() * 4) { throw new HgRemoteConnectionException(String.format("Bad number of nodeids in result (shall be factor 4), expected %d, got %d", nodes.size()*4, parseResult.size())); } ArrayList<RemoteBranch> rv = new ArrayList<RemoteBranch>(nodes.size()); for (int i = 0; i < nodes.size(); i++) { RemoteBranch rb = new RemoteBranch(parseResult.get(i*4), parseResult.get(i*4 + 1), parseResult.get(i*4 + 2), parseResult.get(i*4 + 3)); rv.add(rb); } return rv; } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("branches").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("branches").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } /* * XXX need to describe behavior when roots arg is empty; our RepositoryComparator code currently returns empty lists when * no common elements found, which in turn means we need to query changes starting with NULL nodeid. * * WireProtocol wiki: roots = a list of the latest nodes on every service side changeset branch that both the client and server know about. * * Perhaps, shall be named 'changegroup' * Changegroup: * http://mercurial.selenic.com/wiki/Merge * http://mercurial.selenic.com/wiki/WireProtocol * * according to latter, bundleformat data is sent through zlib * (there's no header like HG10?? with the server output, though, * as one may expect according to http://mercurial.selenic.com/wiki/BundleFormat) */ public HgBundle getChanges(List<Nodeid> roots) throws HgRemoteConnectionException, HgRuntimeException { List<Nodeid> _roots = roots.isEmpty() ? Collections.singletonList(Nodeid.NULL) : roots; StringBuilder sb = appendNodeidListArgument("roots", _roots, null); HttpURLConnection c = null; try { URL u = new URL(url, url.getPath() + "?cmd=changegroup&" + sb.toString()); c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } File tf = writeBundle(c.getInputStream(), false, "HG10GZ" /*didn't see any other that zip*/); if (debug) { System.out.printf("Wrote bundle %s for roots %s\n", tf, sb); } return getLookupHelper().loadBundle(tf); } catch (MalformedURLException ex) { // XXX in fact, this exception might be better to be re-thrown as RuntimeEx, // as there's little user can do about this issue (URLs are constructed by our code) throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("changegroup").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("changegroup").setServerInfo(getLocation()); } catch (HgRepositoryNotFoundException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("changegroup").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } public void unbundle(HgBundle bundle, List<Nodeid> remoteHeads) throws HgRemoteConnectionException, HgRuntimeException { if (remoteHeads == null) { // TODO collect heads from bundle: // bundle.inspectChangelog(new HeadCollector(for each c : if collected has c.p1 or c.p2, remove them. Add c)) // or get from remote server??? throw Internals.notImplemented(); } StringBuilder sb = appendNodeidListArgument("heads", remoteHeads, null); HttpURLConnection c = null; DataSerializer.DataSource bundleData = bundle.new BundleSerializer(); try { URL u = new URL(url, url.getPath() + "?cmd=unbundle&" + sb.toString()); c = setupConnection(u.openConnection()); c.setRequestMethod("POST"); c.setRequestProperty("Content-Length", String.valueOf(bundleData.serializeLength())); c.setRequestProperty("Content-Type", "application/mercurial-0.1"); c.setDoOutput(true); c.connect(); OutputStream os = c.getOutputStream(); bundleData.serialize(new OutputStreamSerializer(os)); os.flush(); os.close(); if (debug) { dumpResponseHeader(u, c); dumpResponse(c); } checkResponseOk(c, "Push", "unbundle"); } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("unbundle").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("unbundle").setServerInfo(getLocation()); } catch (HgIOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("unbundle").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } public Bookmarks getBookmarks() throws HgRemoteConnectionException, HgRuntimeException { final String actionName = "Get remote bookmarks"; final List<Pair<String, String>> values = listkeys("bookmarks", actionName); ArrayList<Pair<String, Nodeid>> rv = new ArrayList<Pair<String, Nodeid>>(); for (Pair<String, String> l : values) { if (l.second().length() != Nodeid.SIZE_ASCII) { sessionContext.getLog().dump(getClass(), Severity.Warn, "%s: bad nodeid '%s', ignored", actionName, l.second()); continue; } Nodeid n = Nodeid.fromAscii(l.second()); String bm = new String(l.first()); rv.add(new Pair<String, Nodeid>(bm, n)); } return new Bookmarks(rv); } public Outcome updateBookmark(String name, Nodeid oldRev, Nodeid newRev) throws HgRemoteConnectionException, HgRuntimeException { initCapabilities(); if (!remoteCapabilities.contains("pushkey")) { return new Outcome(Failure, "Server doesn't support pushkey protocol"); } if (pushkey("Update remote bookmark", "bookmarks", name, oldRev.toString(), newRev.toString())) { return new Outcome(Success, String.format("Bookmark %s updated to %s", name, newRev.shortNotation())); } return new Outcome(Failure, String.format("Bookmark update (%s: %s -> %s) failed", name, oldRev.shortNotation(), newRev.shortNotation())); } public Phases getPhases() throws HgRemoteConnectionException, HgRuntimeException { initCapabilities(); if (!remoteCapabilities.contains("pushkey")) { // old server defaults to publishing return new Phases(true, Collections.<Nodeid>emptyList()); } final List<Pair<String, String>> values = listkeys("phases", "Get remote phases"); boolean publishing = false; ArrayList<Nodeid> draftRoots = new ArrayList<Nodeid>(); for (Pair<String, String> l : values) { if ("publishing".equalsIgnoreCase(l.first())) { publishing = Boolean.parseBoolean(l.second()); continue; } Nodeid root = Nodeid.fromAscii(l.first()); int ph = Integer.parseInt(l.second()); if (ph == HgPhase.Draft.mercurialOrdinal()) { draftRoots.add(root); } else { assert false; sessionContext.getLog().dump(getClass(), Severity.Error, "Unexpected phase value %d for revision %s", ph, root); } } return new Phases(publishing, draftRoots); } public Outcome updatePhase(HgPhase from, HgPhase to, Nodeid n) throws HgRemoteConnectionException, HgRuntimeException { initCapabilities(); if (!remoteCapabilities.contains("pushkey")) { return new Outcome(Failure, "Server doesn't support pushkey protocol"); } if (pushkey("Update remote phases", "phases", n.toString(), String.valueOf(from.mercurialOrdinal()), String.valueOf(to.mercurialOrdinal()))) { return new Outcome(Success, String.format("Phase of %s updated to %s", n.shortNotation(), to.name())); } return new Outcome(Failure, String.format("Phase update (%s: %s -> %s) failed", n.shortNotation(), from.name(), to.name())); } @Override public String toString() { return getClass().getSimpleName() + '[' + getLocation() + ']'; } private void initCapabilities() throws HgRemoteConnectionException { if (remoteCapabilities == null) { remoteCapabilities = new HashSet<String>(); // say hello to server, check response try { URL u = new URL(url, url.getPath() + "?cmd=hello"); HttpURLConnection c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } BufferedReader r = new BufferedReader(new InputStreamReader(c.getInputStream(), "US-ASCII")); String line = r.readLine(); c.disconnect(); final String capsPrefix = "capabilities:"; if (line == null || !line.startsWith(capsPrefix)) { // for whatever reason, some servers do not respond to hello command (e.g. svnkit) // but respond to 'capabilities' instead. Try it. // TODO [post-1.0] tests needed u = new URL(url, url.getPath() + "?cmd=capabilities"); c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } r = new BufferedReader(new InputStreamReader(c.getInputStream(), "US-ASCII")); line = r.readLine(); c.disconnect(); if (line == null || line.trim().length() == 0) { return; } } else { line = line.substring(capsPrefix.length()).trim(); } String[] caps = line.split("\\s"); remoteCapabilities.addAll(Arrays.asList(caps)); c.disconnect(); } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("hello").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("hello").setServerInfo(getLocation()); } } } private HgLookup getLookupHelper() { if (lookupHelper == null) { lookupHelper = new HgLookup(sessionContext); } return lookupHelper; } private List<Pair<String,String>> listkeys(String namespace, String actionName) throws HgRemoteConnectionException, HgRuntimeException { HttpURLConnection c = null; try { URL u = new URL(url, url.getPath() + "?cmd=listkeys&namespace=" + namespace); c = setupConnection(u.openConnection()); c.connect(); if (debug) { dumpResponseHeader(u, c); } checkResponseOk(c, actionName, "listkeys"); ArrayList<Pair<String, String>> rv = new ArrayList<Pair<String, String>>(); BufferedReader r = new BufferedReader(new InputStreamReader(c.getInputStream(), EncodingHelper.getUTF8())); String l; while ((l = r.readLine()) != null) { int sep = l.indexOf('\t'); if (sep == -1) { sessionContext.getLog().dump(getClass(), Severity.Warn, "%s: bad line '%s', ignored", actionName, l); continue; } rv.add(new Pair<String,String>(l.substring(0, sep), l.substring(sep+1))); } r.close(); return rv; } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("listkeys").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("listkeys").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } private boolean pushkey(String opName, String namespace, String key, String oldValue, String newValue) throws HgRemoteConnectionException, HgRuntimeException { HttpURLConnection c = null; try { final String p = String.format("%s?cmd=pushkey&namespace=%s&key=%s&old=%s&new=%s", url.getPath(), namespace, key, oldValue, newValue); URL u = new URL(url, p); c = setupConnection(u.openConnection()); c.setRequestMethod("POST"); c.connect(); if (debug) { dumpResponseHeader(u, c); } checkResponseOk(c, opName, "pushkey"); final InputStream is = c.getInputStream(); int rv = is.read(); is.close(); return rv == '1'; } catch (MalformedURLException ex) { throw new HgRemoteConnectionException("Bad URL", ex).setRemoteCommand("pushkey").setServerInfo(getLocation()); } catch (IOException ex) { throw new HgRemoteConnectionException("Communication failure", ex).setRemoteCommand("pushkey").setServerInfo(getLocation()); } finally { if (c != null) { c.disconnect(); } } } private void checkResponseOk(HttpURLConnection c, String opName, String remoteCmd) throws HgRemoteConnectionException, IOException { if (c.getResponseCode() != 200) { String m = c.getResponseMessage() == null ? "unknown reason" : c.getResponseMessage(); String em = String.format("%s failed: %s (HTTP error:%d)", opName, m, c.getResponseCode()); throw new HgRemoteConnectionException(em).setRemoteCommand(remoteCmd).setServerInfo(getLocation()); } } private HttpURLConnection setupConnection(URLConnection urlConnection) { urlConnection.setRequestProperty("User-Agent", "hg4j/1.0.0"); urlConnection.addRequestProperty("Accept", "application/mercurial-0.1"); if (authInfo != null) { urlConnection.addRequestProperty("Authorization", "Basic " + authInfo); } if (sslContext != null) { ((HttpsURLConnection) urlConnection).setSSLSocketFactory(sslContext.getSocketFactory()); } return (HttpURLConnection) urlConnection; } private StringBuilder appendNodeidListArgument(String key, List<Nodeid> values, StringBuilder sb) { if (sb == null) { sb = new StringBuilder(20 + values.size() * 41); } sb.append(key); sb.append('='); for (Nodeid n : values) { sb.append(n.toString()); sb.append('+'); } if (sb.charAt(sb.length() - 1) == '+') { // strip last space sb.setLength(sb.length() - 1); } return sb; } private void dumpResponseHeader(URL u, HttpURLConnection c) { System.out.printf("Query (%d bytes):%s\n", u.getQuery().length(), u.getQuery()); System.out.println("Response headers:"); final Map<String, List<String>> headerFields = c.getHeaderFields(); for (String s : headerFields.keySet()) { System.out.printf("%s: %s\n", s, c.getHeaderField(s)); } } private void dumpResponse(HttpURLConnection c) throws IOException { if (c.getContentLength() > 0) { final Object content = c.getContent(); System.out.println(content); } } private static File writeBundle(InputStream is, boolean decompress, String header) throws IOException { InputStream zipStream = decompress ? new InflaterInputStream(is) : is; File tf = File.createTempFile("hg4j-bundle-", null); FileOutputStream fos = new FileOutputStream(tf); fos.write(header.getBytes()); int r; byte[] buf = new byte[8*1024]; while ((r = zipStream.read(buf)) != -1) { fos.write(buf, 0, r); } fos.close(); zipStream.close(); return tf; } public static final class Range { /** * Root of the range, earlier revision */ public final Nodeid start; /** * Head of the range, later revision. */ public final Nodeid end; /** * @param from - root/base revision * @param to - head/tip revision */ public Range(Nodeid from, Nodeid to) { start = from; end = to; } } public static final class RemoteBranch { public final Nodeid head, root, p1, p2; public RemoteBranch(Nodeid h, Nodeid r, Nodeid parent1, Nodeid parent2) { head = h; root = r; p1 = parent1; p2 = parent2; } @Override public boolean equals(Object obj) { if (this == obj) { return true; } if (false == obj instanceof RemoteBranch) { return false; } RemoteBranch o = (RemoteBranch) obj; // in fact, p1 and p2 are not supposed to be null, ever (at least for RemoteBranch created from server output) return head.equals(o.head) && root.equals(o.root) && (p1 == null && o.p1 == null || p1.equals(o.p1)) && (p2 == null && o.p2 == null || p2.equals(o.p2)); } } public static final class Bookmarks implements Iterable<Pair<String, Nodeid>> { private final List<Pair<String, Nodeid>> bm; private Bookmarks(List<Pair<String, Nodeid>> bookmarks) { bm = bookmarks; } public Iterator<Pair<String, Nodeid>> iterator() { return bm.iterator(); } } public static final class Phases { private final boolean pub; private final List<Nodeid> droots; private Phases(boolean publishing, List<Nodeid> draftRoots) { pub = publishing; droots = draftRoots; } /** * Non-publishing servers may (shall?) respond with a list of draft roots. * This method doesn't make sense when {@link #isPublishingServer()} is <code>true</code> * * @return list of draft roots on remote server */ public List<Nodeid> draftRoots() { return droots; } /** * @return <code>true</code> if revisions on remote server shall be deemed published (either * old server w/o explicit setting, or a new one with <code>phases.publish == true</code>) */ public boolean isPublishingServer() { return pub; } } }