morpheus-cli 0.9.9 → 0.9.10

data/lib/morpheus/cli/mixins/accounts_helper.rb

@@ -1,8 +1,15 @@
+require 'table_print'
+require 'morpheus/cli/mixins/print_helper'
+
 # Mixin for Morpheus::Cli command classes 
 # Provides common methods for fetching and printing accounts, roles, and users.
 # The including class must establish @accounts_interface, @roles_interface, @users_interface
 module Morpheus::Cli::AccountsHelper
 
+  def self.included(klass)
+    klass.include Morpheus::Cli::PrintHelper
+  end
+
   def find_account_by_id(id)
     raise "#{self.class} has not defined @accounts_interface" if @accounts_interface.nil?
     begin
@@ -182,12 +189,4 @@ module Morpheus::Cli::AccountsHelper
     print reset
   end
 
-  def print_red_alert(msg)
-    print red, bold, "\n#{msg}\n\n", reset
-  end
-
-  def print_green_success(msg)
-    print green, bold, "\n#{msg}\n\n", reset
-  end
-
 end

data/lib/morpheus/cli/mixins/print_helper.rb

@@ -0,0 +1,67 @@
+require 'term/ansicolor'
+require 'json'
+
+module Morpheus::Cli::PrintHelper
+
+  def self.included(klass)
+    klass.include Term::ANSIColor
+  end
+
+  def print_red_alert(msg)
+    print red, bold, "\n#{msg}\n\n", reset
+  end
+
+  def print_green_success(msg)
+    print green, bold, "\n#{msg}\n\n", reset
+  end
+
+  def print_errors(response, options={})
+    begin
+      if options[:json]
+        print red, "\n"
+        print JSON.pretty_generate(response)
+        print reset, "\n\n"
+      else
+        if !response['success']
+          print red,bold, "\n"
+          if response['msg']
+            puts response['msg']
+          end
+          if response['errors']
+            response['errors'].each do |key, value|
+              print "* #{key}: #{value}\n"
+            end
+          end
+          print reset, "\n"
+        else
+          # this should not really happen
+          print cyan,bold, "\nSuccess!"
+        end
+      end
+    ensure
+      print reset
+    end
+  end
+
+  def print_rest_exception(e, options={})
+    if e.response.code == 400
+      response = JSON.parse(e.response.to_s)
+      print_errors(response, options)
+    else
+      print_red_alert "Error Communicating with the Appliance. Please try again later. #{e}"
+      if options[:json]
+        begin
+          response = JSON.parse(e.response.to_s)
+          print red, "\n"
+          print JSON.pretty_generate(response)
+          print reset, "\n\n"
+        rescue TypeError, JSON::ParserError => ex
+          #print_red_alert "Failed to parse JSON response: #{ex}"
+        ensure
+          print reset
+        end
+      end
+    end
+  end
+
+end

data/lib/morpheus/cli/mixins/provisioning_helper.rb

@@ -0,0 +1,461 @@
+require 'morpheus/cli/mixins/print_helper'
+require 'morpheus/cli/option_types'
+# Mixin for Morpheus::Cli command classes 
+# Provides common methods for provisioning instances
+module Morpheus::Cli::ProvisioningHelper
+
+  def self.included(klass)
+    klass.include Morpheus::Cli::PrintHelper
+  end
+
+  # This recreates the behavior of multi_disk.js
+  # returns array of volumes based on service plan options (plan_info)
+  def prompt_instance_volumes(plan_info, options={}, api_client=nil, api_params={})
+    #puts "Configure Volumes:"
+    no_prompt = (options[:no_prompt] || (options[:options] && options[:options][:no_prompt]))
+
+    volumes = []
+
+    plan_size = nil
+    if plan_info['maxStorage']
+      plan_size = plan_info['maxStorage'].to_i / (1024 * 1024 * 1024)
+    end
+
+    root_storage_types = []
+    if plan_info['rootStorageTypes']
+      plan_info['rootStorageTypes'].each do |opt|
+        root_storage_types << {'name' => opt['name'], 'value' => opt['id']}
+      end
+    end
+
+    storage_types = []
+    if plan_info['storageTypes']
+      plan_info['storageTypes'].each do |opt|
+        storage_types << {'name' => opt['name'], 'value' => opt['id']}
+      end
+    end
+
+    datastore_options = []
+    if plan_info['supportsAutoDatastore']
+      if plan_info['autoOptions']
+        plan_info['autoOptions'].each do |opt|
+          datastore_options << {'name' => opt['name'], 'value' => opt['id']}
+        end
+      end
+    end
+    if plan_info['datastores']
+      plan_info['datastores'].each do |k, v|
+        v.each do |opt|
+          datastore_options << {'name' => "#{k}: #{opt['name']}", 'value' => opt['id']}
+        end
+      end
+    end
+
+    #puts "Configure Root Volume"
+
+    field_context = "rootVolume"
+
+    if root_storage_types.empty?
+      # this means there's no configuration, just send a single root volume to the server
+      storage_type_id = nil
+      storage_type = nil
+    else
+      v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'storageType', 'type' => 'select', 'fieldLabel' => 'Root Storage Type', 'selectOptions' => root_storage_types, 'required' => true, 'skipSingleOption' => true, 'description' => 'Choose a storage type.'}], options[:options])
+      storage_type_id = v_prompt[field_context]['storageType']
+      storage_type = plan_info['storageTypes'].find {|i| i['id'] == storage_type_id.to_i }
+    end
+
+    # sometimes the user chooses sizeId from a list of size options (AccountPrice) and other times it is free form
+    root_custom_size_options = []
+    if plan_info['rootCustomSizeOptions'] && plan_info['rootCustomSizeOptions'][storage_type_id.to_s]
+      plan_info['rootCustomSizeOptions'][storage_type_id.to_s].each do |opt|
+        root_custom_size_options << {'name' => opt['value'], 'value' => opt['key']}
+      end
+    end
+
+    volume_label = 'root'
+    volume = {
+      'id' => -1, 
+      'rootVolume' => true,
+      'name' => volume_label,
+      'size' => plan_size,
+      'sizeId' => nil,
+      'storageType' => storage_type_id,
+      'datastoreId' => nil
+    }
+
+    if plan_info['rootDiskCustomizable'] && storage_type && storage_type['customLabel']
+      v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'name', 'type' => 'text', 'fieldLabel' => 'Root Volume Label', 'required' => true, 'description' => 'Enter a volume label.', 'defaultValue' => volume_label}], options[:options])
+      volume['name'] = v_prompt[field_context]['name']
+    end
+    if plan_info['rootDiskCustomizable'] && storage_type && storage_type['customSize']
+      if root_custom_size_options.empty?
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'size', 'type' => 'number', 'fieldLabel' => 'Root Volume Size (GB)', 'required' => true, 'description' => 'Enter a volume size (GB).', 'defaultValue' => plan_size}], options[:options])
+        volume['size'] = v_prompt[field_context]['size']
+        volume['sizeId'] = nil #volume.delete('sizeId')
+      else
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'sizeId', 'type' => 'select', 'fieldLabel' => 'Root Volume Size', 'selectOptions' => root_custom_size_options, 'required' => true, 'description' => 'Choose a volume size.'}], options[:options])
+        volume['sizeId'] = v_prompt[field_context]['sizeId']
+        volume['size'] = nil #volume.delete('size')
+      end
+    else
+      # might need different logic here ? =o
+      volume['size'] = plan_size 
+      volume['sizeId'] = nil #volume.delete('sizeId')
+    end
+    if !datastore_options.empty?
+      v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'datastoreId', 'type' => 'select', 'fieldLabel' => 'Root Datastore', 'selectOptions' => datastore_options, 'required' => true, 'description' => 'Choose a datastore.'}], options[:options])
+      volume['datastoreId'] = v_prompt[field_context]['datastoreId']
+    end
+
+    volumes << volume
+
+    if plan_info['addVolumes']
+      volume_index = 1
+      
+      has_another_volume = options[:options] && options[:options]["dataVolume#{volume_index}"]
+      add_another_volume = has_another_volume || (!no_prompt && Morpheus::Cli::OptionTypes.confirm("Add data volume?"))
+      
+      while add_another_volume do
+        
+        #puts "Configure Data #{volume_index} Volume"
+
+        field_context = "dataVolume#{volume_index}"
+
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'storageType', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Storage Type", 'selectOptions' => storage_types, 'required' => true, 'skipSingleOption' => true, 'description' => 'Choose a storage type.'}], options[:options])
+        storage_type_id = v_prompt[field_context]['storageType']
+        storage_type = plan_info['storageTypes'].find {|i| i['id'] == storage_type_id.to_i }
+        
+
+        # sometimes the user chooses sizeId from a list of size options (AccountPrice) and other times it is free form
+        custom_size_options = []
+        if plan_info['customSizeOptions'] && plan_info['customSizeOptions'][storage_type_id.to_s]
+          plan_info['customSizeOptions'][storage_type_id.to_s].each do |opt|
+            custom_size_options << {'name' => opt['value'], 'value' => opt['key']}
+          end
+        end
+
+        volume_label = (volume_index == 1 ? 'data' : "data #{volume_index}")
+        volume = {
+          'id' => -1, 
+          'rootVolume' => false,
+          'name' => volume_label,
+          'size' => plan_size,
+          'sizeId' => nil,
+          'storageType' => storage_type_id,
+          'datastoreId' => nil
+        }
+
+        if plan_info['customizeVolume'] && storage_type['customLabel']
+          v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'name', 'type' => 'text', 'fieldLabel' => "Disk #{volume_index} Volume Label", 'required' => true, 'description' => 'Enter a volume label.', 'defaultValue' => volume_label}], options[:options])
+          volume['name'] = v_prompt[field_context]['name']
+        end
+        if plan_info['customizeVolume'] && storage_type['customSize']
+          if custom_size_options.empty?
+            v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'size', 'type' => 'number', 'fieldLabel' => "Disk #{volume_index} Volume Size (GB)", 'required' => true, 'description' => 'Enter a volume size (GB).', 'defaultValue' => plan_size}], options[:options])
+            volume['size'] = v_prompt[field_context]['size']
+            volume['sizeId'] = nil #volume.delete('sizeId')
+          else
+            v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'sizeId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Volume Size", 'selectOptions' => custom_size_options, 'required' => true, 'description' => 'Choose a volume size.'}], options[:options])
+            volume['sizeId'] = v_prompt[field_context]['sizeId']
+            volume['size'] = nil #volume.delete('size')
+          end
+        else
+          # might need different logic here ? =o
+          volume['size'] = plan_size 
+          volume['sizeId'] = nil #volume.delete('sizeId')
+        end
+        if !datastore_options.empty?
+          v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'datastoreId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Datastore", 'selectOptions' => datastore_options, 'required' => true, 'description' => 'Choose a datastore.'}], options[:options])
+          volume['datastoreId'] = v_prompt[field_context]['datastoreId']
+        end
+
+        volumes << volume
+
+        # todo: should maxDisk check consider the root volume too?
+        if plan_info['maxDisk'] && volume_index >= plan_info['maxDisk']
+          add_another_volume = false
+        else
+          volume_index += 1
+          has_another_volume = options[:options] && options[:options]["dataVolume#{volume_index}"]
+          add_another_volume = has_another_volume || (!no_prompt && Morpheus::Cli::OptionTypes.confirm("Add another data volume?"))
+        end
+
+      end
+
+    end
+
+    return volumes
+  end
+
+
+  # This recreates the behavior of multi_disk.js
+  # returns array of volumes based on service plan options (plan_info)
+  def prompt_resize_instance_volumes(current_volumes, plan_info, options={}, api_client=nil, api_params={})
+    #puts "Configure Volumes:"
+    no_prompt = (options[:no_prompt] || (options[:options] && options[:options][:no_prompt]))
+
+    current_root_volume = current_volumes[0]
+
+    volumes = []
+
+    plan_size = nil
+    if plan_info['maxStorage']
+      plan_size = plan_info['maxStorage'].to_i / (1024 * 1024 * 1024)
+    end
+
+    root_storage_types = []
+    if plan_info['rootStorageTypes']
+      plan_info['rootStorageTypes'].each do |opt|
+        root_storage_types << {'name' => opt['name'], 'value' => opt['id']}
+      end
+    end
+
+    storage_types = []
+    if plan_info['storageTypes']
+      plan_info['storageTypes'].each do |opt|
+        storage_types << {'name' => opt['name'], 'value' => opt['id']}
+      end
+    end
+
+    datastore_options = []
+    if plan_info['supportsAutoDatastore']
+      if plan_info['autoOptions']
+        plan_info['autoOptions'].each do |opt|
+          datastore_options << {'name' => opt['name'], 'value' => opt['id']}
+        end
+      end
+    end
+    if plan_info['datastores']
+      plan_info['datastores'].each do |k, v|
+        v.each do |opt|
+          datastore_options << {'name' => "#{k}: #{opt['name']}", 'value' => opt['id']}
+        end
+      end
+    end
+
+    #puts "Configure Root Volume"
+
+    field_context = "rootVolume"
+
+    if root_storage_types.empty?
+      # this means there's no configuration, just send a single root volume to the server
+      storage_type_id = nil
+      storage_type = nil
+    else
+      #v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'storageType', 'type' => 'select', 'fieldLabel' => 'Root Storage Type', 'selectOptions' => root_storage_types, 'required' => true, 'skipSingleOption' => true, 'description' => 'Choose a storage type.'}], options[:options])
+      #storage_type_id = v_prompt[field_context]['storageType']
+      storage_type_id = current_root_volume['type'] || current_root_volume['storageType']
+      storage_type = plan_info['storageTypes'].find {|i| i['id'] == storage_type_id.to_i }
+    end
+
+    # sometimes the user chooses sizeId from a list of size options (AccountPrice) and other times it is free form
+    root_custom_size_options = []
+    if plan_info['rootCustomSizeOptions'] && plan_info['rootCustomSizeOptions'][storage_type_id.to_s]
+      plan_info['rootCustomSizeOptions'][storage_type_id.to_s].each do |opt|
+        root_custom_size_options << {'name' => opt['value'], 'value' => opt['key']}
+      end
+    end
+
+    volume = {
+      'id' => current_root_volume['id'], 
+      'rootVolume' => true,
+      'name' => current_root_volume['name'],
+      'size' => current_root_volume['size'] > plan_size ? current_root_volume['size'] : plan_size,
+      'sizeId' => nil,
+      'storageType' => storage_type_id,
+      'datastoreId' => current_root_volume['datastoreId']
+    }
+
+    if plan_info['rootDiskCustomizable'] && storage_type && storage_type['customLabel']
+      v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'name', 'type' => 'text', 'fieldLabel' => 'Root Volume Label', 'required' => true, 'description' => 'Enter a volume label.', 'defaultValue' => volume['name']}], options[:options])
+      volume['name'] = v_prompt[field_context]['name']
+    end
+    if plan_info['rootDiskCustomizable'] && storage_type && storage_type['customSize']
+      if root_custom_size_options.empty?
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'size', 'type' => 'number', 'fieldLabel' => 'Root Volume Size (GB)', 'required' => true, 'description' => 'Enter a volume size (GB).', 'defaultValue' => volume['size']}], options[:options])
+        volume['size'] = v_prompt[field_context]['size']
+        volume['sizeId'] = nil #volume.delete('sizeId')
+      else
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'sizeId', 'type' => 'select', 'fieldLabel' => 'Root Volume Size', 'selectOptions' => root_custom_size_options, 'required' => true, 'description' => 'Choose a volume size.'}], options[:options])
+        volume['sizeId'] = v_prompt[field_context]['sizeId']
+        volume['size'] = nil #volume.delete('size')
+      end
+    else
+      # might need different logic here ? =o
+      volume['size'] = plan_size 
+      volume['sizeId'] = nil #volume.delete('sizeId')
+    end
+    # if !datastore_options.empty?
+    #   v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'datastoreId', 'type' => 'select', 'fieldLabel' => 'Root Datastore', 'selectOptions' => datastore_options, 'required' => true, 'description' => 'Choose a datastore.'}], options[:options])
+    #   volume['datastoreId'] = v_prompt[field_context]['datastoreId']
+    # end
+
+    volumes << volume
+
+    # modify or delete existing data volumes
+    (1..(current_volumes.size-1)).each do |volume_index|
+      current_volume = current_volumes[volume_index]
+      
+      if current_volume
+
+        field_context = "dataVolume#{volume_index}"
+
+        if no_prompt
+          volume_action = 'keep'
+        else
+          action_options = [{'name' => 'Modify', 'value' => 'modify'}, {'name' => 'Keep', 'value' => 'keep'}, {'name' => 'Delete', 'value' => 'delete'}]
+          v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'action', 'type' => 'select', 'fieldLabel' => "Modify/Keep/Delete volume '#{current_volume['name']}'", 'selectOptions' => action_options, 'required' => true, 'description' => 'Modify, Keep or Delete existing data volume?'}], options[:options])
+          volume_action = v_prompt[field_context]['action']
+        end
+
+        if volume_action == 'delete'
+          # deleted volume is just excluded from post params
+          next
+        elsif volume_action == 'keep'
+          volume = {
+            'id' => current_volume['id'].to_i, 
+            'rootVolume' => false,
+            'name' => current_volume['name'],
+            'size' => current_volume['size'] > plan_size ? current_volume['size'] : plan_size,
+            'sizeId' => nil,
+            'storageType' => (current_volume['type'] || current_volume['storageType']),
+            'datastoreId' => current_volume['datastoreId']
+          }
+          volumes << volume
+        else
+        
+          # v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'storageType', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Storage Type", 'selectOptions' => storage_types, 'required' => true, 'skipSingleOption' => true, 'description' => 'Choose a storage type.'}], options[:options])
+          # storage_type_id = v_prompt[field_context]['storageType']
+          storage_type_id = current_volume['type'] || current_volume['storageType']
+          storage_type = plan_info['storageTypes'].find {|i| i['id'] == storage_type_id.to_i }
+          
+          # sometimes the user chooses sizeId from a list of size options (AccountPrice) and other times it is free form
+          custom_size_options = []
+          if plan_info['customSizeOptions'] && plan_info['customSizeOptions'][storage_type_id.to_s]
+            plan_info['customSizeOptions'][storage_type_id.to_s].each do |opt|
+              custom_size_options << {'name' => opt['value'], 'value' => opt['key']}
+            end
+          end
+
+          volume = {
+            'id' => current_volume['id'].to_i, 
+            'rootVolume' => false,
+            'name' => current_volume['name'],
+            'size' => current_volume['size'] > plan_size ? current_volume['size'] : plan_size,
+            'sizeId' => nil,
+            'storageType' => (current_volume['type'] || current_volume['storageType']),
+            'datastoreId' => current_volume['datastoreId']
+          }
+
+          if plan_info['customizeVolume'] && storage_type['customLabel']
+            v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'name', 'type' => 'text', 'fieldLabel' => "Disk #{volume_index} Volume Label", 'required' => true, 'description' => 'Enter a volume label.', 'defaultValue' => volume['name']}], options[:options])
+            volume['name'] = v_prompt[field_context]['name']
+          end
+          if plan_info['customizeVolume'] && storage_type['customSize']
+            if custom_size_options.empty?
+              v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'size', 'type' => 'number', 'fieldLabel' => "Disk #{volume_index} Volume Size (GB)", 'required' => true, 'description' => 'Enter a volume size (GB).', 'defaultValue' => volume['size']}], options[:options])
+              volume['size'] = v_prompt[field_context]['size']
+              volume['sizeId'] = nil #volume.delete('sizeId')
+            else
+              v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'sizeId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Volume Size", 'selectOptions' => custom_size_options, 'required' => true, 'description' => 'Choose a volume size.'}], options[:options])
+              volume['sizeId'] = v_prompt[field_context]['sizeId']
+              volume['size'] = nil #volume.delete('size')
+            end
+          else
+            # might need different logic here ? =o
+            volume['size'] = plan_size 
+            volume['sizeId'] = nil #volume.delete('sizeId')
+          end
+          # if !datastore_options.empty?
+          #   v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'datastoreId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Datastore", 'selectOptions' => datastore_options, 'required' => true, 'description' => 'Choose a datastore.'}], options[:options])
+          #   volume['datastoreId'] = v_prompt[field_context]['datastoreId']
+          # end
+
+          volumes << volume
+
+        end
+
+      end
+    end
+
+
+    if plan_info['addVolumes']
+      volume_index = current_volumes.size
+      
+      has_another_volume = options[:options] && options[:options]["dataVolume#{volume_index}"]
+      add_another_volume = has_another_volume || (!no_prompt && Morpheus::Cli::OptionTypes.confirm("Add data volume?"))
+      
+      while add_another_volume do
+        
+        #puts "Configure Data #{volume_index} Volume"
+
+        field_context = "dataVolume#{volume_index}"
+
+        v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'storageType', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Storage Type", 'selectOptions' => storage_types, 'required' => true, 'skipSingleOption' => true, 'description' => 'Choose a storage type.'}], options[:options])
+        storage_type_id = v_prompt[field_context]['storageType']
+        storage_type = plan_info['storageTypes'].find {|i| i['id'] == storage_type_id.to_i }
+        
+
+        # sometimes the user chooses sizeId from a list of size options (AccountPrice) and other times it is free form
+        custom_size_options = []
+        if plan_info['customSizeOptions'] && plan_info['customSizeOptions'][storage_type_id.to_s]
+          plan_info['customSizeOptions'][storage_type_id.to_s].each do |opt|
+            custom_size_options << {'name' => opt['value'], 'value' => opt['key']}
+          end
+        end
+
+        volume_label = (volume_index == 1 ? 'data' : "data #{volume_index}")
+        volume = {
+          'id' => -1, 
+          'rootVolume' => false,
+          'name' => volume_label,
+          'size' => plan_size,
+          'sizeId' => nil,
+          'storageType' => storage_type_id,
+          'datastoreId' => nil
+        }
+
+        if plan_info['customizeVolume'] && storage_type['customLabel']
+          v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'name', 'type' => 'text', 'fieldLabel' => "Disk #{volume_index} Volume Label", 'required' => true, 'description' => 'Enter a volume label.', 'defaultValue' => volume_label}], options[:options])
+          volume['name'] = v_prompt[field_context]['name']
+        end
+        if plan_info['customizeVolume'] && storage_type['customSize']
+          if custom_size_options.empty?
+            v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'size', 'type' => 'number', 'fieldLabel' => "Disk #{volume_index} Volume Size (GB)", 'required' => true, 'description' => 'Enter a volume size (GB).', 'defaultValue' => plan_size}], options[:options])
+            volume['size'] = v_prompt[field_context]['size']
+            volume['sizeId'] = nil #volume.delete('sizeId')
+          else
+            v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'sizeId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Volume Size", 'selectOptions' => custom_size_options, 'required' => true, 'description' => 'Choose a volume size.'}], options[:options])
+            volume['sizeId'] = v_prompt[field_context]['sizeId']
+            volume['size'] = nil #volume.delete('size')
+          end
+        else
+          # might need different logic here ? =o
+          volume['size'] = plan_size 
+          volume['sizeId'] = nil #volume.delete('sizeId')
+        end
+        if !datastore_options.empty?
+          v_prompt = Morpheus::Cli::OptionTypes.prompt([{'fieldContext' => field_context, 'fieldName' => 'datastoreId', 'type' => 'select', 'fieldLabel' => "Disk #{volume_index} Datastore", 'selectOptions' => datastore_options, 'required' => true, 'description' => 'Choose a datastore.'}], options[:options])
+          volume['datastoreId'] = v_prompt[field_context]['datastoreId']
+        end
+
+        volumes << volume
+
+        # todo: should maxDisk check consider the root volume too?
+        if plan_info['maxDisk'] && volume_index >= plan_info['maxDisk']
+          add_another_volume = false
+        else
+          volume_index += 1
+          has_another_volume = options[:options] && options[:options]["dataVolume#{volume_index}"]
+          add_another_volume = has_another_volume || (!no_prompt && Morpheus::Cli::OptionTypes.confirm("Add another data volume?"))
+        end
+
+      end
+
+    end
+
+    return volumes
+  end
+
+end