hopsoft/ellington

require "delegate"

module Ellington
  class LineList < SimpleDelegator
    attr_reader :route_class

    def initialize(route_class)
      @route_class = route_class
      @inner_list = UniqueTypeArray.new
      super @inner_list
    end

    def push(line)
      value = inner_list << line
      line.route_class = route_class
      value
    end
    alias_method :<<, :push

    def add(line)
      push(line).last
    end

    def find_by_type(type)
      find { |entry| entry.is_a? type }
    end

    protected

    attr_reader :inner_list
  end
end