lita-digitalocean 0.0.1 → 1.0.0

data/lib/lita/handlers/digitalocean/region.rb

@@ -0,0 +1,23 @@
+module Lita
+  module Handlers
+    class Digitalocean < Handler
+      class Region < Base
+        do_route /^do\s+regions?\s+list$/i, :list, {
+          t("help.regions.list_key") => t("help.regions.list_value")
+        }
+
+        def list(response)
+          do_response = do_call(response) do |client|
+            client.regions.list
+          end or return
+
+          messages = do_response[:regions].map { |region|  t("regions.details", region) }
+
+          response.reply(*messages)
+        end
+      end
+
+      Lita.register_handler(Region)
+    end
+  end
+end

data/lib/lita/handlers/digitalocean/size.rb

@@ -0,0 +1,23 @@
+module Lita
+  module Handlers
+    class Digitalocean < Handler
+      class Size < Base
+        do_route /^do\s+sizes\s+list$/, :list, {
+          t("help.sizes.list_key") => t("help.sizes.list_value")
+        }
+
+        def list(response)
+          do_response = do_call(response) do |client|
+            client.sizes.list
+          end or return
+
+          messages = do_response[:sizes].map { |size| t("sizes.details", size) }
+
+          response.reply(*messages)
+        end
+      end
+
+      Lita.register_handler(Size)
+    end
+  end
+end

data/lib/lita/handlers/digitalocean/ssh_key.rb

@@ -0,0 +1,94 @@
+module Lita
+  module Handlers
+    class Digitalocean < Handler
+      class SSHKey < Base
+        do_route /^do\s+ssh\s+keys?\s+add\s+.+$/i, :add, {
+          t("help.ssh_keys.add_key") => t("help.ssh_keys.add_value")
+        }
+
+        do_route /^do\s+ssh\s+keys?\s+delete\s+(\d+)$/i, :delete, {
+          t("help.ssh_keys.delete_key") => t("help.ssh_keys.delete_value")
+        }
+
+        do_route /^do\s+ssh\s+keys?\s+edit\s+(\d+)\s+.+$/i, :edit, {
+          t("help.ssh_keys.edit_key") => t("help.ssh_keys.edit_value")
+        }, { name: {}, public_key: {} }
+
+        do_route /^do\s+ssh\s+keys?\s+list$/i, :list, {
+          t("help.ssh_keys.list_key") => t("help.ssh_keys.list_value")
+        }
+
+        do_route /^do\s+ssh\s+keys?\s+show\s+(\d+)$/i, :show, {
+          t("help.ssh_keys.show_key") => t("help.ssh_keys.show_value"),
+        }
+
+        def add(response)
+          name, public_key = response.args[3..4]
+
+          unless name && public_key
+            return response.reply("#{t('format')}: #{t('help.ssh_keys.add_key')}")
+          end
+
+          do_response = do_call(response) do |client|
+            client.ssh_keys.add(name: name, ssh_pub_key: public_key)
+          end or return
+
+          response.reply(t("ssh_keys.add.created", do_response[:ssh_key]))
+        end
+
+        def delete(response)
+          key_id = response.matches[0][0]
+
+          do_call(response) do |client|
+            client.ssh_keys.delete(key_id)
+          end or return
+
+          response.reply(t("ssh_keys.delete.deleted", key_id: key_id))
+        end
+
+        def edit(response)
+          kwargs = {}
+
+          if (name = response.extensions[:kwargs][:name])
+            kwargs[:name] = name
+          end
+
+          if (public_key = response.extensions[:kwargs][:public_key])
+            kwargs[:ssh_pub_key] = public_key
+          end
+
+          do_response = do_call(response) do |client|
+            client.ssh_keys.edit(response.matches[0][0], kwargs)
+          end or return
+
+          response.reply(t("ssh_keys.edit.updated", do_response[:ssh_key]))
+        end
+
+        def list(response)
+          do_response = do_call(response) do |client|
+            client.ssh_keys.list
+          end or return
+
+          if do_response[:ssh_keys].empty?
+            response.reply(t("ssh_keys.list.empty"))
+          else
+            do_response[:ssh_keys].each do |key|
+              response.reply("#{key[:id]} (#{key[:name]})")
+            end
+          end
+        end
+
+        def show(response)
+          do_response = do_call(response) do |client|
+            client.ssh_keys.show(response.matches[0][0])
+          end or return
+
+          key = do_response[:ssh_key]
+          response.reply("#{key[:id]} (#{key[:name]}): #{key[:ssh_pub_key]}")
+        end
+      end
+
+      Lita.register_handler(SSHKey)
+    end
+  end
+end

data/lita-digitalocean.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-digitalocean"
-  spec.version       = "0.0.1"
+  spec.version       = "1.0.0"
   spec.authors       = ["Jimmy Cuadra"]
   spec.email         = ["jimmy@jimmycuadra.com"]
   spec.description   = %q{A Lita handler for managing DigitalOcean services.}
@@ -14,8 +14,9 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_runtime_dependency "lita", ">= 3.1"
-  spec.add_runtime_dependency "digital_ocean", ">= 1.3.0"
+  spec.add_runtime_dependency "lita", ">= 3.2"
+  spec.add_runtime_dependency "lita-keyword-arguments"
+  spec.add_runtime_dependency "digital_ocean", ">= 1.5.0"
 
   spec.add_development_dependency "bundler", "~> 1.3"
   spec.add_development_dependency "rake"

data/locales/en.yml

@@ -5,26 +5,163 @@ en:
         credentials_missing: >-
           client_id and api_key must be set in Lita's configuration to
           use the DigitalOcean commands.
+        domains:
+          create:
+            created: "Created new DNS record set for %{name}."
+          delete:
+            deleted: "Deleted DNS record set."
+          list:
+            detail: "ID: %{id}, Name: %{name}"
+          show:
+            details: >-
+              ID: %{id}, Name: %{name}, TTL: %{ttl}, Live Zone File: %{live_zone_file},
+              Error: %{error}, Zone File With Error: %{zone_file_with_error}
+        domain_records:
+          create:
+            created: "Created new DNS record: %{id}"
+          delete:
+            deleted: "Deleted DNS record."
+          edit:
+            updated: "Updated DNS record."
+          list:
+            detail: "ID: %{id}, Record Type: %{record_type}, Data: %{data}"
+          show:
+            details: >-
+              ID: %{id}, Record Type: %{record_type}, Data: %{data}, Name: %{name},
+              Priority: %{priority}, Port: %{port}, Weight: %{weight}
+        droplets:
+          create:
+            created: "Created new droplet: %{id} (%{name})"
+          delete:
+            deleted: "Deleted droplet: %{id}"
+          list:
+            detail: "ID: %{id}, Name: %{name}, IP: %{ip_address}"
+          password_reset:
+            reset: "Password reset for droplet: %{id}"
+          power_cycle:
+            cycled: "Power cycled for droplet: %{id}"
+          power_off:
+            powered_off: "Powered off droplet: %{id}"
+          power_on:
+            powered_on: "Powered on droplet: %{id}"
+          reboot:
+            rebooted: "Rebooted droplet: %{id}"
+          rebuild:
+            rebuilt: "Rebuilt droplet: %{id}"
+          resize:
+            resized: "Resized droplet: %{id}"
+          restore:
+            restored: "Restored droplet: %{id}"
+          show:
+            details: >-
+              ID: %{id}, Image ID: %{image_id}, Name: %{name}, Region ID: %{region_id}, Size ID:
+              %{size_id}, Backups active: %{backups_active}, Backups: %{formatted_backups},
+              Snapshots: %{formatted_snapshots}, IP address: %{ip_address}, Private IP address:
+              %{private_ip_address}, Locked: %{locked}, Status: %{status}
+          shutdown:
+            shut_down: "Shut down droplet: %{id}"
+          snapshot:
+            snapshotted: "Snapshotted droplet: %{id}"
         error: "DigitalOcean API error: %{message}"
         format: Format
         help:
+          domains:
+            create_key: do domains create NAME IP
+            create_value: Creates a new DNS record set for domain NAME and IP.
+            delete_key: do domains delete DOMAIN_NAME_OR_ID
+            delete_value: Deletes the DNS record set for domain DOMAIN_NAME_OR_ID.
+            list_key: do domains list
+            list_value: Lists all DNS record sets.
+            show_key: do domains show DOMAIN_NAME_OR_ID
+            show_value: Shows the DNS record set details for domain DOMAIN_NAME_OR_ID.
+          domain_records:
+            create_key: >-
+              do domain records create DOMAIN_NAME_OR_ID TYPE DATA [--name NAME]
+              [--priority PRIORITY] [--port PORT] [--weight WEIGHT]
+            create_value: Creates a new DNS record for domain DOMAIN_NAME_OR_ID.
+            delete_key: do domain records delete DOMAIN_NAME_OR_ID DOMAIN_RECORD_ID
+            delete_value: Deletes the DNS record DOMAIN_RECORD_ID for domain DOMAIN_NAME_OR_ID.
+            edit_key: >-
+              do domain records edit DOMAIN_NAME_OR_ID DOMAIN_RECORD_ID TYPE DATA [--name NAME]
+              [--priority PRIORITY] [--port PORT] [--weight WEIGHT]
+            edit_value: Edits DNS record DOMAIN_RECORD_ID for domain DOMAIN_NAME_OR_ID.
+            list_key: do domain records list DOMAIN_NAME_OR_ID
+            list_value: Lists all DNS records for domain DOMAIN_NAME_OR_ID.
+            show_key: do domain records show DOMAIN_NAME_OR_ID DOMAIN_RECORD_ID
+            show_value: Shows DNS record DOMAIN_RECORD_ID for domain DOMAIN_NAME_OR_ID.
+          droplets:
+            create_key: >-
+              do droplets create HOSTNAME SIZE_ID_OR_SLUG IMAGE_ID_OR_SLUG REGION_ID_OR_SLUG
+              [--ssh-key-ids SSH_KEY_IDS] [--private-networking] [--backups-enabled]
+            create_value: >-
+              Creates a droplet. The optional ssh_key_ids field is a comma-separated list of
+              SSH key IDs.
+            delete_key: do droplets delete ID [--scrub]
+            delete_value: Deletes a droplet, optionally writing zeroes to the disk first.
+            list_key: do droplets list
+            list_value: Lists all droplets.
+            password_reset_key: do droplets password reset ID
+            password_reset_value: Resets the root password for a droplet.
+            power_cycle_key: do droplets power cycle ID
+            power_cycle_value: Powers off and then powers on a droplet.
+            power_off_key: do droplets power off ID
+            power_off_value: Powers off a droplet.
+            power_on_key: do droplets power on ID
+            power_on_value: Powers on a droplet.
+            reboot_key: do droplets reboot ID
+            reboot_value: Reboots a droplet.
+            resize_key: do droplets resize ID SIZE_ID_OR_SLUG
+            resize_value: Resizes a droplet.
+            restore_key: do droplets restore ID IMAGE_ID
+            restore_value: Restores a droplet from an image.
+            show_key: do droplets show ID
+            show_value: Shows the details of a droplet.
+            shutdown_key: do droplets shutdown ID
+            shutdown_value: Shuts down a droplet.
+            snapshot_key: do droplets snapshot ID [NAME]
+            snapshot_value: Takes a snapshot of a droplet, optionally naming the snapshot.
+          images:
+            delete_key: do images delete ID_OR_SLUG
+            delete_value: Deletes an image with the given ID or slug.
+            list_key: "do images list [FILTER]"
+            list_value: >-
+              Lists available images, optionally filtered by FILTER ("global" or "my_images").
+            show_key: do images show ID_OR_SLUG
+            show_value: Shows the details of an image with the given ID or slug.
+          regions:
+            list_key: do regions list
+            list_value: Lists all available regions.
           ssh_keys:
             add_key: do ssh keys add NAME PUBLIC_KEY
             add_value: Adds a new SSH key.
             delete_key: do ssh keys delete ID
-            delete_value: Delete the SSH key with the given ID.
-            edit_key: "do ssh keys edit ID [name=NAME] [public_key=PUBLIC_KEY]"
-            edit_value: Change the NAME and/or PUBLIC_KEY of the SSH key with the given ID.
+            delete_value: Deletes the SSH key with the given ID.
+            edit_key: "do ssh keys edit ID [--name NAME] [--public-key PUBLIC_KEY]"
+            edit_value: Changes the NAME and/or PUBLIC_KEY of the SSH key with the given ID.
             list_key: do ssh keys list
             list_value: Lists all SSH keys.
             show_key: do ssh keys show ID
             show_value: Shows the public key for SSH key with the given ID.
+          sizes:
+            list_key: do sizes list
+            list_value: Lists all the possible image sizes.
+        images:
+          details: >-
+            ID: %{id}, Name: %{name}, Slug: %{slug}, Distribution: %{distribution},
+            Public: %{public}, Regions: %{formatted_regions},
+            Region Slugs: %{formatted_region_slugs}
+          delete:
+            deleted: "Deleted image: %{image_id}"
+        regions:
+          details: "ID: %{id}, Name: %{name}, Slug: %{slug}"
+        sizes:
+          details: "ID: %{id}, Name: %{name}, Slug: %{slug}"
         ssh_keys:
           add:
-            created: "Created new SSH key: %{message}"
+            created: "Created new SSH key: %{id} (%{name}): %{ssh_pub_key}"
           delete:
             deleted: "Deleted SSH key: %{key_id}"
           edit:
-            updated: "Updated SSH key: %{message}"
+            updated: "Updated SSH key: %{id} (%{name}): %{ssh_pub_key}"
           list:
             empty: No SSH keys have been added yet.

data/spec/lita/handlers/digitalocean/base_spec.rb

@@ -0,0 +1,39 @@
+require "spec_helper"
+
+handler_class = Class.new(Lita::Handlers::Digitalocean::Base) do
+  route /do droplets list/, :list, command: true
+
+  def self.default_config(config)
+  end
+
+  def list(response)
+    do_call(response) do |client|
+      { status: "error", message: "Something went wrong" }
+    end
+  end
+end
+
+describe handler_class, lita_handler: true do
+  describe "#do_call" do
+    it "responds with an error if the DigitalOcean API key is not set" do
+      send_command("do droplets list")
+      expect(replies.last).to include("client_id and api_key must be set")
+    end
+
+    it "responds with an error if the DigitalOcean client ID is not set" do
+      Lita.config.handlers.digitalocean.api_key = "secret"
+      send_command("do droplets list")
+      expect(replies.last).to include("client_id and api_key must be set")
+    end
+
+    it "responds with an error if the DigitalOcean API responds with an error" do
+      Lita.config.handlers.digitalocean.tap do |config|
+        config.api_key = "secret"
+        config.client_id = "secret"
+      end
+
+      send_command("do droplets list")
+      expect(replies.last).to eq("DigitalOcean API error: Something went wrong")
+    end
+  end
+end

data/spec/lita/handlers/digitalocean/domain_record_spec.rb

@@ -0,0 +1,118 @@
+require "spec_helper"
+
+describe Lita::Handlers::Digitalocean::DomainRecord, lita_handler: true do
+  it do
+    routes_command(
+      "do domain records create example.com txt 'some value' --name example.com"
+    ).to(:create)
+  end
+  it { routes_command("do domain records delete example.com 123").to(:delete) }
+  it do
+    routes_command(
+      "do domain records edit example.com 123 txt 'some value' name=example.com"
+    ).to(:edit)
+  end
+  it { routes_command("do domain records list example.com").to(:list) }
+  it { routes_command("do domain records show example.com 123").to(:show) }
+
+  let(:client) { instance_double("::DigitalOcean::API", domains: client_domains) }
+  let(:client_domains) { instance_double("::DigitalOcean::Resource::Domain") }
+
+  before do
+    Lita.config.handlers.digitalocean = Lita::Config.new
+    Lita.config.handlers.digitalocean.tap do |config|
+      config.client_id = "CLIENT_ID"
+      config.api_key = "API_KEY"
+    end
+
+    allow(Lita::Authorization).to receive(:user_in_group?).with(
+      user,
+      :digitalocean_admins
+    ).and_return(true)
+
+    allow(::DigitalOcean::API).to receive(:new).and_return(client)
+  end
+
+  describe "#create" do
+    it "creates a new domain record" do
+      allow(client_domains).to receive(:create_record).with(
+        "example.com",
+        data: "@",
+        name: "foo",
+        port: "456",
+        priority: "123",
+        record_type: "srv",
+        weight: "789"
+      ).and_return(status: "OK", record: { id: 123 })
+      send_command("do domain records create example.com srv @ --name foo --priority 123 --port 456 --weight 789")
+      expect(replies.last).to eq("Created new DNS record: 123")
+    end
+  end
+
+  describe "#delete" do
+    it "deletes a domain record" do
+      allow(client_domains).to receive(:delete_record).with("123", "456").and_return(status: "OK")
+      send_command("do domain records delete 123 456")
+      expect(replies.last).to eq("Deleted DNS record.")
+    end
+  end
+
+  describe "#edit" do
+    it "edits a domain record" do
+      allow(client_domains).to receive(:edit_record).with(
+        "example.com",
+        "123",
+        data: "example.com",
+        record_type: "cname",
+        name: "www.example.com"
+      ).and_return(status: "OK")
+      send_command(
+        "do domain records edit example.com 123 cname example.com --name www.example.com"
+      )
+      expect(replies.last).to eq("Updated DNS record.")
+    end
+  end
+
+  describe "#list" do
+    it "responds with a list of all domain records for the given record set" do
+      allow(client_domains).to receive(:list_records).with("example.com").and_return(
+        status: "OK",
+        records: [{
+          id: 123,
+          record_type: "A",
+          data: "10.0.0.0"
+        }, {
+          id: 456,
+          record_type: "CNAME",
+          data: "@"
+        }]
+      )
+      send_command("do domain records list example.com")
+      expect(replies).to eq([
+        "ID: 123, Record Type: A, Data: 10.0.0.0",
+        "ID: 456, Record Type: CNAME, Data: @"
+      ])
+    end
+  end
+
+  describe "#show" do
+    it "responds with the details of the domain record" do
+      allow(client_domains).to receive(:show_record).with("example.com", "123").and_return(
+        status: "OK",
+        record: {
+          id: 123,
+          record_type: "A",
+          data: "10.0.0.0",
+          name: "example.com",
+          priority: 123,
+          port: 456,
+          weight: 789
+        }
+      )
+      send_command("do domain records show example.com 123")
+      expect(replies.last).to eq(
+"ID: 123, Record Type: A, Data: 10.0.0.0, Name: example.com, Priority: 123, Port: 456, Weight: 789"
+      )
+    end
+  end
+end