Rakefile

require 'yaml'
environment = "lappis"

ssh_config_file = "config/#{environment}/ssh_config"
ips_file = "config/#{environment}/ips.yaml"
passwd_file = "config/#{environment}/passwd.yaml"
certificate_domains_file = "config/#{environment}/certificate_domains.yaml"
external_ips_file = "config/#{environment}/external_ips.yaml"

ENV['CHAKE_SSH_CONFIG'] = ssh_config_file
ENV['CHAKE_RSYNC_OPTIONS'] = " --exclude backups"

require "chake"

ips ||= YAML.load_file(ips_file)
passwords ||= YAML.load_file(passwd_file)
crt_domains ||= YAML.load_file(certificate_domains_file)
external_ips ||= YAML.load_file(external_ips_file)

$nodes.each do |node|
  node.data['peers'] = ips
  node.data['passwd'] = passwords
  node.data['crt_domains'] = crt_domains
  node.data['external_ips'] = external_ips
end

require 'optparse'

task :create do
  # to use this task you need to write the command like the following example
  # rake create [cookbook_name] -- --ip [virtual_manhine_ip] --name [service_name]

  ARGV.each { |a| task a.to_sym do ; end }
  cookbook_name = ARGV[1].to_s.strip
  sh 'mkdir', '-p', "cookbooks/#{cookbook_name}/files/default"
  sh 'mkdir', '-p', "cookbooks/#{cookbook_name}/templates/default"
  sh 'mkdir', '-p', "cookbooks/#{cookbook_name}/recipes"
  sh 'touch', "cookbooks/#{cookbook_name}/recipes/default.rb"

  options = {}
  opts = OptionParser.new
  opts.banner = "Usage: rake create [cookbook_name] -- [options]"
  opts.on("-i IP", "--ip IP") { |ip| options[:ip] = ip }
  opts.on("-n NAME", "--name NAME") { |name| options[:name] = name }
  args = opts.order!(ARGV){}
  opts.parse!(args)

  unless options[:ip].nil?
    options[:name] ||= cookbook_name
    unless ips.has_key?(options[:name])
      open(ips_file, 'a') do |f|
        f.puts "#{options[:name]}: #{options[:ip]}"
      end
    end
  end

  exit
end

def ssh_cmd(cmd, host)
  sh 'ssh', '-F', ENV['CHAKE_SSH_CONFIG'], host, cmd
end

def scp_cmd(file, dest, host, flags='')
  sh 'scp', flags,'-F', ENV['CHAKE_SSH_CONFIG'], file, "#{host}:#{dest}"
end
Rake.add_rakelib 'lib/tasks'