From 817ed2be0600c51a711fa1d105fbd1f4219cd43b Mon Sep 17 00:00:00 2001
From: Doug Hellmann <doug@doughellmann.com>
Date: Wed, 3 May 2017 20:27:46 +0000
Subject: [PATCH] add support for available volumes

Create an image from the volume and download the image.

Change-Id: Idc1c44c74f43f8d9b259e9da4895c94f25464973
Signed-off-by: Doug Hellmann <doug@doughellmann.com>
---
 downpour/resolver.py | 39 ++++++++++++++++++++++++++-------------
 1 file changed, 26 insertions(+), 13 deletions(-)

diff --git a/downpour/resolver.py b/downpour/resolver.py
index 224ecf9..eaa64e5 100644
--- a/downpour/resolver.py
+++ b/downpour/resolver.py
@@ -13,6 +13,9 @@
 # under the License.
 
 import itertools
+import logging
+
+LOG = logging.getLogger(__name__)
 
 
 class Resolver:
@@ -122,22 +125,32 @@ class Resolver:
         if ('volume', volume.name) in self._memo:
             return
         self._memo.add(('volume', volume.name))
-        # FIXME(dhellmann): For now this only creates new empty
-        # volumes, and doesn't handle cases like booting from a volume
-        # or creating a volume from an image.
-        #
-        # FIXME(dhellmann): Need to snapshot the volume and then
-        # download the results.
         if save_state:
-            self._downloader.add_volume(volume)
+            image_name = 'volume-capture-{}'.format(volume.id)
+            image_info = self.cloud.get_image(image_name)
+            if image_info:
+                LOG.info('found existing capture of volume %s in %s',
+                         volume.name, image_name)
+            else:
+                LOG.info('creating image from volume %s as %s',
+                         volume.name, image_name)
+                image_info = self.cloud.create_image(
+                    name=image_name,
+                    volume=volume.id,
+                    wait=True,
+                )
+            yield from self.image(image_info)
+        os_volume = {
+            'display_name': volume.display_name,
+            'display_description': volume.display_description,
+            'size': volume.size,
+            'state': 'present',
+        }
+        if save_state:
+            os_volume['image'] = image_name
         yield {
             'name': 'Create volume {}'.format(volume.name),
-            'os_volume': {
-                'display_name': volume.display_name,
-                'display_description': volume.display_description,
-                'size': volume.size,
-                'state': 'present',
-            },
+            'os_volume': os_volume,
             'register': self._mk_var_name('vol'),
         }
         yield self._map_uuids('volume', volume.name, volume.id, 'volume.id')