Bst artifact subcommands take artifact ref names.becky/artifact_access

Changes made to allow artifact checkout/push/pull to take
artifact references as well as element names.
Added test to ensure this works

4 files changed, 114 insertions, 46 deletions

diff --git a/src/buildstream/_artifact.py b/src/buildstream/_artifact.py
index b0c09622f..493ca5d26 100644
--- a/src/buildstream/_artifact.py
+++ b/src/buildstream/_artifact.py
@@ -75,7 +75,6 @@ class Artifact():
     #
     def get_files(self):
         files_digest = self._get_field_digest("files")
-
         return CasBasedDirectory(self._cas, digest=files_digest)
 
     # get_buildtree():
diff --git a/src/buildstream/_frontend/cli.py b/src/buildstream/_frontend/cli.py
index bf92161bf..12b9439cf 100644
--- a/src/buildstream/_frontend/cli.py
+++ b/src/buildstream/_frontend/cli.py
@@ -948,7 +948,7 @@ def workspace_list(app):
 #############################################################
 #                     Artifact Commands                     #
 #############################################################
-@cli.group(short_help="Manipulate cached artifacts")
+@cli.group(short_help="Manipulate cached artifacts.")
 def artifact():
     """Manipulate cached artifacts"""
 
diff --git a/src/buildstream/_stream.py b/src/buildstream/_stream.py
index 0f320c569..28002c0d2 100644
--- a/src/buildstream/_stream.py
+++ b/src/buildstream/_stream.py
@@ -31,7 +31,8 @@ import tempfile
 from contextlib import contextmanager, suppress
 from fnmatch import fnmatch
 
-from ._artifactelement import verify_artifact_ref
+from ._artifact import Artifact
+from ._artifactelement import verify_artifact_ref, ArtifactElement
 from ._exceptions import StreamError, ImplError, BstError, ArtifactElementError, ArtifactError
 from ._message import Message, MessageType
 from ._scheduler import Scheduler, SchedStatus, TrackQueue, FetchQueue, \
@@ -518,8 +519,7 @@ class Stream():
 
         # if pulling we need to ensure dependency artifacts are also pulled
         selection = PipelineSelection.RUN if pull else PipelineSelection.NONE
-        elements, _ = self._load((target,), (), selection=selection, use_artifact_config=True)
-
+        elements, _ = self._load((target,), (), selection=selection, use_artifact_config=True, load_refs=True)
         target = elements[-1]
 
         self._check_location_writable(location, force=force, tar=tar)
@@ -533,48 +533,58 @@ class Stream():
             self._run()
 
         # Stage deps into a temporary sandbox first
-        try:
-            with target._prepare_sandbox(scope=scope, directory=None,
-                                         integrate=integrate) as sandbox:
-
-                # Copy or move the sandbox to the target directory
-                sandbox_vroot = sandbox.get_virtual_directory()
-
-                if not tar:
-                    with target.timed_activity("Checking out files in '{}'"
-                                               .format(location)):
-                        try:
-                            if hardlinks:
-                                self._checkout_hardlinks(sandbox_vroot, location)
-                            else:
-                                sandbox_vroot.export_files(location)
-                        except OSError as e:
-                            raise StreamError("Failed to checkout files: '{}'"
-                                              .format(e)) from e
-                else:
-                    if location == '-':
-                        mode = 'w|' + compression
-                        with target.timed_activity("Creating tarball"):
-                            # Save the stdout FD to restore later
-                            saved_fd = os.dup(sys.stdout.fileno())
-                            try:
-                                with os.fdopen(sys.stdout.fileno(), 'wb') as fo:
-                                    with tarfile.open(fileobj=fo, mode=mode) as tf:
-                                        sandbox_vroot.export_to_tar(tf, '.')
-                            finally:
-                                # No matter what, restore stdout for further use
-                                os.dup2(saved_fd, sys.stdout.fileno())
-                                os.close(saved_fd)
+        if isinstance(target, ArtifactElement):
+            try:
+                key = target._get_cache_key()
+                artifact = Artifact(target, self._context, strong_key=key)
+                virdir = artifact.get_files()
+                self._export_artifact(tar, location, compression, target, hardlinks, virdir)
+            except AttributeError as e:
+                raise ArtifactError("Artifact reference '{}' seems to be invalid. "
+                                    "Note that an Element name can also be used.".format(artifact))from e
+        else:
+            try:
+                with target._prepare_sandbox(scope=scope, directory=None,
+                                             integrate=integrate) as sandbox:
+                    # Copy or move the sandbox to the target directory
+                    virdir = sandbox.get_virtual_directory()
+                    self._export_artifact(tar, location, compression, target, hardlinks, virdir)
+            except BstError as e:
+                raise StreamError("Error while staging dependencies into a sandbox"
+                                  ": '{}'".format(e), detail=e.detail, reason=e.reason) from e
+
+    def _export_artifact(self, tar, location, compression, target, hardlinks, virdir):
+        if not tar:
+            with target.timed_activity("Checking out files in '{}'"
+                                       .format(location)):
+                try:
+                    if hardlinks:
+                        self._checkout_hardlinks(virdir, location)
                     else:
-                        mode = 'w:' + compression
-                        with target.timed_activity("Creating tarball '{}'"
-                                                   .format(location)):
-                            with tarfile.open(location, mode=mode) as tf:
-                                sandbox_vroot.export_to_tar(tf, '.')
-
-        except BstError as e:
-            raise StreamError("Error while staging dependencies into a sandbox"
-                              ": '{}'".format(e), detail=e.detail, reason=e.reason) from e
+                        virdir.export_files(location)
+                except OSError as e:
+                    raise StreamError("Failed to checkout files: '{}'"
+                                      .format(e)) from e
+        else:
+            if location == '-':
+                mode = 'w|' + compression
+                with target.timed_activity("Creating tarball"):
+                    # Save the stdout FD to restore later
+                    saved_fd = os.dup(sys.stdout.fileno())
+                    try:
+                        with os.fdopen(sys.stdout.fileno(), 'wb') as fo:
+                            with tarfile.open(fileobj=fo, mode=mode) as tf:
+                                virdir.export_to_tar(tf, '.')
+                    finally:
+                        # No matter what, restore stdout for further use
+                        os.dup2(saved_fd, sys.stdout.fileno())
+                        os.close(saved_fd)
+            else:
+                mode = 'w:' + compression
+                with target.timed_activity("Creating tarball '{}'"
+                                           .format(location)):
+                    with tarfile.open(location, mode=mode) as tf:
+                        virdir.export_to_tar(tf, '.')
 
     # artifact_log()
     #
diff --git a/tests/frontend/buildcheckout.py b/tests/frontend/buildcheckout.py
index dd4a461ea..50b2daf4f 100644
--- a/tests/frontend/buildcheckout.py
+++ b/tests/frontend/buildcheckout.py
@@ -303,6 +303,65 @@ def test_build_checkout_tarball(datafiles, cli):
 
 
 @pytest.mark.datafiles(DATA_DIR)
+def test_build_checkout_using_ref(datafiles, cli):
+    project = str(datafiles)
+    checkout = os.path.join(cli.directory, 'checkout')
+
+    result = cli.run(project=project, args=['build', 'checkout-deps.bst'])
+    result.assert_success()
+
+    checkout_args = ['artifact', 'checkout', '--directory', checkout,
+                     'test/checkout-deps/602d6859b2e4bdc2af7f3ff34113c4e49219291c0b5c3be8642f85e94322ce49']
+
+    result = cli.run(project=project, args=checkout_args)
+    result.assert_success()
+
+    filename = os.path.join(checkout, 'etc', 'buildstream', 'config')
+    assert os.path.exists(filename)
+
+
+@pytest.mark.datafiles(DATA_DIR)
+def test_build_checkout_tarball_using_ref(datafiles, cli):
+    project = str(datafiles)
+    checkout = os.path.join(cli.directory, 'checkout.tar')
+
+    result = cli.run(project=project, args=['build', 'checkout-deps.bst'])
+    result.assert_success()
+
+    builddir = os.path.join(cli.directory, 'build')
+    assert os.path.isdir(builddir)
+    assert not os.listdir(builddir)
+
+    checkout_args = ['artifact', 'checkout', '--deps', 'none', '--tar', checkout,
+                     'test/checkout-deps/602d6859b2e4bdc2af7f3ff34113c4e49219291c0b5c3be8642f85e94322ce49']
+
+    result = cli.run(project=project, args=checkout_args)
+    result.assert_success()
+
+    tar = tarfile.TarFile(checkout)
+    assert os.path.join('.', 'etc', 'buildstream', 'config') in tar.getnames()
+
+
+@pytest.mark.datafiles(DATA_DIR)
+def test_build_checkout_invalid_ref(datafiles, cli):
+    project = str(datafiles)
+    checkout = os.path.join(cli.directory, 'checkout.tar')
+
+    result = cli.run(project=project, args=['build', 'checkout-deps.bst'])
+    result.assert_success()
+
+    builddir = os.path.join(cli.directory, 'build')
+    assert os.path.isdir(builddir)
+    assert not os.listdir(builddir)
+
+    checkout_args = ['artifact', 'checkout', '--deps', 'none', '--tar', checkout,
+                     'test/checkout-deps/602d6859b2e4bdc2af7f3ff34113c4e49219291c0b5c3be8642f85e94322ce48']
+
+    result = cli.run(project=project, args=checkout_args)
+    assert "seems to be invalid. Note that an Element name can also be used." in result.stderr
+
+
+@pytest.mark.datafiles(DATA_DIR)
 def test_build_checkout_no_tar_no_directory(datafiles, cli, tmpdir):
     project = str(datafiles)
     runtestdir = str(tmpdir)