aboutsummaryrefslogtreecommitdiff
path: root/gitosis/repository.py
blob: 092e41d6a9fecdc558d7079e873dfbfaa8362081 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
import errno
import os
import re
import subprocess
import sys
 
from gitosis import util
 
class GitError(Exception):
    """git failed"""
 
    def __str__(self):
        return '%s%s' % (self.__doc__''.join(self.args))
 
class GitInitError(Exception):
    """git init failed"""
 
def init(
    path,
    template=None,
    _git=None,
    ):
    """
    Create a git repository at C{path} (if missing).
 
    Leading directories of C{path} must exist.
 
    @param path: Path of repository create.
 
    @type path: str
 
    @param template: Template directory, to pass to C{git init}.
 
    @type template: str
    """
    if _git is None:
        _git = 'git'
 
    util.mkdir(path0750)
    args = [ 
        _git, 
        '--git-dir=.', 
        'init', 
        ]
    if template is not None:
        args.append('--template=%s' % template)
    returncode = subprocess.call(
        args=args,
        cwd=path,
        stdout=sys.stderr,
        close_fds=True,
        )
    if returncode != 0:
        raise GitInitError('exit status %d' % returncode)
 
 
class GitFastImportError(GitError):
    """git fast-import failed"""
    pass
 
def fast_import(
    git_dir,
    commit_msg,
    committer,
    files,
    parent=None,
    ):
    """
    Create an initial commit.
    """
    child = subprocess.Popen(
        args=[ 
            'git', 
            '--git-dir=.', 
            'fast-import', 
            '--quiet', 
            '--date-format=now', 
            ],
        cwd=git_dir,
        stdin=subprocess.PIPE,
        close_fds=True,
        )
    files = list(files)
    for index(pathcontent) in enumerate(files):
        child.stdin.write("""\
blob
mark :%(mark)d
data %(len)d
%(content)s
""" % dict(
            mark=index+1,
            len=len(content),
            content=content,
            ))
    child.stdin.write("""\
commit refs/heads/master
committer %(committer)s now
data %(commit_msg_len)d
%(commit_msg)s
""" % dict(
            committer=committer,
            commit_msg_len=len(commit_msg),
            commit_msg=commit_msg,
            ))
    if parent is not None:
        assert not parent.startswith(':')
        child.stdin.write("""\
from %(parent)s
""" % dict(
                parent=parent,
                ))
    for index(pathcontent) in enumerate(files):
        child.stdin.write('M 100644 :%d %s\n' % (index+1path))
    child.stdin.close()
    returncode = child.wait()
    if returncode != 0:
        raise GitFastImportError(
            'git fast-import failed''exit status %d' % returncode)
 
class GitExportError(GitError):
    """Export failed"""
    pass
 
class GitReadTreeError(GitExportError):
    """git read-tree failed"""
 
class GitCheckoutIndexError(GitExportError):
    """git checkout-index failed"""
 
def export(git_dir, path):
    try:
        os.mkdir(path)
    except OSErrore:
        if e.errno == errno.EEXIST:
            pass
        else:
            raise
    returncode = subprocess.call(
        args=[ 
            'git', 
            '--git-dir=%s' % git_dir, 
            'read-tree', 
            'HEAD', 
            ],
        close_fds=True,
        )
    if returncode != 0:
        raise GitReadTreeError('exit status %d' % returncode)
    # jumping through hoops to be compatible with git versions 
    # that don't have --work-tree= 
    env = {}
    env.update(os.environ)
    env['GIT_WORK_TREE'] = '.'
    returncode = subprocess.call(
        args=[ 
            'git', 
            '--git-dir=%s' % os.path.abspath(git_dir), 
            'checkout-index', 
            '-a', 
            '-f', 
            ],
        cwd=path,
        close_fds=True,
        env=env,
        )
    if returncode != 0:
        raise GitCheckoutIndexError('exit status %d' % returncode)
 
class GitHasInitialCommitError(GitError):
    """Check for initial commit failed"""
 
class GitRevParseError(GitError):
    """rev-parse failed"""
 
def has_initial_commit(git_dir):
    child = subprocess.Popen(
        args=[ 
            'git', 
            '--git-dir=.', 
            'rev-parse', 
            'HEAD', 
            ],
        cwd=git_dir,
        stdout=subprocess.PIPE,
        close_fds=True,
        )
    got = child.stdout.read()
    returncode = child.wait()
    if returncode != 0:
        raise GitRevParseError('exit status %d' % returncode)
    if got == 'HEAD\n':
        return False
    elif re.match('^[0-9a-f]{40}\n$'got):
        return True
    else:
        raise GitHasInitialCommitError('Unknown git HEAD: %r' % got)