krates-plugin-vagrant 0.3.1

data/lib/kontena/machine/vagrant/Vagrantfile.node.rb.erb

@@ -0,0 +1,38 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+# Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
+VAGRANTFILE_API_VERSION = "2"
+
+Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
+  config.vm.provider :virtualbox do |v|
+    v.check_guest_additions = false
+    v.functional_vboxsf     = false
+  end
+
+  if Vagrant.has_plugin?("vagrant-vbguest") then
+    config.vbguest.auto_update = false
+  end
+
+  config.vm.define "<%= name %>" do |docker|
+    docker.vm.box = "coreos-<%= coreos_channel %>"
+    docker.vm.box_url = "http://stable.release.core-os.net/amd64-usr/current/coreos_production_vagrant.json"
+    <% if network_address == "dhcp" %>
+    docker.vm.network "private_network", type: "dhcp"
+    <% else  %>
+    docker.vm.network "private_network", ip: "<%= network_address %>"
+    <% end %>
+    docker.vm.hostname = "<%= name %>"
+    docker.vm.provider "virtualbox" do |vb|
+      vb.name = "<%= name %>"
+      vb.gui = false
+      vb.cpus = <%= cpus %>
+      vb.memory = <%= memory %>
+      vb.auto_nat_dns_proxy = false
+      vb.customize ["modifyvm", :id, "--natdnsproxy1", "off" ]
+      vb.customize ["modifyvm", :id, "--natdnshostresolver1", "off" ]
+    end
+    docker.vm.provision :file, :source => "<%= cloudinit %>", :destination => "/tmp/vagrantfile-user-data"
+    docker.vm.provision :shell, :inline => "mv /tmp/vagrantfile-user-data /var/lib/coreos-vagrant/", :privileged => true
+  end
+end

data/lib/kontena/machine/vagrant/cloudinit.yml

@@ -0,0 +1,92 @@
+#cloud-config
+write_files:
+  - path: /etc/kontena-agent.env
+    permissions: 0600
+    owner: root
+    content: |
+      KONTENA_URI=<%= master_uri %>
+      KONTENA_TOKEN=<%= grid_token %>
+      KONTENA_PEER_INTERFACE=eth1
+      KONTENA_VERSION=<%= version %>
+  - path: /etc/systemd/system/docker.service.d/50-kontena.conf
+    content: |
+        [Service]
+        Environment='DOCKER_OPTS=--insecure-registry="10.81.0.0/16" --bip="172.17.43.1/16" -s overlay2'
+        Environment='DOCKER_CGROUPS=--exec-opt native.cgroupdriver=cgroupfs'
+  - path: /etc/sysctl.d/99-inotify.conf
+    owner: root
+    permissions: 0644
+    content: |
+      fs.inotify.max_user_instances = 8192
+coreos:
+  update:
+      reboot-strategy: off
+  units:
+    - name: 00-eth.network
+      runtime: true
+      content: |
+        [Match]
+        Name=eth*
+        [Network]
+        DHCP=yes
+        DNS=172.17.43.1
+        DNS=172.28.128.1
+        DNS=8.8.8.8
+        DOMAINS=kontena.local
+        [DHCP]
+        UseDNS=false
+    - name: etcd2.service
+      command: start
+      enable: true
+      content: |
+        Description=etcd 2.0
+        After=docker.service
+        [Service]
+        Restart=always
+        RestartSec=5
+        ExecStart=/usr/bin/docker logs --tail=10 -f kontena-etcd
+    - name: 50-docker.network
+      mask: true
+    - name: 50-docker-veth.network
+      mask: true
+    - name: zz-default.network
+      runtime: false
+      content: |
+        # default should only match real network interfaces
+        [Match]
+        Name=eth*
+
+        [Network]
+        DHCP=yes
+
+        [DHCP]
+        UseMTU=true
+        UseDomains=true
+    - name: kontena-agent.service
+      command: start
+      enable: true
+      content: |
+        [Unit]
+        Description=kontena-agent
+        After=network-online.target
+        After=docker.service
+        Description=Kontena Agent
+        Documentation=http://www.kontena.io/
+        Requires=network-online.target
+        Requires=docker.service
+
+        [Service]
+        Restart=always
+        RestartSec=5
+        TimeoutStartSec=0
+        EnvironmentFile=/etc/kontena-agent.env
+        ExecStartPre=-/usr/bin/docker stop kontena-agent
+        ExecStartPre=-/usr/bin/docker rm kontena-agent
+        ExecStartPre=/usr/bin/docker pull kontena/agent:${KONTENA_VERSION}
+        ExecStart=/usr/bin/docker run --name kontena-agent \
+            --env-file /etc/kontena-agent.env \
+            -v=/var/run/docker.sock:/var/run/docker.sock \
+            -v=/etc/kontena-agent.env:/etc/kontena.env \
+            --net=host \
+            kontena/agent:${KONTENA_VERSION}
+        ExecStop=/usr/bin/docker stop kontena-agent

data/lib/kontena/machine/vagrant/common.rb

@@ -0,0 +1,21 @@
+require 'fileutils'
+require 'erb'
+require 'open3'
+require 'json'
+
+module Kontena
+  module Machine
+    module Vagrant
+      module Common
+
+        def erb(template, vars)
+          ERB.new(template).result(OpenStruct.new(vars).instance_eval { binding })
+        end
+
+        def run_command(cmd)
+          exit $?.exitstatus unless system(cmd)
+        end
+      end
+    end
+  end
+end

data/lib/kontena/machine/vagrant/master_destroyer.rb

@@ -0,0 +1,21 @@
+require_relative 'common'
+
+module Kontena
+  module Machine
+    module Vagrant
+      class MasterDestroyer
+        include Kontena::Machine::Vagrant::Common
+        include Kontena::Cli::ShellSpinner
+
+        def run!
+          vagrant_path = "#{Dir.home}/.kontena/vagrant_master"
+          Dir.chdir(vagrant_path) do
+            spinner "Triggering termination of Kontena Master from Vagrant"
+            run_command('vagrant destroy -f')
+            FileUtils.remove_entry_secure(vagrant_path)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kontena/machine/vagrant/master_provisioner.rb

@@ -0,0 +1,96 @@
+require_relative 'common'
+
+module Kontena
+  module Machine
+    module Vagrant
+      class MasterProvisioner
+        include RandomName
+        include Kontena::Machine::Common
+        include Kontena::Machine::Vagrant::Common
+        include Kontena::Cli::Common
+        include Kontena::Cli::ShellSpinner
+
+        API_URL = 'http://192.168.66.100:8080'
+        attr_reader :client
+
+        def initialize
+          @client = Excon.new(API_URL)
+        end
+
+        def run!(opts)
+          name = opts[:name] || generate_name
+          version = opts[:version]
+          memory = opts[:memory] || 1024
+          vault_secret = opts[:vault_secret]
+          vault_iv = opts[:vault_iv]
+          initial_admin_code = opts[:initial_admin_code]
+          coreos_channel = opts[:coreos_channel]
+          vagrant_path = "#{Dir.home}/.kontena/vagrant_master/"
+          if Dir.exist?(vagrant_path)
+            puts "Oops... cannot create Kontena Master because installation path already exists."
+            puts "If you are sure that no Kontena Masters exist on this machine, remove folder: #{vagrant_path}"
+            abort
+          end
+          FileUtils.mkdir_p(vagrant_path)
+
+          template = File.join(__dir__ , '/Vagrantfile.master.rb.erb')
+          cloudinit_template = File.join(__dir__ , '/cloudinit.yml')
+          vars = {
+            name: name,
+            server_name: name.sub('kontena-master-', ''),
+            version: version,
+            memory: memory,
+            vault_secret: vault_secret,
+            initial_admin_code: initial_admin_code,
+            vault_iv: vault_iv,
+            coreos_channel: coreos_channel,
+            cloudinit: "#{vagrant_path}/cloudinit.yml"
+          }
+          spinner "Creating Vagrant config " do
+            vagrant_data = erb(File.read(template), vars)
+            cloudinit = erb(File.read(cloudinit_template), vars)
+            File.write("#{vagrant_path}/Vagrantfile", vagrant_data)
+            File.write("#{vagrant_path}/cloudinit.yml", cloudinit)
+          end
+
+          Dir.chdir(vagrant_path) do
+            spinner "Triggering CoreOS Container Linux box update"
+            system('vagrant box update')
+            spinner "Executing 'vagrant up'"
+            run_command('vagrant up')
+            spinner "'vagrant up' executed successfully"
+          end
+
+          spinner "Waiting for #{name.colorize(:cyan)} to start " do
+            sleep 1 until master_running?
+          end
+
+          master_version = nil
+          spinner "Retrieving #{name.colorize(:cyan)} version" do
+            master_version = JSON.parse(client.get(path: '/'))["version"] rescue nil
+          end
+
+          spinner "Kontena Master #{master_version} is now running at #{API_URL}"
+
+          {
+            name: name.sub('kontena-master-', ''),
+            public_ip: API_URL.split('//').last,
+            provider: 'vagrant',
+            version: master_version,
+            code: opts[:initial_admin_code]
+          }
+        end
+
+        def master_running?
+          client.get(path: '/').status == 200
+        rescue
+          false
+        end
+
+        def generate_name
+          "kontena-master-#{super}-#{rand(1..99)}"
+        end
+      end
+    end
+  end
+end

data/lib/kontena/machine/vagrant/node_destroyer.rb

@@ -0,0 +1,35 @@
+require_relative 'common'
+
+module Kontena
+  module Machine
+    module Vagrant
+      class NodeDestroyer
+        include RandomName
+        include Kontena::Machine::Vagrant::Common
+        include Kontena::Cli::ShellSpinner
+
+        attr_reader :client, :api_client
+
+        # @param [Kontena::Client] api_client Kontena api client
+        def initialize(api_client)
+          @api_client = api_client
+        end
+
+        def run!(grid, name)
+          vagrant_path = "#{Dir.home}/.kontena/#{grid}/#{name}"
+          Dir.chdir(vagrant_path) do
+            spinner "Triggering termination of Vagrant machine #{name.colorize(:cyan)}"
+            run_command('vagrant destroy -f')
+            FileUtils.remove_entry_secure(vagrant_path)
+          end
+          node = api_client.get("grids/#{grid}/nodes")['nodes'].find{|n| n['name'] == name}
+          if node
+            spinner "Removing node #{name.colorize(:cyan)} from grid #{grid.colorize(:cyan)} " do
+              api_client.delete("nodes/#{grid}/#{name}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kontena/machine/vagrant/node_provisioner.rb

@@ -0,0 +1,82 @@
+require_relative 'common'
+module Kontena
+  module Machine
+    module Vagrant
+      class NodeProvisioner
+        include RandomName
+        include Kontena::Machine::Vagrant::Common
+        include Kontena::Cli::ShellSpinner
+
+        attr_reader :client, :api_client
+
+        # @param [Kontena::Client] api_client Kontena api client
+        def initialize(api_client)
+          @api_client = api_client
+        end
+
+        def run!(opts)
+          grid = opts[:grid]
+          if opts[:name]
+            name = "#{opts[:name]}-#{opts[:instance_number]}"
+          else
+            name = generate_name
+          end
+          version = opts[:version]
+          vagrant_path = "#{Dir.home}/.kontena/#{grid}/#{name}"
+          FileUtils.mkdir_p(vagrant_path)
+
+          template = File.join(__dir__ , '/Vagrantfile.node.rb.erb')
+          cloudinit_template = File.join(__dir__ , '/cloudinit.yml')
+          vars = {
+            name: name,
+            version: version,
+            cpus: opts[:cpus] || 1,
+            memory: opts[:memory] || 1024,
+            network_address: opts[:network_address],
+            master_uri: opts[:master_uri],
+            grid_token: opts[:grid_token],
+            coreos_channel: opts[:coreos_channel],
+            cloudinit: "#{vagrant_path}/cloudinit.yml"
+          }
+          spinner "Generating Vagrant config" do
+            vagrant_data = erb(File.read(template), vars)
+            cloudinit = erb(File.read(cloudinit_template), vars)
+            File.write("#{vagrant_path}/Vagrantfile", vagrant_data)
+            File.write("#{vagrant_path}/cloudinit.yml", cloudinit)
+          end
+
+          node = nil
+          Dir.chdir(vagrant_path) do
+            spinner "Triggering CoreOS Container Linux box update"
+            system('vagrant box update')
+            spinner "Executing 'vagrant up'"
+            run_command('vagrant up')
+            spinner "'vagrant up' executed successfully"
+            spinner "Waiting for node #{name.colorize(:cyan)} to join grid #{grid.colorize(:cyan)} " do
+              sleep 1 until node = node_exists_in_grid?(grid, name)
+            end
+          end
+          set_labels(
+            node,
+            [
+              "provider=vagrant"
+            ]
+          )
+        end
+
+        def generate_name
+          "#{super}-#{rand(1..99)}"
+        end
+
+        def node_exists_in_grid?(grid, name)
+          api_client.get("grids/#{grid}/nodes")['nodes'].find{|n| n['name'] == name}
+        end
+
+        def set_labels(node, labels)
+          data = {labels: labels}
+          api_client.put("nodes/#{node['id']}", data, {}, {'Kontena-Grid-Token' => node['grid']['token']})
+        end
+      end
+    end
+  end
+end

data/lib/kontena/plugin/vagrant.rb

@@ -0,0 +1,7 @@
+module Kontena
+  module Plugin
+    module Vagrant
+      VERSION = "0.3.1"
+    end
+  end
+end

data/lib/kontena/plugin/vagrant/master/create_command.rb

@@ -0,0 +1,44 @@
+require 'securerandom'
+
+module Kontena::Plugin::Vagrant::Master
+  class CreateCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    option "--name", "NAME", "Set master name"
+    option "--memory", "MEMORY", "How much memory node has"
+    option "--version", "VERSION", "Define installed Kontena version", default: 'latest'
+    option "--vault-secret", "VAULT_SECRET", "Secret key for Vault"
+    option "--vault-iv", "VAULT_IV", "Initialization vector for Vault"
+    option "--coreos-channel", "CHANNEL", "CoreOS release channel", default: 'stable'
+
+    def execute
+      require_relative '../../../machine/vagrant'
+      mem = ask_instance_memory
+      provisioner.run!(
+        name: name,
+        memory: mem,
+        version: version,
+        vault_secret: vault_secret || SecureRandom.hex(24),
+        vault_iv: vault_iv || SecureRandom.hex(24),
+        initial_admin_code: SecureRandom.hex(16),
+        coreos_channel: coreos_channel
+      )
+    end
+
+    def ask_instance_memory
+      if self.memory.nil?
+        prompt.select("Choose a size") do |menu|
+          %w(512 1024 2048).each do |mem|
+            menu.choice "#{mem}MB", mem
+          end
+        end
+      else
+        self.memory
+      end
+    end
+
+    def provisioner
+      Kontena::Machine::Vagrant::MasterProvisioner.new
+    end
+  end
+end