Contents

#          Copyright (c) 2008 Michael Fellinger [e-mail]
# All files in this distribution are subject to the terms of the Ruby license.

require(File.join(File.dirname(__FILE__), 'blankslate'))

class OrderedSet < BlankSlate
  def initialize(*args)
    if args.size == 1
      @set = args.shift
    else
      @set = *args
    end

    @set ||= []
    @set = [@set] unless ::Array === @set
    @set.uniq!
  end

  def method_missing(meth, *args, &block)
    case meth.to_s
    when /push|unshift|\<\</
      @set.delete *args
    when '[]='
      @set.map! do |e|
        if ::Array === args.last
          args.last.include?(e) ? nil : e
        else
          args.last == e ? nil : e
        end
      end
    end
    @set.__send__(meth, *args, &block)
  ensure
    @set.compact! if meth == :[]=
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
ramaze-0.3.9	lib/ramaze/snippets/ordered_set.rb
ramaze-0.3.9.1	lib/ramaze/snippets/ordered_set.rb