sunzi-vps 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 22de99d93f1dcc0b6189b27a8f76827c611f85b142f75ac8220725a909f1746b
-  data.tar.gz: 4d10928a3be79b0c81e103eff8b3fcd2dc9c88eaec980584d275c91c9d341539
+  metadata.gz: 3659a8a697bee5226a4b7c322f72968b1c5bad29d08a993baf0c48225c2da733
+  data.tar.gz: 0ff944661b1dcd0d225a7308ba1bfb3db6568e0ee559685cdb75d0ab9f010296
 SHA512:
-  metadata.gz: 655ef8b565505126a3a20438c1f79699d84f50aba835680cf3ea9ca9565935d27a6456260e5ef14c788ec2eee4fac34f8e2fcf5c8864ef449c5bae4cbe1b4a2d
-  data.tar.gz: 236e67762ce6e2aa1dbf82d39f795b5f8d612d35d19b3dd14c0d6b4c43e4fbe05bd25c0d691046cf22620bbdc8d6eaf0b1c6e426be6eca76f806835d99121c82
+  metadata.gz: 1ccdc5db32aa135bb6049d5d4c7a4aac1a2c91384053d198327c2797e7486689c770a4aefac68636669a48a22ed221f7e32879f603351e74702d7b17e8353458
+  data.tar.gz: 28b16c5384243e6c7f71579bed968f7bbcef804db03022ed43a22425f9f46488664731704f726c28d8e75c85b90ca090524d6ddc92934a0ff337b0d7fae8f720

data/README.md

@@ -2,6 +2,8 @@
 
 sunzi-vps is a simple command line utility to create and/or destroy VPS instances interactively.
 
+Works with [Linode](https://www.linode.com/), [DigitalOcean](https://www.digitalocean.com/) and [Vultr](https://www.vultr.com/)
+
 It works as a plugin for [sunzi](https://github.com/kenn/sunzi), but can be used as a standalone tool.
 
 ![Sunzi for Linode](http://farm8.staticflickr.com/7210/6783789868_ab89010d5c.jpg)

data/lib/sunzi/vps/api.rb

@@ -3,10 +3,12 @@ require 'sunzi/vps/mapping'
 require 'sunzi/vps/compute/base'
 require 'sunzi/vps/compute/linode'
 require 'sunzi/vps/compute/digital_ocean'
+require 'sunzi/vps/compute/vultr'
 
 require 'sunzi/vps/dns/base'
 require 'sunzi/vps/dns/linode'
 require 'sunzi/vps/dns/digital_ocean'
+require 'sunzi/vps/dns/vultr'
 
 module Sunzi
   module Vps
@@ -25,6 +27,9 @@ module Sunzi
             ::Linode.new(api_key: config.api_key)
           when 'digital_ocean'
             DropletKit::Client.new(access_token: config.api_key)
+          when 'vultr'
+            Vultr.api_key = config.api_key
+            Vultr
           end
         end
       end

data/lib/sunzi/vps/cli.rb

@@ -5,18 +5,18 @@ module Sunzi
   module Vps
     class Cli < Thor
 
-      desc 'init [provider]', 'Generate VPS config file (provider: linode, digital_ocean)'
+      desc 'init [provider]', "Generate VPS config file (provider: #{Mapping.keys.join(', ')})"
       def init(provider)
         Sunzi::Vps::Init.new.run(provider)
       end
 
-      desc 'up [provider]', 'Set up a new instance (provider: linode, digital_ocean)'
+      desc 'up [provider]', "Set up a new instance (provider: #{Mapping.keys.join(', ')})"
       def up(provider)
         api = Sunzi::Vps::Api.new(provider)
         api.compute.up
       end
 
-      desc 'down [provider]', 'Tear down an existing instance (provider: linode, digital_ocean)'
+      desc 'down [provider]', "Tear down an existing instance (provider: #{Mapping.keys.join(', ')})"
       def down(provider)
         api = Sunzi::Vps::Api.new(provider)
         api.compute.down

data/lib/sunzi/vps/compute/base.rb

@@ -36,7 +36,7 @@ module Sunzi
           create_file instance_config_path, YAML.dump(@instance)
 
           # Register IP to DNS
-          api.dns.add(@fqdn, @public_ip)
+          api.dns.add(@fqdn, @instance[:public_ip])
         end
 
         def down
@@ -56,7 +56,7 @@ module Sunzi
           do_down
 
           # Delete DNS record
-          api.dns.delete @instance.send(ip_key)
+          api.dns.delete @instance.public_ip
 
           # Remove the instance config file
           remove_file instance_config_path

data/lib/sunzi/vps/compute/digital_ocean.rb

@@ -40,8 +40,8 @@ module Sunzi
             droplet = client.droplets.find(id: @droplet_id)
           end
 
-          @public_ip = droplet.networks.v4.first.ip_address
-          say "Done. ip address = #{@public_ip}"
+          public_ip = droplet.networks.v4.first.ip_address
+          say "Done. ip address = #{public_ip}"
 
           @instance = {
             droplet_id: @droplet_id,
@@ -49,7 +49,7 @@ module Sunzi
             host: @host,
             fqdn: @fqdn,
             name: @name,
-            ip_address: @public_ip,
+            public_ip:  public_ip,
             size:       @attributes[:size],
             region:     @attributes[:region],
             image:      @attributes[:image],
@@ -80,10 +80,6 @@ module Sunzi
           say 'deleting droplet...'
           client.droplets.delete(id: @instance[:droplet_id])
         end
-
-        def ip_key
-          :ip_address
-        end
       end
     end
   end

data/lib/sunzi/vps/compute/linode.rb

@@ -74,7 +74,7 @@ module Sunzi
           # Add a private IP
           say "Adding a private IP..."
           result = client.linode.ip.list(:LinodeID => @linodeid)
-          @public_ip = result.first.ipaddress
+          public_ip = result.first.ipaddress
           result = client.linode.ip.addprivate(:LinodeID => @linodeid)
           result = client.linode.ip.list(:LinodeID => @linodeid).find{|i| i.ispublic == 0 }
           @private_ip = result.ipaddress
@@ -98,7 +98,7 @@ module Sunzi
             :root_diskid => @root_diskid,
             :swap_diskid => @swap_diskid,
             :config_id => @config_id,
-            :public_ip => @public_ip,
+            :public_ip => public_ip,
             :private_ip => @private_ip,
           }
 
@@ -136,10 +136,6 @@ module Sunzi
           client.linode.delete(@linode_id_hash.merge(:skipChecks => 1))
         end
 
-        def ip_key
-          :public_ip
-        end
-
         def wait_for(action)
           begin
             sleep 3

data/lib/sunzi/vps/compute/vultr.rb

@@ -0,0 +1,91 @@
+module Sunzi
+  module Vps
+    class Compute
+      class Vultr < Base
+
+        def do_up
+          result = choose(:plan, client::Plans.list)
+          choose(:region, client::Regions.list, ids: result['available_locations'])
+
+          # Choose an image
+          result = client::OS.list
+          if config.distributions_filter
+            result[:result].reject!{|id, hash| hash['name'] !~ Regexp.new(config.distributions_filter, Regexp::IGNORECASE) }
+          end
+          choose(:os, result)
+
+          # Go ahead?
+          proceed?
+
+          # Create
+          say 'creating a new instance...'
+          result = client::Server.create(
+            label: @name,
+            DCID: @attributes[:region],
+            VPSPLANID: @attributes[:plan],
+            OSID: @attributes[:os]
+          )
+
+          subid = result[:result]['SUBID']
+
+          say "Created a new instance (id: #{subid}). Booting..."
+
+          begin
+            sleep 3
+            result = client::Server.list_ipv4(SUBID: subid)
+            public_ip = result[:result][subid].first['ip']
+          end while public_ip == '0.0.0.0'
+
+          say "Done. ip address = #{public_ip}"
+
+          @instance = {
+            subid: subid,
+            env:  @env,
+            host: @host,
+            fqdn: @fqdn,
+            name: @name,
+            public_ip:  public_ip,
+            plan:       @attributes[:plan],
+            region:     @attributes[:region],
+            os:         @attributes[:os],
+          }
+
+        end
+
+        def choose(key, result, ids: nil)
+          abort "API returned for #{key}!" if result[:status] != 200
+          result = result[:result].map{|id,hash| hash.merge(id: id) }
+
+          # Filter out items not included in ids
+          if ids
+            ids = ids.map(&:to_s)
+            result.reject!{|h| !ids.include?(h[:id].to_s) }
+          end
+
+          rows = result.map do |h|
+            h.values.map do |v|
+              next v unless v.is_a?(Array)
+              if v.size > 5
+                v.first(5).join(', ') + ', ...'
+              else
+                v.join(', ')
+              end
+            end
+          end
+
+          table = Terminal::Table.new headings: result.first.keys, rows: rows
+          say table
+
+          @attributes[key] = ask("which #{key}?: ", default: result.first[:id], limited_to: result.map{|i| i[:id] })
+          result.find{|i| i[:id].to_s == @attributes[key] }
+        end
+
+        def do_down
+          say 'deleting instance...'
+          result = client::Server.destroy(SUBID: @instance[:subid])
+          abort result[:result] if result[:status] != 200
+        end
+      end
+    end
+  end
+end

data/lib/sunzi/vps/dependency.rb

@@ -4,6 +4,7 @@ module Sunzi
       [
         ['linode', '~> 0.7'],
         ['droplet_kit', '~> 2.2'],
+        ['vultr', '~> 0.3'],
       ].each do |name, version|
         Sunzi::Dependency.new(name, version)
       end

data/lib/sunzi/vps/dns/vultr.rb

@@ -0,0 +1,31 @@
+module Sunzi
+  module Vps
+    class DNS
+      class Vultr < Base
+
+        def verify
+          result = client::DNS.list
+          domain = result[:result].find{|h| h['domain'] == @zone }
+          abort_with "zone for #{@zone} was not found on Vultr DNS!" unless domain
+        end
+
+        def add(fqdn, ip)
+          say 'adding the public IP to Vultr DNS...'
+          client::DNS.create_record(domain: @zone, name: fqdn.sub('.' + @zone, ''), type: 'A', data: ip)
+        end
+
+        def delete(ip)
+          say 'deleting the public IP from Vultr DNS...'
+
+          result = client::DNS.records(domain: @zone)
+          hash = result[:result].find{|i| i['type'] == 'A' && i['data'] == ip }
+
+          abort_with "ip address #{ip} was not found on Vultr DNS!" unless hash['RECORDID']
+
+          client::DNS.delete_record(domain: @zone, RECORDID: hash['RECORDID'])
+        end
+
+      end
+    end
+  end
+end

data/lib/sunzi/vps/mapping.rb

@@ -9,6 +9,10 @@ module Sunzi
         gem: 'droplet_kit',
         klass: 'DigitalOcean',
       },
+      vultr: {
+        gem: 'vultr',
+        klass: 'Vultr',
+      },
     }.freeze
   end
 end

data/sunzi-vps.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name          = 'sunzi-vps'
-  spec.version       = '0.1.0' # retrieve this value by: Gem.loaded_specs['sunzi'].version.to_s
+  spec.version       = '0.2.0' # retrieve this value by: Gem.loaded_specs['sunzi'].version.to_s
   spec.authors       = ['Kenn Ejima']
   spec.email         = ['kenn.ejima@gmail.com']
   spec.summary       = %q{Sunzi VPS}
@@ -24,4 +24,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'webmock', '~> 3.2'
   spec.add_development_dependency 'linode', '~> 0.7'
   spec.add_development_dependency 'droplet_kit', '~> 2.2'
+  spec.add_development_dependency 'vultr', '~> 0.3'
 end

data/templates/digital_ocean.yml

@@ -1,6 +1,5 @@
 ---
 api_key: your_api_key
-client_id: your_client_id
 
 # add / remove environments
 environments:

data/templates/vultr.yml

@@ -0,0 +1,17 @@
+---
+api_key: your_api_key
+
+# add / remove environments
+environments:
+  - production
+  - staging
+fqdn:
+  zone: example.com
+  production: '%{host}.example.com'
+  staging: '%{host}.staging.example.com'
+name:
+  production: 'example-%{host}'
+  staging: 'example-staging-%{host}'
+
+# filter out large lists by keyword
+distributions_filter: debian

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sunzi-vps
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Kenn Ejima
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-01-18 00:00:00.000000000 Z
+date: 2018-01-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sunzi
@@ -150,6 +150,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: vultr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
 description: Simple CLI to create and/or destroy VPS instances
 email:
 - kenn.ejima@gmail.com
@@ -170,15 +184,18 @@ files:
 - lib/sunzi/vps/compute/base.rb
 - lib/sunzi/vps/compute/digital_ocean.rb
 - lib/sunzi/vps/compute/linode.rb
+- lib/sunzi/vps/compute/vultr.rb
 - lib/sunzi/vps/dependency.rb
 - lib/sunzi/vps/dns/base.rb
 - lib/sunzi/vps/dns/digital_ocean.rb
 - lib/sunzi/vps/dns/linode.rb
+- lib/sunzi/vps/dns/vultr.rb
 - lib/sunzi/vps/init.rb
 - lib/sunzi/vps/mapping.rb
 - sunzi-vps.gemspec
 - templates/digital_ocean.yml
 - templates/linode.yml
+- templates/vultr.yml
 homepage: https://github.com/kenn/sunzi-vps
 licenses:
 - MIT