X-Git-Url: https://git.distorted.org.uk/~mdw/stgit/blobdiff_plain/575a7e7cb31c21fcf13aa7622200496161286d10..7354d6941b555dd15a0fe9c9e46d4394cf92f01f:/stgit/git.py diff --git a/stgit/git.py b/stgit/git.py index 56374bc..3137797 100644 --- a/stgit/git.py +++ b/stgit/git.py @@ -18,7 +18,7 @@ along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA """ -import sys, os, glob, popen2 +import sys, os, popen2 from stgit.utils import * @@ -92,6 +92,8 @@ def get_commit(id_hash): """Commit objects factory. Save/look-up them in the __commits dictionary """ + global __commits + if id_hash in __commits: return __commits[id_hash] else: @@ -112,7 +114,7 @@ def get_conflicts(): return None def _input(cmd, file_desc): - p = popen2.Popen3(cmd) + p = popen2.Popen3(cmd, True) while True: line = file_desc.readline() if not line: @@ -123,14 +125,14 @@ def _input(cmd, file_desc): raise GitException, '%s failed' % str(cmd) def _output(cmd): - p=popen2.Popen3(cmd) + p=popen2.Popen3(cmd, True) string = p.fromchild.read() if p.wait(): raise GitException, '%s failed' % str(cmd) return string def _output_one_line(cmd, file_desc = None): - p=popen2.Popen3(cmd) + p=popen2.Popen3(cmd, True) if file_desc != None: for line in file_desc: p.tochild.write(line) @@ -141,7 +143,7 @@ def _output_one_line(cmd, file_desc = None): return string def _output_lines(cmd): - p=popen2.Popen3(cmd) + p=popen2.Popen3(cmd, True) lines = p.fromchild.readlines() if p.wait(): raise GitException, '%s failed' % str(cmd) @@ -173,7 +175,7 @@ def __tree_status(files = [], tree_id = 'HEAD', unknown = False, noexclude = True): """Returns a list of pairs - [status, filename] """ - os.system('git-update-index --refresh > /dev/null') + refresh_index() cache_files = [] @@ -214,29 +216,129 @@ def local_changes(): """ return len(__tree_status()) != 0 +# HEAD value cached +__head = None + def get_head(): - """Returns a string representing the HEAD + """Verifies the HEAD and returns the SHA1 id that represents it """ - return read_string(head_link) + global __head + + if not __head: + __head = rev_parse('HEAD') + return __head def get_head_file(): """Returns the name of the file pointed to by the HEAD link """ - # valid link - if os.path.islink(head_link) and os.path.isfile(head_link): - return os.path.basename(os.readlink(head_link)) - else: - raise GitException, 'Invalid .git/HEAD link. Git tree not initialised?' + return os.path.basename(_output_one_line('git-symbolic-ref HEAD')) + +def set_head_file(ref): + """Resets HEAD to point to a new ref + """ + # head cache flushing is needed since we might have a different value + # in the new head + __clear_head_cache() + if __run('git-symbolic-ref HEAD', [ref]) != 0: + raise GitException, 'Could not set head to "%s"' % ref def __set_head(val): """Sets the HEAD value """ - write_string(head_link, val) + global __head + + if not __head or __head != val: + if __run('git-update-ref HEAD', [val]) != 0: + raise GitException, 'Could not update HEAD to "%s".' % val + __head = val + +def __clear_head_cache(): + """Sets the __head to None so that a re-read is forced + """ + global __head + + __head = None + +def refresh_index(): + """Refresh index with stat() information from the working directory. + """ + __run('git-update-index -q --unmerged --refresh') def rev_parse(git_id): - """Parse the string and return an SHA1 id + """Parse the string and return a verified SHA1 id + """ + try: + return _output_one_line(['git-rev-parse', '--verify', git_id]) + except GitException: + raise GitException, 'Unknown revision: %s' % git_id + +def branch_exists(branch): + """Existance check for the named branch + """ + for line in _output_lines(['git-rev-parse', '--symbolic', '--all']): + if line.strip() == branch: + return True + return False + +def create_branch(new_branch, tree_id = None): + """Create a new branch in the git repository + """ + new_head = os.path.join('refs', 'heads', new_branch) + if branch_exists(new_head): + raise GitException, 'Branch "%s" already exists' % new_branch + + current_head = get_head() + set_head_file(new_head) + __set_head(current_head) + + # a checkout isn't needed if new branch points to the current head + if tree_id: + switch(tree_id) + + if os.path.isfile(os.path.join(base_dir, 'MERGE_HEAD')): + os.remove(os.path.join(base_dir, 'MERGE_HEAD')) + +def switch_branch(name): + """Switch to a git branch + """ + global __head + + new_head = os.path.join('refs', 'heads', name) + if not branch_exists(new_head): + raise GitException, 'Branch "%s" does not exist' % name + + tree_id = rev_parse(new_head + '^0') + if tree_id != get_head(): + refresh_index() + if __run('git-read-tree -u -m', [get_head(), tree_id]) != 0: + raise GitException, 'git-read-tree failed (local changes maybe?)' + __head = tree_id + set_head_file(new_head) + + if os.path.isfile(os.path.join(base_dir, 'MERGE_HEAD')): + os.remove(os.path.join(base_dir, 'MERGE_HEAD')) + +def delete_branch(name): + """Delete a git branch + """ + branch_head = os.path.join('refs', 'heads', name) + if not branch_exists(branch_head): + raise GitException, 'Branch "%s" does not exist' % name + os.remove(os.path.join(base_dir, branch_head)) + +def rename_branch(from_name, to_name): + """Rename a git branch """ - return _output(['git-rev-parse', git_id]).strip() + from_head = os.path.join('refs', 'heads', from_name) + if not branch_exists(from_head): + raise GitException, 'Branch "%s" does not exist' % from_name + to_head = os.path.join('refs', 'heads', to_name) + if branch_exists(to_head): + raise GitException, 'Branch "%s" already exists' % to_name + + if get_head_file() == from_name: + set_head_file(to_head) + os.rename(os.path.join(base_dir, from_head), os.path.join(base_dir, to_head)) def add(names): """Add the files or recursively add the directory contents @@ -265,11 +367,6 @@ def add(names): def rm(files, force = False): """Remove a file from the repository """ - if force: - git_opt = '--force-remove' - else: - git_opt = '--remove' - if not force: for f in files: if os.path.exists(f): @@ -368,12 +465,13 @@ def merge(base, head1, head2): """Perform a 3-way merge between base, head1 and head2 into the local tree """ + refresh_index() if __run('git-read-tree -u -m', [base, head1, head2]) != 0: raise GitException, 'git-read-tree failed (local changes maybe?)' # this can fail if there are conflicts - if os.system('git-merge-index -o -q gitmergeonefile.py -a') != 0: - raise GitException, 'git-merge-cache failed (possible conflicts)' + if __run('git-merge-index -o -q gitmergeonefile.py -a') != 0: + raise GitException, 'git-merge-index failed (possible conflicts)' def status(files = [], modified = False, new = False, deleted = False, conflict = False, unknown = False, noexclude = False): @@ -410,7 +508,7 @@ def diff(files = [], rev1 = 'HEAD', rev2 = None, out_fd = None): if rev2: diff_str = _output(['git-diff-tree', '-p', rev1, rev2] + files) else: - os.system('git-update-index --refresh > /dev/null') + refresh_index() diff_str = _output(['git-diff-index', '-p', rev1] + files) if out_fd: @@ -470,6 +568,7 @@ def checkout(files = [], tree_id = None, force = False): def switch(tree_id): """Switch the tree to the given id """ + refresh_index() if __run('git-read-tree -u -m', [get_head(), tree_id]) != 0: raise GitException, 'git-read-tree failed (local changes maybe?)' @@ -495,6 +594,9 @@ def pull(repository = 'origin', refspec = None): """Pull changes from the remote repository. At the moment, just use the 'git pull' command """ + # 'git pull' updates the HEAD + __clear_head_cache() + args = [repository] if refspec: args.append(refspec) @@ -502,17 +604,35 @@ def pull(repository = 'origin', refspec = None): if __run('git pull', args) != 0: raise GitException, 'Failed "git pull %s"' % repository -def apply_patch(filename = None): - """Apply a patch onto the current index. There must not be any - local changes in the tree, otherwise the command fails +def apply_patch(filename = None, base = None): + """Apply a patch onto the current or given index. There must not + be any local changes in the tree, otherwise the command fails """ - os.system('git-update-index --refresh > /dev/null') - - if filename: - if __run('git-apply --index', [filename]) != 0: - raise GitException, 'Patch does not apply cleanly' - else: - _input('git-apply --index', sys.stdin) + def __apply_patch(): + if filename: + return __run('git-apply --index', [filename]) == 0 + else: + try: + _input('git-apply --index', sys.stdin) + except GitException: + return False + return True + + refresh_index() + + if base: + orig_head = get_head() + switch(base) + + if not __apply_patch(): + if base: + switch(orig_head) + raise GitException, 'Patch does not apply cleanly' + elif base: + top = commit(message = 'temporary commit used for applying a patch', + parents = [base]) + switch(orig_head) + merge(base, orig_head, top) def clone(repository, local_dir): """Clone a remote repository. At the moment, just use the