Vendor the external cookbooks

Knife-Zero doesn't include Berkshelf support, so vendoring everything in
the repo is convenient again

cookbooks/elasticsearch/libraries/provider_plugin.rb

@@ -0,0 +1,108 @@
+# Chef Provider for installing an elasticsearch plugin
+class ElasticsearchCookbook::PluginProvider < Chef::Provider::LWRPBase
+  include ElasticsearchCookbook::Helpers
+  include Chef::Mixin::ShellOut
+
+  provides :elasticsearch_plugin
+
+  def whyrun_supported?
+    true # we only use core Chef resources that also support whyrun, or guard
+  end
+
+  def action_install
+    return if plugin_exists(new_resource.plugin_name)
+
+    # since install can take a URL argument instead
+    url_or_name = new_resource.url || new_resource.plugin_name
+    manage_plugin("install #{url_or_name}")
+  end # action
+
+  def action_remove
+    return unless plugin_exists(new_resource.plugin_name)
+
+    manage_plugin("remove #{new_resource.plugin_name}")
+  end # action
+
+  def manage_plugin(arguments)
+    es_user = find_es_resource(Chef.run_context, :elasticsearch_user, new_resource)
+    es_install = find_es_resource(Chef.run_context, :elasticsearch_install, new_resource)
+    es_conf = find_es_resource(Chef.run_context, :elasticsearch_configure, new_resource)
+
+    assert_state_is_valid(es_user, es_install, es_conf) unless whyrun_mode?
+
+    # shell_out! automatically raises on error, logs command output
+    # required for package installs that show up with parent dir owned by root
+    plugin_dir_exists = ::File.exist?(es_conf.path_plugins)
+    unless plugin_dir_exists
+      cmd_str = "mkdir -p #{es_conf.path_plugins}"
+      if whyrun_mode?
+        Chef::Log.info("Would run command: #{cmd_str}")
+      else
+        shell_out_as_user!(cmd_str, Chef.run_context)
+        new_resource.updated_by_last_action(true)
+      end
+    end
+
+    unless plugin_exists(new_resource.plugin_name)
+      cmd_str = "#{es_conf.path_bin}/elasticsearch-plugin #{arguments.chomp(' ')} #{new_resource.options}".chomp(' ')
+      if whyrun_mode?
+        Chef::Log.info("Would run command: #{cmd_str}")
+      else
+        command_array = cmd_str.split(' ')
+        shell_out_as_user!(command_array, Chef.run_context)
+        new_resource.updated_by_last_action(true)
+      end
+    end
+  end
+
+  def plugin_exists(name)
+    es_conf = find_es_resource(Chef.run_context, :elasticsearch_configure, new_resource)
+    path = es_conf.path_plugins
+
+    Dir.entries(path).any? do |plugin|
+      next if plugin =~ /^\./
+      name == plugin
+    end
+  rescue
+    false
+  end
+
+  def assert_state_is_valid(_es_user, _es_install, es_conf)
+    unless es_conf.path_plugins # we do not check existence (may not exist if no plugins installed)
+      raise "Could not determine the plugin directory (#{es_conf.path_plugins}). Please check elasticsearch_configure[#{es_conf.name}]."
+    end
+
+    unless es_conf.path_bin && ::File.exist?(es_conf.path_bin)
+      raise "Could not determine the binary directory (#{es_conf.path_bin}). Please check elasticsearch_configure[#{es_conf.name}]."
+    end
+
+    true
+  end
+
+  def shell_out_as_user!(command, run_ctx)
+    es_install = find_es_resource(run_ctx, :elasticsearch_install, new_resource)
+    es_conf = find_es_resource(run_ctx, :elasticsearch_configure, new_resource)
+    es_svc = find_es_resource(run_ctx, :elasticsearch_service, new_resource)
+
+    # we need to figure out the env file path to set environment for plugins
+    default_config_name = es_svc.service_name || es_svc.instance_name || es_conf.instance_name || 'elasticsearch'
+    include_file_resource = find_exact_resource(run_ctx, :template, "elasticsearch.in.sh-#{default_config_name}")
+    env = { 'ES_INCLUDE' => include_file_resource.path }
+
+    # Add HTTP Proxy vars unless explicitly told not to
+    if new_resource.chef_proxy
+      env['ES_JAVA_OPTS'] = "#{ENV['ES_JAVA_OPTS']} #{get_java_proxy_arguments}"
+    end
+
+    # See this link for an explanation:
+    # https://www.elastic.co/guide/en/elasticsearch/plugins/2.1/plugin-management.html
+    if es_install.type == 'package' || es_install.type == 'repository'
+      # package installations should install plugins as root
+      shell_out!(command, :env => env, :timeout => 1200)
+    else
+      # non-package installations should install plugins as the ES user
+      es_user = find_es_resource(run_ctx, :elasticsearch_user, new_resource)
+      shell_out!(command, user: es_user.username, group: es_user.groupname, :env => env, :timeout => 1200)
+    end
+  end
+end # provider