staticpages-plugin-digitalocean 0.3.5

data/lib/kontena/plugin/digital_ocean/nodes/restart_command.rb

@@ -0,0 +1,34 @@
+require 'kontena/plugin/digital_ocean/prompts'
+
+module Kontena::Plugin::DigitalOcean::Nodes
+  class RestartCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Plugin::DigitalOcean::Prompts
+
+    parameter "[NAME]", "Node name"
+    option "--token", "TOKEN", "DigitalOcean API token", environment_variable: "DO_TOKEN"
+
+    def execute
+      suppress_warnings # until DO merges resource_kit pr #32
+      require 'kontena/machine/digital_ocean'
+      require_api_url
+      require_current_grid
+      do_token = ask_do_token
+
+      node_name = ask_node(require_token)
+      client = DropletKit::Client.new(access_token: do_token)
+      droplet = client.droplets.all.find{|d| d.name == node_name}
+      if droplet
+        spinner "Restarting DigitalOcean droplet #{pastel.cyan(name)} " do
+          client.droplet_actions.reboot(droplet_id: droplet.id)
+          sleep 1 until client.droplets.find(id: droplet.id).status == 'active'
+        end
+      else
+        exit_with_error "Cannot find droplet #{pastel.cyan(name)} in DigitalOcean"
+      end
+    ensure
+      resume_warnings
+    end
+  end
+end

data/lib/kontena/plugin/digital_ocean/nodes/terminate_command.rb

@@ -0,0 +1,36 @@
+require 'kontena/plugin/digital_ocean/prompts'
+
+module Kontena::Plugin::DigitalOcean::Nodes
+  class TerminateCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Plugin::DigitalOcean::Prompts
+
+    parameter "[NAME]", "Node name"
+    option "--token", "TOKEN", "DigitalOcean API token", environment_variable: "DO_TOKEN"
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+
+    def execute
+      suppress_warnings # until DO merges resource_kit pr #32
+      require 'kontena/machine/digital_ocean'
+      require_api_url
+      require_current_grid
+      token = require_token
+      node_name = ask_node(token)
+      do_token = ask_do_token
+      confirm_command(node_name) unless forced?
+
+      grid = client(require_token).get("grids/#{current_grid}")
+      destroyer = destroyer(client(token), do_token)
+      destroyer.run!(grid, node_name)
+    ensure
+      resume_warnings
+    end
+
+    # @param [Kontena::Client] client
+    # @param [String] token
+    def destroyer(client, token)
+      Kontena::Machine::DigitalOcean::NodeDestroyer.new(client, token)
+    end
+  end
+end

data/lib/kontena/plugin/digital_ocean/prompts.rb

@@ -0,0 +1,112 @@
+module Kontena::Plugin::DigitalOcean::Prompts
+
+  # Until DO merges https://github.com/digitalocean/resource_kit/pull/32
+  def suppress_warnings
+    @original_verbosity = $VERBOSE
+    $VERBOSE = nil
+  end
+
+  def resume_warnings
+    $VERBOSE = @original_verbosity
+  end
+
+  def ask_ssh_key(do_token)
+    manager = Kontena::Machine::DigitalOcean::SshKeyManager.new(do_token)
+
+    if self.ssh_key
+      public_key = File.read(self.ssh_key).strip
+    else
+      keys = manager.list
+      key = :new
+
+      default_path = File.join(Dir.home, '.ssh', 'id_rsa.pub')
+      default = File.exist?(default_path) ? File.read(default_path).strip : nil
+
+      unless keys.empty?
+        key = prompt.select("Choose SSH key:") do |menu|
+          i = 1
+          keys.each do |item|
+            menu.choice "#{item.name} (#{item.fingerprint})" , item
+            menu.default i if item.public_key == default
+            i += 1
+          end
+          menu.choice "Create new SSH key", :new
+        end
+      end
+
+      if key == :new
+
+        public_key = prompt.ask('SSH public key: (enter an ssh key in OpenSSH format "ssh-xxx xxxxx key_name")', default: default) do |q|
+          q.validate /^ssh-rsa \S+ \S+$/
+        end
+      else
+        return key.id
+      end
+    end
+    manager.find_or_create_by_public_key(public_key).id
+  end
+
+  def ask_do_token
+    if self.token.nil?
+      prompt.mask('DigitalOcean API token:')
+    else
+      self.token
+    end
+  end
+
+  def ask_droplet_region(do_token)
+    if self.region.nil?
+      prompt.select("Choose a datacenter region:") do |menu|
+        do_client = DropletKit::Client.new(access_token: do_token)
+        do_client.regions.all.sort_by{|r| r.slug }.each{ |region|
+          menu.choice region.name, region.slug
+        }
+      end
+    else
+      self.region
+    end
+  end
+
+  def ask_droplet_size(do_token, do_region)
+    if self.size.nil?
+      prompt.select("Choose droplet size:") do |menu|
+        do_client = DropletKit::Client.new(access_token: do_token)
+        do_client.sizes.all.to_a.select{ |s| s.memory > 1000 }.sort_by{|s| s.memory }.each{ |size|
+          #p size
+          if size.regions.include?(do_region)
+            memory = size.memory.to_i / 1024
+            menu.choice "#{size.slug}: #{memory}GB/#{size.vcpus}CPU/#{size.disk}GB ($#{size.price_monthly.to_i}/mo)", size.slug
+          end
+        }
+      end
+    else
+      self.size
+    end
+  end
+
+  def ask_node(token)
+    if self.name.nil?
+      nodes = client(token).get("grids/#{current_grid}/nodes")
+      nodes = nodes['nodes'].select{ |n|
+        n['labels'] && n['labels'].include?('provider=digitalocean'.freeze)
+      }
+      raise "Did not find any nodes with label provider=digitalocean" if nodes.size == 0
+      prompt.select("Select node:") do |menu|
+        nodes.sort_by{|n| n['node_number'] }.reverse.each do |node|
+          initial = node['initial_member'] ? '(initial) ' : ''
+          menu.choice "#{node['name']} #{initial}", node['name']
+        end
+      end
+    else
+      self.name
+    end
+  end
+
+  def ask_channel
+    prompt.select('Select Container Linux channel:') do |menu|
+      menu.choice 'Stable', 'stable'
+      menu.choice 'Beta', 'beta'
+      menu.choice 'Alpha', 'alpha'
+    end
+  end
+end

data/lib/kontena/plugin/digital_ocean.rb

@@ -0,0 +1,7 @@
+module Kontena
+  module Plugin
+    module DigitalOcean
+      VERSION = "0.3.5"
+    end
+  end
+end

data/lib/kontena/plugin/digital_ocean_command.rb

@@ -0,0 +1,4 @@
+class Kontena::Plugin::DigitalOceanCommand < Kontena::Command
+  subcommand 'master', 'DigitalOcean master related commands', load_subcommand('kontena/plugin/digital_ocean/master_command')
+  subcommand 'node', 'DigitalOcean node related commands', load_subcommand('kontena/plugin/digital_ocean/node_command')
+end

data/lib/kontena_cli_plugin.rb

@@ -0,0 +1,5 @@
+require 'kontena_cli'
+require 'kontena/plugin/digital_ocean'
+require 'kontena/cli/subcommand_loader'
+
+Kontena::MainCommand.register("digitalocean", "DigitalOcean specific commands", Kontena::Cli::SubcommandLoader.new('kontena/plugin/digital_ocean_command'))

data/staticpages-plugin-digitalocean.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'kontena/plugin/digital_ocean'
+
+Gem::Specification.new do |spec|
+  spec.name          = "staticpages-plugin-digitalocean"
+  spec.version       = Kontena::Plugin::DigitalOcean::VERSION
+  spec.authors       = ["Pavel Tsurbeleu"]
+  spec.email         = ["staticpagesio@gmail.com"]
+
+  spec.summary       = "StaticPages DigitalOcean plugin"
+  spec.description   = "StaticPages DigitalOcean plugin"
+  spec.homepage      = "https://staticpages.io"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.require_paths = ["lib"]
+
+  # NOTE: Exclude files not relevant to the plugin
+  spec.files        -= %w[ Dockerfile Makefile docker-compose.yml docker-compose.dev.yml Jenkinsfile .dockerignore ]
+
+  spec.add_runtime_dependency 'kontena-cli', '1.5.2'
+  spec.add_runtime_dependency 'droplet_kit', '~> 2.2'
+  spec.add_runtime_dependency 'activesupport', '~> 4.0'
+  spec.add_runtime_dependency 'pastel', '0.7.2'
+  spec.add_development_dependency "bundler", "~> 1.11"
+  spec.add_development_dependency "rake", "~> 10.0"
+end

metadata

@@ -0,0 +1,153 @@
+--- !ruby/object:Gem::Specification
+name: staticpages-plugin-digitalocean
+version: !ruby/object:Gem::Version
+  version: 0.3.5
+platform: ruby
+authors:
+- Pavel Tsurbeleu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-07-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: kontena-cli
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.5.2
+- !ruby/object:Gem::Dependency
+  name: droplet_kit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: pastel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.2
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: StaticPages DigitalOcean plugin
+email:
+- staticpagesio@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".gitmodules"
+- ".rspec"
+- Gemfile
+- LICENSE.txt
+- README.md
+- lib/kontena/machine/digital_ocean.rb
+- lib/kontena/machine/digital_ocean/cloudinit.yml
+- lib/kontena/machine/digital_ocean/cloudinit_master.yml
+- lib/kontena/machine/digital_ocean/master_destroyer.rb
+- lib/kontena/machine/digital_ocean/master_provisioner.rb
+- lib/kontena/machine/digital_ocean/node_destroyer.rb
+- lib/kontena/machine/digital_ocean/node_provisioner.rb
+- lib/kontena/machine/digital_ocean/ssh_key_manager.rb
+- lib/kontena/plugin/digital_ocean.rb
+- lib/kontena/plugin/digital_ocean/master/create_command.rb
+- lib/kontena/plugin/digital_ocean/master/terminate_command.rb
+- lib/kontena/plugin/digital_ocean/master_command.rb
+- lib/kontena/plugin/digital_ocean/node_command.rb
+- lib/kontena/plugin/digital_ocean/nodes/create_command.rb
+- lib/kontena/plugin/digital_ocean/nodes/restart_command.rb
+- lib/kontena/plugin/digital_ocean/nodes/terminate_command.rb
+- lib/kontena/plugin/digital_ocean/prompts.rb
+- lib/kontena/plugin/digital_ocean_command.rb
+- lib/kontena_cli_plugin.rb
+- staticpages-plugin-digitalocean.gemspec
+homepage: https://staticpages.io
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.4
+signing_key: 
+specification_version: 4
+summary: StaticPages DigitalOcean plugin
+test_files: []