Added new compose element

This creates a selective composition of its dependencies.

2 files changed, 186 insertions, 0 deletions

diff --git a/buildstream/plugins/elements/compose.py b/buildstream/plugins/elements/compose.py
new file mode 100644
index 000000000..af6fe68e3
--- /dev/null
+++ b/buildstream/plugins/elements/compose.py
@@ -0,0 +1,160 @@
+#!/usr/bin/env python3
+#
+#  Copyright (C) 2017 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>
+
+"""Compose element
+
+This element creates a selective composition of it's dependencies.
+
+This is normally used at near the end of a pipeline to prepare
+something for later deployment.
+
+Since this element's output includes all of its input, it may only
+depend on its dependencies as `build` type dependencies.
+
+The default configuration and possible options are as such:
+  .. literalinclude:: ../../../buildstream/plugins/elements/compose.yaml
+     :language: yaml
+"""
+
+import os
+import re
+from buildstream import utils
+from buildstream import Element, ElementError, Scope
+
+
+# Element implementation for the 'compose' kind.
+class ComposeElement(Element):
+
+    def configure(self, node):
+        # We name this variable 'integration' only to avoid
+        # collision with the Element.integrate() method.
+        self.integration = self.node_get_member(node, bool, 'integrate')
+        self.include = self.node_get_member(node, list, 'include')
+        self.include_orphans = self.node_get_member(node, bool, 'include-orphans')
+
+    def preflight(self):
+        # Assert that the user did not list any runtime dependencies
+        runtime_deps = list(self.dependencies(Scope.RUN, recurse=False))
+        if runtime_deps:
+            raise ElementError("{}: Only build type dependencies supported by compose elements"
+                               .format(self))
+
+        # Assert that the user did not specify any sources, as they will
+        # be ignored by this element type anyway
+        sources = list(self.sources())
+        if sources:
+            raise ElementError("Compose elements may not have sources")
+
+    def get_unique_key(self):
+        # The output of this element will be effected depending
+        # on the splitting rules defined by the elements in this
+        # composition.
+        #
+        # As such, we include the split rules themselves in the
+        # cache key calculation.
+        #
+        include_rules = [
+            {
+                'element': elt.name,
+                'splits': [
+                    {
+                        'domain': domain,
+                        'rules': rules
+                    }
+                    for domain, rules in sorted(self.splits(elt))
+                ]
+            }
+            for elt in self.dependencies(Scope.BUILD)
+        ]
+        return {
+            'integrate': self.integration,
+            'include': include_rules,
+            'orphans': self.include_orphans
+        }
+
+    def assemble(self, sandbox):
+
+        # Stage deps in the sandbox root
+        with self.timed_activity("Staging dependencies", silent_nested=True):
+            for dep in self.dependencies(Scope.BUILD):
+                dep.stage(sandbox)
+
+        # Make a snapshot of all the files.
+        basedir = sandbox.get_directory()
+        snapshot = list(utils.list_relative_paths(basedir))
+        manifest = []
+
+        # Run any integration commands provided by the dependencies
+        # once they are all staged and ready
+        if self.integration:
+            with self.timed_activity("Integrating sandbox", silent_nested=True):
+
+                for dep in self.dependencies(Scope.BUILD):
+                    dep.integrate(sandbox)
+
+                integration_files = [
+                    path for path in utils.list_relative_paths(basedir)
+                    if path not in snapshot
+                ]
+                self.info("Integration produced {} new files".format(len(integration_files)))
+
+        manifest += integration_files
+
+        # The remainder of this is expensive, make an early exit if
+        # we're not being selective about what is to be included.
+        if not self.include and self.include_orphans:
+            return '/'
+
+        # XXX We should be moving things outside of the build sandbox
+        # instead of into a subdir. The element assemble() method should
+        # support this in some way.
+        #
+        installdir = os.path.join(basedir, 'buildstream', 'install')
+        stagedir = os.path.join(os.sep, 'buildstream', 'install')
+        os.makedirs(installdir, exist_ok=True)
+
+        # We already saved the manifest for created files in the integration phase,
+        # now collect the rest of the manifest.
+        #
+        with self.timed_activity("Creating composition", silent_nested=True):
+            for elt in self.dependencies(Scope.BUILD):
+                elt.stage(sandbox, path=stagedir,
+                          splits=self.include,
+                          orphans=self.include_orphans)
+
+            self.status("Moving {} integration files".format(len(integration_files)))
+            utils.move_files(basedir, installdir, integration_files)
+
+        # And we're done
+        return os.path.join(os.sep, 'buildstream', 'install')
+
+    # Generator for extracting the split rules to be included
+    # for a given element (which should be in the dependency chain)
+    def splits(self, element):
+        bstdata = element.get_public_data('bst')
+        splits = bstdata.get('split-rules')
+        for domain, rules in self.node_items(splits):
+            if not self.include or domain in self.include:
+                yield (domain, rules)
+
+
+# Plugin entry point
+def setup():
+    return ComposeElement
diff --git a/buildstream/plugins/elements/compose.yaml b/buildstream/plugins/elements/compose.yaml
new file mode 100644
index 000000000..45ffd4e24
--- /dev/null
+++ b/buildstream/plugins/elements/compose.yaml
@@ -0,0 +1,26 @@
+
+# Compose element configuration
+config:
+
+  # Whether to run the integration commands for the
+  # staged dependencies.
+  #
+  integrate: True
+
+  # A list of domains to include from each artifact, as
+  # they were defined in the element's 'split-rules'.
+  #
+  # Since domains can be added, it is not an error to
+  # specify domains which may not exist for all of the
+  # elements in this composition.
+  #
+  # The default empty list indicates that all domains
+  # from each dependency should be included.
+  #
+  include: []
+
+  # Whether to include orphan files which are not
+  # included by any of the 'split-rules' present on
+  # a given element.
+  #
+  include-orphans: True