module Cucumber
  module Tree
    class Feature
      attr_accessor :file
      
      def initialize(header, &proc)
        @header = header
        @scenarios = []
        instance_eval(&proc) if block_given?
      end

      def add_scenario(name, &proc)
        scenario = Scenario.new(self, name, &proc)
        @scenarios << scenario
        scenario
      end
      
      def add_row_scenario(template_scenario, values, line)
        scenario = RowScenario.new(self, template_scenario, values, line)
        @scenarios << scenario
        scenario
      end

      def Scenario(name, &proc)
        add_scenario(name, &proc)
      end
      
      def Table(matrix = [], &proc)
        table = Table.new(matrix)
        proc.call(table)
        template_scenario = @scenarios.last
        matrix[1..-1].each do |row|
          @scenarios << RowScenario.new(self, template_scenario, row, row.line)
        end
      end

      def accept(visitor)
        visitor.visit_header(@header)
        @scenarios.each do |scenario|
          visitor.visit_scenario(scenario)
        end
      end
    end
  end
end