/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to __init__.py

Commit initial content.

Show diffs side-by-side

added added

removed removed

Lines of Context:
__init__.py
18
18
 
19
19
"""A GIT branch and repository format implementation for bzr."""
20
20
 
21
 
try:
22
 
    import dulwich as git
23
 
except ImportError:
24
 
    import os, sys
25
 
    sys.path.insert(0, os.path.join(os.path.dirname(__file__), "dulwich"))
26
 
    import dulwich as git
27
 
from bzrlib import bzrdir
28
 
from bzrlib.foreign import ForeignVcs, VcsMappingRegistry, foreign_vcs_registry
29
 
from bzrlib.plugins.git.dir import LocalGitBzrDirFormat, RemoteGitBzrDirFormat
30
 
from bzrlib.transport import register_lazy_transport
31
 
from bzrlib.commands import Command, register_command
32
 
from bzrlib.option import Option
33
 
 
34
 
bzrdir.format_registry.register(
35
 
    'git', LocalGitBzrDirFormat,
36
 
    help='GIT repository.', 
37
 
    native=False, experimental=True,
38
 
    )
39
 
 
40
 
bzrdir.BzrDirFormat.register_control_format(LocalGitBzrDirFormat)
41
 
bzrdir.BzrDirFormat.register_control_format(RemoteGitBzrDirFormat)
42
 
 
43
 
register_lazy_transport("git://", 'bzrlib.plugins.git.remote',
44
 
                        'GitSmartTransport')
45
 
 
46
 
 
47
 
class ForeignGit(ForeignVcs):
48
 
    """Foreign Git."""
49
 
 
50
 
 
51
 
git_mapping_registry = VcsMappingRegistry()
52
 
git_mapping_registry.register_lazy('git-experimental', "bzrlib.plugins.git.mapping",
53
 
                                   "BzrGitMappingExperimental")
54
 
foreign_vcs_registry.register("git", ForeignGit(git_mapping_registry), 
55
 
                                      "Stupid content tracker")
56
 
 
57
 
 
58
 
class cmd_git_serve(Command):
59
 
    """Provide access to a Bazaar branch using the git protocol.
60
 
 
61
 
    This command is experimental and doesn't do much yet.
62
 
    """
63
 
    takes_options = [
64
 
        Option('inet',
65
 
               help='serve on stdin/out for use from inetd or sshd'),
66
 
        Option('directory',
67
 
               help='serve contents of directory',
68
 
               type=unicode)
69
 
    ]
70
 
 
71
 
    def run(self, inet=None, port=None, directory=None):
72
 
        from dulwich.server import TCPGitServer
73
 
        from bzrlib.plugins.git.server import BzrBackend
74
 
        from bzrlib.trace import warning
75
 
        import os
76
 
 
77
 
        warning("server support in bzr-git is experimental.")
78
 
 
79
 
        if directory is None:
80
 
            directory = os.getcwd()
81
 
 
82
 
        backend = BzrBackend(directory)
83
 
 
84
 
        if inet:
85
 
            #def send_fn(data):
86
 
            #    sys.stdout.write(data)
87
 
            #    sys.stdout.flush()
88
 
            #server = GitServer(sys.stdin.read, send_fn)
89
 
            raise NotImplementedError
90
 
        else:
91
 
            server = TCPGitServer(backend, 'localhost')
92
 
            server.serve_forever()
93
 
 
94
 
register_command(cmd_git_serve)
95
 
 
96
 
 
97
 
def test_suite():
98
 
    from bzrlib.plugins.git import tests
99
 
    return tests.test_suite()
 
21
 
 
22
import stgit
 
23
import stgit.git as git
 
24
 
 
25
from bzrlib.decorators import *
 
26
import bzrlib.branch
 
27
import bzrlib.bzrdir
 
28
import bzrlib.errors as errors
 
29
import bzrlib.repository
 
30
from bzrlib.revision import Revision
 
31
 
 
32
 
 
33
def gitrevid_from_bzr(revision_id):
 
34
    return revision_id[4:]
 
35
 
 
36
 
 
37
def bzrrevid_from_git(revision_id):
 
38
    return "git:" + revision_id
 
39
 
 
40
 
 
41
class GitLock(object):
 
42
    """A lock that thunks through to Git."""
 
43
 
 
44
    def lock_write(self):
 
45
        pass
 
46
 
 
47
    def lock_read(self):
 
48
        pass
 
49
 
 
50
    def unlock(self):
 
51
        pass
 
52
 
 
53
 
 
54
class GitLockableFiles(bzrlib.lockable_files.LockableFiles):
 
55
    """Git specific lockable files abstraction."""
 
56
 
 
57
    def __init__(self, lock):
 
58
        self._lock = lock
 
59
        self._transaction = None
 
60
        self._lock_mode = None
 
61
        self._lock_count = 0
 
62
 
 
63
 
 
64
class GitDir(bzrlib.bzrdir.BzrDir):
 
65
    """An adapter to the '.git' dir used by git."""
 
66
 
 
67
    def __init__(self, transport, lockfiles, format):
 
68
        self._format = format
 
69
        self.root_transport = transport
 
70
        self.transport = transport.clone('.git')
 
71
        self._lockfiles = lockfiles
 
72
 
 
73
    def get_branch_transport(self, branch_format):
 
74
        if branch_format is None:
 
75
            return self.transport
 
76
        if isinstance(branch_format, GitBzrDirFormat):
 
77
            return self.transport
 
78
        raise errors.IncompatibleFormat(branch_format, self._format)
 
79
 
 
80
    get_repository_transport = get_branch_transport
 
81
    get_workingtree_transport = get_branch_transport
 
82
 
 
83
    def is_supported(self):
 
84
        return True
 
85
 
 
86
    def open_branch(self, ignored=None):
 
87
        """'crate' a branch for this dir."""
 
88
        return GitBranch(self, self._lockfiles)
 
89
 
 
90
    def open_repository(self, shared=False):
 
91
        """'open' a repository for this dir."""
 
92
        return GitRepository(self._gitrepo, self, self._lockfiles)
 
93
 
 
94
 
 
95
class GitBzrDirFormat(bzrlib.bzrdir.BzrDirFormat):
 
96
    """The .git directory control format."""
 
97
 
 
98
    @classmethod
 
99
    def _known_formats(self):
 
100
        return set([GitBzrDirFormat()])
 
101
 
 
102
    def open(self, transport, _create=False, _found=None):
 
103
        """Open this directory.
 
104
        
 
105
        :param _create: create the git dir on the fly. private to GitDirFormat.
 
106
        """
 
107
        # we dont grok readonly - git isn't integrated with transport.
 
108
        url = transport.base
 
109
        if url.startswith('readonly+'):
 
110
            url = url[len('readonly+'):]
 
111
        if url.startswith('file://'):
 
112
            url = url[len('file://'):]
 
113
        url = url.encode('utf8')
 
114
        lockfiles = GitLockableFiles(GitLock())
 
115
        return GitDir(transport, lockfiles, self)
 
116
 
 
117
    @classmethod
 
118
    def probe_transport(klass, transport):
 
119
        """Our format is present if the transport ends in '.not/'."""
 
120
        # little ugly, but works
 
121
        format = klass() 
 
122
        # try a manual probe first, its a little faster perhaps ?
 
123
        if transport.has('.git'):
 
124
            return format
 
125
        # delegate to the main opening code. This pays a double rtt cost at the
 
126
        # moment, so perhaps we want probe_transport to return the opened thing
 
127
        # rather than an openener ? or we could return a curried thing with the
 
128
        # dir to open already instantiated ? Needs more thought.
 
129
        try:
 
130
            format.open(transport)
 
131
            return format
 
132
        except Exception, e:
 
133
            raise errors.NotBranchError(path=transport.base)
 
134
        raise errors.NotBranchError(path=transport.base)
 
135
 
 
136
 
 
137
bzrlib.bzrdir.BzrDirFormat.register_control_format(GitBzrDirFormat)
 
138
 
 
139
 
 
140
class GitBranch(bzrlib.branch.Branch):
 
141
    """An adapter to git repositories for bzr Branch objects."""
 
142
 
 
143
    def __init__(self, gitdir, lockfiles):
 
144
        self.bzrdir = gitdir
 
145
        self.control_files = lockfiles
 
146
        self.repository = GitRepository(gitdir, lockfiles)
 
147
        self.base = gitdir.root_transport.base
 
148
 
 
149
    def lock_write(self):
 
150
        self.control_files.lock_write()
 
151
 
 
152
    @needs_read_lock
 
153
    def last_revision(self):
 
154
        # perhaps should escape this ?
 
155
        return bzrrevid_from_git(git.get_head())
 
156
 
 
157
    def lock_read(self):
 
158
        self.control_files.lock_read()
 
159
 
 
160
    def unlock(self):
 
161
        self.control_files.unlock()
 
162
 
 
163
 
 
164
class GitRepository(bzrlib.repository.Repository):
 
165
    """An adapter to git repositories for bzr."""
 
166
 
 
167
    def __init__(self, gitdir, lockfiles):
 
168
        self.bzrdir = gitdir
 
169
        self.control_files = lockfiles
 
170
 
 
171
    def get_revision(self, revision_id):
 
172
        raw = stgit.git._output_lines('git-rev-list --header --max-count=1 %s' % gitrevid_from_bzr(revision_id))
 
173
        # first field is the rev itself.
 
174
        # then its 'field value'
 
175
        # until the EOF??
 
176
        parents = []
 
177
        log = []
 
178
        in_log = False
 
179
        committer = None
 
180
        for field in raw[1:]:
 
181
            #if field.startswith('author '):
 
182
            #    committer = field[7:]
 
183
            if field.startswith('parent '):
 
184
                parents.append(bzrrevid_from_git(field.split()[1]))
 
185
            elif field.startswith('committer '):
 
186
                commit_fields = field.split()
 
187
                if committer is None:
 
188
                    committer = ' '.join(commit_fields[1:-3])
 
189
                timestamp = commit_fields[-2]
 
190
                timezone = commit_fields[-1]
 
191
            elif in_log:
 
192
                log.append(field)
 
193
            elif field == '\n':
 
194
                in_log = True
 
195
 
 
196
        log = ''.join(log)
 
197
        result = Revision(revision_id)
 
198
        result.parent_ids = parents
 
199
        result.message = log
 
200
        result.inventory_sha1 = ""
 
201
        result.timezone = timezone and int(timezone)
 
202
        result.timestamp = float(timestamp)
 
203
        result.committer = committer 
 
204
        return result