create_command.rb in kontena-plugin-digitalocean-0.2.0.pre2

- old
+ new
@@ -1,33 +1,53 @@
+require_relative '../prompts'
+
 module Kontena::Plugin::DigitalOcean::Nodes
   class CreateCommand < Kontena::Command
     include Kontena::Cli::Common
     include Kontena::Cli::GridOptions
+    include Kontena::Plugin::DigitalOcean::Prompts
 
     parameter "[NAME]", "Node name"
-    option "--token", "TOKEN", "DigitalOcean API token", required: true
+    option "--token", "TOKEN", "DigitalOcean API token", environment_variable: 'DO_TOKEN'
+    option "--region", "REGION", "Region"
     option "--ssh-key", "SSH_KEY", "Path to ssh public key", default: '~/.ssh/id_rsa.pub'
-    option "--size", "SIZE", "Droplet size", default: '1gb'
-    option "--region", "REGION", "Region", default: 'ams2'
+    option "--size", "SIZE", "Droplet size"
+    option "--count", "COUNT", "How many droplets to create"
     option "--version", "VERSION", "Define installed Kontena version", default: 'latest'
 
     def execute
       require_api_url
       require_current_grid
 
-      require 'kontena/machine/digital_ocean'
+      do_token = ask_do_token
+
+      require_relative '../../../machine/digital_ocean'
+
+      do_region = ask_droplet_region(do_token)
+      do_size = ask_droplet_size(do_token, do_region)
+      do_count = ask_droplet_count
+
       grid = fetch_grid
-      provisioner = provisioner(client(require_token), token)
+      provisioner = provisioner(client(require_token), do_token)
       provisioner.run!(
         master_uri: api_url,
         grid_token: grid['token'],
         grid: current_grid,
         ssh_key: ssh_key,
         name: name,
-        size: size,
-        region: region,
+        size: do_size,
+        count: do_count,
+        region: do_region,
         version: version
       )
+    end
+
+    def ask_droplet_count
+      if self.count.nil?
+        prompt.ask('How many droplets?: ', default: 1)
+      else
+        self.count
+      end
     end
 
     # @param [Kontena::Client] client
     # @param [String] token
     def provisioner(client, token)