| # |
| # Copyright (C) 2016 Codethink Limited |
| # |
| # This program is free software; you can redistribute it and/or |
| # modify it under the terms of the GNU Lesser General Public |
| # License as published by the Free Software Foundation; either |
| # version 2 of the License, or (at your option) any later version. |
| # |
| # This library 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 |
| # Lesser General Public License for more details. |
| # |
| # You should have received a copy of the GNU Lesser General Public |
| # License along with this library. If not, see <http://www.gnu.org/licenses/>. |
| # |
| # Authors: |
| # Tristan Van Berkom <tristan.vanberkom@codethink.co.uk> |
| |
| """ |
| git - stage files from a git repository |
| ======================================= |
| |
| **Host dependencies:** |
| |
| * git |
| |
| **Usage:** |
| |
| .. code:: yaml |
| |
| # Specify the git source kind |
| kind: git |
| |
| # Optionally specify a relative staging directory |
| # directory: path/to/stage |
| |
| # Specify the repository url, using an alias defined |
| # in your project configuration is recommended. |
| url: upstream:foo.git |
| |
| # Optionally specify a symbolic tracking branch or tag, this |
| # will be used to update the 'ref' when refreshing the pipeline. |
| track: master |
| |
| # Specify the commit ref, this must be specified in order to |
| # checkout sources and build, but can be automatically updated |
| # if the 'track' attribute was specified. |
| ref: d63cbb6fdc0bbdadc4a1b92284826a6d63a7ebcd |
| |
| # Optionally specify whether submodules should be checked-out. |
| # If not set, this will default to 'True' |
| checkout-submodules: True |
| |
| # If your repository has submodules, explicitly specifying the |
| # url from which they are to be fetched allows you to easily |
| # rebuild the same sources from a different location. This is |
| # especially handy when used with project defined aliases which |
| # can be redefined at a later time. |
| # You may also explicitly specify whether to check out this |
| # submodule. If 'checkout' is set, it will override |
| # 'checkout-submodules' with the value set below. |
| submodules: |
| plugins/bar: |
| url: upstream:bar.git |
| checkout: True |
| plugins/baz: |
| url: upstream:baz.git |
| checkout: False |
| |
| """ |
| |
| import os |
| import errno |
| import re |
| import shutil |
| from collections import Mapping |
| from io import StringIO |
| |
| from configparser import RawConfigParser |
| |
| from buildstream import Source, SourceError, Consistency, SourceFetcher |
| from buildstream import utils |
| |
| GIT_MODULES = '.gitmodules' |
| |
| |
| # Because of handling of submodules, we maintain a GitMirror |
| # for the primary git source and also for each submodule it |
| # might have at a given time |
| # |
| class GitMirror(SourceFetcher): |
| |
| def __init__(self, source, path, url, ref, *, parent=None): |
| |
| super().__init__() |
| self.source = source |
| self.parent = parent |
| self.url = url |
| self.mirror = os.path.join(source.get_mirror_directory(), utils.url_directory_name(url)) |
| self.mark_download_url(url) |
| |
| self._path = path |
| self._ref = ref |
| self._alias_override = None |
| |
| # Ensures that the mirror exists |
| def ensure(self, alias_override=None): |
| |
| # Unfortunately, git does not know how to only clone just a specific ref, |
| # so we have to download all of those gigs even if we only need a couple |
| # of bytes. |
| if not os.path.exists(self.mirror): |
| |
| # Do the initial clone in a tmpdir just because we want an atomic move |
| # after a long standing clone which could fail overtime, for now do |
| # this directly in our git directory, eliminating the chances that the |
| # system configured tmpdir is not on the same partition. |
| # |
| with self.source.tempdir() as tmpdir: |
| url = self.source.translate_url(self.url, alias_override=alias_override) |
| self.source.call([self.source.host_git, 'clone', '--mirror', '-n', url, tmpdir], |
| fail="Failed to clone git repository {}".format(url), |
| fail_temporarily=True) |
| |
| # Attempt atomic rename into destination, this will fail if |
| # another process beat us to the punch |
| try: |
| os.rename(tmpdir, self.mirror) |
| except OSError as e: |
| |
| # When renaming and the destination repo already exists, os.rename() |
| # will fail with ENOTEMPTY, since an empty directory will be silently |
| # replaced |
| if e.errno == errno.ENOTEMPTY: |
| self.source.status("{}: Discarding duplicate clone of {}" |
| .format(self.source, url)) |
| else: |
| raise SourceError("{}: Failed to move cloned git repository {} from '{}' to '{}': {}" |
| .format(self.source, url, tmpdir, self.mirror, e)) from e |
| |
| def _fetch(self, alias_override=None): |
| url = self.source.translate_url(self.url, alias_override=alias_override) |
| |
| if alias_override: |
| remote_name = utils.url_directory_name(alias_override) |
| _, remotes = self.source.check_output( |
| [self.source.host_git, 'remote'], |
| fail="Failed to retrieve list of remotes in {}".format(self.mirror), |
| cwd=self.mirror |
| ) |
| if remote_name not in remotes: |
| self.source.call( |
| [self.source.host_git, 'remote', 'add', remote_name, url], |
| fail="Failed to add remote {} with url {}".format(remote_name, url), |
| cwd=self.mirror |
| ) |
| else: |
| remote_name = "origin" |
| |
| self.source.call([self.source.host_git, 'fetch', remote_name, '--prune'], |
| fail="Failed to fetch from remote git repository: {}".format(url), |
| fail_temporarily=True, |
| cwd=self.mirror) |
| |
| def fetch(self, alias_override=None): |
| self._alias_override = alias_override |
| self.ensure(alias_override) |
| if not self.has_ref(): |
| self._fetch(alias_override) |
| self.assert_ref() |
| |
| def has_ref(self): |
| if not self.ref: |
| return False |
| |
| # If the mirror doesnt exist, we also dont have the ref |
| if not os.path.exists(self.mirror): |
| return False |
| |
| # Check if the ref is really there |
| rc = self.source.call([self.source.host_git, 'cat-file', '-t', self.ref], cwd=self.mirror) |
| return rc == 0 |
| |
| def assert_ref(self): |
| if not self.has_ref(): |
| raise SourceError("{}: expected ref '{}' was not found in git repository: '{}'" |
| .format(self.source, self.ref, self.url)) |
| |
| def latest_commit(self, tracking): |
| _, output = self.source.check_output( |
| [self.source.host_git, 'rev-parse', tracking], |
| fail="Unable to find commit for specified branch name '{}'".format(tracking), |
| cwd=self.mirror) |
| return output.rstrip('\n') |
| |
| def stage(self, directory): |
| fullpath = os.path.join(directory, self.path) |
| |
| # Using --shared here avoids copying the objects into the checkout, in any |
| # case we're just checking out a specific commit and then removing the .git/ |
| # directory. |
| self.source.call([self.source.host_git, 'clone', '--no-checkout', '--shared', self.mirror, fullpath], |
| fail="Failed to create git mirror {} in directory: {}".format(self.mirror, fullpath), |
| fail_temporarily=True) |
| |
| self.source.call([self.source.host_git, 'checkout', '--force', self.ref], |
| fail="Failed to checkout git ref {}".format(self.ref), |
| cwd=fullpath) |
| |
| # Remove .git dir |
| shutil.rmtree(os.path.join(fullpath, ".git")) |
| |
| def init_workspace(self, directory): |
| fullpath = os.path.join(directory, self.path) |
| url = self.source.translate_url(self.url) |
| |
| self.source.call([self.source.host_git, 'clone', '--no-checkout', self.mirror, fullpath], |
| fail="Failed to clone git mirror {} in directory: {}".format(self.mirror, fullpath), |
| fail_temporarily=True) |
| |
| self.source.call([self.source.host_git, 'remote', 'set-url', 'origin', url], |
| fail='Failed to add remote origin "{}"'.format(url), |
| cwd=fullpath) |
| |
| self.source.call([self.source.host_git, 'checkout', '--force', self.ref], |
| fail="Failed to checkout git ref {}".format(self.ref), |
| cwd=fullpath) |
| |
| def _read_gitmodules(self): |
| modules = "{}:{}".format(self.ref, GIT_MODULES) |
| exit_code, output = self.source.check_output( |
| [self.source.host_git, 'show', modules], cwd=self.mirror) |
| |
| # If git show reports error code 128 here, we take it to mean there is |
| # no .gitmodules file to display for the given revision. |
| if exit_code == 128: |
| return |
| elif exit_code != 0: |
| raise SourceError( |
| "{plugin}: Failed to show gitmodules at ref {ref}".format( |
| plugin=self, ref=self.ref)) |
| |
| content = '\n'.join([l.strip() for l in output.splitlines()]) |
| |
| io = StringIO(content) |
| parser = RawConfigParser() |
| parser.read_file(io) |
| |
| for section in parser.sections(): |
| # validate section name against the 'submodule "foo"' pattern |
| if re.match(r'submodule "(.*)"', section): |
| yield (parser, section) |
| |
| # List the submodules (path/url tuples) present at the given ref of this repo |
| def submodule_list(self): |
| for parser, section in self._read_gitmodules(): |
| path = parser.get(section, 'path') |
| url = parser.get(section, 'url') |
| |
| yield (path, url) |
| |
| # Fetch the ref which this mirror requires its submodule to have, |
| # at the given ref of this mirror. |
| def submodule_ref(self, submodule, ref=None): |
| if not ref: |
| ref = self.ref |
| |
| # list objects in the parent repo tree to find the commit |
| # object that corresponds to the submodule |
| _, output = self.source.check_output([self.source.host_git, 'ls-tree', ref, submodule], |
| fail="ls-tree failed for commit {} and submodule: {}".format( |
| ref, submodule), |
| cwd=self.mirror) |
| |
| # read the commit hash from the output |
| fields = output.split() |
| if len(fields) >= 2 and fields[1] == 'commit': |
| submodule_commit = output.split()[2] |
| |
| # fail if the commit hash is invalid |
| if len(submodule_commit) != 40: |
| raise SourceError("{}: Error reading commit information for submodule '{}'" |
| .format(self.source, submodule)) |
| |
| return submodule_commit |
| |
| else: |
| detail = "The submodule '{}' is defined either in the BuildStream source\n".format(submodule) + \ |
| "definition, or in a .gitmodules file. But the submodule was never added to the\n" + \ |
| "underlying git repository with `git submodule add`." |
| |
| self.source.warn("{}: Ignoring inconsistent submodule '{}'" |
| .format(self.source, submodule), detail=detail) |
| |
| return None |
| |
| def get_submodule_path(self, url): |
| real_url = self.source.translate_url(url) |
| for parser, section in self._read_gitmodules(): |
| parsed_url = parser.get(section, 'url') |
| if parsed_url == real_url: |
| return parser.get(section, 'path') |
| |
| raise SourceError("{}: No submodule found with url '{}'".format(self.source, real_url)) |
| |
| @property |
| def path(self): |
| if self._path is None: |
| self._path = self.parent.get_submodule_path(self.url) |
| |
| return self._path |
| |
| @property |
| def ref(self): |
| # The top-level GitMirror may have ref as None, submodules don't. |
| if self._ref is None and self.parent: |
| self._ref = self.parent.submodule_ref(self.path) |
| |
| return self._ref |
| |
| @ref.setter |
| def ref(self, ref): |
| self._ref = ref |
| |
| |
| # A SourceFetcher that may also check for, and have submodules of its own. |
| class TopLevelGitMirror(GitMirror): |
| def __init__(self, source, path, url, ref): |
| super().__init__(source, path, url, ref) |
| self.auto_submodules = [] |
| |
| def fetch(self, alias_override=None): |
| super().fetch(alias_override) |
| self.refresh_submodules() |
| |
| # auto_submodules do not have aliases, so don't need an override |
| for mirror in self.auto_submodules: |
| mirror.fetch() |
| |
| # Refreshes the GitMirror objects for submodules |
| # |
| # Assumes that we have our mirror and we have the ref which we point to |
| # |
| def refresh_submodules(self): |
| self.ensure() |
| |
| excluded_paths = list([s.path for s in self.source.manual_submodules]) |
| submodules = [] |
| |
| # XXX Here we should issue a warning if either: |
| # A.) A submodule exists but is not defined in the element configuration |
| # B.) The element configuration configures submodules which dont exist at the current ref |
| # |
| for path, url in self.submodule_list(): |
| if path in excluded_paths: |
| continue |
| else: |
| self.source.warn("Unexpected submodule detected with path '{}' and url '{}'" |
| .format(path, url)) |
| |
| ref = self.submodule_ref(path) |
| if ref is not None: |
| mirror = GitMirror(self, path, url, ref) |
| submodules.append(mirror) |
| |
| self.auto_submodules = submodules |
| |
| |
| class GitSource(Source): |
| # pylint: disable=attribute-defined-outside-init |
| |
| def configure(self, node): |
| ref = self.node_get_member(node, str, 'ref', None) |
| |
| config_keys = ['url', 'track', 'ref', 'submodules', 'checkout-submodules'] |
| self.node_validate(node, config_keys + Source.COMMON_CONFIG_KEYS) |
| |
| self.original_url = self.node_get_member(node, str, 'url') |
| self.mirror = TopLevelGitMirror(self, '', self.original_url, ref) |
| self.tracking = self.node_get_member(node, str, 'track', None) |
| self.checkout_submodules = self.node_get_member(node, bool, 'checkout-submodules', True) |
| self.manual_submodules = [] |
| |
| # Parse a dict of submodule overrides, stored in the submodule_overrides |
| # and submodule_checkout_overrides dictionaries. |
| self.submodule_overrides = {} |
| self.submodule_checkout_overrides = {} |
| modules = self.node_get_member(node, Mapping, 'submodules', {}) |
| for path, _ in self.node_items(modules): |
| submodule = self.node_get_member(modules, Mapping, path) |
| url = self.node_get_member(submodule, str, 'url', None) |
| submodule_mirror = GitMirror(self, None, url, None, parent=self.mirror) |
| self.manual_submodules.append(submodule_mirror) |
| |
| self.submodule_overrides[path] = url |
| if 'checkout' in submodule: |
| checkout = self.node_get_member(submodule, bool, 'checkout') |
| self.submodule_checkout_overrides[path] = checkout |
| |
| self.mark_download_url(self.original_url) |
| |
| def preflight(self): |
| # Check if git is installed, get the binary at the same time |
| self.host_git = utils.get_host_tool('git') |
| |
| def get_unique_key(self): |
| # Here we want to encode the local name of the repository and |
| # the ref, if the user changes the alias to fetch the same sources |
| # from another location, it should not effect the cache key. |
| key = [self.original_url, self.mirror.ref] |
| |
| # Only modify the cache key with checkout_submodules if it's something |
| # other than the default behaviour. |
| if self.checkout_submodules is False: |
| key.append({"checkout_submodules": self.checkout_submodules}) |
| |
| # We want the cache key to change if the source was |
| # configured differently, and submodules count. |
| if self.submodule_overrides: |
| key.append(self.submodule_overrides) |
| |
| if self.submodule_checkout_overrides: |
| key.append({"submodule_checkout_overrides": self.submodule_checkout_overrides}) |
| |
| return key |
| |
| def get_consistency(self): |
| if self.have_all_refs(): |
| return Consistency.CACHED |
| elif self.mirror.ref is not None: |
| return Consistency.RESOLVED |
| return Consistency.INCONSISTENT |
| |
| def load_ref(self, node): |
| self.mirror.ref = self.node_get_member(node, str, 'ref', None) |
| |
| def get_ref(self): |
| return self.mirror.ref |
| |
| def set_ref(self, ref, node): |
| node['ref'] = self.mirror.ref = ref |
| |
| def track(self): |
| |
| # If self.tracking is not specified it's not an error, just silently return |
| if not self.tracking: |
| # Is there a better way to check if a ref is given. |
| if self.mirror.ref is None: |
| detail = 'Without a tracking branch ref can not be updated. Please ' + \ |
| 'provide a ref or a track.' |
| raise SourceError("{}: No track or ref".format(self), |
| detail=detail, reason="track-attempt-no-track") |
| return None |
| |
| with self.timed_activity("Tracking {} from {}" |
| .format(self.tracking, self.mirror.url), |
| silent_nested=True): |
| self.mirror.ensure() |
| self.mirror._fetch() |
| |
| # Update self.mirror.ref and node.ref from the self.tracking branch |
| ret = self.mirror.latest_commit(self.tracking) |
| |
| return ret |
| |
| def init_workspace(self, directory): |
| # XXX: may wish to refactor this as some code dupe with stage() |
| self.mirror.refresh_submodules() |
| |
| with self.timed_activity('Setting up workspace "{}"'.format(directory), silent_nested=True): |
| self.mirror.init_workspace(directory) |
| for mirror in self.submodules: |
| mirror.init_workspace(directory) |
| |
| def stage(self, directory): |
| |
| # Need to refresh submodule list here again, because |
| # it's possible that we did not load in the main process |
| # with submodules present (source needed fetching) and |
| # we may not know about the submodule yet come time to build. |
| # |
| self.mirror.refresh_submodules() |
| |
| # Stage the main repo in the specified directory |
| # |
| with self.timed_activity("Staging {}".format(self.mirror.url), silent_nested=True): |
| self.mirror.stage(directory) |
| for mirror in self.submodules: |
| if mirror.path in self.submodule_checkout_overrides: |
| checkout = self.submodule_checkout_overrides[mirror.path] |
| else: |
| checkout = self.checkout_submodules |
| |
| if checkout: |
| mirror.stage(directory) |
| |
| def get_source_fetchers(self): |
| return [self.mirror] + self.manual_submodules |
| |
| ########################################################### |
| # Local Functions # |
| ########################################################### |
| @property |
| def submodules(self): |
| return self.manual_submodules + self.mirror.auto_submodules |
| |
| def have_all_refs(self): |
| if not self.mirror.has_ref(): |
| return False |
| |
| self.mirror.refresh_submodules() |
| for mirror in self.submodules: |
| if not os.path.exists(mirror.mirror): |
| return False |
| if not mirror.has_ref(): |
| return False |
| |
| return True |
| |
| |
| # Plugin entry point |
| def setup(): |
| return GitSource |
