blob: 93594b62309c41d2070698c76b927473759e2568 [file] [log] [blame]
#
# 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>
"""
import - Import sources directly
================================
Import elements produce artifacts directly from its sources
without any kind of processing. These are typically used to
import an SDK to build on top of or to overlay your build with
some configuration data.
The empty configuration is as such:
.. literalinclude:: ../../../buildstream/plugins/elements/import.yaml
:language: yaml
"""
import os
import shutil
from buildstream import Element, BuildElement, ElementError
# Element implementation for the 'import' kind.
class ImportElement(BuildElement):
# pylint: disable=attribute-defined-outside-init
def configure(self, node):
self.source = self.node_subst_member(node, 'source')
self.target = self.node_subst_member(node, 'target')
def preflight(self):
# Assert that we have at least one source to fetch.
sources = list(self.sources())
if not sources:
raise ElementError("{}: An import element must have at least one source.".format(self))
def get_unique_key(self):
return {
'source': self.source,
'target': self.target
}
def configure_sandbox(self, sandbox):
pass
def stage(self, sandbox):
pass
def assemble(self, sandbox):
# Stage sources into the input directory
# Do not mount workspaces as the files are copied from outside the sandbox
self._stage_sources_in_sandbox(sandbox, 'input', mount_workspaces=False)
rootdir = sandbox.get_directory()
inputdir = os.path.join(rootdir, 'input')
outputdir = os.path.join(rootdir, 'output')
# The directory to grab
inputdir = os.path.join(inputdir, self.source.lstrip(os.sep))
inputdir = inputdir.rstrip(os.sep)
# The output target directory
outputdir = os.path.join(outputdir, self.target.lstrip(os.sep))
outputdir = outputdir.rstrip(os.sep)
# Ensure target directory parent
os.makedirs(os.path.dirname(outputdir), exist_ok=True)
if not os.path.exists(inputdir):
raise ElementError("{}: No files were found inside directory '{}'"
.format(self, self.source))
# Move it over
shutil.move(inputdir, outputdir)
# And we're done
return '/output'
def prepare(self, sandbox):
# We inherit a non-default prepare from BuildElement.
Element.prepare(self, sandbox)
def generate_script(self):
build_root = self.get_variable('build-root')
install_root = self.get_variable('install-root')
commands = []
# The directory to grab
inputdir = os.path.join(build_root, self.normal_name, self.source.lstrip(os.sep))
inputdir = inputdir.rstrip(os.sep)
# The output target directory
outputdir = os.path.join(install_root, self.target.lstrip(os.sep))
outputdir = outputdir.rstrip(os.sep)
# Ensure target directory parent exists but target directory doesn't
commands.append("mkdir -p {}".format(os.path.dirname(outputdir)))
commands.append("[ ! -e {} ] || rmdir {}".format(outputdir, outputdir))
# Move it over
commands.append("mv {} {}".format(inputdir, outputdir))
script = ""
for cmd in commands:
script += "(set -ex; {}\n) || exit 1\n".format(cmd)
return script
# Plugin entry point
def setup():
return ImportElement