327 lines
10 KiB
Python
327 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright 2010-2014 The pygit2 contributors
|
|
#
|
|
# This file is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License, version 2,
|
|
# as published by the Free Software Foundation.
|
|
#
|
|
# In addition to the permissions in the GNU General Public License,
|
|
# the authors give you unlimited permission to link the compiled
|
|
# version of this file into combinations with other programs,
|
|
# and to distribute those combinations without any restriction
|
|
# coming from the use of this file. (The General Public License
|
|
# restrictions do apply in other respects; for example, they cover
|
|
# modification of the file, and distribution when not linked into
|
|
# a combined executable.)
|
|
#
|
|
# This file is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; see the file COPYING. If not, write to
|
|
# the Free Software Foundation, 51 Franklin Street, Fifth Floor,
|
|
# Boston, MA 02110-1301, USA.
|
|
|
|
# Import from the future
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
from _pygit2 import Oid, Tree, Diff
|
|
|
|
from .ffi import ffi, C, to_str, is_string, strings_to_strarray
|
|
from .errors import check_error, GitError
|
|
|
|
class Index(object):
|
|
|
|
def __init__(self, path=None):
|
|
"""Create a new Index
|
|
|
|
If path is supplied, the read and write methods will use that path
|
|
to read from and write to.
|
|
"""
|
|
cindex = ffi.new('git_index **')
|
|
err = C.git_index_open(cindex, to_str(path))
|
|
check_error(err)
|
|
|
|
self._index = cindex[0]
|
|
self._cindex = cindex
|
|
|
|
@classmethod
|
|
def from_c(cls, repo, ptr):
|
|
index = cls.__new__(cls);
|
|
index._repo = repo
|
|
index._index = ptr[0]
|
|
index._cindex = ptr
|
|
|
|
return index
|
|
|
|
@property
|
|
def _pointer(self):
|
|
return bytes(ffi.buffer(self._cindex)[:])
|
|
|
|
def __del__(self):
|
|
C.git_index_free(self._index)
|
|
|
|
def __len__(self):
|
|
return C.git_index_entrycount(self._index)
|
|
|
|
def __contains__(self, path):
|
|
err = C.git_index_find(ffi.NULL, self._index, to_str(path))
|
|
if err == C.GIT_ENOTFOUND:
|
|
return False
|
|
|
|
check_error(err)
|
|
return True
|
|
|
|
def __getitem__(self, key):
|
|
centry = ffi.NULL
|
|
if is_string(key):
|
|
centry = C.git_index_get_bypath(self._index, to_str(key), 0)
|
|
elif not key >= 0:
|
|
raise ValueError(key)
|
|
else:
|
|
centry = C.git_index_get_byindex(self._index, key)
|
|
|
|
if centry == ffi.NULL:
|
|
raise KeyError(key)
|
|
|
|
return IndexEntry._from_c(centry)
|
|
|
|
def __iter__(self):
|
|
return IndexIterator(self)
|
|
|
|
def read(self, force=True):
|
|
"""Update the contents the Index
|
|
|
|
Update the contents by reading from a file
|
|
|
|
Arguments:
|
|
|
|
force: if True (the default) allways reload. If False, only if the file has changed
|
|
"""
|
|
|
|
err = C.git_index_read(self._index, force)
|
|
check_error(err, True)
|
|
|
|
def write(self):
|
|
"""Write the contents of the Index to disk
|
|
"""
|
|
err = C.git_index_write(self._index)
|
|
check_error(err, True)
|
|
|
|
def clear(self):
|
|
err = C.git_index_clear(self._index)
|
|
check_error(err)
|
|
|
|
def read_tree(self, tree):
|
|
"""Replace the contents of the Index with those of a tree
|
|
|
|
The tree will be read recursively and all its children will also be
|
|
inserted into the Index.
|
|
"""
|
|
if is_string(tree):
|
|
tree = self._repo[tree]
|
|
|
|
if isinstance(tree, Oid):
|
|
if not hasattr(self, '_repo'):
|
|
raise TypeError("id given but no associated repository")
|
|
|
|
tree = self._repo[tree]
|
|
elif not isinstance(tree, Tree):
|
|
raise TypeError("argument must be Oid or Tree")
|
|
|
|
tree_cptr = ffi.new('git_tree **')
|
|
ffi.buffer(tree_cptr)[:] = tree._pointer[:]
|
|
err = C.git_index_read_tree(self._index, tree_cptr[0])
|
|
check_error(err)
|
|
|
|
def write_tree(self, repo=None):
|
|
"""Create a tree out of the Index
|
|
|
|
The contents of the index will be written out to the object
|
|
database. If there is no associated repository, 'repo' must be
|
|
passed. If there is an associated repository and 'repo' is
|
|
passed, then that repository will be used instead.
|
|
|
|
It returns the id of the resulting tree.
|
|
"""
|
|
coid = ffi.new('git_oid *')
|
|
if repo:
|
|
err = C.git_index_write_tree_to(coid, self._index, repo._repo)
|
|
else:
|
|
err = C.git_index_write_tree(coid, self._index)
|
|
|
|
check_error(err)
|
|
return Oid(raw=bytes(ffi.buffer(coid)[:]))
|
|
|
|
|
|
def remove(self, path):
|
|
"""Remove an entry from the Index.
|
|
"""
|
|
err = C.git_index_remove(self._index, to_str(path), 0)
|
|
check_error(err, True)
|
|
|
|
def add_all(self, pathspecs=[]):
|
|
"""Add or update index entries matching files in the working directory.
|
|
|
|
If pathspecs are specified, only files matching those pathspecs will be added.
|
|
"""
|
|
arr, refs = strings_to_strarray(pathspecs)
|
|
err = C.git_index_add_all(self._index, arr, 0, ffi.NULL, ffi.NULL)
|
|
check_error(err, True)
|
|
|
|
def add(self, path_or_entry):
|
|
"""Add or update an entry in the Index
|
|
|
|
If a path is given, that file will be added. The path must be
|
|
relative to the root of the worktree and the Index must be associated
|
|
with a repository.
|
|
|
|
If an IndexEntry is given, that entry will be added or update in the Index
|
|
without checking for the existence of the path or id.
|
|
"""
|
|
|
|
if is_string(path_or_entry):
|
|
path = path_or_entry
|
|
err = C.git_index_add_bypath(self._index, to_str(path))
|
|
elif isinstance(path_or_entry, IndexEntry):
|
|
entry = path_or_entry
|
|
centry, str_ref = entry._to_c()
|
|
err = C.git_index_add(self._index, centry)
|
|
else:
|
|
raise AttributeError('argument must be string or IndexEntry')
|
|
|
|
check_error(err, True)
|
|
|
|
def diff_to_workdir(self, flags=0, context_lines=3, interhunk_lines=0):
|
|
"""Diff the index against the working directory
|
|
|
|
Return a :py:class:`~pygit2.Diff` object with the differences
|
|
between the index and the working copy.
|
|
|
|
Arguments:
|
|
|
|
flags: a GIT_DIFF_* constant.
|
|
|
|
context_lines: the number of unchanged lines that define the
|
|
boundary of a hunk (and to display before and after)
|
|
|
|
interhunk_lines: the maximum number of unchanged lines between hunk
|
|
boundaries before the hunks will be merged into a one
|
|
"""
|
|
if not hasattr(self, '_repo'):
|
|
raise ValueError('diff needs an associated repository')
|
|
|
|
copts = ffi.new('git_diff_options *')
|
|
err = C.git_diff_init_options(copts, 1)
|
|
check_error(err)
|
|
|
|
copts.flags = flags
|
|
copts.context_lines = context_lines
|
|
copts.interhunk_lines = interhunk_lines
|
|
|
|
cdiff = ffi.new('git_diff **')
|
|
err = C.git_diff_index_to_workdir(cdiff, self._repo._repo, self._index, copts)
|
|
check_error(err)
|
|
|
|
return Diff.from_c(bytes(ffi.buffer(cdiff)[:]), self._repo)
|
|
|
|
def diff_to_tree(self, tree, flags=0, context_lines=3, interhunk_lines=0):
|
|
"""Diff the index against a tree
|
|
|
|
Return a :py:class:`~pygit2.Diff` object with the differences between the
|
|
index and the given tree.
|
|
|
|
Arguments:
|
|
|
|
tree: the tree to diff.
|
|
|
|
flags: a GIT_DIFF_* constant.
|
|
|
|
context_lines: the number of unchanged lines that define the boundary
|
|
of a hunk (and to display before and after)
|
|
|
|
interhunk_lines: the maximum number of unchanged lines between hunk
|
|
boundaries before the hunks will be merged into a one.
|
|
"""
|
|
|
|
if not hasattr(self, '_repo'):
|
|
raise ValueError('diff needs an associated repository')
|
|
|
|
if not isinstance(tree, Tree):
|
|
raise TypeError('tree must be a Tree')
|
|
|
|
copts = ffi.new('git_diff_options *')
|
|
err = C.git_diff_init_options(copts, 1)
|
|
check_error(err)
|
|
|
|
copts.flags = flags
|
|
copts.context_lines = context_lines
|
|
copts.interhunk_lines = interhunk_lines
|
|
|
|
ctree = ffi.new('git_tree **')
|
|
ffi.buffer(ctree)[:] = tree._pointer[:]
|
|
|
|
cdiff = ffi.new('git_diff **')
|
|
err = C.git_diff_tree_to_index(cdiff, self._repo._repo, ctree[0], self._index, copts)
|
|
check_error(err)
|
|
|
|
return Diff.from_c(bytes(ffi.buffer(cdiff)[:]), self._repo)
|
|
|
|
class IndexEntry(object):
|
|
__slots__ = ['id', 'path', 'mode', '_index']
|
|
|
|
def __init__(self, path, object_id, mode):
|
|
self.path = path
|
|
self.id = object_id
|
|
self.mode = mode
|
|
|
|
@property
|
|
def hex(self):
|
|
return self.id.hex
|
|
|
|
def _to_c(self):
|
|
"""Convert this entry into the C structure
|
|
|
|
The first returned arg is the pointer, the second is the reference to
|
|
the string we allocated, which we need to exist past this function
|
|
"""
|
|
centry = ffi.new('git_index_entry *')
|
|
# basically memcpy()
|
|
ffi.buffer(ffi.addressof(centry, 'id'))[:] = self.id.raw[:]
|
|
centry.mode = self.mode
|
|
path = ffi.new('char[]', to_str(self.path))
|
|
centry.path = path
|
|
|
|
return centry, path
|
|
|
|
@classmethod
|
|
def _from_c(cls, centry):
|
|
entry = cls.__new__(cls)
|
|
entry.path = ffi.string(centry.path).decode()
|
|
entry.mode = centry.mode
|
|
entry.id = Oid(raw=bytes(ffi.buffer(ffi.addressof(centry, 'id'))[:]))
|
|
|
|
return entry
|
|
|
|
class IndexIterator(object):
|
|
|
|
def __init__(self, index):
|
|
self.index = index
|
|
self.n = 0
|
|
self.max = len(index)
|
|
|
|
def next(self):
|
|
return self.__next__()
|
|
|
|
def __next__(self):
|
|
if self.n >= self.max:
|
|
raise StopIteration
|
|
|
|
entry = self.index[self.n]
|
|
self.n += 1
|
|
|
|
return entry
|