From b55a9606e9455056a4280b06ef3785964af6d3df Mon Sep 17 00:00:00 2001
From: Ben Howard <ben.howard@canonical.com>
Date: Thu, 26 Sep 2013 08:18:29 -0600
Subject: Added support for formating the ephemeral disk on Windows Azure.

---
 cloudinit/sources/DataSourceAzure.py | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

(limited to 'cloudinit/sources/DataSourceAzure.py')

diff --git a/cloudinit/sources/DataSourceAzure.py b/cloudinit/sources/DataSourceAzure.py
index a77c3d9a..3f96a0a5 100644
--- a/cloudinit/sources/DataSourceAzure.py
+++ b/cloudinit/sources/DataSourceAzure.py
@@ -44,8 +44,17 @@ BUILTIN_DS_CONFIG = {
         'policy': True,
         'command': BOUNCE_COMMAND,
         'hostname_command': 'hostname',
-    }
+        },
+    'ephemeral_disk': '/dev/sdb1',
+    'disk_setup': {
+        'ephemeral0': {'table_type': 'mbr',
+                       'layout': True,
+                       'overwrite': False}
+         },
+    'fs_setup': [{'filesystem': 'ext4', 'device': '/dev/sdb',
+                  'partition': 'auto'}],
 }
+
 DS_CFG_PATH = ['datasource', DS_NAME]
 
 
@@ -111,11 +120,22 @@ class DataSourceAzureNet(sources.DataSource):
         if seed:
             self.metadata['random_seed'] = seed
 
+
         # now update ds_cfg to reflect contents pass in config
         usercfg = util.get_cfg_by_path(self.cfg, DS_CFG_PATH, {})
         self.ds_cfg = util.mergemanydict([usercfg, self.ds_cfg])
         mycfg = self.ds_cfg
 
+        # Put the disk format elements into the config object
+        if 'disk_setup' in mycfg:
+            self.cfg['disk_setup'] = mycfg['disk_setup']
+
+        if 'fs_setup' in mycfg:
+            self.cfg['fs_setup'] = mycfg['fs_setup']
+
+        if 'ephemeral_disk' in mycfg:
+            self.cfg['ephemeral_disk'] = mycfg['ephemeral_disk']
+
         # walinux agent writes files world readable, but expects
         # the directory to be protected.
         write_files(mycfg['data_dir'], files, dirmode=0700)
@@ -161,9 +181,12 @@ class DataSourceAzureNet(sources.DataSource):
         pubkeys = pubkeys_from_crt_files(fp_files)
 
         self.metadata['public-keys'] = pubkeys
-
         return True
 
+    def device_name_to_device(self, name):
+        if 'ephemeral0' in name and 'ephemeral_disk' in self.cfg:
+            return self.cfg['ephemeral_disk']
+
     def get_config_obj(self):
         return self.cfg
 
-- 
cgit v1.2.3