Tristan Van Berkom pushed to branch master at BuildStream / buildstream
Commits:
-
f8b06acc
by Tristan Van Berkom at 2018-09-02T09:33:39Z
-
559b6fbc
by Tristan Van Berkom at 2018-09-02T09:34:10Z
-
43ad22d7
by Tristan Van Berkom at 2018-09-02T09:34:10Z
-
16462e9c
by Tristan Van Berkom at 2018-09-02T09:36:49Z
-
1f9c4147
by Tristan Van Berkom at 2018-09-02T09:37:12Z
-
f9b2f1a9
by Tristan Van Berkom at 2018-09-02T09:37:21Z
-
88460cd2
by Tristan Van Berkom at 2018-09-03T06:23:38Z
5 changed files:
- buildstream/element.py
- buildstream/plugin.py
- buildstream/plugins/sources/git.py
- buildstream/source.py
- tests/frontend/project/sources/fetch_source.py
Changes:
... | ... | @@ -245,7 +245,7 @@ class Element(Plugin): |
245 | 245 |
# Collect the composited element configuration and
|
246 | 246 |
# ask the element to configure itself.
|
247 | 247 |
self.__config = self.__extract_config(meta)
|
248 |
- self.configure(self.__config)
|
|
248 |
+ self._configure(self.__config)
|
|
249 | 249 |
|
250 | 250 |
# Extract Sandbox config
|
251 | 251 |
self.__sandbox_config = self.__extract_sandbox_config(meta)
|
... | ... | @@ -179,6 +179,7 @@ class Plugin(): |
179 | 179 |
self.__provenance = provenance # The Provenance information
|
180 | 180 |
self.__type_tag = type_tag # The type of plugin (element or source)
|
181 | 181 |
self.__unique_id = _plugin_register(self) # Unique ID
|
182 |
+ self.__configuring = False # Whether we are currently configuring
|
|
182 | 183 |
|
183 | 184 |
# Infer the kind identifier
|
184 | 185 |
modulename = type(self).__module__
|
... | ... | @@ -682,7 +683,32 @@ class Plugin(): |
682 | 683 |
else:
|
683 | 684 |
yield log
|
684 | 685 |
|
686 |
+ # _configure():
|
|
687 |
+ #
|
|
688 |
+ # Calls configure() for the plugin, this must be called by
|
|
689 |
+ # the core instead of configure() directly, so that the
|
|
690 |
+ # _get_configuring() state is up to date.
|
|
691 |
+ #
|
|
692 |
+ # Args:
|
|
693 |
+ # node (dict): The loaded configuration dictionary
|
|
694 |
+ #
|
|
695 |
+ def _configure(self, node):
|
|
696 |
+ self.__configuring = True
|
|
697 |
+ self.configure(node)
|
|
698 |
+ self.__configuring = False
|
|
699 |
+ |
|
700 |
+ # _get_configuring():
|
|
701 |
+ #
|
|
702 |
+ # Checks whether the plugin is in the middle of having
|
|
703 |
+ # its Plugin.configure() method called
|
|
704 |
+ #
|
|
705 |
+ # Returns:
|
|
706 |
+ # (bool): Whether we are currently configuring
|
|
707 |
+ def _get_configuring(self):
|
|
708 |
+ return self.__configuring
|
|
709 |
+ |
|
685 | 710 |
# _preflight():
|
711 |
+ #
|
|
686 | 712 |
# Calls preflight() for the plugin, and allows generic preflight
|
687 | 713 |
# checks to be added
|
688 | 714 |
#
|
... | ... | @@ -690,6 +716,7 @@ class Plugin(): |
690 | 716 |
# SourceError: If it's a Source implementation
|
691 | 717 |
# ElementError: If it's an Element implementation
|
692 | 718 |
# ProgramNotFoundError: If a required host tool is not found
|
719 |
+ #
|
|
693 | 720 |
def _preflight(self):
|
694 | 721 |
self.preflight()
|
695 | 722 |
|
... | ... | @@ -100,13 +100,14 @@ INCONSISTENT_SUBMODULE = "inconsistent-submodules" |
100 | 100 |
#
|
101 | 101 |
class GitMirror(SourceFetcher):
|
102 | 102 |
|
103 |
- def __init__(self, source, path, url, ref):
|
|
103 |
+ def __init__(self, source, path, url, ref, *, primary=False):
|
|
104 | 104 |
|
105 | 105 |
super().__init__()
|
106 | 106 |
self.source = source
|
107 | 107 |
self.path = path
|
108 | 108 |
self.url = url
|
109 | 109 |
self.ref = ref
|
110 |
+ self.primary = primary
|
|
110 | 111 |
self.mirror = os.path.join(source.get_mirror_directory(), utils.url_directory_name(url))
|
111 | 112 |
self.mark_download_url(url)
|
112 | 113 |
|
... | ... | @@ -124,7 +125,8 @@ class GitMirror(SourceFetcher): |
124 | 125 |
# system configured tmpdir is not on the same partition.
|
125 | 126 |
#
|
126 | 127 |
with self.source.tempdir() as tmpdir:
|
127 |
- url = self.source.translate_url(self.url, alias_override=alias_override)
|
|
128 |
+ url = self.source.translate_url(self.url, alias_override=alias_override,
|
|
129 |
+ primary=self.primary)
|
|
128 | 130 |
self.source.call([self.source.host_git, 'clone', '--mirror', '-n', url, tmpdir],
|
129 | 131 |
fail="Failed to clone git repository {}".format(url),
|
130 | 132 |
fail_temporarily=True)
|
... | ... | @@ -146,7 +148,9 @@ class GitMirror(SourceFetcher): |
146 | 148 |
.format(self.source, url, tmpdir, self.mirror, e)) from e
|
147 | 149 |
|
148 | 150 |
def _fetch(self, alias_override=None):
|
149 |
- url = self.source.translate_url(self.url, alias_override=alias_override)
|
|
151 |
+ url = self.source.translate_url(self.url,
|
|
152 |
+ alias_override=alias_override,
|
|
153 |
+ primary=self.primary)
|
|
150 | 154 |
|
151 | 155 |
if alias_override:
|
152 | 156 |
remote_name = utils.url_directory_name(alias_override)
|
... | ... | @@ -307,7 +311,7 @@ class GitSource(Source): |
307 | 311 |
self.node_validate(node, config_keys + Source.COMMON_CONFIG_KEYS)
|
308 | 312 |
|
309 | 313 |
self.original_url = self.node_get_member(node, str, 'url')
|
310 |
- self.mirror = GitMirror(self, '', self.original_url, ref)
|
|
314 |
+ self.mirror = GitMirror(self, '', self.original_url, ref, primary=True)
|
|
311 | 315 |
self.tracking = self.node_get_member(node, str, 'track', None)
|
312 | 316 |
|
313 | 317 |
# At this point we now know if the source has a ref and/or a track.
|
... | ... | @@ -327,6 +331,11 @@ class GitSource(Source): |
327 | 331 |
for path, _ in self.node_items(modules):
|
328 | 332 |
submodule = self.node_get_member(modules, Mapping, path)
|
329 | 333 |
url = self.node_get_member(submodule, str, 'url', None)
|
334 |
+ |
|
335 |
+ # Make sure to mark all URLs that are specified in the configuration
|
|
336 |
+ if url:
|
|
337 |
+ self.mark_download_url(url, primary=False)
|
|
338 |
+ |
|
330 | 339 |
self.submodule_overrides[path] = url
|
331 | 340 |
if 'checkout' in submodule:
|
332 | 341 |
checkout = self.node_get_member(submodule, bool, 'checkout')
|
... | ... | @@ -28,6 +28,18 @@ Abstract Methods |
28 | 28 |
For loading and configuration purposes, Sources must implement the
|
29 | 29 |
:ref:`Plugin base class abstract methods <core_plugin_abstract_methods>`.
|
30 | 30 |
|
31 |
+.. attention::
|
|
32 |
+ |
|
33 |
+ In order to ensure that all configuration data is processed at
|
|
34 |
+ load time, it is important that all URLs have been processed during
|
|
35 |
+ :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>`.
|
|
36 |
+ |
|
37 |
+ Source implementations *must* either call
|
|
38 |
+ :func:`Source.translate_url() <buildstream.source.Source.translate_url>` or
|
|
39 |
+ :func:`Source.mark_download_url() <buildstream.source.Source.mark_download_url>`
|
|
40 |
+ for every URL that has been specified in the configuration during
|
|
41 |
+ :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>`
|
|
42 |
+ |
|
31 | 43 |
Sources expose the following abstract methods. Unless explicitly mentioned,
|
32 | 44 |
these methods are mandatory to implement.
|
33 | 45 |
|
... | ... | @@ -184,6 +196,13 @@ class SourceFetcher(): |
184 | 196 |
fetching and substituting aliases.
|
185 | 197 |
|
186 | 198 |
*Since: 1.2*
|
199 |
+ |
|
200 |
+ .. attention::
|
|
201 |
+ |
|
202 |
+ When implementing a SourceFetcher, remember to call
|
|
203 |
+ :func:`Source.mark_download_url() <buildstream.source.Source.mark_download_url>`
|
|
204 |
+ for every URL found in the configuration data at
|
|
205 |
+ :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>` time.
|
|
187 | 206 |
"""
|
188 | 207 |
def __init__(self):
|
189 | 208 |
self.__alias = None
|
... | ... | @@ -206,7 +225,7 @@ class SourceFetcher(): |
206 | 225 |
Implementors should raise :class:`.SourceError` if the there is some
|
207 | 226 |
network error or if the source reference could not be matched.
|
208 | 227 |
"""
|
209 |
- raise ImplError("Source fetcher '{}' does not implement fetch()".format(type(self)))
|
|
228 |
+ raise ImplError("SourceFetcher '{}' does not implement fetch()".format(type(self)))
|
|
210 | 229 |
|
211 | 230 |
#############################################################
|
212 | 231 |
# Public Methods #
|
... | ... | @@ -277,8 +296,11 @@ class Source(Plugin): |
277 | 296 |
self.__element_kind = meta.element_kind # The kind of the element owning this source
|
278 | 297 |
self.__directory = meta.directory # Staging relative directory
|
279 | 298 |
self.__consistency = Consistency.INCONSISTENT # Cached consistency state
|
299 |
+ |
|
300 |
+ # The alias_override is only set on a re-instantiated Source
|
|
280 | 301 |
self.__alias_override = alias_override # Tuple of alias and its override to use instead
|
281 |
- self.__expected_alias = None # A hacky way to store the first alias used
|
|
302 |
+ self.__expected_alias = None # The primary alias
|
|
303 |
+ self.__marked_urls = set() # Set of marked download URLs
|
|
282 | 304 |
|
283 | 305 |
# FIXME: Reconstruct a MetaSource from a Source instead of storing it.
|
284 | 306 |
self.__meta = meta # MetaSource stored so we can copy this source later.
|
... | ... | @@ -289,7 +311,7 @@ class Source(Plugin): |
289 | 311 |
self.__config = self.__extract_config(meta)
|
290 | 312 |
self.__first_pass = meta.first_pass
|
291 | 313 |
|
292 |
- self.configure(self.__config)
|
|
314 |
+ self._configure(self.__config)
|
|
293 | 315 |
|
294 | 316 |
COMMON_CONFIG_KEYS = ['kind', 'directory']
|
295 | 317 |
"""Common source config keys
|
... | ... | @@ -351,10 +373,10 @@ class Source(Plugin): |
351 | 373 |
Args:
|
352 | 374 |
ref (simple object): The internal source reference to set, or ``None``
|
353 | 375 |
node (dict): The same dictionary which was previously passed
|
354 |
- to :func:`~buildstream.source.Source.configure`
|
|
376 |
+ to :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>`
|
|
355 | 377 |
|
356 |
- See :func:`~buildstream.source.Source.get_ref` for a discussion on
|
|
357 |
- the *ref* parameter.
|
|
378 |
+ See :func:`Source.get_ref() <buildstream.source.Source.get_ref>`
|
|
379 |
+ for a discussion on the *ref* parameter.
|
|
358 | 380 |
|
359 | 381 |
.. note::
|
360 | 382 |
|
... | ... | @@ -384,8 +406,8 @@ class Source(Plugin): |
384 | 406 |
backend store allows one to query for a new ref from a symbolic
|
385 | 407 |
tracking data without downloading then that is desirable.
|
386 | 408 |
|
387 |
- See :func:`~buildstream.source.Source.get_ref` for a discussion on
|
|
388 |
- the *ref* parameter.
|
|
409 |
+ See :func:`Source.get_ref() <buildstream.source.Source.get_ref>`
|
|
410 |
+ for a discussion on the *ref* parameter.
|
|
389 | 411 |
"""
|
390 | 412 |
# Allow a non implementation
|
391 | 413 |
return None
|
... | ... | @@ -435,7 +457,7 @@ class Source(Plugin): |
435 | 457 |
:class:`.SourceError`
|
436 | 458 |
|
437 | 459 |
Default implementation is to call
|
438 |
- :func:`~buildstream.source.Source.stage`.
|
|
460 |
+ :func:`Source.stage() <buildstream.source.Source.stage>`.
|
|
439 | 461 |
|
440 | 462 |
Implementors overriding this method should assume that *directory*
|
441 | 463 |
already exists.
|
... | ... | @@ -453,8 +475,15 @@ class Source(Plugin): |
453 | 475 |
is recommended.
|
454 | 476 |
|
455 | 477 |
Returns:
|
456 |
- list: A list of SourceFetchers. If SourceFetchers are not supported,
|
|
457 |
- this will be an empty list.
|
|
478 |
+ iterable: The Source's SourceFetchers, if any.
|
|
479 |
+ |
|
480 |
+ .. note::
|
|
481 |
+ |
|
482 |
+ Implementors can implement this as a generator.
|
|
483 |
+ |
|
484 |
+ The :func:`SourceFetcher.fetch() <buildstream.source.SourceFetcher.fetch>`
|
|
485 |
+ method will be called on the returned fetchers one by one,
|
|
486 |
+ before consuming the next fetcher in the list.
|
|
458 | 487 |
|
459 | 488 |
*Since: 1.2*
|
460 | 489 |
"""
|
... | ... | @@ -477,17 +506,27 @@ class Source(Plugin): |
477 | 506 |
os.makedirs(directory, exist_ok=True)
|
478 | 507 |
return directory
|
479 | 508 |
|
480 |
- def translate_url(self, url, *, alias_override=None):
|
|
509 |
+ def translate_url(self, url, *, alias_override=None, primary=True):
|
|
481 | 510 |
"""Translates the given url which may be specified with an alias
|
482 | 511 |
into a fully qualified url.
|
483 | 512 |
|
484 | 513 |
Args:
|
485 |
- url (str): A url, which may be using an alias
|
|
514 |
+ url (str): A URL, which may be using an alias
|
|
486 | 515 |
alias_override (str): Optionally, an URI to override the alias with. (*Since: 1.2*)
|
516 |
+ primary (bool): Whether this is the primary URL for the source. (*Since: 1.2*)
|
|
487 | 517 |
|
488 | 518 |
Returns:
|
489 |
- str: The fully qualified url, with aliases resolved
|
|
519 |
+ str: The fully qualified URL, with aliases resolved
|
|
520 |
+ .. note::
|
|
521 |
+ |
|
522 |
+ This must be called for every URL in the configuration during
|
|
523 |
+ :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>` if
|
|
524 |
+ :func:`Source.mark_download_url() <buildstream.source.Source.mark_download_url>`
|
|
525 |
+ is not called.
|
|
490 | 526 |
"""
|
527 |
+ # Ensure that the download URL is also marked
|
|
528 |
+ self.mark_download_url(url, primary=primary)
|
|
529 |
+ |
|
491 | 530 |
# Alias overriding can happen explicitly (by command-line) or
|
492 | 531 |
# implicitly (the Source being constructed with an __alias_override).
|
493 | 532 |
if alias_override or self.__alias_override:
|
... | ... | @@ -506,25 +545,55 @@ class Source(Plugin): |
506 | 545 |
url = override_url + url_body
|
507 | 546 |
return url
|
508 | 547 |
else:
|
509 |
- # Sneakily store the alias if it hasn't already been stored
|
|
510 |
- if not self.__expected_alias and url and utils._ALIAS_SEPARATOR in url:
|
|
511 |
- self.mark_download_url(url)
|
|
512 |
- |
|
513 | 548 |
project = self._get_project()
|
514 | 549 |
return project.translate_url(url, first_pass=self.__first_pass)
|
515 | 550 |
|
516 |
- def mark_download_url(self, url):
|
|
551 |
+ def mark_download_url(self, url, *, primary=True):
|
|
517 | 552 |
"""Identifies the URL that this Source uses to download
|
518 | 553 |
|
519 |
- This must be called during :func:`~buildstream.plugin.Plugin.configure` if
|
|
520 |
- :func:`~buildstream.source.Source.translate_url` is not called.
|
|
521 |
- |
|
522 | 554 |
Args:
|
523 |
- url (str): The url used to download
|
|
555 |
+ url (str): The URL used to download
|
|
556 |
+ primary (bool): Whether this is the primary URL for the source
|
|
557 |
+ |
|
558 |
+ .. note::
|
|
559 |
+ |
|
560 |
+ This must be called for every URL in the configuration during
|
|
561 |
+ :func:`Plugin.configure() <buildstream.plugin.Plugin.configure>` if
|
|
562 |
+ :func:`Source.translate_url() <buildstream.source.Source.translate_url>`
|
|
563 |
+ is not called.
|
|
524 | 564 |
|
525 | 565 |
*Since: 1.2*
|
526 | 566 |
"""
|
527 |
- self.__expected_alias = _extract_alias(url)
|
|
567 |
+ # Only mark the Source level aliases on the main instance, not in
|
|
568 |
+ # a reinstantiated instance in mirroring.
|
|
569 |
+ if not self.__alias_override:
|
|
570 |
+ if primary:
|
|
571 |
+ expected_alias = _extract_alias(url)
|
|
572 |
+ |
|
573 |
+ assert (self.__expected_alias is None or
|
|
574 |
+ self.__expected_alias == expected_alias), \
|
|
575 |
+ "Primary URL marked twice with different URLs"
|
|
576 |
+ |
|
577 |
+ self.__expected_alias = expected_alias
|
|
578 |
+ |
|
579 |
+ # Enforce proper behaviour of plugins by ensuring that all
|
|
580 |
+ # aliased URLs have been marked at Plugin.configure() time.
|
|
581 |
+ #
|
|
582 |
+ if self._get_configuring():
|
|
583 |
+ # Record marked urls while configuring
|
|
584 |
+ #
|
|
585 |
+ self.__marked_urls.add(url)
|
|
586 |
+ else:
|
|
587 |
+ # If an unknown aliased URL is seen after configuring,
|
|
588 |
+ # this is an error.
|
|
589 |
+ #
|
|
590 |
+ # It is still possible that a URL that was not mentioned
|
|
591 |
+ # in the element configuration can be marked, this is
|
|
592 |
+ # the case for git submodules which might be automatically
|
|
593 |
+ # discovered.
|
|
594 |
+ #
|
|
595 |
+ assert (url in self.__marked_urls or not _extract_alias(url)), \
|
|
596 |
+ "URL was not seen at configure time: {}".format(url)
|
|
528 | 597 |
|
529 | 598 |
def get_project_directory(self):
|
530 | 599 |
"""Fetch the project base directory
|
... | ... | @@ -15,14 +15,17 @@ from buildstream import Source, Consistency, SourceError, SourceFetcher |
15 | 15 |
|
16 | 16 |
|
17 | 17 |
class FetchFetcher(SourceFetcher):
|
18 |
- def __init__(self, source, url):
|
|
18 |
+ def __init__(self, source, url, primary=False):
|
|
19 | 19 |
super().__init__()
|
20 | 20 |
self.source = source
|
21 | 21 |
self.original_url = url
|
22 |
+ self.primary = primary
|
|
22 | 23 |
self.mark_download_url(url)
|
23 | 24 |
|
24 | 25 |
def fetch(self, alias_override=None):
|
25 |
- url = self.source.translate_url(self.original_url, alias_override=alias_override)
|
|
26 |
+ url = self.source.translate_url(self.original_url,
|
|
27 |
+ alias_override=alias_override,
|
|
28 |
+ primary=self.primary)
|
|
26 | 29 |
with open(self.source.output_file, "a") as f:
|
27 | 30 |
success = url in self.source.fetch_succeeds and self.source.fetch_succeeds[url]
|
28 | 31 |
message = "Fetch {} {} from {}\n".format(self.original_url,
|
... | ... | @@ -37,12 +40,21 @@ class FetchSource(Source): |
37 | 40 |
# Read config to know which URLs to fetch
|
38 | 41 |
def configure(self, node):
|
39 | 42 |
self.original_urls = self.node_get_member(node, list, 'urls')
|
40 |
- self.fetchers = [FetchFetcher(self, url) for url in self.original_urls]
|
|
41 | 43 |
self.output_file = self.node_get_member(node, str, 'output-text')
|
42 | 44 |
self.fetch_succeeds = {}
|
43 | 45 |
if 'fetch-succeeds' in node:
|
44 | 46 |
self.fetch_succeeds = {x[0]: x[1] for x in self.node_items(node['fetch-succeeds'])}
|
45 | 47 |
|
48 |
+ # First URL is the primary one for this test
|
|
49 |
+ #
|
|
50 |
+ primary = True
|
|
51 |
+ self.fetchers = []
|
|
52 |
+ for url in self.original_urls:
|
|
53 |
+ self.mark_download_url(url, primary=primary)
|
|
54 |
+ fetcher = FetchFetcher(self, url, primary=primary)
|
|
55 |
+ self.fetchers.append(fetcher)
|
|
56 |
+ primary = False
|
|
57 |
+ |
|
46 | 58 |
def get_source_fetchers(self):
|
47 | 59 |
return self.fetchers
|
48 | 60 |
|