tugboat 0.0.3 → 0.0.4

data/lib/tugboat/middleware/ssh_droplet.rb

@@ -11,6 +11,14 @@ module Tugboat
           "-o", "UserKnownHostsFile=/dev/null",
           "-i", env["config"].ssh_key_path.to_s]
 
+        if env["user_droplet_ssh_port"]
+          options.push("-p", env["user_droplet_ssh_port"].to_s)
+        elsif env["config"].ssh_port
+          options.push("-p", env["config"].ssh_port.to_s)
+        else
+          options.push("-p", "22")
+        end
+
         host_string = "#{env["config"].ssh_user}@#{env["droplet_ip"]}"
 
         options << host_string

data/lib/tugboat/version.rb

@@ -1,3 +1,3 @@
 module Tugboat
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/spec/cli/authorize_cli_spec.rb

@@ -0,0 +1,37 @@
+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}").
+           to_return(:status => 200)
+    end
+
+    it "asks the right questions and checks credentials" do
+
+      $stdout.should_receive(:print).exactly(6).times
+      $stdout.should_receive(:print).with("Enter your client key: ")
+      $stdin.should_receive(:gets).and_return(client_key)
+      $stdout.should_receive(:print).with("Enter your API key: ")
+      $stdin.should_receive(:gets).and_return(api_key)
+      $stdout.should_receive(:print).with("Enter your SSH key path (optional, defaults to ~/.ssh/id_rsa): ")
+      $stdin.should_receive(:gets).and_return(ssh_key_path)
+      $stdout.should_receive(:print).with("Enter your SSH user (optional, defaults to #{ENV['USER']}): ")
+      $stdin.should_receive(:gets).and_return(ssh_user)
+      $stdout.should_receive(:print).with("Enter your SSH port number (optional, defaults to 22): ")
+      $stdin.should_receive(:gets).and_return(ssh_port)
+
+      @cli.authorize
+
+      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/create_cli_spec.rb

@@ -0,0 +1,40 @@
+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").
+         to_return(:status => 200, :body => '{"status":"OK"}')
+
+      @cli.create(droplet_name)
+
+      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&name=#{droplet_name}&region_id&size_id&ssh_key_ids")).to have_been_made
+    end
+
+    it "with args" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/new?api_key=#{api_key}&client_id=#{client_key}&image_id=2672&name=#{droplet_name}&region_id=2&size_id=64&ssh_key_ids=1234").
+         to_return(:status => 200, :body => '{"status":"OK"}')
+
+      @cli.options = @cli.options.merge(:image => 2672, :size => 64, :region => 2, :keys => "1234")
+      @cli.create(droplet_name)
+
+      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=2672&name=foo&region_id=2&size_id=64&ssh_key_ids=1234")).to have_been_made
+    end
+  end
+
+
+end
+

data/spec/cli/destroy_cli_spec.rb

@@ -0,0 +1,93 @@
+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}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      stub_request(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.destroy("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy 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(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "destroys a droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      stub_request(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.options = @cli.options.merge(:id => droplet_id)
+      @cli.destroy
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      expect(a_request(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+
+    it "destroys 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(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      $stdin.should_receive(:gets).and_return("y")
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      @cli.destroy
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)\nWarning! Potentially destructive action. Please confirm [y/n]: Queuing destroy 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(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+
+    it "destroys a droplet with confirm flag 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"))
+
+      stub_request(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      @cli.options = @cli.options.merge(:confirm => true)
+      @cli.destroy
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing destroy 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(:delete, "https://api.digitalocean.com/droplets/100823/destroy?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+  end
+
+end

data/spec/cli/droplets_cli_spec.rb

@@ -0,0 +1,28 @@
+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}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      @cli.droplets
+
+      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)
+      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/halt_cli_spec.rb

@@ -0,0 +1,70 @@
+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}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      stub_request(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.halt("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing shutdown 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(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "halts a droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      stub_request(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:id => droplet_id)
+      @cli.halt
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+Queuing shutdown for 100823 (foo)...done
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+      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 "halts 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(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      @cli.halt
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+Queuing shutdown 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(:put, "https://api.digitalocean.com/droplets/100823/shutdown?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+  end
+
+end

data/spec/cli/images_cli_spec.rb

@@ -0,0 +1,53 @@
+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").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      @cli.images
+
+      expect($stdout.string).to eq <<-eos
+My Images:
+NYTD Backup 1-18-2012 (id: 466, distro: Ubuntu)
+NLP Final (id: 478, distro: Ubuntu)
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=my_images")).to have_been_made
+    end
+
+    it "shows a global list" do
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=my_images").
+           to_return(:status => 200, :body => fixture("show_images"))
+
+      stub_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=global").
+           to_return(:status => 200, :body => fixture("show_images_global"))
+
+      @cli.options = @cli.options.merge(:global => true)
+      @cli.images
+
+      expect($stdout.string).to eq <<-eos
+My Images:
+NYTD Backup 1-18-2012 (id: 466, distro: Ubuntu)
+NLP Final (id: 478, distro: Ubuntu)
+
+Global Images:
+NYTD Backup 1-18-2012 (id: 466, distro: Ubuntu)
+NLP Final (id: 478, distro: Ubuntu)
+Global Final (id: 479, distro: Ubuntu)
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=my_images")).to have_been_made
+      expect(a_request(:get, "https://api.digitalocean.com/images?api_key=#{api_key}&client_id=#{client_key}&filter=global")).to have_been_made
+    end
+  end
+
+end
+

data/spec/cli/info_cli_spec.rb

@@ -0,0 +1,93 @@
+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}").
+           to_return(:status => 200, :body => fixture("show_droplets"))
+
+      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"))
+
+      @cli.info("foo")
+
+      expect($stdout.string).to eq <<-eos
+Droplet fuzzy name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+
+Name:             foo
+ID:               100823
+Status:           \e[32mactive\e[0m
+IP:               33.33.33.10
+Region ID:        1
+Image ID:         420
+Size ID:          33
+Backups Active:   false
+      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?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "shows a droplet with an id" do
+      stub_request(:get, "https://api.digitalocean.com/droplets/#{droplet_id}?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?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:id => droplet_id)
+      @cli.info
+
+      expect($stdout.string).to eq <<-eos
+Droplet id provided. Finding Droplet...done\e[0m, 100823 (foo)
+
+Name:             foo
+ID:               100823
+Status:           \e[32mactive\e[0m
+IP:               33.33.33.10
+Region ID:        1
+Image ID:         420
+Size ID:          33
+Backups Active:   false
+      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/#{droplet_id}?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+    it "shows 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?api_key=#{api_key}&client_id=#{client_key}").
+           to_return(:status => 200, :body => fixture("show_droplet"))
+
+      @cli.options = @cli.options.merge(:name => droplet_name)
+      @cli.info
+
+      expect($stdout.string).to eq <<-eos
+Droplet name provided. Finding droplet ID...done\e[0m, 100823 (foo)
+
+Name:             foo
+ID:               100823
+Status:           \e[32mactive\e[0m
+IP:               33.33.33.10
+Region ID:        1
+Image ID:         420
+Size ID:          33
+Backups Active:   false
+      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?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+
+  end
+
+end

data/spec/cli/keys_cli_spec.rb

@@ -0,0 +1,27 @@
+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}").
+           to_return(:status => 200, :body => fixture("show_keys"))
+
+      @cli.keys
+
+      expect($stdout.string).to eq <<-eos
+SSH Keys:
+office-imac (id: 10)
+macbook-air (id: 11)
+      eos
+
+      expect(a_request(:get, "https://api.digitalocean.com/ssh_keys?api_key=#{api_key}&client_id=#{client_key}")).to have_been_made
+    end
+  end
+end
+