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 && 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