class TrustyCms::AdminUI::RegionSet

def [](region)
  @regions[region.to_sym]
end

def add(region=nil, partial=nil, options={})
  raise ArgumentError, "You must specify a region and a partial" unless region and partial
  if options[:before]
    index = @regions[region].empty? ? 0 : (@regions[region].index(options[:before]) || @regions[region].size)
    self[region].insert(index, partial)
  elsif options[:after]
    index = @regions[region].empty? ? 0 : (@regions[region].index(options[:after]) || @regions[region].size - 1)
    self[region].insert(index + 1, partial)
  else
    self[region] << partial
  end
end

def initialize
  @regions = Hash.new do |h,k|
    h[k] = []
  end
  yield self if block_given?
end

def method_missing(method, *args, &block)
  if args.empty?
    self[method]
  else
    super
  end
end