[WIP] Feature/swarm

libraries/_autoload.rb

@@ -0,0 +1,27 @@
+begin
+  gem 'docker-api', '= 1.33.6'
+rescue LoadError
+  unless defined?(ChefSpec)
+    run_context = Chef::RunContext.new(Chef::Node.new, {}, Chef::EventDispatch::Dispatcher.new)
+
+    require 'chef/resource/chef_gem'
+
+    docker = Chef::Resource::ChefGem.new('docker-api', run_context)
+    docker.version '= 1.33.6'
+    docker.run_action(:install)
+  end
+end
+
+begin
+  gem 'docker-swarm-sdk', '~> 1.2.4'
+rescue LoadError
+  unless defined?(ChefSpec)
+    run_context = Chef::RunContext.new(Chef::Node.new, {}, Chef::EventDispatch::Dispatcher.new)
+
+    require 'chef/resource/chef_gem'
+
+    docker = Chef::Resource::ChefGem.new('docker-swarm-sdk', run_context)
+    docker.version '= 1.2.4'
+    docker.run_action(:install)
+  end
+end

libraries/docker_swarm_base.rb

@@ -0,0 +1,14 @@
+module DockerCookbook
+  # We override the DockerBase to use the Docker Swarm SDK Connection object
+  class DockerSwarmBase < DockerBase
+    require_relative 'helpers_base'
+    include DockerHelpers::Base
+    require_relative 'helpers_swarm_base'
+    include DockerHelpers::SwarmBase
+
+    property :host, [String, nil], default: lazy { default_host }, desired_state: false
+
+    def swarm_info
+    end
+  end
+end

libraries/docker_swarm_manager.rb

@@ -0,0 +1,46 @@
+require 'docker-swarm'
+
+module DockerCookbook
+  require_relative 'docker_swarm_node_base'
+  # Docker swarm manager init and join
+  class DockerSwarmManager < DockerSwarmNodeBase
+    require_relative 'helpers_swarm_cluster'
+
+    include DockerHelpers::SwarmCluster
+
+    # Resource properties
+    resource_name :docker_swarm_manager
+
+    property :first_manager, [TrueClass, FalseClass], default: false
+
+    declare_action_class.class_eval do
+      def whyrun_supported?
+        true
+      end
+    end
+
+    #########
+    # Actions
+    #########
+
+    default_action :create
+
+    action :create do
+      ensure_swarm_available!
+      return if cluster_initialized?
+
+      if new_resource.first_manager
+        converge_by 'Running swarm init' do
+          manager_init
+        end
+      else
+        if join_address.nil? || join_token.nil?
+          raise 'Provide join_address & join_token or set `first_manager true`'
+        end
+        converge_by 'Running swarm join' do
+          cluster_join(:manager, join_address, join_token)
+        end
+      end
+    end
+  end
+end

libraries/docker_swarm_node_base.rb

@@ -0,0 +1,44 @@
+require 'docker-swarm'
+
+module DockerCookbook
+  # Initialize a docker swarm
+  class DockerSwarmNodeBase < DockerSwarmBase
+    require_relative 'helpers_swarm_cluster'
+
+    include DockerHelpers::SwarmCluster
+
+    property :cluster_name, String, name_propery: true
+    property :listen_address, String, default: '0.0.0.0'
+
+    property :join_address, String
+    property :join_token, String
+    property :force_leave, [TrueClass, FalseClass], default: true
+
+    property :drain_opts, Hash, default: {}
+
+    declare_action_class.class_eval do
+      def whyrun_supported?
+        true
+      end
+    end
+
+    action :drain do
+      ensure_swarm_available!
+      return unless cluster_initialized?
+      return if current_swarm_node.running_tasks == 0
+
+      converge_by 'Draining node' do
+        current_swarm_node.drain(drain_opts)
+      end
+    end
+
+    action :destroy do
+      ensure_swarm_available!
+      return unless cluster_initialized?
+
+      converge_by "Leaving the cluster (force=#{force_leave})" do
+        cluster_leave(force_leave)
+      end
+    end
+  end
+end

libraries/docker_swarm_overlay_network.rb

@@ -0,0 +1,46 @@
+require 'docker-swarm'
+
+module DockerCookbook
+  # Initialize a docker swarm
+  class DockerSwarmOverlayNetwork < DockerSwarmBase
+    require_relative 'helpers_swarm_network'
+
+    include DockerHelpers::SwarmNetwork
+
+    # Resource properties
+    resource_name :docker_swarm_overlay_network
+    property :network_name, String, name_propery: true
+
+    declare_action_class.class_eval do
+      def whyrun_supported?
+        true
+      end
+    end
+
+    #########
+    # Actions
+    #########
+
+    default_action :create
+
+    action :create do
+      ensure_swarm_available!
+      return if network_find(new_resource.name)
+
+      converge_by 'Create overlay network' do
+        create_overlay_network(new_resource.name)
+      end
+    end
+
+    action :destroy do
+      ensure_swarm_available!
+
+      network = network_find(new_resource.name)
+      return unless network
+
+      converge_by 'Destroying network' do
+        remove_overlay_network(network)
+      end
+    end
+  end
+end

libraries/docker_swarm_service.rb

@@ -0,0 +1,108 @@
+require 'docker-swarm'
+
+module DockerCookbook
+  # Initialize a docker swarm
+  class DockerSwarmService < DockerSwarmBase
+    require_relative 'helpers_swarm_service'
+
+    include DockerHelpers::SwarmService
+
+    resource_name :docker_swarm_service
+
+    property :service_name, String, name_propery: true
+
+    property :image, String, required: true
+    property :user, String, required: true, default: 'root'
+    property :networks, [Array, nil], required: true,
+             coerce: proc { |v| Array(v).empty? ? nil : Array(v) }
+    property :hostname, String
+
+    property :environment, Hash, default: {}
+    property :log_driver, String, required: true, default: 'json-file'
+    property :log_driver_opts, Hash,
+             default: { 'max-file' => '3', 'max-size' => '10M' }
+
+    property :limits, Hash, default: {}
+    property :reservations, Hash, default: {}
+
+    property :restart_condition, String, required: true, default: 'on-failure'
+    property :restart_delay, Integer, required: true, default: 1
+    property :restart_attempts, Integer, required: true, default: 3
+
+    property :health_start_period, Integer, required: true, default: 10
+
+    property :global, [TrueClass, FalseClass], required: true, default: false
+    property :replicas, Integer, default: 1
+    property :ports, Array, default: []
+    property :labels, Hash, default: {}
+    property :mounts, Array, default: []
+
+    %i(update rollback).each do |kind|
+      property "#{kind}_parallelism".to_sym, Integer, required: true, default: 1
+      property "#{kind}_delay".to_sym, Integer, required: true, default: 1
+      property "#{kind}_monitor".to_sym, Integer, required: true, default: 15
+      property "#{kind}_max_failure_ratio".to_sym, Float,
+               required: true,
+               default: 0.0
+      property "#{kind}_failure_action".to_sym, %w(pause continue rollback),
+               required: true,
+               default: 'pause'
+    end
+
+    declare_action_class.class_eval do
+      def whyrun_supported?
+        true
+      end
+    end
+
+    #########
+    # Actions
+    #########
+
+    default_action :create
+
+    action :create do
+      ensure_swarm_available!
+
+      if current_service.nil?
+        converge_by('Create service') { create_service }
+      elsif current_service_changed?
+        converge_by('Update service') { update_service }
+      end
+    end
+
+    action :update do
+      ensure_swarm_available!
+
+      # To avoid using a stale version of the current service spec when
+      # executing two actions against the same resource in the same run (and
+      # have an out of sequence ForceUpdate counter). We reset our cached
+      # service spec before every update
+      reset_current_service
+
+      if current_service_changed?
+        converge_by('Update service') { update_service }
+      end
+    end
+
+    action :force_update do
+      ensure_swarm_available!
+
+      # See #update comment
+      reset_current_service
+
+      converge_by('Update service [forced]') do
+        update_service(true)
+      end
+    end
+
+    action :destroy do
+      ensure_swarm_available!
+      return unless current_service
+
+      converge_by 'Delete service' do
+        current_service.remove
+      end
+    end
+  end
+end

libraries/docker_swarm_worker.rb

@@ -0,0 +1,38 @@
+require 'docker-swarm'
+
+module DockerCookbook
+  require_relative 'docker_swarm_node_base'
+  # Initialize a docker swarm worker
+  class DockerSwarmWorker < DockerSwarmNodeBase
+    require_relative 'helpers_swarm_cluster'
+
+    include DockerHelpers::SwarmCluster
+
+    # Resource properties
+    resource_name :docker_swarm_worker
+
+    declare_action_class.class_eval do
+      def whyrun_supported?
+        true
+      end
+    end
+
+    #########
+    # Actions
+    #########
+
+    default_action :create
+
+    action :create do
+      ensure_swarm_available!
+      return if cluster_initialized?
+
+      if join_address.nil? || join_token.nil?
+        raise 'Provide join_address & join_token'
+      end
+      converge_by 'Running swarm join' do
+        cluster_join(:worker, join_address, join_token)
+      end
+    end
+  end
+end