| #!/usr/bin/env python2.7 | |
| # | |
| # Copyright (c) 2016 ARM Limited | |
| # All rights reserved | |
| # | |
| # The license below extends only to copyright in the software and shall | |
| # not be construed as granting a license to any other intellectual | |
| # property including but not limited to intellectual property relating | |
| # to a hardware implementation of the functionality of the software | |
| # licensed hereunder. You may use the software subject to the license | |
| # terms below provided that you ensure that this notice is replicated | |
| # unmodified and in its entirety in all distributions of the software, | |
| # modified or unmodified, in source code or in binary form. | |
| # | |
| # Redistribution and use in source and binary forms, with or without | |
| # modification, are permitted provided that the following conditions are | |
| # met: redistributions of source code must retain the above copyright | |
| # notice, this list of conditions and the following disclaimer; | |
| # redistributions in binary form must reproduce the above copyright | |
| # notice, this list of conditions and the following disclaimer in the | |
| # documentation and/or other materials provided with the distribution; | |
| # neither the name of the copyright holders nor the names of its | |
| # contributors may be used to endorse or promote products derived from | |
| # this software without specific prior written permission. | |
| # | |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| # | |
| # Authors: Andreas Sandberg | |
| from abc import * | |
| import os | |
| import subprocess | |
| from region import * | |
| from style import modified_regions | |
| class AbstractRepo(object): | |
| __metaclass__ = ABCMeta | |
| def file_path(self, fname): | |
| """Get the absolute path to a file relative within the repository. The | |
| input file name must be a valid path within the repository. | |
| """ | |
| return os.path.join(self.repo_base(), fname) | |
| def in_repo(self, fname): | |
| """Check if a path points to something within the repository base. Not | |
| that this does not check for the presence of the object in the | |
| file system as it could exist in the index without being in | |
| the file system. | |
| """ | |
| fname = os.path.abspath(fname) | |
| repo_path = os.path.abspath(self.repo_base()) | |
| return os.path.commonprefix([repo_path, fname]) == repo_path | |
| def repo_path(self, fname): | |
| """Get the path of a file relative to the repository base. The input | |
| file name is assumed to be an absolute path or a path relative | |
| to the current working directory. | |
| """ | |
| return os.path.relpath(fname, self.repo_base()) | |
| def get_file(self, name): | |
| """Get the contents of a file in the file system using a path relative | |
| to the repository root. | |
| """ | |
| with open(self.file_path(name), "r") as f: | |
| return f.read() | |
| @abstractmethod | |
| def repo_base(self): | |
| """Get the path to the base of the repository""" | |
| pass | |
| @abstractmethod | |
| def staged_files(self): | |
| """Get a tuple describing the files that have been staged for a | |
| commit: (list of new, list of modified) | |
| """ | |
| pass | |
| @abstractmethod | |
| def staged_regions(self, fname, context=0): | |
| """Get modified regions that will be committed by the next commit | |
| command | |
| """ | |
| pass | |
| @abstractmethod | |
| def modified_regions(self, fname, context=0): | |
| """Get modified regions that have been staged for commit or are | |
| present in the file system. | |
| """ | |
| pass | |
| class GitRepo(AbstractRepo): | |
| def __init__(self): | |
| self.git = "git" | |
| self._head_revision = None | |
| self._repo_base = None | |
| def repo_base(self): | |
| if self._repo_base is None: | |
| self._repo_base = subprocess.check_output( | |
| [ self.git, "rev-parse", "--show-toplevel" ]).rstrip("\n") | |
| return self._repo_base | |
| def staged_files(self): | |
| added = [] | |
| modified = [] | |
| for action, fname in self.status(filter="MA", cached=True): | |
| if action == "M": | |
| modified.append(fname) | |
| elif action == "A": | |
| added.append(fname) | |
| return added, modified | |
| def staged_regions(self, fname, context=0): | |
| if self.file_status(fname, cached=True) in ("", "A", ): | |
| return all_regions | |
| old = self.file_from_head(self.repo_path(fname)).split("\n") | |
| new = self.file_from_index(self.repo_path(fname)).split("\n") | |
| return modified_regions(old, new, context=context) | |
| def modified_regions(self, fname, context=0): | |
| if self.file_status(fname) in ("", "A", ): | |
| return all_regions | |
| old = self.file_from_head(self.repo_path(fname)).split("\n") | |
| new = self.get_file(self.repo_path(fname)).split("\n") | |
| return modified_regions(old, new, context=context) | |
| def head_revision(self): | |
| if self._head_revision is not None: | |
| return self._head_revision | |
| try: | |
| self._head_revision = subprocess.check_output( | |
| [ self.git, "rev-parse", "--verify", "HEAD" ], | |
| stderr=subprocess.PIPE).rstrip("\n") | |
| except subprocess.CalledProcessError: | |
| # Assume that the repo is empty and use the semi-magic | |
| # empty tree revision if git rev-parse returned an error. | |
| self._head_revision = "4b825dc642cb6eb9a060e54bf8d69288fbee4904" | |
| return self._head_revision | |
| def file_status(self, fname, cached=False): | |
| status = self.status(files=[fname], cached=cached) | |
| assert len(status) <= 1 | |
| if status: | |
| return status[0][0] | |
| else: | |
| # No information available for the file. This usually | |
| # means that it hasn't been added to the | |
| # repository/commit. | |
| return "" | |
| def status(self, filter=None, files=[], cached=False): | |
| cmd = [ self.git, "diff-index", "--name-status" ] | |
| if cached: | |
| cmd.append("--cached") | |
| if filter: | |
| cmd += [ "--diff-filter=%s" % filter ] | |
| cmd += [ self.head_revision(), "--" ] + files | |
| status = subprocess.check_output(cmd).rstrip("\n") | |
| if status: | |
| return [ f.split("\t") for f in status.split("\n") ] | |
| else: | |
| return [] | |
| def file_from_index(self, name): | |
| return subprocess.check_output( | |
| [ self.git, "show", ":%s" % (name, ) ]) | |
| def file_from_head(self, name): | |
| return subprocess.check_output( | |
| [ self.git, "show", "%s:%s" % (self.head_revision(), name) ]) | |
| class MercurialRepo(AbstractRepo): | |
| def __init__(self): | |
| self.hg = "hg" | |
| self._repo_base = None | |
| def repo_base(self): | |
| if self._repo_base is None: | |
| self._repo_base = subprocess.check_output( | |
| [ self.hg, "root" ]).rstrip("\n") | |
| return self._repo_base | |
| def staged_files(self): | |
| added = [] | |
| modified = [] | |
| for action, fname in self.status(): | |
| if action == "M": | |
| modified.append(fname) | |
| elif action == "A": | |
| added.append(fname) | |
| return added, modified | |
| def staged_regions(self, fname, context=0): | |
| return self.modified_regions(fname, context=context) | |
| def modified_regions(self, fname, context=0): | |
| old = self.file_from_tip(fname).split("\n") | |
| new = self.get_file(fname).split("\n") | |
| return modified_regions(old, new, context=context) | |
| def status(self, filter=None): | |
| files = subprocess.check_output([ self.hg, "status" ]).rstrip("\n") | |
| if files: | |
| return [ f.split(" ") for f in files.split("\n") ] | |
| else: | |
| return [] | |
| def file_from_tip(self, name): | |
| return subprocess.check_output([ self.hg, "cat", name ]) | |
| def detect_repo(path="."): | |
| """Auto-detect the revision control system used for a source code | |
| directory. The code starts searching for repository meta data | |
| directories in path and then continues towards the root directory | |
| until root is reached or a metadatadirectory has been found. | |
| Returns: List of repository helper classes that can interface with | |
| the detected revision control system(s). | |
| """ | |
| _repo_types = ( | |
| (".git", GitRepo), | |
| (".hg", MercurialRepo), | |
| ) | |
| repo_types = [] | |
| for repo_dir, repo_class in _repo_types: | |
| if os.path.exists(os.path.join(path, repo_dir)): | |
| repo_types.append(repo_class) | |
| if repo_types: | |
| return repo_types | |
| else: | |
| parent_dir = os.path.abspath(os.path.join(path, "..")) | |
| if not os.path.samefile(parent_dir, path): | |
| return detect_repo(path=parent_dir) | |
| else: | |
| # We reached the root directory without finding a meta | |
| # data directory. | |
| return [] |