-# server.py -- Implementation of the server side git protocols
-# Copryight (C) 2008 Jelmer Vernooij <jelmer@samba.org>
+# client.py -- Implementation of the server side git protocols
+# Copyright (C) 2008-2009 Jelmer Vernooij <jelmer@samba.org>
+# Copyright (C) 2008 John Carr
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
+"""Client side support for the Git protocol."""
+
+__docformat__ = 'restructuredText'
+
import os
import select
import socket
import subprocess
+from dulwich.errors import (
+ ChecksumMismatch,
+ )
from dulwich.protocol import (
Protocol,
TCP_GIT_PORT,
+ ZERO_SHA,
extract_capabilities,
)
from dulwich.pack import (
write_pack_data,
)
-class SimpleFetchGraphWalker(object):
- def __init__(self, local_heads, get_parents):
- self.heads = set(local_heads)
- self.get_parents = get_parents
- self.parents = {}
-
- def ack(self, ref):
- if ref in self.heads:
- self.heads.remove(ref)
- if ref in self.parents:
- for p in self.parents[ref]:
- self.ack(p)
-
- def next(self):
- if self.heads:
- ret = self.heads.pop()
- ps = self.get_parents(ret)
- self.parents[ret] = ps
- self.heads.update(ps)
- return ret
- return None
+def _fileno_can_read(fileno):
+ """Check if a file descriptor is readable."""
+ return len(select.select([fileno], [], [], 0)[0]) > 0
CAPABILITIES = ["multi_ack", "side-band-64k", "ofs-delta"]
"""
- def __init__(self, fileno, read, write, thin_packs=None, include_tag=None,
- shallow=None):
- self.proto = Protocol(read, write)
- self.fileno = fileno
+ def __init__(self, can_read, read, write, thin_packs=True,
+ report_activity=None):
+ """Create a new GitClient instance.
+
+ :param can_read: Function that returns True if there is data available
+ to be read.
+ :param read: Callback for reading data, takes number of bytes to read
+ :param write: Callback for writing data
+ :param thin_packs: Whether or not thin packs should be retrieved
+ :param report_activity: Optional callback for reporting transport
+ activity.
+ """
+ self.proto = Protocol(read, write, report_activity)
+ self._can_read = can_read
self._capabilities = list(CAPABILITIES)
- if thin_packs is None:
- thin_packs = True
if thin_packs:
self._capabilities.append("thin-pack")
- if include_tag is None:
- include_tag = True
- if include_tag:
- self._capabilities.append("include-tag")
- if shallow is None:
- shallow = True
- if shallow:
- self._capabilities.append("shallow")
def capabilities(self):
return " ".join(self._capabilities)
(sha, ref) = pkt.rstrip("\n").split(" ", 1)
if server_capabilities is None:
(ref, server_capabilities) = extract_capabilities(ref)
- if not (ref == "capabilities^{}" and sha == "0" * 40):
- refs[ref] = sha
+ refs[ref] = sha
return refs, server_capabilities
- def send_pack(self, path, generate_pack_contents):
- refs, server_capabilities = self.read_refs()
- changed_refs = [] # FIXME
- if not changed_refs:
+ def send_pack(self, path, determine_wants, generate_pack_contents):
+ """Upload a pack to a remote repository.
+
+ :param path: Repository path
+ :param generate_pack_contents: Function that can return the shas of the
+ objects to upload.
+ """
+ old_refs, server_capabilities = self.read_refs()
+ new_refs = determine_wants(old_refs)
+ if not new_refs:
self.proto.write_pkt_line(None)
- return
- self.proto.write_pkt_line("%s %s %s\0%s" % (changed_refs[0][0], changed_refs[0][1], changed_refs[0][2], self.capabilities()))
+ return {}
want = []
- have = []
- for changed_ref in changed_refs[:]:
- self.proto.write_pkt_line("%s %s %s" % changed_refs)
- want.append(changed_refs[1])
- if changed_refs[0] != "0"*40:
- have.append(changed_refs[0])
+ have = [x for x in old_refs.values() if not x == ZERO_SHA]
+ sent_capabilities = False
+ for refname in set(new_refs.keys() + old_refs.keys()):
+ old_sha1 = old_refs.get(refname, ZERO_SHA)
+ new_sha1 = new_refs.get(refname, ZERO_SHA)
+ if old_sha1 != new_sha1:
+ if sent_capabilities:
+ self.proto.write_pkt_line("%s %s %s" % (old_sha1, new_sha1, refname))
+ else:
+ self.proto.write_pkt_line("%s %s %s\0%s" % (old_sha1, new_sha1, refname, self.capabilities()))
+ sent_capabilities = True
+ if not new_sha1 in (have, ZERO_SHA):
+ want.append(new_sha1)
self.proto.write_pkt_line(None)
- shas = generate_pack_contents(want, have, None)
- write_pack_data(self.write, shas, len(shas))
-
- def fetch_pack(self, path, determine_wants, graph_walker, pack_data, progress):
+ if not want:
+ return new_refs
+ objects = generate_pack_contents(have, want)
+ (entries, sha) = write_pack_data(self.proto.write_file(), objects,
+ len(objects))
+
+ # read the final confirmation sha
+ client_sha = self.proto.read(20)
+ if not client_sha in (None, "", sha):
+ raise ChecksumMismatch(sha, client_sha)
+
+ return new_refs
+
+ def fetch(self, path, target, determine_wants=None, progress=None):
+ """Fetch into a target repository.
+
+ :param path: Path to fetch from
+ :param target: Target repository to fetch into
+ :param determine_wants: Optional function to determine what refs
+ to fetch
+ :param progress: Optional progress function
+ :return: remote refs
+ """
+ if determine_wants is None:
+ determine_wants = target.object_store.determine_wants_all
+ f, commit = target.object_store.add_pack()
+ try:
+ return self.fetch_pack(path, determine_wants,
+ target.get_graph_walker(), f.write, progress)
+ finally:
+ commit()
+
+ def fetch_pack(self, path, determine_wants, graph_walker, pack_data,
+ progress):
"""Retrieve a pack from a git smart server.
:param determine_wants: Callback that returns list of commits to fetch
wants = determine_wants(refs)
if not wants:
self.proto.write_pkt_line(None)
- return
+ return refs
+ assert isinstance(wants, list) and type(wants[0]) == str
self.proto.write_pkt_line("want %s %s\n" % (wants[0], self.capabilities()))
for want in wants[1:]:
self.proto.write_pkt_line("want %s\n" % want)
have = graph_walker.next()
while have:
self.proto.write_pkt_line("have %s\n" % have)
- if len(select.select([self.fileno], [], [], 0)[0]) > 0:
+ if self._can_read():
pkt = self.proto.read_pkt_line()
parts = pkt.rstrip("\n").split(" ")
if parts[0] == "ACK":
progress(pkt)
else:
raise AssertionError("Invalid sideband channel %d" % channel)
+ return refs
class TCPGitClient(GitClient):
+ """A Git Client that works over TCP directly (i.e. git://)."""
- def __init__(self, host, port=TCP_GIT_PORT, thin_packs=None):
+ def __init__(self, host, port=None, *args, **kwargs):
self._socket = socket.socket(type=socket.SOCK_STREAM)
+ if port is None:
+ port = TCP_GIT_PORT
self._socket.connect((host, port))
self.rfile = self._socket.makefile('rb', -1)
self.wfile = self._socket.makefile('wb', 0)
self.host = host
- super(TCPGitClient, self).__init__(self._socket.fileno(), self.rfile.read, self.wfile.write, thin_packs=thin_packs)
+ super(TCPGitClient, self).__init__(lambda: _fileno_can_read(self._socket.fileno()), self.rfile.read, self.wfile.write, *args, **kwargs)
+
+ def send_pack(self, path, changed_refs, generate_pack_contents):
+ """Send a pack to a remote host.
- def send_pack(self, path):
+ :param path: Path of the repository on the remote host
+ """
self.proto.send_cmd("git-receive-pack", path, "host=%s" % self.host)
- super(TCPGitClient, self).send_pack(path)
+ return super(TCPGitClient, self).send_pack(path, changed_refs, generate_pack_contents)
def fetch_pack(self, path, determine_wants, graph_walker, pack_data, progress):
+ """Fetch a pack from the remote host.
+
+ :param path: Path of the reposiutory on the remote host
+ :param determine_wants: Callback that receives available refs dict and
+ should return list of sha's to fetch.
+ :param graph_walker: GraphWalker instance used to find missing shas
+ :param pack_data: Callback for writing pack data
+ :param progress: Callback for writing progress
+ """
self.proto.send_cmd("git-upload-pack", path, "host=%s" % self.host)
- super(TCPGitClient, self).fetch_pack(path, determine_wants, graph_walker, pack_data, progress)
+ return super(TCPGitClient, self).fetch_pack(path, determine_wants,
+ graph_walker, pack_data, progress)
class SubprocessGitClient(GitClient):
+ """Git client that talks to a server using a subprocess."""
- def __init__(self, thin_packs=None):
+ def __init__(self, *args, **kwargs):
self.proc = None
- self._thin_packs = thin_packs
+ self._args = args
+ self._kwargs = kwargs
- def _connect(self, service, *args):
+ def _connect(self, service, *args, **kwargs):
argv = [service] + list(args)
self.proc = subprocess.Popen(argv, bufsize=0,
stdin=subprocess.PIPE,
def write_fn(data):
self.proc.stdin.write(data)
self.proc.stdin.flush()
- return GitClient(self.proc.stdout.fileno(), read_fn, write_fn, thin_packs=self._thin_packs)
+ return GitClient(lambda: _fileno_can_read(self.proc.stdout.fileno()), read_fn, write_fn, *args, **kwargs)
- def send_pack(self, path):
- client = self._connect("git-receive-pack", path)
- client.send_pack(path)
+ def send_pack(self, path, changed_refs, generate_pack_contents):
+ """Upload a pack to the server.
- def fetch_pack(self, path, determine_wants, graph_walker, pack_data, progress):
+ :param path: Path to the git repository on the server
+ :param changed_refs: Dictionary with new values for the refs
+ :param generate_pack_contents: Function that returns an iterator over
+ objects to send
+ """
+ client = self._connect("git-receive-pack", path)
+ return client.send_pack(path, changed_refs, generate_pack_contents)
+
+ def fetch_pack(self, path, determine_wants, graph_walker, pack_data,
+ progress):
+ """Retrieve a pack from the server
+
+ :param path: Path to the git repository on the server
+ :param determine_wants: Function that receives existing refs
+ on the server and returns a list of desired shas
+ :param graph_walker: GraphWalker instance
+ :param pack_data: Function that can write pack data
+ :param progress: Function that can write progress texts
+ """
client = self._connect("git-upload-pack", path)
- client.fetch_pack(path, determine_wants, graph_walker, pack_data, progress)
+ return client.fetch_pack(path, determine_wants, graph_walker, pack_data,
+ progress)
class SSHSubprocess(object):
class SSHGitClient(GitClient):
- def __init__(self, host, port=None):
+ def __init__(self, host, port=None, username=None, *args, **kwargs):
self.host = host
self.port = port
+ self.username = username
+ self._args = args
+ self._kwargs = kwargs
- def send_pack(self, path):
- remote = get_ssh_vendor().connect_ssh(self.host, ["git-receive-pack %s" % path], port=self.port)
- client = GitClient(remote.proc.stdout.fileno(), remote.recv, remote.send)
- client.send_pack(path)
+ def send_pack(self, path, determine_wants, generate_pack_contents):
+ remote = get_ssh_vendor().connect_ssh(self.host, ["git-receive-pack '%s'" % path], port=self.port, username=self.username)
+ client = GitClient(lambda: _fileno_can_read(remote.proc.stdout.fileno()), remote.recv, remote.send, *self._args, **self._kwargs)
+ return client.send_pack(path, determine_wants, generate_pack_contents)
+
+ def fetch_pack(self, path, determine_wants, graph_walker, pack_data,
+ progress):
+ remote = get_ssh_vendor().connect_ssh(self.host, ["git-upload-pack '%s'" % path], port=self.port, username=self.username)
+ client = GitClient(lambda: _fileno_can_read(remote.proc.stdout.fileno()), remote.recv, remote.send, *self._args, **self._kwargs)
+ return client.fetch_pack(path, determine_wants, graph_walker, pack_data,
+ progress)
- def fetch_pack(self, path, determine_wants, graph_walker, pack_data, progress):
- remote = get_ssh_vendor().connect_ssh(self.host, ["git-upload-pack %s" % path], port=self.port)
- client = GitClient(remote.proc.stdout.fileno(), remote.recv, remote.send)
- client.fetch_pack(path, determine_wants, graph_walker, pack_data, progress)
+def get_transport_and_path(uri):
+ """Obtain a git client from a URI or path.
+
+ :param uri: URI or path
+ :return: Tuple with client instance and relative path.
+ """
+ from dulwich.client import TCPGitClient, SSHGitClient, SubprocessGitClient
+ for handler, transport in (("git://", TCPGitClient), ("git+ssh://", SSHGitClient)):
+ if uri.startswith(handler):
+ host, path = uri[len(handler):].split("/", 1)
+ return transport(host), "/"+path
+ # if its not git or git+ssh, try a local url..
+ return SubprocessGitClient(), uri