tugboat 0.0.9 → 0.2.0

data/lib/tugboat/middleware/check_configuration.rb

@@ -10,6 +10,12 @@ module Tugboat
           exit 1
         end
 
+        # If the user passes the global `-q/--quiet` flag, redirect
+        # stdout
+        if env["user_quiet"]
+          $stdout = File.new('/dev/null', 'w')
+        end
+
         @app.call(env)
       end
     end

data/lib/tugboat/middleware/create_droplet.rb

@@ -26,12 +26,18 @@ module Tugboat
             droplet_private_networking = env["create_droplet_private_networking"] :
             droplet_private_networking = env["config"].default_private_networking
 
+        env["create_droplet_backups_enabled"] ?
+            droplet_backups_enabled = env["create_droplet_backups_enabled"] :
+            droplet_backups_enabled = env["config"].default_backups_enabled
+
+
         req = ocean.droplets.create :name               => env["create_droplet_name"],
                                     :size_id            => droplet_size_id,
                                     :image_id           => droplet_image_id,
                                     :region_id          => droplet_region_id,
                                     :ssh_key_ids        => droplet_ssh_key_id,
-                                    :private_networking => droplet_private_networking
+                                    :private_networking => droplet_private_networking,
+                                    :backups_enabled    => droplet_backups_enabled
 
         if req.status == "ERROR"
           say req.error_message, :red

data/lib/tugboat/middleware/destroy_image.rb

@@ -0,0 +1,23 @@
+module Tugboat
+  module Middleware
+    class DestroyImage < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        say "Queuing destroy image for #{env["image_id"]} #{env["image_name"]}...", nil, false
+
+        req = ocean.images.delete env["image_id"]
+
+        if req.status == "ERROR"
+          say "#{req.status}: #{req.error_message}", :red
+          exit 1
+        end
+
+        say "done", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/tugboat/middleware/find_droplet.rb

@@ -72,7 +72,7 @@ module Tugboat
 
           ocean.droplets.list.droplets.each_with_index do |d, i|
             # Check to see if one of the droplet names have the fuzzy string.
-            if d.name.include? user_fuzzy_name
+            if d.name.upcase.include? user_fuzzy_name.upcase
               found_droplets << d
             end
           end

data/lib/tugboat/middleware/find_image.rb

@@ -0,0 +1,113 @@
+module Tugboat
+  module Middleware
+    # Check if the client has set-up configuration yet.
+    class FindImage < Base
+      def call(env)
+        ocean = env["ocean"]
+        user_fuzzy_name = env['user_image_fuzzy_name']
+        user_image_name = env['user_image_name']
+        user_image_id = env['user_image_id']
+
+        # First, if nothing is provided to us, we should quit and
+        # let the user know.
+        if !user_fuzzy_name && !user_image_name && !user_image_id
+          say "Tugboat attempted to find an image with no arguments. Try `tugboat help`", :red
+          exit 1
+        end
+
+        # If you were to `tugboat restart foo -n foo-server-001` then we'd use
+        # 'foo-server-001' without looking up the fuzzy name.
+        #
+        # This is why we check in this order.
+
+        # Easy for us if they provide an id. Just set it to the image_id
+        if user_image_id
+          say "Image id provided. Finding Image...", nil, false
+          req = ocean.images.show user_image_id
+
+          if req.status == "ERROR"
+            say "#{req.status}: #{req.error_message}", :red
+            exit 1
+          end
+
+          env["image_id"] = req.image.id
+          env["image_name"] = "(#{req.image.name})"
+        end
+
+        # If they provide a name, we need to get the ID for it.
+        # This requires a lookup.
+        if user_image_name && !env["image_id"]
+          say "Image name provided. Finding image ID...", nil, false
+
+          # Look for the image by an exact name match.
+          ocean.images.list.images.each do |d|
+            if d.name == user_image_name
+              env["image_id"] = d.id
+              env["image_name"] = "(#{d.name})"
+            end
+          end
+
+          # If we coulnd't find it, tell the user and drop out of the
+          # sequence.
+          if !env["image_id"]
+            say "error\nUnable to find a image named '#{user_image_name}'.", :red
+            exit 1
+          end
+        end
+
+        # We only need to "fuzzy find" a image if a fuzzy name is provided,
+        # and we don't want to fuzzy search if an id or name is provided
+        # with a flag.
+        #
+        # This requires a lookup.
+        if user_fuzzy_name && !env["image_id"]
+          say "Image fuzzy name provided. Finding image ID...", nil, false
+
+          found_images = []
+          choices = []
+
+          ocean.images.list.images.each_with_index do |d, i|
+            # Check to see if one of the image names have the fuzzy string.
+            if d.name.upcase.include? user_fuzzy_name.upcase
+              found_images << d
+            end
+          end
+
+          # Check to see if we have more then one image, and prompt
+          # a user to choose otherwise.
+          if found_images.length == 1
+            env["image_id"] = found_images.first.id
+            env["image_name"] = "(#{found_images.first.name})"
+          elsif found_images.length > 1
+            # Did we run the multiple questionairre?
+            did_run_multiple = true
+
+            say "Multiple images found."
+            say
+            found_images.each_with_index do |d, i|
+              say "#{i}) #{d.name} (#{d.id})"
+              choices << i.to_s
+            end
+            say
+            choice = ask "Please choose a image:", :limited_to => choices
+            env["image_id"] = found_images[choice.to_i].id
+            env["image_name"] = found_images[choice.to_i].name
+          end
+
+          # If we coulnd't find it, tell the user and drop out of the
+          # sequence.
+          if !env["image_id"]
+            say "error\nUnable to find an image named '#{user_fuzzy_name}'.", :red
+            exit 1
+          end
+        end
+
+        if !did_run_multiple
+          say "done#{CLEAR}, #{env["image_id"]} #{env["image_name"]}", :green
+        end
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/tugboat/middleware/info_image.rb

@@ -0,0 +1,25 @@
+module Tugboat
+  module Middleware
+    class InfoImage < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        req = ocean.images.show env["image_id"]
+
+        if req.status == "ERROR"
+          say "#{req.status}: #{req.error_message}", :red
+          exit 1
+        end
+
+        image = req.image
+
+        say
+        say "Name:             #{image.name}"
+        say "ID:               #{image.id}"
+        say "Distribution:     #{image.distribution}"
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/tugboat/middleware/rebuild_droplet.rb

@@ -0,0 +1,23 @@
+module Tugboat
+  module Middleware
+    class RebuildDroplet < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        say "Queuing rebuild for droplet #{env["droplet_id"]} #{env["droplet_name"]} with image #{env["image_id"]} #{env["image_name"]}...", nil, false
+        
+        req = ocean.droplets.rebuild env["droplet_id"],
+                                     :image_id => env["image_id"]
+
+        if req.status == "ERROR"
+          say req.error_message, :red
+          exit 1
+        end
+
+        say "done", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/tugboat/version.rb

@@ -1,3 +1,3 @@
 module Tugboat
-  VERSION = "0.0.9"
+  VERSION = "0.2.0"
 end

data/spec/cli/authorize_cli_spec.rb

@@ -34,12 +34,15 @@ describe Tugboat::CLI do
       $stdin.should_receive(:gets).and_return(ssh_key_id)
       $stdout.should_receive(:print).with("Enter your default for private networking (optional, defaults to false): ")
       $stdin.should_receive(:gets).and_return(private_networking)
+      $stdout.should_receive(:print).with("Enter your default for enabling backups (optional, defaults to false): ")
+      $stdin.should_receive(:gets).and_return(backups_enabled)
+
 
       @cli.authorize
 
       expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
 
-      File.read(tmp_path).should include "image: '#{image}'", "region: '#{region}'", "size: '#{size}'", "ssh_user: #{ssh_user}", "ssh_key_path: #{ssh_key_path}", "ssh_port: '#{ssh_port}'", "ssh_key: '#{ssh_key_id}'", "private_networking: '#{private_networking}'"
+      File.read(tmp_path).should include "image: '#{image}'", "region: '#{region}'", "size: '#{size}'", "ssh_user: #{ssh_user}", "ssh_key_path: #{ssh_key_path}", "ssh_port: '#{ssh_port}'", "ssh_key: '#{ssh_key_id}'", "private_networking: '#{private_networking}'", "backups_enabled: '#{backups_enabled}'"
 
     end
 
@@ -66,6 +69,9 @@ describe Tugboat::CLI do
       $stdin.should_receive(:gets).and_return('')
       $stdout.should_receive(:print).with("Enter your default for private networking (optional, defaults to false): ")
       $stdin.should_receive(:gets).and_return('')
+      $stdout.should_receive(:print).with("Enter your default for enabling backups (optional, defaults to false): ")
+      $stdin.should_receive(:gets).and_return('')
+
 
       @cli.authorize
 

data/spec/cli/create_cli_spec.rb

@@ -5,7 +5,7 @@ describe Tugboat::CLI do
 
   describe "create a droplet" do
     it "with a name, uses defaults from configuration" do
-      stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=#{image}&name=#{droplet_name}&private_networking=#{private_networking}&region_id=#{region}&size_id=#{size}&ssh_key_ids=#{ssh_key_id}").
+      stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=#{image}&name=#{droplet_name}&backups_enabled=#{backups_enabled}&private_networking=#{private_networking}&region_id=#{region}&size_id=#{size}&ssh_key_ids=#{ssh_key_id}").
          to_return(:status => 200, :body => '{"status":"OK"}')
 
       @cli.create(droplet_name)
@@ -13,11 +13,11 @@ describe Tugboat::CLI do
       expect($stdout.string).to eq <<-eos
 Queueing creation of droplet '#{droplet_name}'...done
       eos
-      expect(a_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=#{image}&name=#{droplet_name}&private_networking=#{private_networking}&region_id=#{region}&size_id=#{size}&ssh_key_ids=#{ssh_key_id}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=#{image}&name=#{droplet_name}&backups_enabled=#{backups_enabled}&private_networking=#{private_networking}&region_id=#{region}&size_id=#{size}&ssh_key_ids=#{ssh_key_id}")).to have_been_made
     end
 
     it "with args does not use defaults from configuration" do
-      stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=555&name=foo&private_networking=#{private_networking}&region_id=3&size_id=666&ssh_key_ids=4321").
+      stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=555&name=foo&backups_enabled=#{backups_enabled}&private_networking=#{private_networking}&region_id=3&size_id=666&ssh_key_ids=4321").
          to_return(:status => 200, :body => '{"status":"OK"}')
 
       @cli.options = @cli.options.merge(:image => '555', :size => '666', :region => '3', :keys => '4321')
@@ -27,7 +27,7 @@ Queueing creation of droplet '#{droplet_name}'...done
 Queueing creation of droplet '#{droplet_name}'...done
       eos
 
-      expect(a_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=555&name=foo&private_networking=#{private_networking}&region_id=3&size_id=666&ssh_key_ids=4321")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=555&name=foo&backups_enabled=#{backups_enabled}&private_networking=#{private_networking}&region_id=3&size_id=666&ssh_key_ids=4321")).to have_been_made
     end
   end
 

data/spec/cli/destroy_image_cli_spec.rb

@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "destroy image" do
+    it "destroys an image with a fuzzy name" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.destroy_image("NLP Final")
+
+      expect($stdout.string).to eq <<-eos
+Image fuzzy name provided. Finding image ID...done\e[0m, 478 (NLP Final)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy image for 478 (NLP Final)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "destroys an image with an id" do
+      stub_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.options = @cli.options.merge(:id => 478)
+      @cli.destroy_image
+
+      expect($stdout.string).to eq <<-eos
+Image id provided. Finding Image...done\e[0m, 478 (NLP Final)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy image for 478 (NLP Final)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+
+    it "destroys an image with a name" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.options = @cli.options.merge(:name => "NLP Final")
+      @cli.destroy_image
+
+      expect($stdout.string).to eq <<-eos
+Image name provided. Finding image ID...done\e[0m, 478 (NLP Final)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy image for 478 (NLP Final)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+
+    it "destroys an image with confirm flag set" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      @cli.options = @cli.options.merge(:name => "NLP Final")
+      @cli.options = @cli.options.merge(:confirm => true)
+      @cli.destroy_image("NLP Final")
+
+      expect($stdout.string).to eq <<-eos
+Image name provided. Finding image ID...done\e[0m, 478 (NLP Final)\nQueuing destroy image for 478 (NLP Final)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images/478/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+  end
+
+end

data/spec/cli/droplets_cli_spec.rb

@@ -30,6 +30,18 @@ You don't appear to have any droplets.
 Try creating one with \e[32m`tugboat create`\e[0m
       eos
 
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+    it "shows no output when --quiet is set" do
+      stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplets_empty"))
+
+      @cli.options = @cli.options.merge(:quiet => true)
+      @cli.droplets
+
+      # Should be /dev/null not stringIO
+      expect($stdout).to be_a File
+
       expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
     end
   end

data/spec/cli/info_image_cli_spec.rb

@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "show" do
+    it "shows an image with a fuzzy name" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      @cli.info_image("NLP Final")
+
+      expect($stdout.string).to eq <<-eos
+Image fuzzy name provided. Finding image ID...done\e[0m, 478 (NLP Final)\n\nName:             NLP Final\nID:               478\nDistribution:     Ubuntu
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "shows an image with an id" do
+      stub_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      @cli.options = @cli.options.merge(:id => 478)
+      @cli.info_image
+
+      expect($stdout.string).to eq <<-eos
+Image id provided. Finding Image...done\e[0m, 478 (NLP Final)\n\nName:             NLP Final\nID:               478\nDistribution:     Ubuntu
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made.times(2)
+    end
+
+    it "shows an image with a name" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_image"))
+
+      @cli.options = @cli.options.merge(:name => "NLP Final")
+      @cli.info_image
+
+      expect($stdout.string).to eq <<-eos
+Image name provided. Finding image ID...done\e[0m, 478 (NLP Final)\n\nName:             NLP Final\nID:               478\nDistribution:     Ubuntu
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images/478?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+  end
+
+end