|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 # Copyright (C) 2007 Daniel Holth <dholth@fastmail.fm> |
|
4 # This is a stripped-down version of the original bzr-svn transport.py, |
|
5 # Copyright (C) 2006 Jelmer Vernooij <jelmer@samba.org> |
|
6 |
|
7 # This program is free software; you can redistribute it and/or modify |
|
8 # it under the terms of the GNU General Public License as published by |
|
9 # the Free Software Foundation; either version 2 of the License, or |
|
10 # (at your option) any later version. |
|
11 |
|
12 # This program is distributed in the hope that it will be useful, |
|
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
15 # GNU General Public License for more details. |
|
16 |
|
17 # You should have received a copy of the GNU General Public License |
|
18 # along with this program; if not, write to the Free Software |
|
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
|
20 |
|
21 from cStringIO import StringIO |
|
22 import os |
|
23 from tempfile import mktemp |
|
24 |
|
25 from svn.core import SubversionException, Pool |
|
26 import svn.ra |
|
27 import svn.core |
|
28 |
|
29 # Some older versions of the Python bindings need to be |
|
30 # explicitly initialized. But what we want to do probably |
|
31 # won't work worth a darn against those libraries anyway! |
|
32 svn.ra.initialize() |
|
33 |
|
34 svn_config = svn.core.svn_config_get_config(None) |
|
35 |
|
36 |
|
37 def _create_auth_baton(pool): |
|
38 """Create a Subversion authentication baton. """ |
|
39 import svn.client |
|
40 # Give the client context baton a suite of authentication |
|
41 # providers.h |
|
42 providers = [ |
|
43 svn.client.get_simple_provider(pool), |
|
44 svn.client.get_username_provider(pool), |
|
45 svn.client.get_ssl_client_cert_file_provider(pool), |
|
46 svn.client.get_ssl_client_cert_pw_file_provider(pool), |
|
47 svn.client.get_ssl_server_trust_file_provider(pool), |
|
48 ] |
|
49 return svn.core.svn_auth_open(providers, pool) |
|
50 |
|
51 |
|
52 # # The SVN libraries don't like trailing slashes... |
|
53 # return url.rstrip('/') |
|
54 |
|
55 |
|
56 class SvnRaCallbacks(svn.ra.callbacks2_t): |
|
57 """Remote access callbacks implementation for bzr-svn.""" |
|
58 def __init__(self, pool): |
|
59 svn.ra.callbacks2_t.__init__(self) |
|
60 self.auth_baton = _create_auth_baton(pool) |
|
61 self.pool = pool |
|
62 |
|
63 def open_tmp_file(self, pool): |
|
64 return mktemp(prefix='tailor-svn') |
|
65 |
|
66 class NotBranchError(SubversionException): |
|
67 pass |
|
68 |
|
69 class SvnRaTransport(object): |
|
70 """ |
|
71 Open an ra connection to a Subversion repository. |
|
72 """ |
|
73 def __init__(self, url="", ra=None): |
|
74 self.pool = Pool() |
|
75 self.svn_url = url |
|
76 |
|
77 # Only Subversion 1.4 has reparent() |
|
78 if ra is None or not hasattr(svn.ra, 'reparent'): |
|
79 self.callbacks = SvnRaCallbacks(self.pool) |
|
80 try: |
|
81 ver = svn.ra.version() |
|
82 try: # Older SVN bindings |
|
83 self.ra = svn.ra.open2(self.svn_url.encode('utf8'), self.callbacks, None, svn_config, None) |
|
84 except TypeError, e: |
|
85 self.ra = svn.ra.open2(self.svn_url.encode('utf8'), self.callbacks, svn_config, None) |
|
86 except SubversionException, (_, num): |
|
87 if num == svn.core.SVN_ERR_RA_ILLEGAL_URL: |
|
88 raise NotBranchError(url) |
|
89 if num == svn.core.SVN_ERR_RA_LOCAL_REPOS_OPEN_FAILED: |
|
90 raise NotBranchError(url) |
|
91 if num == svn.core.SVN_ERR_BAD_URL: |
|
92 raise NotBranchError(url) |
|
93 raise |
|
94 |
|
95 else: |
|
96 self.ra = ra |
|
97 svn.ra.reparent(self.ra, self.svn_url.encode('utf8')) |
|
98 |
|
99 class Reporter: |
|
100 def __init__(self, (reporter, report_baton)): |
|
101 self._reporter = reporter |
|
102 self._baton = report_baton |
|
103 |
|
104 def set_path(self, path, revnum, start_empty, lock_token, pool=None): |
|
105 svn.ra.reporter2_invoke_set_path(self._reporter, self._baton, |
|
106 path, revnum, start_empty, lock_token, pool) |
|
107 |
|
108 def delete_path(self, path, pool=None): |
|
109 svn.ra.reporter2_invoke_delete_path(self._reporter, self._baton, |
|
110 path, pool) |
|
111 |
|
112 def link_path(self, path, url, revision, start_empty, lock_token, |
|
113 pool=None): |
|
114 svn.ra.reporter2_invoke_link_path(self._reporter, self._baton, |
|
115 path, url, revision, start_empty, lock_token, |
|
116 pool) |
|
117 |
|
118 def finish_report(self, pool=None): |
|
119 svn.ra.reporter2_invoke_finish_report(self._reporter, |
|
120 self._baton, pool) |
|
121 |
|
122 def abort_report(self, pool=None): |
|
123 svn.ra.reporter2_invoke_abort_report(self._reporter, |
|
124 self._baton, pool) |
|
125 |
|
126 def do_update(self, revnum, path, *args, **kwargs): |
|
127 return self.Reporter(svn.ra.do_update(self.ra, revnum, path, *args, **kwargs)) |
|
128 |
|
129 def clone(self, offset=None): |
|
130 """See Transport.clone().""" |
|
131 if offset is None: |
|
132 return self.__class__(self.base) |
|
133 |
|
134 return SvnRaTransport(urlutils.join(self.base, offset), ra=self.ra) |