tugboat 0.0.5 → 0.0.6

data/lib/tugboat/middleware/resize_droplet.rb

@@ -0,0 +1,24 @@
+module Tugboat
+  module Middleware
+    class ResizeDroplet < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        say "Queuing resize for #{env["droplet_id"]} #{env["droplet_name"]}...", nil, false
+
+        res = ocean.droplets.resize env["droplet_id"],
+                                    :size_id => env["user_droplet_size"]
+
+        if res.status == "ERROR"
+          say "#{res.status}: #{res.error_message}", :red
+          exit 1
+        end
+
+        say "done", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/tugboat/middleware/ssh_droplet.rb

@@ -19,6 +19,10 @@ module Tugboat
           options.push("-p", "22")
         end
 
+        if env["user_droplet_ssh_opts"]
+          options.concat env["user_droplet_ssh_opts"].split
+        end
+
         ssh_user = env["user_droplet_ssh_user"] || env["config"].ssh_user
         host_string = "#{ssh_user}@#{env["droplet_ip"]}"
 

data/lib/tugboat/middleware/start_droplet.rb

@@ -0,0 +1,21 @@
+module Tugboat
+  module Middleware
+    class StartDroplet < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        say "Queuing start for #{env["droplet_id"]} #{env["droplet_name"]}...", nil, false
+        req = ocean.droplets.power_on env["droplet_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/wait_for_state.rb

@@ -0,0 +1,35 @@
+module Tugboat
+  module Middleware
+    class WaitForState < Base
+      def call(env)
+        ocean = env["ocean"]
+
+        say "Waiting for droplet to become #{env["user_droplet_desired_state"]}.", nil, false
+
+        start_time = Time.now
+
+        req = ocean.droplets.show env["droplet_id"]
+
+        say ".", nil, false
+
+        if req.status == "ERROR"
+          say req.error_message, :red
+          exit 1
+        end
+
+        while req.droplet.status != env["user_droplet_desired_state"] do
+          sleep 2
+          req = ocean.droplets.show env["droplet_id"]
+          say ".", nil, false
+        end
+
+        total_time = (Time.now - start_time).to_i
+
+        say "done#{CLEAR} (#{total_time}s)", :green
+
+        @app.call(env)
+      end
+    end
+  end
+end
+

data/lib/tugboat/version.rb

@@ -1,3 +1,3 @@
 module Tugboat
-  VERSION = "0.0.5"
+  VERSION = "0.0.6"
 end

data/spec/cli/authorize_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "authorize" do
     before do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/create_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "create a droplet" do
     it "with a name" do
       stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id&name=#{droplet_name}&region_id&size_id&ssh_key_ids").

data/spec/cli/destroy_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "destroy" do
     it "destroys a droplet with a fuzzy name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/droplets_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "droplets" do
     it "shows a list" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
@@ -17,7 +13,7 @@ describe Tugboat::CLI do
       expect($stdout.string).to eq <<-eos
 test222 (ip: 33.33.33.10, status: \e[32mactive\e[0m, region: 1, id: 100823)
 test223 (ip: 33.33.33.10, status: \e[32mactive\e[0m, region: 1, id: 100823)
-foo (ip: 33.33.33.10, status: \e[31moff\e[0m, region: 1, id: 100823)
+foo (ip: 33.33.33.10, status: \e[32mactive\e[0m, region: 1, id: 100823)
       eos
 
       expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made

data/spec/cli/halt_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "halt" do
     it "halts a droplet with a fuzzy name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").
@@ -83,6 +79,22 @@ Queuing shutdown for 100823 (foo)...done
       expect(a_request(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
     end
 
+
+    it "does not halt a droplet that is off" 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_inactive"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      expect {@cli.halt}.to raise_error(SystemExit)
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Droplet must be on for this operation to be successful.
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
   end
 
 end

data/spec/cli/help_cli_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "help" do
+    it "shows a help message" do
+      @cli.help
+      expect($stdout.string).to match("Commands:")
+    end
+
+    it "shows a help message for specific commands" do
+      @cli.help "sizes"
+      expect($stdout.string).to match("Usage:")
+    end
+  end
+end

data/spec/cli/images_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "images" do
     it "shows a list" do
       stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=my_images").

data/spec/cli/info_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "show" do
     it "shows a droplet with a fuzzy name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/keys_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "keys" do
     it "shows a list" do
       stub_request(:get, "https://api.digitalocean.com/ssh_keys?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/password_reset_cli_spec.rb

@@ -0,0 +1,83 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "passwordreset" do
+    it "resets the root password given a fuzzy name" 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"))
+      stub_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.password_reset("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing password reset for 100823 (foo)...done
+Your new root password will be emailed to you
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+    end
+
+    it "resets the root password given an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplet"))
+      stub_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.options = @cli.options.merge(:id => 100823)
+      @cli.password_reset
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+Queuing password reset for 100823 (foo)...done
+Your new root password will be emailed to you
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+    end
+
+    it "resets the root password given a name" 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"))
+      stub_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.options = @cli.options.merge(:name => "foo")
+      @cli.password_reset
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing password reset for 100823 (foo)...done
+Your new root password will be emailed to you
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+    end
+
+    it "raises SystemExit when a request fails" 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"))
+      stub_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 500, :body => '{"status":"ERROR","message":"Some error"}')
+
+      expect { @cli.password_reset("foo") }.to raise_error(SystemExit)
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:post, "https://api.digitalocean.com/droplets/100823/password_reset?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+    end
+  end
+end

data/spec/cli/regions_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "regions" do
     it "shows a list" do
       stub_request(:get, "https://api.digitalocean.com/regions?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/resize_cli_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+
+describe Tugboat::CLI do
+  include_context "spec"
+
+  describe "resize" do
+    it "resizes a droplet with a fuzzy name" 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"))
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.options = @cli.options.merge(:size => 123)
+      @cli.resize("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing resize for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123")).
+        to have_been_made
+    end
+
+    it "resizes a droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}").
+        to_return(:status => 200, :body => fixture("show_droplet"))
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.options = @cli.options.merge(:size => 123, :id => 100823)
+      @cli.resize
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+Queuing resize for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123")).
+        to have_been_made
+    end
+
+    it "resizes a droplet with a name" 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"))
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123").
+        to_return(:status => 200, :body => '{"status":"OK","event_id":456}')
+
+      @cli.options = @cli.options.merge(:size => 123, :name => "foo")
+      @cli.resize
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing resize for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123")).
+        to have_been_made
+    end
+
+    it "raises SystemExit when a request fails" 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"))
+      stub_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123").
+        to_return(:status => 500, :body => '{"status":"ERROR","message":"Some error"}')
+
+      @cli.options = @cli.options.merge(:size => 123)
+      expect { @cli.resize("foo") }.to raise_error(SystemExit)
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}")).
+        to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/100823/resize?api_key=#{api_key}&client_id=#{client_key}&size_id=123")).
+        to have_been_made
+    end
+  end
+end

data/spec/cli/restart_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "restarts a droplet" do
     it "with a fuzzy name" do
       stub_request(:get, "https://api.digitalocean.com/droplets?api_key=#{api_key}&client_id=#{client_key}").

data/spec/cli/sizes_cli_spec.rb

@@ -3,10 +3,6 @@ require 'spec_helper'
 describe Tugboat::CLI do
   include_context "spec"
 
-  before :each do
-    @cli = Tugboat::CLI.new
-  end
-
   describe "sizes" do
     it "shows a list" do
       stub_request(:get, "https://api.digitalocean.com/sizes?api_key=#{api_key}&client_id=#{client_key}").