loadbalancer.rb

#!/usr/bin/env ruby

require 'linode'

if ARGV.count != 3 || !["accept","reject"].include?(ARGV[2])
  puts "usage: loadbalancer {node_balancer_label | ALL} {host_label} {accept|reject}"
  exit
end

nb_label = ARGV[0]
cmd_label = ARGV[1]
cmd_action = ARGV[2]

l = Linode.new(:api_key => "LINODE API KEY HERE")

balancers = []
nb_list = l.nodebalancer.list
if nb_label == "ALL"
  balancers = nb_list
else
  nb = nb_list.find {|n| n.label == nb_label}
  if nb
    balancers << nb
  else
    puts "could not find nodebalancer labeled #{nb_label}"
    exit
  end
end

balancers.each do |nb|
  puts "Found NodeBalancer #{nb.label} (#{nb.nodebalancerid})"
end

monitor_nodes = Array.new

balancers.each do |nb|
  config_list = l.nodebalancer.config.list(:NodeBalancerID => nb.nodebalancerid)
  config_list.each do |conf|
    nodes = l.nodebalancer.node.list(:ConfigID => conf.configid)
    nodes.each do |node|
      if node.label == cmd_label
        puts "found node on #{nb.label} ext port #{conf.port}, changing status to #{cmd_action}"
        l.nodebalancer.node.update(:NodeID => node.nodeid, :Mode => cmd_action)
        puts "  status changed to #{cmd_action}"
        monitor_nodes << {:config => conf, :node_id => node.nodeid}
      end
    end
  end
end

print "Waiting for propagation..."

new_status = cmd_action == "reject" ? "DOWN" : "UP"
completed = false
while !completed do
  completed = true
  monitor_nodes.each do |mn|
    nodes = l.nodebalancer.node.list(:ConfigID => mn[:config].configid)
    node = nodes.find {|n| n.nodeid == mn[:node_id]}
    print "."
    completed = false if node.status != new_status
  end
  sleep(2) if !completed
end

puts "\nDone."