crowbar/crowbar-ha

#
# 2009-2013, Opscode, Inc
# 2014, SUSE
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

# This is largely copied/inspired from https://github.com/hw-cookbooks/haproxy

use_inline_resources if defined?(use_inline_resources)

action :create do
  # While there is no way to have an include directive for haproxy
  # configuration file, this provider will only modify attributes !

  if new_resource.port < 1 || new_resource.port > 65535
    raise "Invalid port: #{new_resource.port}."
  end

  if new_resource.servers.empty?
    raise "No server specified."
  end

  new_resource.servers.each do |server|
    raise "One of the servers has no name." if server["name"].nil?
    raise "Server #{server['name']} has no address." if server["address"].nil?
    raise "Server #{server['name']} has no port." if server["port"].nil?
    raise "Server #{server['name']} has invalid port." if (server["port"] < 1 || server["port"] > 65535)
  end

  section = {}
  section["address"] = new_resource.address
  section["port"] = new_resource.port
  section["use_ssl"] = new_resource.use_ssl
  section["terminate_ssl"] = new_resource.terminate_ssl
  if new_resource.use_ssl
    if new_resource.terminate_ssl
      section["mode"] = "http"
      section["pemfile"] = new_resource.pemfile
    else
      section["mode"] = "tcp"
    end
  else
    section["mode"] = new_resource.mode
  end
  section["balance"] = new_resource.balance unless new_resource.balance.empty?

  section["stick"] = new_resource.stick
  section["stick"]["expire"] ||= "30m"

  section["options"] = new_resource.options || []
  if section["options"].empty? || section["options"].include?("defaults")
    section["options"].delete("defaults")
    if section["use_ssl"] && !section["terminate_ssl"]
      section["options"] = [["tcpka", "tcplog"], section["options"]].flatten
    elsif section["mode"] == "http"
      section["options"] = [["tcpka", "httplog", "forwardfor"], section["options"]].flatten
    end
  end
  section["max_connections"] = new_resource.max_connections unless new_resource.max_connections.nil?
  section["default_server"] = new_resource.default_server unless new_resource.default_server.empty?
  section["servers"] = new_resource.servers

  if new_resource.rate_limit && new_resource.rate_limit.to_i != 0
    section["rate_limit"] = new_resource.rate_limit
  end

  node["haproxy"]["sections"][new_resource.type] ||= {}
  node["haproxy"]["sections"][new_resource.type][new_resource.name] = section
end

action :delete do
  node["haproxy"]["sections"].keys.each do |type|
    node["haproxy"]["sections"][type].delete(new_resource.name)
  end
end