solr

#
# Cookbook Name:: solr-component
# Recipe:: default
#

require "pathname"
include_recipe "zookeeper-component"
include_recipe "tomcat"

# Since solr 4.3.0 we need slf4j jar http://wiki.apache.org/solr/SolrLogging#Solr_4.3_and_above
slf4j_url = "#{node["solr"]["slf4j"]["url"]}/slf4j-#{node["solr"]["slf4j"]["version"]}.tar.gz"
remote_file "/tmp/slf4j-#{node["solr"]["slf4j"]["version"]}.tar.gz" do
  source slf4j_url
  action :create_if_missing
end

execute "extract slf4j" do
  command "tar -xzvf /tmp/slf4j-#{node["solr"]["slf4j"]["version"]}.tar.gz -C /tmp"
  creates "/tmp/slf4j-#{node["solr"]["slf4j"]["version"]}"
end

["slf4j-jdk14-#{node["solr"]["slf4j"]["version"]}.jar", "log4j-over-slf4j-#{node["solr"]["slf4j"]["version"]}.jar", "slf4j-api-#{node["solr"]["slf4j"]["version"]}.jar", "jcl-over-slf4j-#{node["solr"]["slf4j"]["version"]}.jar"].each do |file|
  execute "copy #{file} to tomcat" do
    command "cp /tmp/slf4j-#{node["solr"]["slf4j"]["version"]}/#{file} #{node["tomcat"]["lib_dir"]}/"
    creates "#{node["tomcat"]["lib_dir"]}/#{file}"
  end
end

# Extract war file from solr archive
solr_url = "#{node["solr"]["url"]}#{node["solr"]["version"]}/solr-#{node["solr"]["version"]}.tgz"
remote_file "solr_src" do
  path "/opt/solr-#{node["solr"]["version"]}.tgz"
  source solr_url
  action :create_if_missing
end

execute "extract solr_src" do
  command "tar -xzvf /opt/solr-#{node["solr"]["version"]}.tgz -C /opt"
  creates "/opt/solr-#{node["solr"]["version"]}"
end

#Copy sorl.war to webapps
directory "#{node["solr"]["path"]}/webapps" do
  mode 00755
  owner node["tomcat"]["user"]
  group node["tomcat"]["group"]
  recursive true
  action :create
end


#Deploy solr to tomcat
template "#{node["solr"]["path"]}/webapps/solr.xml" do
  owner node["tomcat"]["user"]
  group node["tomcat"]["group"]
  source "solr-app.xml.erb"
end

execute "link solr app" do
  command "ln -s #{node["solr"]["path"]}/webapps/solr.xml #{node["tomcat"]["context_dir"]}/solr.xml"
  creates "#{node["tomcat"]["context_dir"]}/solr.xml"
end

execute "extract solr_war" do
  command "unzip -o /opt/solr-#{node["solr"]["version"]}/dist/solr-#{node["solr"]["version"]}.war -d #{node["solr"]["path"]}/webapps/solr"
  action :run
end

#Create collections
node["solr"]["collection"].each do |collection|
  directory "#{node["solr"]["path"]}/cores/#{collection}/conf/" do
    owner node["tomcat"]["user"]
    group node["tomcat"]["group"]
    mode 00755
    recursive true
    action :create
  end

  template "#{node["solr"]["path"]}/cores/#{collection}/conf/schema.xml" do
    owner node["tomcat"]["user"]
    group node["tomcat"]["group"]
    source "schema.xml.erb"
    variables({
      :collection => "#{collection}"
    })
  end

  template "#{node["solr"]["path"]}/cores/#{collection}/conf/solrconfig.xml" do
    owner node["tomcat"]["user"]
    group node["tomcat"]["group"]
    source "solrconfig.xml.erb"
    variables({
      :collection => "#{collection}"
    })
  end
end

template "#{node["solr"]["path"]}/cores/solr.xml" do
  owner node["tomcat"]["user"]
  group node["tomcat"]["group"]
  source "solr.xml.erb"
  variables({
    :zooip => node["solr"]["zookeeper"]["host"],
    :zooport => node["solr"]["zookeeper"]["port"]
  })
end

execute "ln home" do
  command "ln -s #{node["solr"]["path"]} #{node["tomcat"]["base"]}/solr"
  creates "#{node["tomcat"]["base_dir"]}/solr"
end

#create zkcli
template "#{node["solr"]["path"]}/webapps/zkcli.sh" do
  owner node["tomcat"]["user"]
  group node["tomcat"]["group"]
  mode 00755
  source "zkcli.sh.erb"
  variables({
    :solr_src => "/opt/solr-#{node["solr"]["version"]}/example/cloud-scripts"
  })
end

#populate zookeeper
zoo_connect = "#{node["solr"]["zookeeper"]["host"]}:#{node["solr"]["zookeeper"]["port"]}"
solr_cores = "#{node["solr"]["path"]}/cores"
node["solr"]["collection"].each do |collection|
  execute "upload zoo collection #{collection}" do
    command "sudo #{node["solr"]["path"]}/webapps/zkcli.sh -cmd upconfig -zkhost #{zoo_connect} -d #{solr_cores}/#{collection}/conf/ -n #{collection}"

  end

  execute "link zoo collection #{collection}" do
    command "sudo #{node["solr"]["path"]}/webapps/zkcli.sh -cmd linkconfig -zkhost #{zoo_connect} -collection #{collection} -confname #{collection} -solrhome #{solr_cores}"
  end

  execute "bootstrap zoo collection #{collection}" do
    command "sudo #{node["solr"]["path"]}/webapps/zkcli.sh -cmd bootstrap -zkhost #{zoo_connect} -solrhome #{solr_cores}"
  end
end

execute "change solr owner" do
  command "sudo chown -R #{node["tomcat"]["user"]}:#{node["tomcat"]["group"]} #{node["solr"]["path"]}"
  notifies :restart, "service[tomcat]", :immediately
end

if platform_family?('rhel')
  execute "stop iptables" do
    command "/etc/init.d/iptables stop"
    action :run
  end
end